15 lines
622 B
Go
15 lines
622 B
Go
|
package consumer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go-v2/service/kinesis"
|
||
|
|
)
|
||
|
|
|
||
|
|
// kinesisClient defines the interface of functions needed for the consumer
|
||
|
|
type kinesisClient interface {
|
||
|
|
GetRecords(ctx context.Context, params *kinesis.GetRecordsInput, optFns ...func(*kinesis.Options)) (*kinesis.GetRecordsOutput, error)
|
||
|
|
ListShards(ctx context.Context, params *kinesis.ListShardsInput, optFns ...func(*kinesis.Options)) (*kinesis.ListShardsOutput, error)
|
||
|
|
GetShardIterator(ctx context.Context, params *kinesis.GetShardIteratorInput, optFns ...func(*kinesis.Options)) (*kinesis.GetShardIteratorOutput, error)
|
||
|
|
}
|