动态支持MySQL/Oracle

config/application-oracle.yml

@@ -3,4 +3,4 @@ database:
   user: C##FMMP
   password: vcare~1(^_^)
 export:
-  folder: ./
+  folder: ./logs/

config/application.yml

@@ -1,3 +1,3 @@
 profile:
-#  active: mysql
-   active: oracle
+  active: mysql
+#   active: oracle

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

@@ -13,10 +13,9 @@ public interface ResultHandlerService {
      * ins
      *
      * @param dataList
-     * @param cls
      * @throws Exception
      */
-    <T> void resultHandler(List<T> dataList, Class<T> cls) throws Exception;
+    <T> void resultHandler(List<T> dataList) throws Exception;
 
 
 }

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

@@ -78,16 +78,19 @@ public class CSVResultHandlerServiceImpl implements ResultHandlerService {
     }
 
     @Override
-    public <T> void resultHandler(List<T> dataList, Class<T> cls) throws Exception {
+    public <T> void resultHandler(List<T> dataList) throws Exception {
         checkFile();
 
         logger.debug("start write to csv file size = {}", dataList.size());
 
         long startTime = System.currentTimeMillis();
 
+
+        Class<?> cls = Object.class;
+
         for (T t : dataList) {
+            cls = t.getClass();
             Object[] row;
-
             if (t.getClass().isArray()) {
                 row = (Object[]) t;
             } else if (List.class.isAssignableFrom(t.getClass())) {
@@ -95,7 +98,8 @@ public class CSVResultHandlerServiceImpl implements ResultHandlerService {
                 row = ((List) t).toArray();
             } else {
                 // auto mapper
-                row = resultMapper(t, cls);
+                //noinspection unchecked
+                row = resultMapper(t, (Class<T>) t.getClass());
             }
             Object[] handlerRow = toCsvContent(row);
             writer.printRecord(handlerRow);

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

@@ -29,7 +29,7 @@ public class MySQLResultHandlerServiceImpl implements ResultHandlerService {
 
 
     @Override
-    public <T> void resultHandler(List<T> dataList, Class<T> cls) throws Exception {
+    public <T> void resultHandler(List<T> dataList) throws Exception {
         logger.debug("start write to mysql size = {}, on = {}", dataList.size(), LocalDateTime.now());
         long startTime = System.currentTimeMillis();
         final Connection connection = dataSource.getConnection();

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

@@ -59,7 +59,7 @@ public class SQLStarter {
 
         do {
             List<T> result = queryService.queryByPage(table, page, this.pageCount, cls);
-            resultHandler.resultHandler(result, cls);
+            resultHandler.resultHandler(result);
             currentSize = result.size();
 
             page++;