添加TimerService

src/main/java/com/vcarecity/cvs/FileExporterApp.java

@@ -8,6 +8,7 @@ import com.vcarecity.cvs.module.SQLModule;
 import com.vcarecity.cvs.properties.AppProperties;
 import com.vcarecity.cvs.properties.DbProperties;
 import com.vcarecity.cvs.starter.SQLStarter;
+import com.vcarecity.cvs.type.ResultHandlerEnum;
 import me.yuxiaoyao.config.loader.ConfigLoader;
 
 import java.util.Map;
@@ -39,35 +40,37 @@ public class FileExporterApp {
         boolean isMysql;
         Object properties;
 
-        if (isImportToMysql(args)) {
+        final ResultHandlerEnum type = isImportToMysql(args);
+
+        if (type == ResultHandlerEnum.MYSQL || type == ResultHandlerEnum.MYSQL_PS) {
             properties = "mysql";
-            isMysql = true;
-        } else {
 
+        } else {
             String basePath = table;
             final AppProperties.Export export = appProperties.getExport();
             if (export != null) {
                 basePath = export.getFolder() + table;
             }
             properties = basePath;
-            isMysql = false;
         }
 
 
-        injector = Guice.createInjector(new PropertiesModule(appProperties), new SQLModule(isMysql));
+        injector = Guice.createInjector(new PropertiesModule(appProperties), new SQLModule(type));
 
         final SQLStarter instance = injector.getInstance(SQLStarter.class);
         instance.exportDataAndConvert(table, properties, tableClass);
     }
 
 
-    private static boolean isImportToMysql(String[] args) {
+    private static ResultHandlerEnum isImportToMysql(String[] args) {
+        String type = null;
         for (String arg : args) {
             if (arg.startsWith("--type=")) {
-                return !"csv".equalsIgnoreCase(arg.substring("--type=".length()));
+                type = arg.substring("--type=".length()).trim();
+                break;
             }
         }
-        return true;
+        return ResultHandlerEnum.fromType(type);
     }
 
 

src/main/java/com/vcarecity/cvs/module/SQLModule.java

@@ -13,6 +13,7 @@ import com.vcarecity.cvs.properties.DbProperties;
 import com.vcarecity.cvs.service.ResultHandlerService;
 import com.vcarecity.cvs.service.SQLQueryService;
 import com.vcarecity.cvs.service.impl.*;
+import com.vcarecity.cvs.type.ResultHandlerEnum;
 import com.vcarecity.cvs.util.CreateDatasource;
 import com.zaxxer.hikari.HikariDataSource;
 
@@ -21,10 +22,10 @@ import com.zaxxer.hikari.HikariDataSource;
  */
 
 public class SQLModule extends AbstractModule {
-    private final boolean isMysql;
+    private final ResultHandlerEnum resultHandlerEnum;
 
-    public SQLModule(boolean isMysql) {
-        this.isMysql = isMysql;
+    public SQLModule(ResultHandlerEnum resultHandlerEnum) {
+        this.resultHandlerEnum = resultHandlerEnum;
     }
 
     @Override
@@ -33,11 +34,21 @@ public class SQLModule extends AbstractModule {
         bind(ExportEventListener.class).to(MySQLImportEvent.class);
 
         Class<? extends ResultHandlerService> resultClass;
-        if (isMysql) {
-             resultClass = MySQLResultStatHandlerServiceImpl.class;
-            //resultClass = MySQLResultHandlerServiceImpl.class;
-        } else {
-            resultClass = CSVResultHandlerServiceImpl.class;
+
+        switch (resultHandlerEnum) {
+            case CSV:
+                resultClass = CSVResultHandlerServiceImpl.class;
+                break;
+            case MYSQL:
+                resultClass = MySQLResultStatHandlerServiceImpl.class;
+                break;
+            case MYSQL_PS:
+                resultClass = MySQLResultHandlerServiceImpl.class;
+                break;
+            case DEFAULT:
+            default:
+                resultClass = LogResultHandlerServiceImpl.class;
+                break;
         }
 
         install(new FactoryModuleBuilder()

src/main/java/com/vcarecity/cvs/service/impl/LogResultHandlerServiceImpl.java

@@ -0,0 +1,18 @@
+package com.vcarecity.cvs.service.impl;
+
+import com.vcarecity.cvs.service.ResultHandlerService;
+import lombok.extern.slf4j.Slf4j;
+
+import java.util.List;
+
+/**
+ * @author Kerry on 19/12/03
+ */
+
+@Slf4j
+public class LogResultHandlerServiceImpl implements ResultHandlerService {
+    @Override
+    public <T> void resultHandler(String table, String[] header, List<T> dataList) throws Exception {
+        logger.debug("LogResultHandlerServiceImpl.resultHandler. table = {}, header = {}, dataList = {}", table, header, dataList);
+    }
+}

src/main/java/com/vcarecity/cvs/starter/SQLStarter.java

@@ -62,6 +62,8 @@ public class SQLStarter {
         int currentSize;
         int page = 1;
 
+        long queryCount = 0;
+
         do {
             // query
             List<T> result = queryService.queryByPage(table, page, this.pageCount, cls);
@@ -69,6 +71,9 @@ public class SQLStarter {
             // convert or save
             resultHandler.resultHandler(table, headers, result);
             currentSize = result.size();
+            queryCount += currentSize;
+
+            logger.info("current query count is {}", queryCount);
 
             page++;
 

src/main/java/com/vcarecity/cvs/type/ResultHandlerEnum.java

@@ -0,0 +1,33 @@
+package com.vcarecity.cvs.type;
+
+/**
+ * @author Kerry on 19/12/03
+ */
+
+public enum ResultHandlerEnum {
+    CSV("csv"),
+    MYSQL("mysql"),
+    MYSQL_PS("mysqlps"),
+    DEFAULT("");
+    private String type;
+
+    ResultHandlerEnum(String type) {
+        this.type = type;
+    }
+
+    public String getType() {
+        return type;
+    }
+
+    public static ResultHandlerEnum fromType(String type) {
+        if (type == null) {
+            return ResultHandlerEnum.DEFAULT;
+        }
+        for (ResultHandlerEnum value : values()) {
+            if (value.getType().equalsIgnoreCase(type)) {
+                return value;
+            }
+        }
+        return ResultHandlerEnum.DEFAULT;
+    }
+}