Support implicitGlobalRegion

awslabs · Jun 18, 2024 · 8c9e46b · 8c9e46b
1 parent af41ebf
commit 8c9e46b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/aws-runtime/aws-endpoint/common/src/aws/sdk/kotlin/runtime/endpoint/functions/Functions.kt b/aws-runtime/aws-endpoint/common/src/aws/sdk/kotlin/runtime/endpoint/functions/Functions.kt
@@ -67,6 +67,7 @@ public data class PartitionConfig(
     public val dualStackDnsSuffix: String? = null,
     public val supportsFIPS: Boolean? = null,
     public val supportsDualStack: Boolean? = null,
+    public val implicitGlobalRegion: String? = null,
 ) {
     public fun mergeWith(other: PartitionConfig): PartitionConfig =
         PartitionConfig(
@@ -75,6 +76,7 @@ public data class PartitionConfig(
             other.dualStackDnsSuffix ?: dualStackDnsSuffix,
             other.supportsFIPS ?: supportsFIPS,
             other.supportsDualStack ?: supportsDualStack,
+            other.implicitGlobalRegion ?: implicitGlobalRegion,
         )
 }
 

diff --git a/...ntime/aws-endpoint/common/test/aws/sdk/kotlin/runtime/endpoint/functions/FunctionsTest.kt b/...ntime/aws-endpoint/common/test/aws/sdk/kotlin/runtime/endpoint/functions/FunctionsTest.kt
@@ -100,6 +100,7 @@ class FunctionsTest {
                 "us-west-2" to PartitionConfig(),
                 "aws-global" to PartitionConfig(
                     dnsSuffix = "override.amazonaws.com",
+                    implicitGlobalRegion = "implicit-global-region"
                 ),
             ),
             baseConfig = PartitionConfig(
@@ -144,6 +145,7 @@ class FunctionsTest {
                 dualStackDnsSuffix = "api.aws",
                 supportsFIPS = true,
                 supportsDualStack = true,
+                implicitGlobalRegion = "implicit-global-region"
             ),
             actual = partition(testPartitions, "aws-global"),
         )

diff --git a/...n/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/endpoints/PartitionsGenerator.kt b/...n/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/endpoints/PartitionsGenerator.kt
@@ -81,6 +81,9 @@ class PartitionsGenerator(
                         regionConfig.getBooleanMember("supportsDualStack").ifPresent {
                             write("supportsDualStack = #L,", it.value)
                         }
+                        regionConfig.getStringMember("implicitGlobalRegion").ifPresent {
+                            write("implicitGlobalRegion = #S,", it.value)
+                        }
                     }
                 }
             }
@@ -90,6 +93,7 @@ class PartitionsGenerator(
                 write("dualStackDnsSuffix = #S,", baseConfig.expectStringMember("dualStackDnsSuffix").value)
                 write("supportsFIPS = #L,", baseConfig.expectBooleanMember("supportsFIPS").value)
                 write("supportsDualStack = #L,", baseConfig.expectBooleanMember("supportsDualStack").value)
+                write("implicitGlobalRegion = #S,", baseConfig.expectStringMember("implicitGlobalRegion").value)
             }
         }
     }