codegen: use service closure to cache shape queries in schema generation

Author: kuhe

private/my-local-model-schema/src/schemas/schemas_0.ts

@@ -23,8 +23,8 @@ const _fWMi = "fieldWithMessage";
 const _g = "gamma";
 const _hE = "httpError";
 const _i = "id";
-const _s = "smithy.ts.sdk.synthetic.org.xyz.v1";
-const _st = "streaming";
+const _s = "streaming";
+const _sm = "smithy.ts.sdk.synthetic.org.xyz.v1";
 const _t = "timestamp";
 const n0 = "org.xyz.v1";
 
@@ -100,13 +100,13 @@ export var XYZServiceServiceException: StaticErrorSchema = [-3, n0, _XYZSSE,
 ];
 TypeRegistry.for(n0).registerError(XYZServiceServiceException, __XYZServiceServiceException);
 export var __Unit = "unit" as const;
-export var XYZServiceSyntheticServiceException: StaticErrorSchema = [-3, _s, "XYZServiceSyntheticServiceException", 0, [], []];
-TypeRegistry.for(_s).registerError(XYZServiceSyntheticServiceException, __XYZServiceSyntheticServiceException);
 export var TradeEvents: StaticStructureSchema = [3, n0, _TE,
-  { [_st]: 1 },
+  { [_s]: 1 },
   [_a, _b, _g],
   [() => Alpha, () => __Unit, () => __Unit]
 ];
+export var XYZServiceSyntheticServiceException: StaticErrorSchema = [-3, _sm, "XYZServiceSyntheticServiceException", 0, [], []];
+TypeRegistry.for(_sm).registerError(XYZServiceSyntheticServiceException, __XYZServiceSyntheticServiceException);
 export var GetNumbers: StaticOperationSchema = [9, n0, _GN,
   0, () => GetNumbersRequest, () => GetNumbersResponse
 ];

private/my-local-model-schema/test/index-objects.spec.mjs

