Skip to content

Commit a9eed7d

Browse files
committed
Add serializer
1 parent 90c13e9 commit a9eed7d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,7 @@ public PTransform<PCollection<byte[]>, KinesisIO.Write.Result> buildExternal(
109109
AwsBasicCredentials creds =
110110
AwsBasicCredentials.create(configuration.awsAccessKey, configuration.awsSecretKey);
111111
StaticCredentialsProvider provider = StaticCredentialsProvider.create(creds);
112+
SerializableFunction<byte[], byte[]> serializer = v -> v;
112113
KinesisIO.Write<byte[]> writeTransform =
113114
KinesisIO.<byte[]>write()
114115
.withStreamName(configuration.streamName)
@@ -118,7 +119,8 @@ public PTransform<PCollection<byte[]>, KinesisIO.Write.Result> buildExternal(
118119
.region(configuration.region)
119120
.endpoint(configuration.serviceEndpoint)
120121
.build())
121-
.withPartitioner(p -> configuration.partitionKey);
122+
.withPartitioner(p -> configuration.partitionKey)
123+
.withSerializer(serializer);
122124

123125
return writeTransform;
124126
}

0 commit comments

Comments
 (0)