batch_write_items.go

package dynago

import (
	"context"

	"github.com/aws/aws-sdk-go-v2/service/dynamodb"
	"github.com/aws/aws-sdk-go-v2/service/dynamodb/types"
)

const ChunkSize = 25

/**
* Used to update records to  dynamodb
* @param input slice of record want to  put to DB
* @return error
 */

func (t *Client) BatchWriteItems(ctx context.Context, input []map[string]types.AttributeValue) error {
	items := make([]types.WriteRequest, 0, len(input))
	table := t.TableName
	for _, model := range input {
		items = append(items,
			types.WriteRequest{
				PutRequest: &types.PutRequest{
					Item: model,
				},
			},
		)
	}
	chunkedItems := chunkBy(items, ChunkSize)
	for _, chunkedBatch := range chunkedItems {
		_, err := t.client.BatchWriteItem(ctx, &dynamodb.BatchWriteItemInput{
			RequestItems: map[string][]types.WriteRequest{
				table: chunkedBatch,
			},
		})
		if err != nil {
			return err
		}
	}

	return nil

}