Skip to content

Commit d278434

Browse files
committed
Allow configuration to be serialized
1 parent 7a63a0b commit d278434

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

sdks/java/io/amazon-web-services2/src/main/java/org/apache/beam/sdk/io/aws2/kinesis/KinesisTransformRegistrar.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ private abstract static class CrossLanguageConfiguration {
6161
String streamName;
6262
String awsAccessKey;
6363
String awsSecretKey;
64-
Region region;
64+
String region;
6565
@Nullable String serviceEndpoint;
6666

6767
public void setStreamName(String streamName) {
@@ -77,7 +77,7 @@ public void setAwsSecretKey(String awsSecretKey) {
7777
}
7878

7979
public void setRegion(String region) {
80-
this.region = Region.of(region);
80+
this.region = region;
8181
}
8282

8383
public void setServiceEndpoint(@Nullable String serviceEndpoint) {
@@ -120,7 +120,7 @@ public PTransform<PCollection<byte[]>, KinesisIO.Write.Result> buildExternal(
120120
.withClientConfiguration(
121121
ClientConfiguration.builder()
122122
.credentialsProvider(provider)
123-
.region(configuration.region)
123+
.region(Regions.fromName(configuration.region))
124124
.endpoint(endpoint)
125125
.build())
126126
.withPartitioner(p -> configuration.partitionKey)
@@ -230,7 +230,7 @@ public PTransform<PBegin, PCollection<byte[]>> buildExternal(
230230
.withClientConfiguration(
231231
ClientConfiguration.builder()
232232
.credentialsProvider(provider)
233-
.region(configuration.region)
233+
.region(Regions.fromName(configuration.region))
234234
.endpoint(endpoint)
235235
.build());
236236

0 commit comments

Comments
 (0)