@@ -60,16 +60,16 @@ func (m DBEsdkMiddleware) createRequestInterceptor() middleware.InitializeMiddle
6060// handleRequestInterception handles the interception logic before the DynamoDB operation
6161func (m DBEsdkMiddleware ) handleRequestInterception (ctx context.Context , request interface {}) (context.Context , error ) {
6262 switch v := request .(type ) {
63- // case *dynamodb.PutItemInput:
64- // ctx = middleware.WithStackValue(ctx, ContextKeyOriginalInput, *v)
65- // // Note: this context is not propagated downstream into dafny layer https://issues.amazon.com/CrypTool-5403
66- // transformedRequest, err := m.client.PutItemInputTransform(ctx, awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.PutItemInputTransformInput{
67- // SdkInput: *v,
68- // })
69- // if err != nil {
70- // return nil, err
71- // }
72- // *v = transformedRequest.TransformedInput
63+ case * dynamodb.PutItemInput :
64+ ctx = middleware .WithStackValue (ctx , ContextKeyOriginalInput , * v )
65+ // Note: this context is not propagated downstream into dafny layer https://issues.amazon.com/CrypTool-5403
66+ transformedRequest , err := m .client .PutItemInputTransform (ctx , awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.PutItemInputTransformInput {
67+ SdkInput : * v ,
68+ })
69+ if err != nil {
70+ return nil , err
71+ }
72+ * v = transformedRequest .TransformedInput
7373 case * dynamodb.GetItemInput :
7474 ctx = middleware .WithStackValue (ctx , ContextKeyOriginalInput , * v )
7575 // Note: this context is not propagated downstream into dafny layer https://issues.amazon.com/CrypTool-5403
@@ -215,16 +215,16 @@ func (m DBEsdkMiddleware) createResponseInterceptor() middleware.FinalizeMiddlew
215215// handleResponseInterception handles the interception logic after the DynamoDB operation
216216func (m DBEsdkMiddleware ) handleResponseInterception (ctx context.Context , response interface {}) error {
217217 switch v := response .(type ) {
218- // case *dynamodb.PutItemOutput:
219- // // Note: this context is not propagated downstream into dafny layer https://issues.amazon.com/CrypTool-5403
220- // transformedRequest, err := m.client.PutItemOutputTransform(ctx, awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.PutItemOutputTransformInput{
221- // OriginalInput: middleware.GetStackValue(ctx, ContextKeyOriginalInput).(dynamodb.PutItemInput),
222- // SdkOutput: *v,
223- // })
224- // if err != nil {
225- // return err
226- // }
227- // *v = transformedRequest.TransformedOutput
218+ case * dynamodb.PutItemOutput :
219+ // Note: this context is not propagated downstream into dafny layer https://issues.amazon.com/CrypTool-5403
220+ transformedRequest , err := m .client .PutItemOutputTransform (ctx , awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.PutItemOutputTransformInput {
221+ OriginalInput : middleware .GetStackValue (ctx , ContextKeyOriginalInput ).(dynamodb.PutItemInput ),
222+ SdkOutput : * v ,
223+ })
224+ if err != nil {
225+ return err
226+ }
227+ * v = transformedRequest .TransformedOutput
228228 case * dynamodb.GetItemOutput :
229229 // Note: this context is not propagated downstream into dafny layer https://issues.amazon.com/CrypTool-5403
230230 transformedRequest , err := m .client .GetItemOutputTransform (ctx , awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.GetItemOutputTransformInput {
0 commit comments