-
Notifications
You must be signed in to change notification settings - Fork 77
Expand file tree
/
Copy pathtestDataflow.java
More file actions
72 lines (66 loc) · 2.33 KB
/
Copy pathtestDataflow.java
File metadata and controls
72 lines (66 loc) · 2.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package dataflow;
import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import common.gspCommon;
import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.TBaseType;
import gudusoft.gsqlparser.dlineage.DataFlowAnalyzer;
import gudusoft.gsqlparser.util.SQLUtil;
import junit.framework.TestCase;
public class testDataflow extends TestCase {
public void test1() {
File privateDir = new File(gspCommon.BASE_SQL_DIR, "private/dataflow");
File publicDir = new File(gspCommon.BASE_SQL_DIR, "public/dataflow");
List<File> sqlDirList = new ArrayList<>();
if(privateDir.exists() && privateDir.listFiles()!=null){
sqlDirList.addAll(Arrays.asList(privateDir.listFiles()));
}
if(publicDir.exists() && publicDir.listFiles()!=null){
sqlDirList.addAll(Arrays.asList(publicDir.listFiles()));
}
File[] sqlDirs = sqlDirList.toArray(new File[0]);
for (int i = 0; i < sqlDirs.length; i++) {
File vendorDir = sqlDirs[i];
if(vendorDir.getName().startsWith("."))
continue;
EDbVendor vendor = EDbVendor.valueOf("dbv" + vendorDir.getName());
File[] sqlfiles = vendorDir.listFiles(new FileFilter() {
public boolean accept(File pathname) {
return pathname.getName().endsWith(".sql");
}
});
for (int j = 0; j < sqlfiles.length; j++) {
DataFlowAnalyzer analyzer = new DataFlowAnalyzer(sqlfiles[j], vendor, false);
analyzer.setShowConstantTable(true);
analyzer.getOption().setTraceProcedure(true);
String dataflow = analyzer.generateDataFlow();
File dataflowFile = new File(sqlfiles[j].getParentFile(),
sqlfiles[j].getName().replace(".sql", ".xml"));
if (!dataflowFile.exists())
{
try {
SQLUtil.writeToFile(dataflowFile, dataflow.trim());
} catch (IOException e) {
e.printStackTrace();
}
}
if(!dataflow.equals( SQLUtil.readFile(dataflowFile))){
File dataflowFile1 = new File(sqlfiles[j].getParentFile(),
sqlfiles[j].getName().replace(".sql", "1.xml"));
if (!dataflowFile1.exists()) {
try {
SQLUtil.writeToFile(dataflowFile1, dataflow);
} catch (IOException e) {
e.printStackTrace();
}
}
}
assertTrue(TBaseType.compareStringBuilderToFile(new StringBuilder(dataflow), dataflowFile.getAbsolutePath()));
}
}
}
}