From 066e10ea95cbf8d88b2eb943763bb471bc7cb5ca Mon Sep 17 00:00:00 2001
From: Adam Fowler <adamfowler71@gmail.com>
Date: Wed, 16 Oct 2024 17:50:58 +0100
Subject: [PATCH] Use xmlNameTrait when outputting xmlRootNodeName

---
 Sources/SotoCodeGeneratorLib/AwsService+shapes.swift | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift b/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift
index d8d63a6..0fbbacb 100644
--- a/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift
+++ b/Sources/SotoCodeGeneratorLib/AwsService+shapes.swift
@@ -174,7 +174,12 @@ extension AwsService {
             } else if isInput {
                 // set XML root node name.
                 if serviceProtocolTrait is AwsProtocolsRestXmlTrait {
-                    xmlRootNodeName = payloadMember.key
+                    // use XML name if trait exists
+                    if let xmlNameTrait = payloadMember.value.trait(type: XmlNameTrait.self) {
+                        xmlRootNodeName = xmlNameTrait.value
+                    } else {
+                        xmlRootNodeName = payloadMember.key
+                    }
                 }
                 // currently only support request streaming of blobs
                 if payload is BlobShape,