diff --git a/src/main/java/com/dashboard/aws/lambda/Constants.java b/src/main/java/com/dashboard/aws/lambda/Constants.java new file mode 100644 index 0000000..21cd70d --- /dev/null +++ b/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 tables = List.of("dashboard_record_accumulate", "dashboard_record_measure"); + +} diff --git a/src/main/java/com/dashboard/aws/lambda/handler/MySQLToS3Handler.java b/src/main/java/com/dashboard/aws/lambda/handler/MySQLToS3Handler.java index 6ba44bb..be201a2 100644 --- a/src/main/java/com/dashboard/aws/lambda/handler/MySQLToS3Handler.java +++ b/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, 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> rows = mysqlService.queryOldData(schema, table, startTs, endTs); if (rows.isEmpty()) { diff --git a/src/main/java/com/dashboard/aws/lambda/handler/S3ToMySQLHandler.java b/src/main/java/com/dashboard/aws/lambda/handler/S3ToMySQLHandler.java index f407fba..8199fb8 100644 --- a/src/main/java/com/dashboard/aws/lambda/handler/S3ToMySQLHandler.java +++ b/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, Str List companyIds = mysqlService.getActiveCompanyIds(); logger.info("company id list: {}", companyIds); - List tables = List.of("dashboard_record_accumulate", "dashboard_record_measure"); + List tables = Constants.tables; for (Long companyId : companyIds) { String schema = "data_center_dongjian_" + companyId;