Updating the config to set the cache type.

This commit is contained in:
Sahil Palvia 2017-09-25 11:56:02 -07:00
parent 1deae5cd07
commit b5d5618a94

View file

@ -1290,7 +1290,13 @@ public class KinesisClientLibConfiguration {
}
public KinesisClientLibConfiguration withDataFetchingStrategy(String dataFetchingStrategy) {
recordsFetcherFactory.setDataFetchingStrategy(DataFetchingStrategy.valueOf(dataFetchingStrategy));
switch (dataFetchingStrategy.toUpperCase()) {
case "PREFETCH_CACHED":
recordsFetcherFactory.setDataFetchingStrategy(DataFetchingStrategy.PREFETCH_CACHED);
break;
default:
recordsFetcherFactory.setDataFetchingStrategy(DataFetchingStrategy.DEFAULT);
}
return this;
}