Error Deserializers – Revision 2: Add support for different error cod…

…e sanitization.
aws · Sep 10, 2024 · da7abb4 · da7abb4
1 parent 4c101ad
commit da7abb4
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/.../main/java/software/amazon/smithy/go/codegen/protocol/aws/AwsJson10ProtocolGenerator.java b/.../main/java/software/amazon/smithy/go/codegen/protocol/aws/AwsJson10ProtocolGenerator.java
@@ -235,7 +235,21 @@ func getProtocolErrorInfo(payload []byte) (typ, msg string, v $value:T, err erro
 
                 if jtyp, ok := jv["__type"]; ok {
                     typ = jtyp.(string)
+                } else if jtyp, ok := jv["code"]; ok {
+                    typ = jtyp.(string)
+                }
+                // TODO: Add in Header Check for "x-amzn-errortype"
+
+                hashIndex, colonIndex := 0, len(typ)-1
+                for idx := 0; idx < len(typ); idx++ {
+                    if typ[idx] == '#' && hashIndex == 0 {
+                        hashIndex = idx
+                    }
+                    if typ[idx] == ':' && colonIndex == len(typ)-1 {
+                        colonIndex = idx
+                    }
                 }
+                typ = typ[hashIndex:colonIndex+1]
 
                 if jmsg, ok := jv["message"]; ok {
                     msg = jmsg.(string)