From b114596e8eecf974e7b0a44a6f098c2fb9246c31 Mon Sep 17 00:00:00 2001
From: Kevin Stich <kevin@kstich.com>
Date: Mon, 16 Sep 2024 22:18:59 -0700
Subject: [PATCH] Use most common suffix for IDL inline IO

This commit updates IDL serialization to use the most common suffix
for serializing IDL models when inlining input/output shapes. Before
this update, the first suffix encountered would be used, even if it
was only used that one time.
---
 .../shapes/SmithyIdlModelSerializer.java      | 25 ++++++++++++++-----
 .../inferred-io/shared.smithy                 | 11 ++++++++
 2 files changed, 30 insertions(+), 6 deletions(-)

diff --git a/smithy-model/src/main/java/software/amazon/smithy/model/shapes/SmithyIdlModelSerializer.java b/smithy-model/src/main/java/software/amazon/smithy/model/shapes/SmithyIdlModelSerializer.java
index 53000a44ced..ba7ac270408 100644
--- a/smithy-model/src/main/java/software/amazon/smithy/model/shapes/SmithyIdlModelSerializer.java
+++ b/smithy-model/src/main/java/software/amazon/smithy/model/shapes/SmithyIdlModelSerializer.java
@@ -243,8 +243,8 @@ private Pair<String, String> determineInlineSuffixes(Model fullModel, Collection
             return Pair.of(inlineInputSuffix, inlineOutputSuffix);
         }
 
-        Set<String> inputSuffixes = new HashSet<>();
-        Set<String> outputSuffixes = new HashSet<>();
+        Map<String, Integer> inputSuffixes = new LinkedHashMap<>();
+        Map<String, Integer> outputSuffixes = new LinkedHashMap<>();
         for (Shape shape : shapes) {
             if (!shape.isOperationShape()) {
                 continue;
@@ -255,15 +255,28 @@ private Pair<String, String> determineInlineSuffixes(Model fullModel, Collection
             StructureShape output = fullModel.expectShape(operation.getOutputShape(), StructureShape.class);
 
             if (shapes.contains(input) && input.getId().getName().startsWith(operation.getId().getName())) {
-                inputSuffixes.add(input.getId().getName().substring(operation.getId().getName().length()));
+                String inputSuffix = input.getId().getName().substring(operation.getId().getName().length());
+                int inputCount = inputSuffixes.getOrDefault(inputSuffix, 0);
+                inputSuffixes.put(inputSuffix, ++inputCount);
             }
 
             if (shapes.contains(output) && output.getId().getName().startsWith(operation.getId().getName())) {
-                outputSuffixes.add(output.getId().getName().substring(operation.getId().getName().length()));
+                String outputSuffix = output.getId().getName().substring(operation.getId().getName().length());
+                int outputCount = outputSuffixes.getOrDefault(outputSuffix, 0);
+                outputSuffixes.put(outputSuffix, ++outputCount);
             }
         }
-        String inputSuffix = inputSuffixes.size() == 1 ? inputSuffixes.iterator().next() : inlineInputSuffix;
-        String outputSuffix = outputSuffixes.size() == 1 ? outputSuffixes.iterator().next() : inlineOutputSuffix;
+
+        String inputSuffix = inputSuffixes.entrySet()
+                .stream()
+                .max(Map.Entry.comparingByValue())
+                .map(Map.Entry::getKey)
+                .orElse(inlineInputSuffix);
+        String outputSuffix = outputSuffixes.entrySet()
+                .stream()
+                .max(Map.Entry.comparingByValue())
+                .map(Map.Entry::getKey)
+                .orElse(inlineInputSuffix);
         return Pair.of(inputSuffix, outputSuffix);
     }
 
diff --git a/smithy-model/src/test/resources/software/amazon/smithy/model/shapes/idl-serialization/inferred-io/shared.smithy b/smithy-model/src/test/resources/software/amazon/smithy/model/shapes/idl-serialization/inferred-io/shared.smithy
index 8dc75dc7764..b8d45df4d95 100644
--- a/smithy-model/src/test/resources/software/amazon/smithy/model/shapes/idl-serialization/inferred-io/shared.smithy
+++ b/smithy-model/src/test/resources/software/amazon/smithy/model/shapes/idl-serialization/inferred-io/shared.smithy
@@ -4,6 +4,11 @@ $operationOutputSuffix: "Response"
 
 namespace com.example
 
+operation NotShared {
+    input: NotSharedInput
+    output: NotSharedOutput
+}
+
 operation SharedCustomA {
     input := {}
     output := {}
@@ -13,3 +18,9 @@ operation SharedCustomB {
     input := {}
     output := {}
 }
+
+@input
+structure NotSharedInput {}
+
+@output
+structure NotSharedOutput {}