Fix serializer

etspaceman · Jan 23, 2025 · cf5512a · cf5512a
1 parent 5e6695b
commit cf5512a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kcl-localstack/src/main/scala/kinesis4cats/kcl/localstack/LocalstackKCLConsumer.scala b/kcl-localstack/src/main/scala/kinesis4cats/kcl/localstack/LocalstackKCLConsumer.scala
@@ -27,6 +27,7 @@ import software.amazon.kinesis.common.LeaseCleanupConfig
 import software.amazon.kinesis.leases.LeaseManagementConfig
 import software.amazon.kinesis.leases.dynamodb.DynamoDBLeaseManagementFactory
 import software.amazon.kinesis.leases.dynamodb.DynamoDBLeaseSerializer
+import software.amazon.kinesis.leases.dynamodb.DynamoDBMultiStreamLeaseSerializer
 import software.amazon.kinesis.processor.StreamTracker
 import software.amazon.kinesis.retrieval.polling.PollingConfig
 
@@ -78,7 +79,8 @@ object LocalstackKCLConsumer {
         defaultLeaseManagement.leaseTableDeletionProtectionEnabled(),
         defaultLeaseManagement.leaseTablePitrEnabled(),
         defaultLeaseManagement.tags(),
-        new DynamoDBLeaseSerializer(),
+        if (isMultiStream) new DynamoDBMultiStreamLeaseSerializer()
+        else new DynamoDBLeaseSerializer(),
         defaultLeaseManagement.customShardDetectorProvider(),
         isMultiStream,
         LeaseCleanupConfig