Example usage for com.amazonaws.services.kinesis.model GetShardIteratorRequest getShardIteratorType

Introduction

In this page you can find the example usage for com.amazonaws.services.kinesis.model GetShardIteratorRequest getShardIteratorType.

Prototype

public String getShardIteratorType() 

Source Link

Document

Determines how the shard iterator is used to start reading data records from the shard.

Usage

From source file:com.facebook.presto.kinesis.util.MockKinesisClient.java

License:Apache License

@Override
public GetShardIteratorResult getShardIterator(GetShardIteratorRequest getShardIteratorRequest)
        throws AmazonServiceException, AmazonClientException {
    ShardIterator iter = ShardIterator.fromStreamAndShard(getShardIteratorRequest.getStreamName(),
            getShardIteratorRequest.getShardId());
    if (iter != null) {
        InternalStream theStream = this.getStream(iter.streamId);
        if (theStream != null) {
            String seqAsString = getShardIteratorRequest.getStartingSequenceNumber();
            if (seqAsString != null && !seqAsString.isEmpty()
                    && getShardIteratorRequest.getShardIteratorType().equals("AFTER_SEQUENCE_NUMBER")) {
                int sequence = Integer.parseInt(seqAsString);
                iter.recordIndex = sequence + 1;
            } else {
                iter.recordIndex = 100;/* w  w w. java2  s .c  o m*/
            }

            GetShardIteratorResult result = new GetShardIteratorResult();
            return result.withShardIterator(iter.makeString());
        } else {
            throw new AmazonClientException("Unknown stream or bad shard iterator!");
        }
    } else {
        throw new AmazonClientException("Bad stream or shard iterator!");
    }
}

---

From source file:org.apache.beam.sdk.io.kinesis.AmazonKinesisMock.java

License:Apache License

@Override
public GetShardIteratorResult getShardIterator(GetShardIteratorRequest getShardIteratorRequest) {
    ShardIteratorType shardIteratorType = ShardIteratorType
            .fromValue(getShardIteratorRequest.getShardIteratorType());

    String shardIterator;/*from w  ww. ja v a  2 s.c o m*/
    if (shardIteratorType == ShardIteratorType.TRIM_HORIZON) {
        shardIterator = String.format("%s:%s", getShardIteratorRequest.getShardId(), 0);
    } else {
        throw new RuntimeException("Not implemented");
    }

    return new GetShardIteratorResult().withShardIterator(shardIterator);
}

---