@@ -180,16 +180,16 @@ func (m DBEsdkMiddleware) handleRequestInterception(ctx context.Context, request
180180 return nil , err
181181 }
182182 * v = transformedRequest .TransformedInput
183- // case *dynamodb.UpdateItemInput:
184- // ctx = middleware.WithStackValue(ctx, ContextKeyOriginalInput, *v)
185- // // Note: this context is not propagated downstream into dafny layer so it's left as context.TODO() https://issues.amazon.com/CrypTool-5403
186- // transformedRequest, err := m.client.UpdateItemInputTransform(context.TODO(), awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.UpdateItemInputTransformInput{
187- // SdkInput: *v,
188- // })
189- // if err != nil {
190- // return nil, err
191- // }
192- // *v = transformedRequest.TransformedInput
183+ case * dynamodb.UpdateItemInput :
184+ ctx = middleware .WithStackValue (ctx , ContextKeyOriginalInput , * v )
185+ // Note: this context is not propagated downstream into dafny layer so it's left as context.TODO() https://issues.amazon.com/CrypTool-5403
186+ transformedRequest , err := m .client .UpdateItemInputTransform (context .TODO (), awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.UpdateItemInputTransformInput {
187+ SdkInput : * v ,
188+ })
189+ if err != nil {
190+ return nil , err
191+ }
192+ * v = transformedRequest .TransformedInput
193193 }
194194 return ctx , nil
195195}
@@ -335,16 +335,16 @@ func (m DBEsdkMiddleware) handleResponseInterception(ctx context.Context, respon
335335 return err
336336 }
337337 * v = transformedRequest .TransformedOutput
338- // case *dynamodb.UpdateItemOutput:
339- // // Note: this context is not propagated downstream into dafny layer so it's left as context.TODO() https://issues.amazon.com/CrypTool-5403
340- // transformedRequest, err := m.client.UpdateItemOutputTransform(context.TODO(), awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.UpdateItemOutputTransformInput{
341- // OriginalInput: middleware.GetStackValue(ctx, ContextKeyOriginalInput).(dynamodb.UpdateItemInput),
342- // SdkOutput: *v,
343- // })
344- // if err != nil {
345- // return err
346- // }
347- // *v = transformedRequest.TransformedOutput
338+ case * dynamodb.UpdateItemOutput :
339+ // Note: this context is not propagated downstream into dafny layer so it's left as context.TODO() https://issues.amazon.com/CrypTool-5403
340+ transformedRequest , err := m .client .UpdateItemOutputTransform (context .TODO (), awscryptographydbencryptionsdkdynamodbtransformssmithygeneratedtypes.UpdateItemOutputTransformInput {
341+ OriginalInput : middleware .GetStackValue (ctx , ContextKeyOriginalInput ).(dynamodb.UpdateItemInput ),
342+ SdkOutput : * v ,
343+ })
344+ if err != nil {
345+ return err
346+ }
347+ * v = transformedRequest .TransformedOutput
348348 }
349349 return nil
350350}
0 commit comments