auto commit

Author: rishav-karanjit

Examples/runtimes/go/migration/PlaintextToAWSDBE/awsdbe/migrationexamplestep3.go

@@ -0,0 +1,100 @@
+package awsdbe
+
+import (
+	// Standard imports
+	"context"
+	"fmt"
+	"reflect"
+
+	// AWS SDK imports
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/config"
+	"github.com/aws/aws-sdk-go-v2/service/dynamodb"
+	"github.com/aws/aws-sdk-go-v2/service/dynamodb/types"
+
+	dbesdkdynamodbencryptiontypes "github.com/aws/aws-database-encryption-sdk-dynamodb/releases/go/dynamodb-esdk/awscryptographydbencryptionsdkdynamodbsmithygeneratedtypes"
+	"github.com/aws/aws-database-encryption-sdk-dynamodb/releases/go/dynamodb-esdk/dbesdkmiddleware"
+	"github.com/aws/aws-database-encryption-sdk-dynamodb/releases/go/dynamodb-esdk/examples/utils"
+)
+
+/*
+Migration Step 3: This is an example demonstrating how to update your configuration
+to stop accepting reading plaintext items.
+
+Once you complete Step 3, all items being read by your system are encrypted.
+
+Before you move onto this step, you will need to encrypt all plaintext items in your dataset.
+How you will want to do this depends on your system.
+
+Running this example requires access to the DDB Table whose name
+is provided in CLI arguments.
+This table must be configured with the following
+primary key configuration:
+  - Partition key is named "partition_key" with type (S)
+  - Sort key is named "sort_key" with type (S)
+*/
+func MigrationStep3(kmsKeyID, ddbTableName, partitionKeyValue, sortKeyValue string) {
+	cfg, err := config.LoadDefaultConfig(context.TODO())
+	utils.HandleError(err)
+
+	// 1. Configure your Keyring, attribute actions,
+	// 	  allowedUnsignedAttributes, and encryption configuration for table.
+	//    This is common across all the steps.
+
+	// When creating encryption configuration for your table,
+	// you must use the plaintext override `PlaintextOverrideForbidPlaintextWriteForbidPlaintextRead`.
+	// This plaintext override means:
+
+	listOfTableConfigs := configureTable(kmsKeyID, ddbTableName, dbesdkdynamodbencryptiontypes.PlaintextOverrideForbidPlaintextWriteForbidPlaintextRead)
+
+	// 5. Create DynamoDB client with dbEsdkMiddleware
+	dbEsdkMiddleware, err := dbesdkmiddleware.NewDBEsdkMiddleware(listOfTableConfigs)
+	utils.HandleError(err)
+
+	ddb := dynamodb.NewFromConfig(cfg, dbEsdkMiddleware.CreateMiddleware())
+
+	// 6. Put an item into your table.
+	//    This item will be encrypted.
+	item := map[string]types.AttributeValue{
+		"partition_key": &types.AttributeValueMemberS{Value: partitionKeyValue},
+		"sort_key":      &types.AttributeValueMemberN{Value: sortKeyValue},
+		"attribute1":    &types.AttributeValueMemberS{Value: "this will be encrypted and signed"},
+		"attribute2":    &types.AttributeValueMemberS{Value: "this will never be encrypted, but it will be signed"},
+		"attribute3":    &types.AttributeValueMemberS{Value: "this will never be encrypted nor signed"},
+	}
+
+	putInput := dynamodb.PutItemInput{
+		TableName: &ddbTableName,
+		Item:      item,
+	}
+
+	_, err = ddb.PutItem(context.TODO(), &putInput)
+	utils.HandleError(err)
+
+	// 7. Get an item back from the table.
+	//    If this is an item written in plaintext (i.e. any item written
+	//    during Step 0 or 1), then the item will still be in plaintext.
+	//    If this is an item that was encrypted client-side (i.e. any item written
+	//    during Step 2 or after), then the DDB enhanced client will decrypt the
+	//    item client-sid and surface it in our code as a plaintext item.
+	key := map[string]types.AttributeValue{
+		"partition_key": &types.AttributeValueMemberS{Value: partitionKeyValue},
+		"sort_key":      &types.AttributeValueMemberN{Value: sortKeyValue},
+	}
+
+	getInput := &dynamodb.GetItemInput{
+		TableName:      aws.String(ddbTableName),
+		Key:            key,
+		ConsistentRead: aws.Bool(true),
+	}
+
+	result, err := ddb.GetItem(context.TODO(), getInput)
+	utils.HandleError(err)
+
+	// Demonstrate we get the expected item back
+	if !reflect.DeepEqual(item, result.Item) {
+		panic("Decrypted item does not match original item")
+	}
+	fmt.Println("MigrationStep2 completed successfully")
+}