dynamodb_record.go

package lambdabase

import (
	"context"
	"fmt"

	"github.com/aws/aws-lambda-go/events"
	"github.com/go-nacelle/nacelle/v2"
)

type (
	DynamoDBRecordHandler interface {
		Handle(ctx context.Context, record events.DynamoDBEventRecord, logger nacelle.Logger) error
	}

	dynamoDBRecordHandlerInitializer interface {
		nacelle.Initializer
		DynamoDBRecordHandler
	}

	dynamoDBRecordHandler struct {
		Services *nacelle.ServiceContainer `service:"services"`
		handler  DynamoDBRecordHandler
	}
)

func NewDynamoDBRecordServer(handler DynamoDBRecordHandler) *Server {
	return NewDynamoDBEventServer(&dynamoDBRecordHandler{
		handler: handler,
	})
}

func (s *dynamoDBRecordHandler) Init(ctx context.Context) error {
	return doInit(ctx, s.Services, s.handler)
}

func (h *dynamoDBRecordHandler) Handle(ctx context.Context, records []events.DynamoDBEventRecord, logger nacelle.Logger) error {
	for _, record := range records {
		recordLogger := logger.WithFields(map[string]interface{}{
			"eventId": record.EventID,
		})

		logger.Debug("Handling record")

		if err := h.handler.Handle(ctx, record, recordLogger); err != nil {
			return fmt.Errorf("failed to process DynamoDB record %s (%s)", record.EventID, err.Error())
		}
	}

	logger.Debug("DynamoDB record handled successfully")
	return nil
}