添加TimerService

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

@@ -34,8 +34,8 @@ public class SQLModule extends AbstractModule {
 
         Class<? extends ResultHandlerService> resultClass;
         if (isMysql) {
-            // resultClass = MySQLResultStatHandlerServiceImpl.class;
-            resultClass = MySQLResultHandlerServiceImpl.class;
+             resultClass = MySQLResultStatHandlerServiceImpl.class;
+            //resultClass = MySQLResultHandlerServiceImpl.class;
         } else {
             resultClass = CSVResultHandlerServiceImpl.class;
         }

src/main/java/com/vcarecity/cvs/service/TimerService.java

@@ -0,0 +1,16 @@
+package com.vcarecity.cvs.service;
+
+/**
+ * @author Kerry on 19/12/03
+ */
+
+public interface TimerService {
+
+    /**
+     * 获取时间
+     *
+     * @return
+     */
+    long getUsageTime();
+
+}

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

@@ -2,6 +2,7 @@ package com.vcarecity.cvs.service.impl;
 
 import com.vcarecity.cvs.core.AutoMapper;
 import com.vcarecity.cvs.service.SQLQueryService;
+import com.vcarecity.cvs.service.TimerService;
 import com.zaxxer.hikari.HikariDataSource;
 import lombok.AllArgsConstructor;
 import lombok.Data;
@@ -12,6 +13,7 @@ import java.sql.*;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Set;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.stream.Collectors;
 
 import static com.vcarecity.cvs.FileExporterApp.TABLE_COLUMN;
@@ -21,10 +23,11 @@ import static com.vcarecity.cvs.FileExporterApp.TABLE_COLUMN;
  */
 
 @Slf4j
-public abstract class AbstractSQLQueryService implements SQLQueryService {
+public abstract class AbstractSQLQueryService implements SQLQueryService, TimerService {
 
     private boolean isFirst = true;
 
+    private AtomicLong readTime = new AtomicLong(0L);
 
     protected final HikariDataSource dataSource;
 
@@ -48,12 +51,16 @@ public abstract class AbstractSQLQueryService implements SQLQueryService {
 
         List<T> list = resultMapper(table, resultSet, clazz);
 
-        logger.info("QUERY SUCCESS. size = {}. usageTime = {} /ms", list.size(), (System.currentTimeMillis() - startTime));
-
         resultSet.close();
         preparedStatement.close();
         connection.close();
 
+        final long curReadTime = System.currentTimeMillis() - startTime;
+
+        readTime.addAndGet(curReadTime);
+
+        logger.info("QUERY SUCCESS. size = {}. usageTime = {} /ms", list.size(), curReadTime);
+
         return list;
     }
 
@@ -154,4 +161,8 @@ public abstract class AbstractSQLQueryService implements SQLQueryService {
         return list;
     }
 
+    @Override
+    public long getUsageTime() {
+        return this.readTime.longValue();
+    }
 }

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

@@ -5,6 +5,7 @@ import com.google.inject.assistedinject.Assisted;
 import com.vcarecity.cvs.core.ColumnUpdateMapper;
 import com.vcarecity.cvs.properties.AppProperties;
 import com.vcarecity.cvs.service.ResultHandlerService;
+import com.vcarecity.cvs.service.TimerService;
 import com.vcarecity.cvs.util.CreateDatasource;
 import com.zaxxer.hikari.HikariDataSource;
 import lombok.extern.slf4j.Slf4j;
@@ -13,6 +14,7 @@ import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.util.Arrays;
 import java.util.List;
+import java.util.concurrent.atomic.AtomicLong;
 
 /**
  * @author Kerry on 19/11/21
@@ -20,14 +22,14 @@ import java.util.List;
 
 
 @Slf4j
-public class MySQLResultHandlerServiceImpl implements ResultHandlerService {
+public class MySQLResultHandlerServiceImpl implements ResultHandlerService, TimerService {
 
-    private final String sql;
     private final HikariDataSource dataSource;
 
+    private AtomicLong writeTime = new AtomicLong(0L);
+
     @Inject
     public MySQLResultHandlerServiceImpl(AppProperties properties, @Assisted Object sql) {
-        this.sql = (String) sql;
         this.dataSource = CreateDatasource.createDataSource(properties.getToDb());
     }
 
@@ -37,7 +39,6 @@ public class MySQLResultHandlerServiceImpl implements ResultHandlerService {
 
         final String sql = getSql(table, header);
 
-
         long startTime = System.currentTimeMillis();
 
         logger.debug("{},{}", dataList.size(), sql);
@@ -72,7 +73,12 @@ public class MySQLResultHandlerServiceImpl implements ResultHandlerService {
         } finally {
             connection.close();
         }
-        logger.info("WRITE SUCCESS to mysql usage time {} /ms, size = {}", System.currentTimeMillis() - startTime, dataList.size());
+
+        final long curUseTime = System.currentTimeMillis() - startTime;
+
+        writeTime.addAndGet(curUseTime);
+
+        logger.info("WRITE SUCCESS to mysql usage time {} /ms, size = {}", curUseTime, dataList.size());
     }
 
 
@@ -88,4 +94,9 @@ public class MySQLResultHandlerServiceImpl implements ResultHandlerService {
         return "INSERT INTO " + table + "(" + column + ") VALUES(" + join + ")";
 
     }
+
+    @Override
+    public long getUsageTime() {
+        return this.writeTime.longValue();
+    }
 }

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

@@ -5,6 +5,7 @@ import com.google.inject.assistedinject.Assisted;
 import com.vcarecity.cvs.core.ColumnUpdateMapper;
 import com.vcarecity.cvs.properties.AppProperties;
 import com.vcarecity.cvs.service.ResultHandlerService;
+import com.vcarecity.cvs.service.TimerService;
 import com.vcarecity.cvs.util.CreateDatasource;
 import com.zaxxer.hikari.HikariDataSource;
 import lombok.extern.slf4j.Slf4j;
@@ -14,6 +15,7 @@ import java.sql.Date;
 import java.sql.Statement;
 import java.sql.Timestamp;
 import java.util.List;
+import java.util.concurrent.atomic.AtomicLong;
 
 /**
  * @author Kerry on 19/11/21
@@ -21,26 +23,26 @@ import java.util.List;
 
 
 @Slf4j
-public class MySQLResultStatHandlerServiceImpl implements ResultHandlerService {
+public class MySQLResultStatHandlerServiceImpl implements ResultHandlerService, TimerService {
 
-
-    private final String sql;
     private final HikariDataSource dataSource;
 
+    private AtomicLong writeTime = new AtomicLong(0L);
+
     @Inject
     public MySQLResultStatHandlerServiceImpl(AppProperties properties, @Assisted Object sql) {
-        this.sql = (String) sql;
         this.dataSource = CreateDatasource.createDataSource(properties.getToDb());
     }
 
-
     @Override
     public <T> void resultHandler(String table, String[] header, List<T> dataList) throws Exception {
 
-
+        if (dataList == null || dataList.isEmpty()) {
+            return;
+        }
         long startTime = System.currentTimeMillis();
 
-        StringBuffer sb = new StringBuffer();
+        StringBuilder sb = new StringBuilder();
 
         final String[] newHeaders = ColumnUpdateMapper.updateColumnName(table, header);
         final String column = String.join(",", newHeaders);
@@ -71,19 +73,27 @@ public class MySQLResultStatHandlerServiceImpl implements ResultHandlerService {
                 sb.append("),");
             }
         }
-        final StringBuffer stringBuffer = sb.deleteCharAt(sb.length() - 1);
+        String execSql = sb.deleteCharAt(sb.length() - 1).toString();
 
-        // logger.debug("{}", stringBuffer.toString());
+        // logger.debug("{}", execSql);
 
         final Connection connection = dataSource.getConnection();
 
         final Statement statement = connection.createStatement();
-        statement.execute(stringBuffer.toString());
+        statement.execute(execSql);
         statement.close();
         connection.close();
 
-        logger.info("WRITE SUCCESS to mysql usage time {} /ms, size = {}", System.currentTimeMillis() - startTime, dataList.size());
+        final long curUseTime = System.currentTimeMillis() - startTime;
+
+        writeTime.addAndGet(curUseTime);
+
+        logger.info("WRITE SUCCESS to mysql usage time {} /ms, size = {}", curUseTime, dataList.size());
     }
 
 
+    @Override
+    public long getUsageTime() {
+        return this.writeTime.longValue();
+    }
 }

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

@@ -4,6 +4,7 @@ import com.vcarecity.cvs.core.ColumnUpdateMapper;
 import com.vcarecity.cvs.factory.ResultHandlerFactory;
 import com.vcarecity.cvs.service.ResultHandlerService;
 import com.vcarecity.cvs.service.SQLQueryService;
+import com.vcarecity.cvs.service.TimerService;
 import com.vcarecity.cvs.service.impl.CSVResultHandlerServiceImpl;
 import lombok.extern.slf4j.Slf4j;
 
@@ -77,13 +78,13 @@ public class SQLStarter {
             ((CSVResultHandlerServiceImpl) resultHandler).close();
         }
 
-
         final String[] headers = TABLE_COLUMN.get(table);
         final String[] mysqlHeader = ColumnUpdateMapper.updateColumnName(table, headers);
 
         logger.info("mysql header: {}", String.join(",", mysqlHeader));
-
-        logger.info("stop query database...{} /ms.", (System.currentTimeMillis() - startTime));
+        logger.info("query sql usage time {} /ms", ((TimerService) queryService).getUsageTime());
+        logger.info("write sql usage time {} /ms", ((TimerService) resultHandler).getUsageTime());
+        logger.info("all usage time {} /ms.", (System.currentTimeMillis() - startTime));
     }
 
 }