From ba3fde7e342642ecb2113c253f6e20ade2b1fcf6 Mon Sep 17 00:00:00 2001 From: ravikiran-s Date: Wed, 3 May 2023 16:29:43 +0530 Subject: [PATCH] length check fixes --- clientlibrary/streams/dynamo_adapter_client_test.go | 1 + clientlibrary/streams/dynamodb_adapter_client.go | 12 ++++++------ 2 files changed, 7 insertions(+), 6 deletions(-) create mode 100644 clientlibrary/streams/dynamo_adapter_client_test.go diff --git a/clientlibrary/streams/dynamo_adapter_client_test.go b/clientlibrary/streams/dynamo_adapter_client_test.go new file mode 100644 index 0000000..d813376 --- /dev/null +++ b/clientlibrary/streams/dynamo_adapter_client_test.go @@ -0,0 +1 @@ +package streams diff --git a/clientlibrary/streams/dynamodb_adapter_client.go b/clientlibrary/streams/dynamodb_adapter_client.go index d3b1aaa..b7e42ec 100644 --- a/clientlibrary/streams/dynamodb_adapter_client.go +++ b/clientlibrary/streams/dynamodb_adapter_client.go @@ -238,9 +238,9 @@ func (d DynamodbStreamAdapterClient) convertListStreamsOutput(dOutput *dynamodbs // Set StreamNames field using StreamDescriptionList field from output kinesisOutput.StreamNames = nil if dOutput.Streams != nil { - kinesisOutput.StreamNames = make([]string, len(dOutput.Streams)) - for i, stream := range dOutput.Streams { - kinesisOutput.StreamNames[i] = *stream.StreamArn + kinesisOutput.StreamNames = make([]string, 0) + for _, stream := range dOutput.Streams { + kinesisOutput.StreamNames = append(kinesisOutput.StreamNames, *stream.StreamArn) } } kinesisOutput.ResultMetadata = dOutput.ResultMetadata @@ -296,7 +296,7 @@ func (d DynamodbStreamAdapterClient) convertGetRecordsInput(params *kinesis.GetR func (d DynamodbStreamAdapterClient) convertGetRecordsOutput(dynamoOutput *dynamodbstreams.GetRecordsOutput) *kinesis.GetRecordsOutput { kinesisOutput := kinesis.GetRecordsOutput{} if dynamoOutput != nil && len(dynamoOutput.Records) != 0 { - kinesisOutput.Records = make([]ktypes.Record, len(dynamoOutput.Records)) + kinesisOutput.Records = make([]ktypes.Record, 0) kinesisOutput.ResultMetadata = dynamoOutput.ResultMetadata kinesisOutput.NextShardIterator = dynamoOutput.NextShardIterator } @@ -320,7 +320,7 @@ func (d DynamodbStreamAdapterClient) convertListShardsOutput(output *dynamodbstr } if output != nil { if output.StreamDescription != nil && len(output.StreamDescription.Shards) != 0 { - kinesisOutput.Shards = make([]ktypes.Shard, len(output.StreamDescription.Shards)) + kinesisOutput.Shards = make([]ktypes.Shard, 0) } for _, shard := range output.StreamDescription.Shards { kinesisOutput.Shards = append(kinesisOutput.Shards, ktypes.Shard{ @@ -349,7 +349,7 @@ func (d DynamodbStreamAdapterClient) convertDescribeStreamOutput(output *dynamod StreamName: output.StreamDescription.StreamArn, StreamStatus: ktypes.StreamStatus(output.StreamDescription.StreamStatus), StreamCreationTimestamp: output.StreamDescription.CreationRequestDateTime, - Shards: make([]ktypes.Shard, len(output.StreamDescription.Shards)), + Shards: make([]ktypes.Shard, 0), } } for _, shard := range output.StreamDescription.Shards {