1 /**
 2  * <a> 读xml的管 理类 </a>
 3  *
 4  * @author: feiyuanxing@baidu.com
 5  * @time 2016/12/27
 6  */
 7 public class ReadXmlManger {
 8     private static final Logger logger = LoggerFactory.getLogger(ReadXmlManger.class);
 9 
10     public static String insertTempFile;
11     public static String updateTempFile;
12 
13     /**
14      * 各种字符型的入库 还没有做
15      *
16      * @param args
17      */
18     public static void main(String[] args) {
19         Long starTime = System.currentTimeMillis();
20         String needAnalysisFileDir = "D:\\creditcode\\xml\\";
21         insertTempFile = "D:\\creditcode\\tmp\\tmpInsert.sql";
22         updateTempFile = "D:\\\\creditcode\\\\tmp\\\\tmpDelete.sql";
23 
24         if (args.length == 3) {
25             needAnalysisFileDir = args[0];
26             insertTempFile = args[1];
27             updateTempFile = args[2];
28             System.out.println("argument is right:" + args.length + Arrays.toString(args));
29         }
30         // else {
31         //            System.out.println("argument is not right:" + args.length + Arrays.toString(args) + " ，over...");
32         //            System.exit(1);
33         //            return;
34         //
35         //        }
36         // 对各种连接做初始化
37         ImportDataUtil.getConnet();
38         // 加载文件路径到队列中
39         ImportDataUtil.getXmlDirToQueue(needAnalysisFileDir);
40         // 执行将数据中的md5 列加载得到内存中
41         ImportDataUtil.getSourecXmlList();
42 
43         // 创建初始化流
44         ImportDataUtil.createWriter(insertTempFile, updateTempFile);
45         logger.info("begin compare...");
46         // 读xml到队列中
47         XmlData.pool.execute(new ReadXmlThread(CreditCode.class));
48         // 从队列中取值比较完后写到txt中
49         XmlData.pool.execute(new CompareThread(CreditCode.class));
50         // 等待所有子线程执行完
51         while (true) {
52             if (XmlData.pool.isShutdown()) {
53                 // 将txt中的数据load到数据库中
54                 loadData();
55                 Long endTime = System.currentTimeMillis();
56                 logger.info("compare over ,all read " + XmlData.atomicIntegerSC.get() + " ;");
57                 logger.info("Import the data complete, takes " + (endTime - starTime) / 1000 + " seconds;");
58                 return;
59             }
60         }
61     }
62 
63     // 都执行完执行
64     public static void loadData() {
65         // 将最后的新增加数据load到source文件中
66         ImportDataUtil.dumpInsertBuilder2File();
67 
68         // 将txt文件load到数据中  false未启用
69         CreditCode creditCode = null;
70         logger.info("Start the load data for the last time....");
71         logger.info("The mission were increased： " + XmlData.currentAddNum.get() + " tiao data");
72         ImportDataUtil.loadFile(insertTempFile, CreditCode.class);
73 
74         // 清空md5表
75         ImportDataUtil.clearTmpTable();
76         // 将集合中剩余的md5写到文件中
77         logger.info("Began to set in the rest of the md5 written to a file ");
78         ImportDataUtil.dumpUpdateBuilder2File();
79         // load进数据 and 级联删除数据
80         logger.info("Delete data load and start cascading delete data...");
81         ImportDataUtil.sourceFile(updateTempFile);
82         XmlData.context.close();
83 
84     }
85 
86 }

1 BeanListHandler
2 CompareThread
3 ImportDataUtil
4 ReadXmlThread
5 WriteFileUtil
6 XmlColumn
7 XmlData
8 XmlSAXParse