代码完善

src/main/java/com/dashboard/aws/lambda/Constants.java

@@ -0,0 +1,9 @@
+package com.dashboard.aws.lambda;
+
+import java.util.List;
+
+public class Constants {
+
+    public static final List<String> tables = List.of("dashboard_record_accumulate", "dashboard_record_measure");
+
+}

src/main/java/com/dashboard/aws/lambda/handler/MySQLToS3Handler.java

@@ -2,6 +2,7 @@ package com.dashboard.aws.lambda.handler;
 
 import com.amazonaws.services.lambda.runtime.Context;
 import com.amazonaws.services.lambda.runtime.RequestHandler;
+import com.dashboard.aws.lambda.Constants;
 import com.dashboard.aws.lambda.service.MySQLService;
 import com.dashboard.aws.lambda.service.S3Service;
 import com.dashboard.aws.lambda.util.CsvUtil;
@@ -36,7 +37,7 @@ public class MySQLToS3Handler implements RequestHandler<Map<String, Object>, Str
             for (Long companyId : companyIds) {
                 String schema = "data_center_dongjian_" + companyId;
 
-                for (String table : List.of("dashboard_record_accumulate", "dashboard_record_measure")) {
+                for (String table : Constants.tables) {
                     // 查询旧数据
                     List<Map<String, Object>> rows = mysqlService.queryOldData(schema, table, startTs, endTs);
                     if (rows.isEmpty()) {

src/main/java/com/dashboard/aws/lambda/handler/S3ToMySQLHandler.java

@@ -2,6 +2,7 @@ package com.dashboard.aws.lambda.handler;
 
 import com.amazonaws.services.lambda.runtime.Context;
 import com.amazonaws.services.lambda.runtime.RequestHandler;
+import com.dashboard.aws.lambda.Constants;
 import com.dashboard.aws.lambda.service.MySQLService;
 import com.dashboard.aws.lambda.service.S3Service;
 import com.dashboard.aws.lambda.util.CsvUtil;
@@ -34,7 +35,7 @@ public class S3ToMySQLHandler implements RequestHandler<Map<String, Object>, Str
             List<Long> companyIds = mysqlService.getActiveCompanyIds();
             logger.info("company id list: {}", companyIds);
 
-            List<String> tables = List.of("dashboard_record_accumulate", "dashboard_record_measure");
+            List<String> tables = Constants.tables;
 
             for (Long companyId : companyIds) {
                 String schema = "data_center_dongjian_" + companyId;