Fixes codegen for multiple account ids

aws · Nov 2, 2022 · fbad6e2 · fbad6e2
1 parent 4f57f90
commit fbad6e2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/private/model/api/endpoint_arn.go b/private/model/api/endpoint_arn.go
@@ -76,11 +76,15 @@ const accountIDWithARNShapeTmplDef = `
 {{ range $_, $name := $.MemberNames -}}
 	{{ $elem := index $.MemberRefs $name -}}
 	{{ if $elem.AccountIDMemberWithARN -}}
-		// updateAccountID returns a pointer to a modified copy of input, 
+		{{ $FunctionName := $name }}
+		{{ if eq $name "AccountId" }}
+  		{{ $FunctionName = "AccountID" }}
+		{{ end }}
+		// update{{ $FunctionName }} returns a pointer to a modified copy of input,
 		// if account id is not provided, we update the account id in modified input
 		// if account id is provided, but doesn't match with the one in ARN, we throw an error
 		// if account id is not updated, we return nil. Note that original input is not modified. 
-		func (s {{ $.ShapeName }}) updateAccountID(accountId string) (interface{}, error) {
+		func (s {{ $.ShapeName }}) update{{ $FunctionName }}(accountId string) (interface{}, error) {
 			if s.{{ $name }} == nil {
 				s.{{ $name }} = aws.String(accountId)
 				return &s, nil