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

List of usage examples for com.amazonaws.services.kinesis.model GetShardIteratorRequest getShardId

Introduction

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

Prototype


public String getShardId() 

Source Link

Document

The shard ID of the Kinesis Data Streams shard to get the iterator for.

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.  j a va 2  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:com.netflix.spectator.tdigest.KinesisTDigestReader.java

License:Apache License

private Counter counter(Registry registry, String name, GetShardIteratorRequest req) {
    return registry.counter("spectator.tdigest." + name, "stream", req.getStreamName(), "shard",
            req.getShardId());
}

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;//w w w .  j ava  2 s.co  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);
}