luciano/amazon-kinesis-client
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Adding guardrails for duplicate shard responses

Browse source
This commit is contained in:
Joshua Kim 2021-04-29 17:10:55 -07:00
parent 4107b28aba
commit dfed0e3b50
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/main/java/com/amazonaws/services/kinesis/clientlibrary/proxies/KinesisProxy.java
View file

@ -22,6 +22,7 @@ import java.util.ArrayList;
import java.util.Date; import java.util.Date;
import java.util.EnumSet; import java.util.EnumSet;
import java.util.HashSet; import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
@ -485,10 +486,11 @@ public class KinesisProxy implements IKinesisProxyExtended {
} }
} while (response.getStreamDescription().isHasMoreShards()); } while (response.getStreamDescription().isHasMoreShards());
} }
this.cachedShardMap = shards.stream().collect(Collectors.toMap(Shard::getShardId, Function.identity())); final List<Shard> dedupedShards = new ArrayList<>(new LinkedHashSet<>(shards));
this.cachedShardMap = dedupedShards.stream().collect(Collectors.toMap(Shard::getShardId, Function.identity()));
this.lastCacheUpdateTime = Instant.now(); this.lastCacheUpdateTime = Instant.now();
return shards; return dedupedShards;
} }
/** /**