Example usage for com.amazonaws.services.kinesis.model GetShardIteratorResult GetShardIteratorResult

Introduction

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

Prototype

GetShardIteratorResult

Source Link

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;//from w w  w.j a  v  a2s .  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 w  w .  jav  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);
}

---