1 |
|
package org.apache.torque.task; |
2 |
|
|
3 |
|
import java.io.File; |
4 |
|
import java.io.FileWriter; |
5 |
|
import java.io.IOException; |
6 |
|
import java.io.StringWriter; |
7 |
|
|
8 |
|
import javax.xml.parsers.DocumentBuilder; |
9 |
|
import javax.xml.parsers.DocumentBuilderFactory; |
10 |
|
import javax.xml.parsers.ParserConfigurationException; |
11 |
|
import javax.xml.transform.OutputKeys; |
12 |
|
import javax.xml.transform.Result; |
13 |
|
import javax.xml.transform.Transformer; |
14 |
|
import javax.xml.transform.TransformerFactory; |
15 |
|
import javax.xml.transform.dom.DOMSource; |
16 |
|
import javax.xml.transform.stream.StreamResult; |
17 |
|
import javax.xml.xpath.XPath; |
18 |
|
import javax.xml.xpath.XPathConstants; |
19 |
|
import javax.xml.xpath.XPathFactory; |
20 |
|
|
21 |
|
import org.apache.tools.ant.BuildException; |
22 |
|
import org.apache.tools.ant.Task; |
23 |
|
import org.w3c.dom.Document; |
24 |
|
import org.w3c.dom.Element; |
25 |
|
import org.w3c.dom.NodeList; |
26 |
|
import org.xml.sax.SAXException; |
27 |
|
|
28 |
|
|
29 |
|
|
30 |
|
|
31 |
|
|
32 |
|
@author |
33 |
|
|
34 |
|
|
|
|
| 0% |
Uncovered Elements: 94 (94) |
Complexity: 19 |
Complexity Density: 0.28 |
|
35 |
|
public class TorqueMergeXMLDoc extends Task { |
36 |
|
|
37 |
|
private static final String DESCRIPTION_ATTR = "description"; |
38 |
|
private static final String JAVA_NAME_ATTR = "javaName"; |
39 |
|
private static final String NAME_ATTR = "name"; |
40 |
|
private static final String COLUMN_ELEMENT = "column"; |
41 |
|
|
42 |
|
private File schemaWithDesc; |
43 |
|
private File dbSchema; |
44 |
|
private String schemaWithDescString; |
45 |
|
private String dbSchemaString; |
46 |
|
private Document schemaWithDescDoc; |
47 |
|
private Document dbSchemaDoc; |
48 |
|
|
49 |
|
|
|
|
| 0% |
Uncovered Elements: 2 (2) |
Complexity: 1 |
Complexity Density: 0.5 |
|
50 |
0
|
public void setDbSchemaString(String dbSchemaString) { ... |
51 |
0
|
this.dbSchemaString = dbSchemaString; |
52 |
0
|
dbSchema = new File(dbSchemaString); |
53 |
|
|
54 |
|
} |
55 |
|
|
|
|
| 0% |
Uncovered Elements: 2 (2) |
Complexity: 1 |
Complexity Density: 0.5 |
|
56 |
0
|
public void setSchemaWithDescString(String schemaWithDescString) { ... |
57 |
0
|
this.schemaWithDescString = schemaWithDescString; |
58 |
0
|
schemaWithDesc = new File(schemaWithDescString); |
59 |
|
|
60 |
|
} |
61 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
62 |
0
|
public File getDbSchema() {... |
63 |
0
|
return dbSchema; |
64 |
|
} |
65 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
66 |
0
|
public void setSchemaWithDesc(String schemaWithDescString) {... |
67 |
0
|
this.schemaWithDesc = new File(schemaWithDescString); |
68 |
|
|
69 |
|
} |
70 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
71 |
0
|
public void setDbSchema(String dbSchemaString) {... |
72 |
0
|
this.dbSchema = new File(dbSchemaString); |
73 |
|
} |
74 |
|
|
75 |
|
|
76 |
|
|
77 |
|
|
78 |
|
@param |
79 |
|
@return |
80 |
|
@throws |
81 |
|
@throws |
82 |
|
@throws |
83 |
|
|
|
|
| 0% |
Uncovered Elements: 6 (6) |
Complexity: 1 |
Complexity Density: 0.17 |
|
84 |
0
|
public static Document setDocument(File file)... |
85 |
|
throws ParserConfigurationException, SAXException, IOException { |
86 |
0
|
DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory |
87 |
|
.newInstance(); |
88 |
0
|
DocumentBuilder documentBuilder; |
89 |
0
|
documentBuilder = documentBuilderFactory.newDocumentBuilder(); |
90 |
0
|
Document XMLdocument = documentBuilder.parse(file); |
91 |
0
|
XMLdocument.getDocumentElement().normalize(); |
92 |
0
|
return XMLdocument; |
93 |
|
} |
94 |
|
|
95 |
|
|
96 |
|
|
97 |
|
|
98 |
|
|
99 |
|
@param |
100 |
|
@param |
101 |
|
|
|
|
| 0% |
Uncovered Elements: 3 (3) |
Complexity: 1 |
Complexity Density: 0.33 |
|
102 |
0
|
public void mergeSchemas(File schemaWithDesc, File dbSchema) throws Exception {... |
103 |
|
|
104 |
0
|
schemaWithDescDoc = setDocument(schemaWithDesc); |
105 |
0
|
dbSchemaDoc = setDocument(dbSchema); |
106 |
0
|
dbSchemaDoc = createNewXML(dbSchemaDoc, schemaWithDescDoc); |
107 |
|
|
108 |
|
} |
109 |
|
|
110 |
|
|
111 |
|
|
112 |
|
|
113 |
|
|
114 |
|
|
|
|
| 0% |
Uncovered Elements: 40 (40) |
Complexity: 7 |
Complexity Density: 0.25 |
|
115 |
0
|
public Document createNewXML(Document dbSchemaDoc,... |
116 |
|
Document schemaWithDescDoc) throws Exception { |
117 |
|
|
118 |
0
|
XPath xpath = XPathFactory.newInstance().newXPath(); |
119 |
|
|
120 |
0
|
NodeList listOfTablesInSchema = (NodeList)xpath.evaluate("/database/table", dbSchemaDoc, XPathConstants.NODESET); |
121 |
|
|
122 |
0
|
for (int tableIndex = 0; tableIndex < listOfTablesInSchema.getLength(); tableIndex++) { |
123 |
0
|
Element tableElementInSchema = (Element)listOfTablesInSchema.item(tableIndex); |
124 |
0
|
String tableName = tableElementInSchema.getAttribute(NAME_ATTR); |
125 |
|
|
126 |
0
|
Element tableDescElem = (Element)xpath.evaluate("/database/table[@" + NAME_ATTR + "='" + tableName + "']", schemaWithDescDoc, XPathConstants.NODE); |
127 |
0
|
if (tableDescElem != null) { |
128 |
0
|
String tableDescription = tableDescElem.getAttribute(DESCRIPTION_ATTR); |
129 |
0
|
String tableJavaName = tableDescElem.getAttribute(JAVA_NAME_ATTR); |
130 |
0
|
tableElementInSchema.setAttribute(DESCRIPTION_ATTR, tableDescription); |
131 |
0
|
tableElementInSchema.setAttribute(JAVA_NAME_ATTR, tableJavaName); |
132 |
0
|
NodeList columnElements = tableElementInSchema.getElementsByTagName(COLUMN_ELEMENT); |
133 |
0
|
for (int columnIndex = 0; columnIndex < columnElements.getLength(); columnIndex++) { |
134 |
|
|
135 |
0
|
Element columnElementInSchema = (Element)columnElements.item(columnIndex); |
136 |
0
|
String columnName = columnElementInSchema.getAttribute(NAME_ATTR); |
137 |
0
|
Element columnDescElement = null; |
138 |
0
|
NodeList columnDescElements = tableDescElem.getElementsByTagName(COLUMN_ELEMENT); |
139 |
0
|
for (int columnDescIndex = 0; columnDescIndex < columnDescElements.getLength(); columnDescIndex++) { |
140 |
0
|
Element element = (Element)columnDescElements.item(columnDescIndex); |
141 |
0
|
if (columnName.equals(element.getAttribute(NAME_ATTR))) { |
142 |
0
|
columnDescElement = element; |
143 |
0
|
break; |
144 |
|
} |
145 |
|
} |
146 |
0
|
if (columnDescElement != null) { |
147 |
0
|
String columnDescription = columnDescElement.getAttribute(DESCRIPTION_ATTR); |
148 |
0
|
String columnJavaName = columnDescElement.getAttribute(JAVA_NAME_ATTR); |
149 |
0
|
columnElementInSchema.setAttribute(DESCRIPTION_ATTR, columnDescription); |
150 |
0
|
columnElementInSchema.setAttribute(JAVA_NAME_ATTR, columnJavaName); |
151 |
|
} |
152 |
|
} |
153 |
|
} |
154 |
|
} |
155 |
0
|
return dbSchemaDoc; |
156 |
|
} |
157 |
|
|
158 |
|
|
159 |
|
|
160 |
|
|
161 |
|
@param |
162 |
|
|
163 |
|
|
|
|
| 0% |
Uncovered Elements: 17 (17) |
Complexity: 2 |
Complexity Density: 0.13 |
|
164 |
0
|
public void writeXMLToFile(Document newXMLDocument) throws Exception {... |
165 |
0
|
File dbSchema = this.getDbSchema(); |
166 |
0
|
TransformerFactory tFactory = TransformerFactory.newInstance(); |
167 |
|
|
168 |
0
|
Transformer transformer = tFactory.newTransformer(); |
169 |
0
|
transformer.setOutputProperty(OutputKeys.DOCTYPE_SYSTEM, |
170 |
|
"database.dtd"); |
171 |
0
|
DOMSource domSource = new DOMSource(newXMLDocument); |
172 |
0
|
StringWriter writer = new StringWriter(); |
173 |
0
|
Result result = new StreamResult(writer); |
174 |
0
|
transformer.transform(domSource, result); |
175 |
0
|
FileWriter fileWriter = new FileWriter(dbSchema); |
176 |
|
|
177 |
0
|
if (dbSchema.exists()) { |
178 |
0
|
StringBuffer bufferedWriter = new StringBuffer(writer |
179 |
|
.toString()); |
180 |
0
|
fileWriter.write(bufferedWriter.toString()); |
181 |
0
|
fileWriter.close(); |
182 |
0
|
System.out.println("The data has been written"); |
183 |
|
} else { |
184 |
0
|
System.out.println("This file is not exist"); |
185 |
|
} |
186 |
|
} |
187 |
|
|
|
|
| 0% |
Uncovered Elements: 11 (11) |
Complexity: 3 |
Complexity Density: 0.33 |
|
188 |
0
|
@Override... |
189 |
|
public void execute() throws BuildException { |
190 |
0
|
setDbSchema(dbSchemaString); |
191 |
0
|
setSchemaWithDesc(schemaWithDescString); |
192 |
0
|
if (!schemaWithDesc.exists()) { |
193 |
0
|
System.out.println("no schema file with description can be located"); |
194 |
0
|
return; |
195 |
|
} |
196 |
0
|
try { |
197 |
0
|
mergeSchemas(schemaWithDesc, dbSchema); |
198 |
0
|
writeXMLToFile(dbSchemaDoc); |
199 |
|
} catch (Exception e) { |
200 |
0
|
throw new BuildException(e); |
201 |
|
} |
202 |
|
} |
203 |
|
|
204 |
|
|
205 |
|
|
206 |
|
} |