auto commit

Author: rishav-karanjit

db-esdk-performance-testing/go/benchmark/config.go

@@ -10,6 +10,12 @@ import (
 	"gopkg.in/yaml.v3"
 )
 
+type KeyringType string
+
+const (
+	RawAESKeying KeyringType = "raw-aes"
+)
+
 // TestConfig represents the configuration for benchmark tests
 type TestConfig struct {
 	DataSizes struct {
@@ -24,6 +30,7 @@ type TestConfig struct {
 	ConcurrencyLevels []int        `yaml:"concurrency_levels"`
 	QuickConfig       *QuickConfig `yaml:"quick_config"`
 	TableName         string       `yaml:"table_name"`
+	Keyring           KeyringType  `yaml:"keyring"`
 }
 
 // QuickConfig represents the quick test configuration

db-esdk-performance-testing/go/benchmark/dbesdk_benchmark.go

@@ -58,6 +58,11 @@ func New(configPath string) (*DBESDKBenchmark, error) {
 	}
 	benchmark.Config = config
 
+	// Setup MPL
+	if err := benchmark.setupMPL(); err != nil {
+		return nil, fmt.Errorf("failed to setup MPL: %w", err)
+	}
+
 	// Setup DB-ESDK
 	if err := benchmark.setupDBESDK(); err != nil {
 		return nil, fmt.Errorf("failed to setup DB-ESDK: %w", err)
@@ -69,6 +74,26 @@ func New(configPath string) (*DBESDKBenchmark, error) {
 	return benchmark, nil
 }
 
+func (b *DBESDKBenchmark) setupMPL() error {
+	// Initialize the material providers client
+	matProvConfig := mpltypes.MaterialProvidersConfig{}
+	matProv, err := mplsmithygenerated.NewClient(matProvConfig)
+	if err != nil {
+		return fmt.Errorf("failed to create material providers client: %w", err)
+	}
+
+	switch b.Config.Keyring {
+	case RawAESKeying:
+		b.Keyring, err = SetupRawAESKeyring(matProv)
+		if err != nil {
+			return fmt.Errorf("failed to create keyring: %w", err)
+		}
+	default:
+		return fmt.Errorf("unsupported keyring type: %s", b.Config.Keyring)
+	}
+	return nil
+}
+
 // setupDBESDK initializes the DynamoDB client with DB-ESDK middleware and creates a default keyring which is AES keyring
 func (b *DBESDKBenchmark) setupDBESDK() error {
 	ddbTableName := b.Config.TableName

db-esdk-performance-testing/go/benchmark/keyringsetup.go

@@ -0,0 +1,30 @@
+package benchmark
+
+import (
+	"context"
+	"crypto/rand"
+	"fmt"
+
+	mplsmithygenerated "github.com/aws/aws-cryptographic-material-providers-library/releases/go/mpl/awscryptographymaterialproviderssmithygenerated"
+	mpltypes "github.com/aws/aws-cryptographic-material-providers-library/releases/go/mpl/awscryptographymaterialproviderssmithygeneratedtypes"
+)
+
+func SetupRawAESKeyring(matProv *mplsmithygenerated.Client) (mpltypes.IKeyring, error) {
+	key := make([]byte, 32)
+	if _, err := rand.Read(key); err != nil {
+		return nil, fmt.Errorf("failed to generate AES-256 key: %w", err)
+	}
+
+	keyringInput := mpltypes.CreateRawAesKeyringInput{
+		KeyName:      "test-aes-256-key",
+		KeyNamespace: "DB-ESDK-performance-test",
+		WrappingKey:  key,
+		WrappingAlg:  mpltypes.AesWrappingAlgAlgAes256GcmIv12Tag16,
+	}
+
+	keyring, err := matProv.CreateRawAesKeyring(context.Background(), keyringInput)
+	if err != nil {
+		return nil, fmt.Errorf("failed to create keyring: %w", err)
+	}
+	return keyring, nil
+}

db-esdk-performance-testing/go/config/test-scenarios.yaml

@@ -41,3 +41,5 @@ concurrency_levels:
   - 16
 
 table_name: "dbesdk-performance-testing" # DynamoDB table name
+
+keyring: "raw-aes"