✨ updated index helper with sorting and pagination

aws/dynamo.go

@@ -8,51 +8,105 @@ import (
 	"github.com/aws/aws-sdk-go-v2/service/dynamodb"
 )
 
-func DynamoInsert(region string, table string, params any) error {
+type DynamoIndexQuery struct {
+	Region     string
+	Table      string
+	WhereField string
+	WhereValue string
+	OrderField string
+	Direction  string
+	SetLimit   *int32
+}
+
+type DynamoIndexResults struct {
+	NextToken *string
+	Items     interface{}
+}
+
+func DynamoIndex(region string, table string) DynamoIndexQuery {
+	var query DynamoIndexQuery
+	query.Region = region
+	query.Table = table
+	return query
+}
+
+func (q DynamoIndexQuery) Where(field string, value string) DynamoIndexQuery {
+	q.WhereField = field
+	q.WhereValue = value
+	return q
+}
+func (q DynamoIndexQuery) Order(field string) DynamoIndexQuery {
+	q.OrderField = field
+	return q
+}
+
+func (q DynamoIndexQuery) Desc() DynamoIndexQuery {
+	q.Direction = "DESC"
+	return q
+}
+func (q DynamoIndexQuery) Asc() DynamoIndexQuery {
+	q.Direction = "ASC"
+	return q
+}
+
+func (q DynamoIndexQuery) Limit(limit int) DynamoIndexQuery {
+	q.SetLimit = aws.Int32(int32(limit))
+	return q
+}
+
+func (q DynamoIndexQuery) Get(out interface{}) (*DynamoIndexResults, error) {
+	config, err := cfg(q.Region)
 
-	config, err := cfg(region)
 	if err != nil {
-		return err
+		return nil, err
 	}
+	query := fmt.Sprintf(`SELECT * FROM "%s" WHERE "%s" = '%s' ORDER BY "%s" %s`,
+		q.Table,
+		q.WhereField,
+		q.WhereValue,
+		q.OrderField,
+		q.Direction,
+	)
+	p, err := dynamodb.NewFromConfig(config).ExecuteStatement(context.Background(),
+		&dynamodb.ExecuteStatementInput{
+			Statement:      aws.String(query),
+			ConsistentRead: aws.Bool(false),
+			Limit:          q.SetLimit,
+		},
+	)
 
-	item, err := attributevalue.MarshalMap(params)
 	if err != nil {
-		return err
+		return nil, err
 	}
 
-	if _, err := dynamodb.NewFromConfig(config).PutItem(context.Background(), &dynamodb.PutItemInput{
-		TableName: aws.String(table),
-		Item:      item,
-	}); err != nil {
-		return err
+	if err := attributevalue.UnmarshalListOfMaps(p.Items, &out); err != nil {
+		return nil, err
 	}
+	var results DynamoIndexResults
+	results.Items = out
+	results.NextToken = p.NextToken
 
-	return nil
+	return &results, nil
 }
 
-func DynamoIndex(region string, table string, out interface{}) error {
+func DynamoInsert(region string, table string, params any) error {
 
 	config, err := cfg(region)
-
 	if err != nil {
 		return err
 	}
 
-	p, err := dynamodb.NewFromConfig(config).ExecuteStatement(context.Background(),
-		&dynamodb.ExecuteStatementInput{
-			Statement:      aws.String(fmt.Sprintf("SELECT * FROM \"%s\"", table)),
-			ConsistentRead: aws.Bool(false),
-		},
-	)
-
+	item, err := attributevalue.MarshalMap(params)
 	if err != nil {
 		return err
 	}
 
-	if err := attributevalue.UnmarshalListOfMaps(p.Items, &out); err != nil {
+	if _, err := dynamodb.NewFromConfig(config).PutItem(context.Background(), &dynamodb.PutItemInput{
+		TableName: aws.String(table),
+		Item:      item,
+	}); err != nil {
 		return err
 	}
 
 	return nil
-
 }