feat: add deleteitem function to write interface

delete_item.go

@@ -14,20 +14,20 @@ import (
 * @param sk the sort key of the record
  * @return true if the record was deleted, false otherwise
 */
-func (t *Client) DeleteItem(ctx context.Context, pk string, sk string) error {
+func (t *Client) DeleteItem(ctx context.Context, pk Attribute, sk Attribute) error {
 
 	//delete item from dynamodb
 	input := &dynamodb.DeleteItemInput{
 		TableName: &t.TableName,
 		Key: map[string]types.AttributeValue{
-			"pk": &types.AttributeValueMemberS{Value: pk},
-			"sk": &types.AttributeValueMemberS{Value: sk},
+			"pk": pk,
+			"sk": sk,
 		},
 	}
-	resp, err := t.client.DeleteItem(ctx, input)
+	_, err := t.client.DeleteItem(ctx, input)
 
-	if err != nil && resp == nil {
-		log.Println("failed to delete record into database. Error:" + err.Error())
+	if err != nil {
+		log.Println("failed to delete record from database. Error:" + err.Error())
 		return err
 	}
 

interface.go

@@ -25,6 +25,7 @@ type WriteAPI interface {
 	// Create or update given item in DynamoDB. Must implemenmt DynamoRecord interface.
 	// DynamoRecord.GetKeys will be called to get values for parition and sort keys.
 	PutItem(ctx context.Context, pk Attribute, sk Attribute, item interface{}) error
+	DeleteItem(ctx context.Context, pk Attribute, sk Attribute) error
 }
 
 type TransactionAPI interface {

tests/delete_item_test.go

@@ -0,0 +1,60 @@
+package tests
+
+import (
+	"context"
+	"testing"
+
+	"github.com/aws/aws-sdk-go-v2/service/dynamodb/types"
+	"github.com/oolio-group/dynago"
+)
+
+type user struct {
+	Pk string `dynamodbav:"pk"`
+	Sk string `dynamodbav:"sk"`
+}
+
+func TestDeleteItem(t *testing.T) {
+	table := prepareTable(t, dynamoEndpoint, "delete_test")
+
+	ctx := context.TODO()
+
+	// Insert a single item
+	item := user{Pk: "users#org1", Sk: "user#1"}
+	err := table.PutItem(ctx, dynago.StringValue(item.Pk), dynago.StringValue(item.Sk), item)
+	if err != nil {
+		t.Fatalf("failed to insert item; got %s", err)
+	}
+
+	// Query the item to ensure it exists
+	var output []user
+	_, err = table.Query(ctx, "pk = :pk and sk = :sk", map[string]types.AttributeValue{
+		":pk": &types.AttributeValueMemberS{Value: item.Pk},
+		":sk": &types.AttributeValueMemberS{Value: item.Sk},
+	}, &output)
+	if err != nil {
+		t.Fatalf("expected query to succeed; got %s", err)
+	}
+	if len(output) == 0 {
+		t.Fatalf("expected item to be found, but found none")
+	}
+
+	// Delete the item
+	err = table.DeleteItem(ctx, dynago.StringValue(item.Pk), dynago.StringValue(item.Sk))
+	if err != nil {
+		t.Fatalf("expected delete to succeed; got %s", err)
+	}
+
+	// Query the item again to ensure it no longer exists
+	var deleteOutput []user
+	_, err = table.Query(ctx, "pk = :pk and sk = :sk", map[string]types.AttributeValue{
+		":pk": &types.AttributeValueMemberS{Value: item.Pk},
+		":sk": &types.AttributeValueMemberS{Value: item.Sk},
+	}, &deleteOutput)
+
+	if err != nil {
+		t.Fatalf("expected query to succeed; got %s", err)
+	}
+	if len(deleteOutput) != 0 {
+		t.Fatalf("expected item to be deleted; found %v", deleteOutput)
+	}
+}