@@ -160,6 +160,7 @@ public void writeAdditionalFiles(
160160 if (supportsComputeInputChecksumsWorkflow ) {
161161 goDelegator .useShapeWriter (service , writer -> {
162162 generateInputComputedChecksumMetadataHelpers (writer , model , symbolProvider , service );
163+ writePackageLevelAddInputChecksumMiddleware (writer );
163164 });
164165 }
165166
@@ -257,16 +258,14 @@ private void writeInputMiddlewareHelper(
257258 writer .openBlock ("func $L(stack *middleware.Stack, options Options) error {" , "}" ,
258259 getAddInputMiddlewareFuncName (operationName ), () -> {
259260 writer .write ("""
260- return $T (stack, $T{
261+ return addInputChecksumMiddleware (stack, $T{
261262 GetAlgorithm: $L,
262263 RequireChecksum: $L,
263264 RequestChecksumCalculation: options.RequestChecksumCalculation,
264265 EnableTrailingChecksum: $L,
265266 EnableComputeSHA256PayloadHash: true,
266267 EnableDecodedContentLengthHeader: $L,
267268 })""" ,
268- SymbolUtils .createValueSymbolBuilder ("AddInputMiddleware" ,
269- AwsGoDependency .SERVICE_INTERNAL_CHECKSUM ).build (),
270269 SymbolUtils .createValueSymbolBuilder ("InputMiddlewareOptions" ,
271270 AwsGoDependency .SERVICE_INTERNAL_CHECKSUM ).build (),
272271 hasRequestAlgorithmMember ?
@@ -279,6 +278,48 @@ private void writeInputMiddlewareHelper(
279278 writer .insertTrailingNewline ();
280279 }
281280
281+ // adapted (service/internal/checksum).AddInputMiddleware to give the service client control over its middleware stack,
282+ // per #2507
283+ private void writePackageLevelAddInputChecksumMiddleware (GoWriter writer ) {
284+ writer .addUseImports (SmithyGoDependency .SMITHY_MIDDLEWARE );
285+ writer .addUseImports (AwsGoDependency .SERVICE_INTERNAL_CHECKSUM );
286+ writer .write ("""
287+ func addInputChecksumMiddleware(stack *middleware.Stack, options internalChecksum.InputMiddlewareOptions) (err error) {
288+ err = stack.Initialize.Add(&internalChecksum.SetupInputContext{
289+ GetAlgorithm: options.GetAlgorithm,
290+ RequireChecksum: options.RequireChecksum,
291+ RequestChecksumCalculation: options.RequestChecksumCalculation,
292+ }, middleware.Before)
293+ if err != nil {
294+ return err
295+ }
296+
297+ stack.Build.Remove("ContentChecksum")
298+
299+ inputChecksum := &internalChecksum.ComputeInputPayloadChecksum{
300+ EnableTrailingChecksum: options.EnableTrailingChecksum,
301+ EnableComputePayloadHash: options.EnableComputeSHA256PayloadHash,
302+ EnableDecodedContentLengthHeader: options.EnableDecodedContentLengthHeader,
303+ }
304+ if err := stack.Finalize.Insert(inputChecksum, "ResolveEndpointV2", middleware.After); err != nil {
305+ return err
306+ }
307+
308+ if options.EnableTrailingChecksum {
309+ trailerMiddleware := &internalChecksum.AddInputChecksumTrailer{
310+ EnableTrailingChecksum: inputChecksum.EnableTrailingChecksum,
311+ EnableComputePayloadHash: inputChecksum.EnableComputePayloadHash,
312+ EnableDecodedContentLengthHeader: inputChecksum.EnableDecodedContentLengthHeader,
313+ }
314+ if err := stack.Finalize.Insert(trailerMiddleware, inputChecksum.ID(), middleware.After); err != nil {
315+ return err
316+ }
317+ }
318+
319+ return nil
320+ }""" );
321+ }
322+
282323 private void writeOutputMiddlewareHelper (
283324 GoWriter writer ,
284325 Model model ,
0 commit comments