CI fixes

smithy-lang · May 25, 2023 · dd21d51 · dd21d51
1 parent 5a4ab98
commit dd21d51
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/aws/rust-runtime/aws-config/external-types.toml b/aws/rust-runtime/aws-config/external-types.toml
@@ -9,6 +9,7 @@ allowed_external_types = [
    "aws_credential_types::provider::SharedCredentialsProvider",
    "aws_sdk_sts::types::_policy_descriptor_type::PolicyDescriptorType",
    "aws_smithy_async::rt::sleep::AsyncSleep",
+   "aws_smithy_async::time::TimeSource",
    "aws_smithy_client::bounds::SmithyConnector",
    "aws_smithy_client::erase::DynConnector",
    "aws_smithy_client::erase::boxclone::BoxCloneService",

diff --git a/aws/rust-runtime/aws-types/external-types.toml b/aws/rust-runtime/aws-types/external-types.toml
@@ -2,6 +2,8 @@ allowed_external_types = [
     "aws_credential_types::cache::CredentialsCache",
     "aws_credential_types::provider::SharedCredentialsProvider",
     "aws_smithy_async::rt::sleep::AsyncSleep",
+    "aws_smithy_async::time::TimeSource",
+    "aws_smithy_async::time::SharedTimeSource",
     "aws_smithy_client::http_connector",
     "aws_smithy_client::http_connector::HttpConnector",
     "aws_smithy_http::endpoint::Endpoint",

diff --git a/aws/rust-runtime/aws-types/src/sdk_config.rs b/aws/rust-runtime/aws-types/src/sdk_config.rs
@@ -573,8 +573,8 @@ impl SdkConfig {
     }
 
     /// Configured time source
-    pub fn time_source(&self) -> Option<SharedTimeSource> {
-        self.time_source.clone()
+    pub fn time_source(&self) -> Option<&SharedTimeSource> {
+        self.time_source.as_ref()
     }
 
     /// Configured app name

diff --git a/aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/SdkConfigDecorator.kt b/aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/SdkConfigDecorator.kt
@@ -77,7 +77,7 @@ class GenericSmithySdkConfigSettings : ClientCodegenDecorator {
                     ${section.serviceConfigBuilder}.set_sleep_impl(${section.sdkConfig}.sleep_impl());
 
                     ${section.serviceConfigBuilder}.set_http_connector(${section.sdkConfig}.http_connector().cloned());
-                    ${section.serviceConfigBuilder}.set_time_source(${section.sdkConfig}.time_source().clone());
+                    ${section.serviceConfigBuilder}.set_time_source(${section.sdkConfig}.time_source().cloned());
                     """,
                 )
             },