diff --git a/java/app/src/main/java/kickstarter/Common.java b/java/app/src/main/java/kickstarter/Common.java index 7c84c18..07b4b37 100644 --- a/java/app/src/main/java/kickstarter/Common.java +++ b/java/app/src/main/java/kickstarter/Common.java @@ -17,7 +17,8 @@ public class Common { public static final String ARG_SERVICE_ACCOUNT_USER = "--service-account-user"; - + public static final String ARG_AWS_REGION = "--aws-region"; + /** * @return returns a new instance of the Jackson ObjectMapper with the JavaTimeModule diff --git a/java/app/src/main/java/kickstarter/DataGeneratorApp.java b/java/app/src/main/java/kickstarter/DataGeneratorApp.java index 84937c8..0111510 100644 --- a/java/app/src/main/java/kickstarter/DataGeneratorApp.java +++ b/java/app/src/main/java/kickstarter/DataGeneratorApp.java @@ -54,6 +54,7 @@ public static void main(String[] args) throws Exception { * Retrieve the value(s) from the command line argument(s) */ String serviceAccountUser = Common.getAppArgumentValue(args, Common.ARG_SERVICE_ACCOUNT_USER); + String awsRegion = Common.getAppArgumentValue(args, Common.ARG_AWS_REGION); // --- Create a blank Flink execution environment (a.k.a. the Flink job graph -- the DAG) StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment(); @@ -208,7 +209,7 @@ public static void main(String[] args) throws Exception { catalogProperties.put("catalog-impl", catalogImpl); catalogProperties.put("io-impl", ioImpl); catalogProperties.put("glue.skip-archive", "true"); - catalogProperties.put("glue.region", "us-east-1"); + catalogProperties.put("glue.region", awsRegion); // --- Use the CatalogLoader since an external metastore is used (AWS Glue Catalog) CatalogLoader catalogLoader = CatalogLoader.custom(catalogName, catalogProperties, new Configuration(false), catalogImpl);