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

Change Shard Iterator State Reset to Recreation

Browse source 
Changed the shard iterator reset to simply recreate the iterator.
This commit is contained in:
Pfifer, Justin 2017-03-27 07:58:28 -07:00 committed by Justin Pfifer
parent e45f59c73b
commit fb4270e98b
1 changed files with 6 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -252,9 +252,10 @@ public class KinesisProxy implements IKinesisProxyExtended {
shardIterationState.update(response.getStreamDescription().getShards());
}
} while (response.getStreamDescription().isHasMoreShards());
this.listOfShardsSinceLastGet.set(shardIterationState.getCollected());
this.listOfShardsSinceLastGet.set(shardIterationState.getShards());
return shardIterationState.getAndReset();
shardIterationState = new ShardIterationState();
return listOfShardsSinceLastGet.get();
}
/**
@ -359,30 +360,23 @@ public class KinesisProxy implements IKinesisProxyExtended {
@Data
static class ShardIterationState {
private List<Shard> collected;
private List<Shard> shards;
private String lastShardId;
public ShardIterationState() {
collected = new ArrayList<>();
shards = new ArrayList<>();
}
public void update(List<Shard> shards) {
if (shards == null || shards.isEmpty()) {
return;
}
collected.addAll(shards);
this.shards.addAll(shards);
Shard lastShard = shards.get(shards.size() - 1);
if (lastShardId == null || lastShardId.compareTo(lastShard.getShardId()) < 0) {
lastShardId = lastShard.getShardId();
}
}
public List<Shard> getAndReset() {
List<Shard> result = collected;
collected = new ArrayList<>();
lastShardId = null;
return result;
}
}
}