@@ -1,5 +1,9 @@
 import {
+  CodedThrottlingError,
   GetNumbersCommand,
+  HaltError,
+  MysteryThrottlingError,
+  RetryableError,
   TradeEventStreamCommand,
   XYZService,
   XYZServiceClient,
@@ -13,5 +17,10 @@ assert(typeof XYZService === "function");
 assert(typeof GetNumbersCommand === "function");
 assert(typeof TradeEventStreamCommand === "function");
 // errors
+assert(CodedThrottlingError.prototype instanceof XYZServiceServiceException);
+assert(HaltError.prototype instanceof XYZServiceServiceException);
+assert(MysteryThrottlingError.prototype instanceof XYZServiceServiceException);
+assert(RetryableError.prototype instanceof XYZServiceServiceException);
+assert(XYZServiceServiceException.prototype instanceof XYZServiceServiceException);
 assert(XYZServiceServiceException.prototype instanceof Error);
 console.log(`XYZService index test passed.`);

private/my-local-model-schema/test/index-types.ts

@@ -15,5 +15,10 @@ export type {
   TradeEventStreamRequest,
   TradeEventStreamResponse,
   Unit,
+  CodedThrottlingError,
+  HaltError,
+  MysteryThrottlingError,
+  RetryableError,
+  XYZServiceServiceException,
   XYZServiceServiceException,
 } from "../dist-types/index.d";

private/smithy-rpcv2-cbor-schema/test/index-objects.spec.mjs

@@ -1,10 +1,12 @@
 import {
+  ComplexError,
   EmptyInputOutputCommand,
   Float16Command,
   FooEnum,
   FractionalSecondsCommand,
   GreetingWithErrorsCommand,
   IntegerEnum,
+  InvalidGreeting,
   NoInputOutputCommand,
   OperationWithDefaultsCommand,
   OptionalInputOutputCommand,
@@ -19,6 +21,7 @@ import {
   SparseNullsOperationCommand,
   TestEnum,
   TestIntEnum,
+  ValidationException,
 } from "../dist-cjs/index.js";
 import assert from "node:assert";
 // clients
@@ -44,5 +47,8 @@ assert(typeof TestIntEnum === "object");
 assert(typeof FooEnum === "object");
 assert(typeof IntegerEnum === "object");
 // errors
+assert(ValidationException.prototype instanceof RpcV2ProtocolServiceException);
+assert(ComplexError.prototype instanceof RpcV2ProtocolServiceException);
+assert(InvalidGreeting.prototype instanceof RpcV2ProtocolServiceException);
 assert(RpcV2ProtocolServiceException.prototype instanceof Error);
 console.log(`RpcV2Protocol index test passed.`);

private/smithy-rpcv2-cbor-schema/test/index-types.ts

@@ -45,7 +45,10 @@ export type {
   TestIntEnum,
   FooEnum,
   IntegerEnum,
+  Unit,
+  ValidationExceptionField,
   ClientOptionalDefaults,
+  ComplexNestedErrorData,
   Defaults,
   EmptyStructure,
   Float16Output,
@@ -64,5 +67,8 @@ export type {
   SparseNullsOperationInputOutput,
   StructureListMember,
   GreetingStruct,
+  ValidationException,
+  ComplexError,
+  InvalidGreeting,
   RpcV2ProtocolServiceException,
 } from "../dist-types/index.d";

private/smithy-rpcv2-cbor/test/index-objects.spec.mjs

@@ -1,10 +1,12 @@
 import {
+  ComplexError,
   EmptyInputOutputCommand,
   Float16Command,
   FooEnum,
   FractionalSecondsCommand,
   GreetingWithErrorsCommand,
   IntegerEnum,
+  InvalidGreeting,
   NoInputOutputCommand,
   OperationWithDefaultsCommand,
   OptionalInputOutputCommand,
@@ -19,6 +21,7 @@ import {
   SparseNullsOperationCommand,
   TestEnum,
   TestIntEnum,
+  ValidationException,
 } from "../dist-cjs/index.js";
 import assert from "node:assert";
 // clients
@@ -44,5 +47,8 @@ assert(typeof TestIntEnum === "object");
 assert(typeof FooEnum === "object");
 assert(typeof IntegerEnum === "object");
 // errors
+assert(ValidationException.prototype instanceof RpcV2ProtocolServiceException);
+assert(ComplexError.prototype instanceof RpcV2ProtocolServiceException);
+assert(InvalidGreeting.prototype instanceof RpcV2ProtocolServiceException);
 assert(RpcV2ProtocolServiceException.prototype instanceof Error);
 console.log(`RpcV2Protocol index test passed.`);

private/smithy-rpcv2-cbor/test/index-types.ts

@@ -45,7 +45,10 @@ export type {
   TestIntEnum,
   FooEnum,
   IntegerEnum,
+  Unit,
+  ValidationExceptionField,
   ClientOptionalDefaults,
+  ComplexNestedErrorData,
   Defaults,
   EmptyStructure,
   Float16Output,
@@ -64,5 +67,8 @@ export type {
   SparseNullsOperationInputOutput,
   StructureListMember,
   GreetingStruct,
+  ValidationException,
+  ComplexError,
+  InvalidGreeting,
   RpcV2ProtocolServiceException,
 } from "../dist-types/index.d";

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/PackageApiValidationGenerator.java

@@ -30,7 +30,7 @@ public final class PackageApiValidationGenerator {
     private final TypeScriptSettings settings;
     private final Model model;
     private final SymbolProvider symbolProvider;
-    private final ServiceClosure serviceClosure;
+    private final ServiceClosure closure;
 
     public PackageApiValidationGenerator(
         TypeScriptWriter writer,
@@ -42,7 +42,7 @@ public PackageApiValidationGenerator(
         this.settings = settings;
         this.model = model;
         this.symbolProvider = symbolProvider;
-        serviceClosure = ServiceClosure.of(model, settings.getService(model));
+        closure = ServiceClosure.of(model, settings.getService(model));
     }
 
     /**
@@ -75,18 +75,18 @@ public void writeTypeIndexTest() {
                 }
 
                 // enums
-                TreeSet<Shape> enumShapes = serviceClosure.getEnums();
+                TreeSet<Shape> enumShapes = closure.getEnums();
                 for (Shape enumShape : enumShapes) {
                     writer.write("$L,", symbolProvider.toSymbol(enumShape).getName());
                 }
 
                 // structure & union types & modeled errors
-                TreeSet<Shape> structuralShapes = serviceClosure.getStructuralNonErrorShapes();
+                TreeSet<Shape> structuralShapes = closure.getStructuralNonErrorShapes();
                 for (Shape structuralShape : structuralShapes) {
                     writer.write("$L,", symbolProvider.toSymbol(structuralShape).getName());
                 }
 
-                TreeSet<Shape> errorShapes = serviceClosure.getErrorShapes();
+                TreeSet<Shape> errorShapes = closure.getErrorShapes();
                 for (Shape errorShape : errorShapes) {
                     writer.write("$L,", symbolProvider.toSymbol(errorShape).getName());
                 }
@@ -95,11 +95,11 @@ public void writeTypeIndexTest() {
                 writer.write("$L,", aggregateClientName + "ServiceException");
 
                 // waiters
-                serviceClosure.getWaiterNames().forEach(waiter -> {
+                closure.getWaiterNames().forEach(waiter -> {
                     writer.write("$L,", waiter);
                 });
                 // paginators
-                serviceClosure.getPaginatorNames().forEach(paginator -> {
+                closure.getPaginatorNames().forEach(paginator -> {
                     writer.write("$L,", paginator);
                 });
             }
@@ -150,7 +150,7 @@ public void writeRuntimeIndexTest() {
 
         // string shapes with enum trait do not generate anything if
         // any enum value does not have a name.
-        TreeSet<Shape> enumShapes = serviceClosure.getEnums().stream()
+        TreeSet<Shape> enumShapes = closure.getEnums().stream()
             .filter(shape -> shape
                 .getTrait(EnumTrait.class)
                 .map(EnumTrait::hasNames)
@@ -173,7 +173,7 @@ public void writeRuntimeIndexTest() {
 
         // modeled errors and synthetic base error
         writer.write("// errors");
-        TreeSet<Shape> errors = serviceClosure.getErrorShapes();
+        TreeSet<Shape> errors = closure.getErrorShapes();
         for (Shape error : errors) {
             Symbol errorSymbol = symbolProvider.toSymbol(error);
             writer.addRelativeImport(errorSymbol.getName(), null, cjsIndex);
@@ -187,7 +187,7 @@ public void writeRuntimeIndexTest() {
         writer.write("assert($L.prototype instanceof Error);", baseExceptionName);
 
         // waiters & paginators
-        TreeSet<String> waiterNames = serviceClosure.getWaiterNames();
+        TreeSet<String> waiterNames = closure.getWaiterNames();
         if (!waiterNames.isEmpty()) {
             writer.write("// waiters");
         }
@@ -198,7 +198,7 @@ public void writeRuntimeIndexTest() {
                 waiter
             );
         });
-        TreeSet<String> paginatorNames = serviceClosure.getPaginatorNames();
+        TreeSet<String> paginatorNames = closure.getPaginatorNames();
         if (!paginatorNames.isEmpty()) {
             writer.write("// paginators");
         }