INNER CODE UNIT · Go

updateStream

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

func updateStream(ctx context.Context, stream string, shards int32) error {
	_, err := kinesisC.UpdateShardCount(ctx, &kinesis.UpdateShardCountInput{
		StreamName:       aws.String(stream),
		TargetShardCount: aws.Int32(shards),
		ScalingType:      ktypes.ScalingTypeUniformScaling,
	})
	if err != nil {
		return err
	}

	for {
		resp, err := kinesisC.DescribeStreamSummary(ctx, &kinesis.DescribeStreamSummaryInput{
			StreamName: aws.String(stream),
		})
		if err != nil {
			return err
		}

View source record →