add inline finalize middleware gen

aws · Nov 8, 2023 · e8a6042 · e8a6042
1 parent 6f75282
commit e8a6042
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/...degen/src/main/java/software/amazon/smithy/go/codegen/GoStackStepMiddlewareGenerator.java b/...degen/src/main/java/software/amazon/smithy/go/codegen/GoStackStepMiddlewareGenerator.java
@@ -137,6 +137,28 @@ public static GoStackStepMiddlewareGenerator createFinalizeStepMiddleware(String
                 SmithyGoTypes.Middleware.FinalizeHandler);
     }
 
+    /**
+     * Create an inline Finalize func.
+     *
+     * @param body is the function body.
+     * @return the generated middleware func.
+     */
+    public static GoWriter.Writable generateFinalizeMiddlewareFunc(GoWriter.Writable body) {
+        return goTemplate("""
+                func(ctx $T, in $T, next $T) (
+                    out $T, metadata $T, err error,
+                ) {
+                    $W
+                }
+                """,
+                GoStdlibTypes.Context.Context,
+                SmithyGoTypes.Middleware.FinalizeInput,
+                SmithyGoTypes.Middleware.FinalizeHandler,
+                SmithyGoTypes.Middleware.FinalizeOutput,
+                SmithyGoTypes.Middleware.Metadata,
+                body);
+    }
+
     /**
      * Create a new DeserializeStep middleware generator with the provided type name.
      *