move model to initialization in SensitiveDataFinder

smithy-lang · Mar 22, 2023 · 292a9bb · 292a9bb
1 parent 8addd74
commit 292a9bb
Show file tree

Hide file tree

Showing 6 changed files with 443 additions and 434 deletions.
diff --git a/...ipt-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java b/...ipt-codegen/src/main/java/software/amazon/smithy/typescript/codegen/CommandGenerator.java
diff --git a/...t-codegen/src/main/java/software/amazon/smithy/typescript/codegen/StructureGenerator.java b/...t-codegen/src/main/java/software/amazon/smithy/typescript/codegen/StructureGenerator.java
@@ -73,7 +73,7 @@ final class StructureGenerator implements Runnable {
     private final StructureShape shape;
     private final boolean includeValidation;
     private final RequiredMemberMode requiredMemberMode;
-    private final SensitiveDataFinder sensitiveDataFinder = new SensitiveDataFinder();
+    private final SensitiveDataFinder sensitiveDataFinder;
 
     /**
      * sets 'includeValidation' to 'false' and requiredMemberMode
@@ -96,6 +96,7 @@ final class StructureGenerator implements Runnable {
         this.shape = shape;
         this.includeValidation = includeValidation;
         this.requiredMemberMode = requiredMemberMode;
+        sensitiveDataFinder = new SensitiveDataFinder(model);
     }
 
     @Override
@@ -189,7 +190,7 @@ private void renderStructureNamespace(StructuredMemberWriter structuredMemberWri
         Symbol symbol = symbolProvider.toSymbol(shape);
         String objectParam = "obj";
 
-        if (sensitiveDataFinder.findsSensitiveData(shape, model)) {
+        if (sensitiveDataFinder.findsSensitiveDataIn(shape)) {
             writer.writeDocs("@internal");
             writer.openBlock("export const $LFilterSensitiveLog = ($L: $L): any => ({", "})",
                     symbol.getName(),