INNER CODE UNIT · Go

listShards

brexhq/substation · cmd/aws/lambda/autoscale/main.go:334

func listShards(ctx context.Context, stream string) (int32, error) {
	var shards int32

	input := kinesis.ListShardsInput{
		StreamName: aws.String(stream),
	}

LOOP:
	for {
		resp, err := kinesisC.ListShards(ctx, &input)
		if err != nil {
			return 0, err
		}

		for _, s := range resp.Shards {
			if end := s.SequenceNumberRange.EndingSequenceNumber; end == nil {
				shards++
			}

View source record →