@@ -180,16 +180,16 @@ func (m DBEsdkMiddleware) handleRequestInterception(ctx context.Context, request
180
180
return nil , err
181
181
}
182
182
* 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
193
193
}
194
194
return ctx , nil
195
195
}
@@ -335,16 +335,16 @@ func (m DBEsdkMiddleware) handleResponseInterception(ctx context.Context, respon
335
335
return err
336
336
}
337
337
* 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
348
348
}
349
349
return nil
350
350
}
0 commit comments