1 |
|
package org.apache.torque.mojo; |
2 |
|
|
3 |
|
import java.io.File; |
4 |
|
import java.io.FileInputStream; |
5 |
|
import java.io.IOException; |
6 |
|
import java.io.InputStream; |
7 |
|
import java.io.OutputStream; |
8 |
|
import java.util.HashSet; |
9 |
|
import java.util.List; |
10 |
|
import java.util.Properties; |
11 |
|
import java.util.Set; |
12 |
|
import java.util.TreeSet; |
13 |
|
|
14 |
|
import org.apache.commons.io.FileUtils; |
15 |
|
import org.apache.commons.io.IOUtils; |
16 |
|
import org.apache.maven.plugin.MojoExecutionException; |
17 |
|
import org.apache.maven.plugin.MojoFailureException; |
18 |
|
import org.apache.tools.ant.DirectoryScanner; |
19 |
|
import org.apache.torque.engine.database.model.Database; |
20 |
|
import org.apache.torque.engine.database.model.Table; |
21 |
|
import org.kuali.core.db.torque.SetUtils; |
22 |
|
import org.kuali.core.db.torque.Utils; |
23 |
|
|
24 |
|
|
25 |
|
|
26 |
|
|
27 |
|
|
28 |
|
|
29 |
|
|
30 |
|
|
31 |
|
|
32 |
|
|
33 |
|
|
34 |
|
|
|
|
| 0% |
Uncovered Elements: 126 (126) |
Complexity: 30 |
Complexity Density: 0.32 |
|
35 |
|
public class IdentifyInvalidDataFiles extends BaseMojo { |
36 |
|
private static final String FS = System.getProperty("file.separator"); |
37 |
|
|
38 |
|
|
39 |
|
|
40 |
|
|
41 |
|
|
42 |
|
private String extension; |
43 |
|
|
44 |
|
|
45 |
|
|
46 |
|
|
47 |
|
|
48 |
|
private File dataDir; |
49 |
|
|
50 |
|
|
51 |
|
|
52 |
|
|
53 |
|
private String dataDirIncludes; |
54 |
|
|
55 |
|
|
56 |
|
|
57 |
|
|
58 |
|
private String dataDirExcludes; |
59 |
|
|
60 |
|
|
61 |
|
|
62 |
|
|
63 |
|
private String schemaXMLFile; |
64 |
|
|
65 |
|
|
66 |
|
|
67 |
|
|
68 |
|
private String targetDatabase; |
69 |
|
|
70 |
|
|
71 |
|
|
72 |
|
|
73 |
|
|
74 |
|
|
75 |
|
private File markedForRemoval; |
76 |
|
|
|
|
| 0% |
Uncovered Elements: 9 (9) |
Complexity: 2 |
Complexity Density: 0.29 |
|
77 |
0
|
protected Set<File> getExportedFiles() throws MojoExecutionException {... |
78 |
0
|
String filename = getProject().getBuild().getDirectory() + "/impex/exported-tables.txt"; |
79 |
0
|
File file = new File(filename); |
80 |
0
|
if (!file.exists()) { |
81 |
0
|
getLog().warn("Could not locate " + file.getAbsolutePath()); |
82 |
0
|
return new HashSet<File>(); |
83 |
|
} |
84 |
0
|
List<String> tablenames = getContents(file); |
85 |
0
|
return getExportedFiles(tablenames); |
86 |
|
} |
87 |
|
|
|
|
| 0% |
Uncovered Elements: 6 (6) |
Complexity: 1 |
Complexity Density: 0.17 |
|
88 |
0
|
protected Set<File> getExportedFiles(List<String> tablenames) {... |
89 |
0
|
Set<File> files = new TreeSet<File>(); |
90 |
0
|
for (String tablename : tablenames) { |
91 |
0
|
String filename = dataDir.getAbsolutePath() + FS + tablename + extension; |
92 |
0
|
File file = new File(filename); |
93 |
0
|
files.add(file); |
94 |
|
} |
95 |
0
|
return files; |
96 |
|
} |
97 |
|
|
|
|
| 0% |
Uncovered Elements: 6 (6) |
Complexity: 2 |
Complexity Density: 0.33 |
|
98 |
0
|
@SuppressWarnings("unchecked")... |
99 |
|
protected List<String> getContents(File file) throws MojoExecutionException { |
100 |
0
|
InputStream in = null; |
101 |
0
|
try { |
102 |
0
|
in = new FileInputStream(file); |
103 |
0
|
return IOUtils.readLines(in); |
104 |
|
} catch (IOException e) { |
105 |
0
|
throw new MojoExecutionException("Unexpected error", e); |
106 |
|
} finally { |
107 |
0
|
IOUtils.closeQuietly(in); |
108 |
|
} |
109 |
|
} |
110 |
|
|
|
|
| 0% |
Uncovered Elements: 35 (35) |
Complexity: 5 |
Complexity Density: 0.17 |
|
111 |
0
|
@Override... |
112 |
|
protected void executeMojo() throws MojoExecutionException, MojoFailureException { |
113 |
0
|
Utils utils = new Utils(); |
114 |
0
|
try { |
115 |
0
|
getLog().info("Examining " + dataDir.getAbsolutePath()); |
116 |
0
|
Database db = utils.getDatabase(schemaXMLFile, targetDatabase); |
117 |
0
|
DirectoryScanner ds = getDirectoryScanner(); |
118 |
0
|
Set<File> existing = getExistingFiles(ds); |
119 |
0
|
Set<File> exported = getExportedFiles(); |
120 |
0
|
Set<File> allowed = getDatabaseFiles(db); |
121 |
0
|
if (exported.size() > 0) { |
122 |
0
|
allowed = exported; |
123 |
|
} |
124 |
0
|
Set<File> invalid = SetUtils.difference(existing, allowed); |
125 |
0
|
getLog().info(existing.size() + " data files currently exist"); |
126 |
0
|
getLog().info(invalid.size() + " of those are invalid"); |
127 |
0
|
StringBuilder sb = new StringBuilder(); |
128 |
0
|
int count = 0; |
129 |
0
|
StringBuilder invalidFiles = new StringBuilder(); |
130 |
0
|
for (File file : invalid) { |
131 |
0
|
if (count != 0) { |
132 |
0
|
sb.append(","); |
133 |
|
} |
134 |
0
|
sb.append("**/src/main/impex/" + file.getName()); |
135 |
0
|
invalidFiles.append(file.getCanonicalPath() + "\n"); |
136 |
0
|
getLog().info("Marked for removal: " + file.getName()); |
137 |
0
|
count++; |
138 |
|
} |
139 |
0
|
Properties properties = getProject().getProperties(); |
140 |
0
|
properties.setProperty("impex.data.invalid", sb.toString()); |
141 |
0
|
if (count > 0) { |
142 |
0
|
properties.setProperty("impex.found.invalid", Boolean.TRUE.toString()); |
143 |
0
|
createFile(markedForRemoval, invalidFiles.toString()); |
144 |
|
} |
145 |
|
} catch (Exception e) { |
146 |
0
|
throw new MojoExecutionException("Error executing mojo", e); |
147 |
|
} |
148 |
|
} |
149 |
|
|
|
|
| 0% |
Uncovered Elements: 5 (5) |
Complexity: 1 |
Complexity Density: 0.2 |
|
150 |
0
|
protected void createFile(File file, String contents) throws IOException {... |
151 |
0
|
OutputStream out = null; |
152 |
0
|
try { |
153 |
0
|
out = FileUtils.openOutputStream(file); |
154 |
0
|
IOUtils.write(contents, out); |
155 |
|
} finally { |
156 |
0
|
IOUtils.closeQuietly(out); |
157 |
|
} |
158 |
|
} |
159 |
|
|
|
|
| 0% |
Uncovered Elements: 9 (9) |
Complexity: 1 |
Complexity Density: 0.11 |
|
160 |
0
|
protected Set<File> getDatabaseFiles(Database db) {... |
161 |
0
|
List<?> tables = db.getTables(); |
162 |
0
|
Set<File> files = new TreeSet<File>(); |
163 |
0
|
for (Object object : tables) { |
164 |
0
|
Table table = (Table) object; |
165 |
0
|
String tableName = table.getName(); |
166 |
0
|
String filename = dataDir.getAbsolutePath() + FS + tableName + extension; |
167 |
0
|
File file = new File(filename); |
168 |
0
|
files.add(file); |
169 |
|
} |
170 |
0
|
return files; |
171 |
|
} |
172 |
|
|
|
|
| 0% |
Uncovered Elements: 10 (10) |
Complexity: 2 |
Complexity Density: 0.25 |
|
173 |
0
|
protected Set<File> getExistingFiles(DirectoryScanner ds) {... |
174 |
0
|
ds.scan(); |
175 |
0
|
String[] relativeFilenames = ds.getIncludedFiles(); |
176 |
0
|
Set<File> files = new TreeSet<File>(); |
177 |
0
|
for (int i = 0; i < relativeFilenames.length; i++) { |
178 |
0
|
String filename = ds.getBasedir().getAbsolutePath() + FS + relativeFilenames[i]; |
179 |
0
|
File file = new File(filename); |
180 |
0
|
files.add(file); |
181 |
|
} |
182 |
0
|
return files; |
183 |
|
} |
184 |
|
|
|
|
| 0% |
Uncovered Elements: 5 (5) |
Complexity: 1 |
Complexity Density: 0.2 |
|
185 |
0
|
protected DirectoryScanner getDirectoryScanner() {... |
186 |
0
|
DirectoryScanner ds = new DirectoryScanner(); |
187 |
0
|
ds.setBasedir(dataDir); |
188 |
0
|
ds.setIncludes(new String[] { dataDirIncludes }); |
189 |
0
|
ds.setExcludes(new String[] { dataDirExcludes }); |
190 |
0
|
return ds; |
191 |
|
} |
192 |
|
|
|
|
| 0% |
Uncovered Elements: 4 (4) |
Complexity: 1 |
Complexity Density: 0.25 |
|
193 |
0
|
protected File getFile(Table table) {... |
194 |
0
|
String tableName = table.getName(); |
195 |
0
|
String filename = dataDir.getAbsolutePath() + FS + tableName + extension; |
196 |
0
|
File file = new File(filename); |
197 |
0
|
return file; |
198 |
|
} |
199 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
200 |
0
|
public File getDataDir() {... |
201 |
0
|
return dataDir; |
202 |
|
} |
203 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
204 |
0
|
public void setDataDir(File dataDir) {... |
205 |
0
|
this.dataDir = dataDir; |
206 |
|
} |
207 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
208 |
0
|
public String getDataDirIncludes() {... |
209 |
0
|
return dataDirIncludes; |
210 |
|
} |
211 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
212 |
0
|
public void setDataDirIncludes(String dataDirIncludes) {... |
213 |
0
|
this.dataDirIncludes = dataDirIncludes; |
214 |
|
} |
215 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
216 |
0
|
public String getDataDirExcludes() {... |
217 |
0
|
return dataDirExcludes; |
218 |
|
} |
219 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
220 |
0
|
public void setDataDirExcludes(String dataDirExcludes) {... |
221 |
0
|
this.dataDirExcludes = dataDirExcludes; |
222 |
|
} |
223 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
224 |
0
|
public String getSchemaXMLFile() {... |
225 |
0
|
return schemaXMLFile; |
226 |
|
} |
227 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
228 |
0
|
public void setSchemaXMLFile(String schemaXMLFile) {... |
229 |
0
|
this.schemaXMLFile = schemaXMLFile; |
230 |
|
} |
231 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
232 |
0
|
public String getExtension() {... |
233 |
0
|
return extension; |
234 |
|
} |
235 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
236 |
0
|
public void setExtension(String extension) {... |
237 |
0
|
this.extension = extension; |
238 |
|
} |
239 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
240 |
0
|
public String getTargetDatabase() {... |
241 |
0
|
return targetDatabase; |
242 |
|
} |
243 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
244 |
0
|
public void setTargetDatabase(String targetDatabase) {... |
245 |
0
|
this.targetDatabase = targetDatabase; |
246 |
|
} |
247 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
248 |
0
|
public File getMarkedForRemoval() {... |
249 |
0
|
return markedForRemoval; |
250 |
|
} |
251 |
|
|
|
|
| 0% |
Uncovered Elements: 1 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
252 |
0
|
public void setMarkedForRemoval(File markedForRemoval) {... |
253 |
0
|
this.markedForRemoval = markedForRemoval; |
254 |
|
} |
255 |
|
|
256 |
|
} |