aws_sdk_sagemaker/operation/
search.rs1#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
4#[non_exhaustive]
5pub struct Search;
6impl Search {
7 pub fn new() -> Self {
9 Self
10 }
11 pub(crate) async fn orchestrate(
12 runtime_plugins: &::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
13 input: crate::operation::search::SearchInput,
14 ) -> ::std::result::Result<
15 crate::operation::search::SearchOutput,
16 ::aws_smithy_runtime_api::client::result::SdkError<
17 crate::operation::search::SearchError,
18 ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
19 >,
20 > {
21 let map_err =
22 |err: ::aws_smithy_runtime_api::client::result::SdkError<
23 ::aws_smithy_runtime_api::client::interceptors::context::Error,
24 ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
25 >| { err.map_service_error(|err| err.downcast::<crate::operation::search::SearchError>().expect("correct error type")) };
26 let context = Self::orchestrate_with_stop_point(runtime_plugins, input, ::aws_smithy_runtime::client::orchestrator::StopPoint::None)
27 .await
28 .map_err(map_err)?;
29 let output = context.finalize().map_err(map_err)?;
30 ::std::result::Result::Ok(output.downcast::<crate::operation::search::SearchOutput>().expect("correct output type"))
31 }
32
33 pub(crate) async fn orchestrate_with_stop_point(
34 runtime_plugins: &::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
35 input: crate::operation::search::SearchInput,
36 stop_point: ::aws_smithy_runtime::client::orchestrator::StopPoint,
37 ) -> ::std::result::Result<
38 ::aws_smithy_runtime_api::client::interceptors::context::InterceptorContext,
39 ::aws_smithy_runtime_api::client::result::SdkError<
40 ::aws_smithy_runtime_api::client::interceptors::context::Error,
41 ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
42 >,
43 > {
44 let input = ::aws_smithy_runtime_api::client::interceptors::context::Input::erase(input);
45 ::aws_smithy_runtime::client::orchestrator::invoke_with_stop_point("sagemaker", "Search", input, runtime_plugins, stop_point).await
46 }
47
48 pub(crate) fn operation_runtime_plugins(
49 client_runtime_plugins: ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
50 client_config: &crate::config::Config,
51 config_override: ::std::option::Option<crate::config::Builder>,
52 ) -> ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins {
53 let mut runtime_plugins = client_runtime_plugins.with_operation_plugin(Self::new());
54 runtime_plugins = runtime_plugins.with_client_plugin(crate::auth_plugin::DefaultAuthOptionsPlugin::new(vec![
55 ::aws_runtime::auth::sigv4::SCHEME_ID,
56 ]));
57 if let ::std::option::Option::Some(config_override) = config_override {
58 for plugin in config_override.runtime_plugins.iter().cloned() {
59 runtime_plugins = runtime_plugins.with_operation_plugin(plugin);
60 }
61 runtime_plugins = runtime_plugins.with_operation_plugin(crate::config::ConfigOverrideRuntimePlugin::new(
62 config_override,
63 client_config.config.clone(),
64 &client_config.runtime_components,
65 ));
66 }
67 runtime_plugins
68 }
69}
70impl ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin for Search {
71 fn config(&self) -> ::std::option::Option<::aws_smithy_types::config_bag::FrozenLayer> {
72 let mut cfg = ::aws_smithy_types::config_bag::Layer::new("Search");
73
74 cfg.store_put(::aws_smithy_runtime_api::client::ser_de::SharedRequestSerializer::new(
75 SearchRequestSerializer,
76 ));
77 cfg.store_put(::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer::new(
78 SearchResponseDeserializer,
79 ));
80
81 cfg.store_put(::aws_smithy_runtime_api::client::auth::AuthSchemeOptionResolverParams::new(
82 ::aws_smithy_runtime_api::client::auth::static_resolver::StaticAuthSchemeOptionResolverParams::new(),
83 ));
84
85 cfg.store_put(::aws_smithy_runtime_api::client::orchestrator::SensitiveOutput);
86 cfg.store_put(::aws_smithy_runtime_api::client::orchestrator::Metadata::new("Search", "sagemaker"));
87 let mut signing_options = ::aws_runtime::auth::SigningOptions::default();
88 signing_options.double_uri_encode = true;
89 signing_options.content_sha256_header = false;
90 signing_options.normalize_uri_path = true;
91 signing_options.payload_override = None;
92
93 cfg.store_put(::aws_runtime::auth::SigV4OperationSigningConfig {
94 signing_options,
95 ..::std::default::Default::default()
96 });
97
98 ::std::option::Option::Some(cfg.freeze())
99 }
100
101 fn runtime_components(
102 &self,
103 _: &::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder,
104 ) -> ::std::borrow::Cow<'_, ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder> {
105 #[allow(unused_mut)]
106 let mut rcb = ::aws_smithy_runtime_api::client::runtime_components::RuntimeComponentsBuilder::new("Search")
107 .with_interceptor(::aws_smithy_runtime::client::stalled_stream_protection::StalledStreamProtectionInterceptor::default())
108 .with_interceptor(SearchEndpointParamsInterceptor)
109 .with_retry_classifier(::aws_smithy_runtime::client::retries::classifiers::TransientErrorClassifier::<
110 crate::operation::search::SearchError,
111 >::new())
112 .with_retry_classifier(::aws_smithy_runtime::client::retries::classifiers::ModeledAsRetryableClassifier::<
113 crate::operation::search::SearchError,
114 >::new())
115 .with_retry_classifier(::aws_runtime::retries::classifiers::AwsErrorCodeClassifier::<
116 crate::operation::search::SearchError,
117 >::new());
118
119 ::std::borrow::Cow::Owned(rcb)
120 }
121}
122
123#[derive(Debug)]
124struct SearchResponseDeserializer;
125impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for SearchResponseDeserializer {
126 fn deserialize_nonstreaming(
127 &self,
128 response: &::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
129 ) -> ::aws_smithy_runtime_api::client::interceptors::context::OutputOrError {
130 let (success, status) = (response.status().is_success(), response.status().as_u16());
131 let headers = response.headers();
132 let body = response.body().bytes().expect("body loaded");
133 #[allow(unused_mut)]
134 let mut force_error = false;
135 ::tracing::debug!(request_id = ?::aws_types::request_id::RequestId::request_id(response));
136 let parse_result = if !success && status != 200 || force_error {
137 crate::protocol_serde::shape_search::de_search_http_error(status, headers, body)
138 } else {
139 crate::protocol_serde::shape_search::de_search_http_response(status, headers, body)
140 };
141 crate::protocol_serde::type_erase_result(parse_result)
142 }
143}
144#[derive(Debug)]
145struct SearchRequestSerializer;
146impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for SearchRequestSerializer {
147 #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
148 fn serialize_input(
149 &self,
150 input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
151 _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
152 ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
153 let input = input.downcast::<crate::operation::search::SearchInput>().expect("correct type");
154 let _header_serialization_settings = _cfg
155 .load::<crate::serialization_settings::HeaderSerializationSettings>()
156 .cloned()
157 .unwrap_or_default();
158 let mut request_builder = {
159 fn uri_base(
160 _input: &crate::operation::search::SearchInput,
161 output: &mut ::std::string::String,
162 ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
163 use ::std::fmt::Write as _;
164 ::std::write!(output, "/").expect("formatting should succeed");
165 ::std::result::Result::Ok(())
166 }
167 #[allow(clippy::unnecessary_wraps)]
168 fn update_http_builder(
169 input: &crate::operation::search::SearchInput,
170 builder: ::http::request::Builder,
171 ) -> ::std::result::Result<::http::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
172 let mut uri = ::std::string::String::new();
173 uri_base(input, &mut uri)?;
174 ::std::result::Result::Ok(builder.method("POST").uri(uri))
175 }
176 let mut builder = update_http_builder(&input, ::http::request::Builder::new())?;
177 builder = _header_serialization_settings.set_default_header(builder, ::http::header::CONTENT_TYPE, "application/x-amz-json-1.1");
178 builder = _header_serialization_settings.set_default_header(
179 builder,
180 ::http::header::HeaderName::from_static("x-amz-target"),
181 "SageMaker.Search",
182 );
183 builder
184 };
185 let body = ::aws_smithy_types::body::SdkBody::from(crate::protocol_serde::shape_search::ser_search_input(&input)?);
186 if let Some(content_length) = body.content_length() {
187 let content_length = content_length.to_string();
188 request_builder = _header_serialization_settings.set_default_header(request_builder, ::http::header::CONTENT_LENGTH, &content_length);
189 }
190 ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
191 }
192}
193#[derive(Debug)]
194struct SearchEndpointParamsInterceptor;
195
196impl ::aws_smithy_runtime_api::client::interceptors::Intercept for SearchEndpointParamsInterceptor {
197 fn name(&self) -> &'static str {
198 "SearchEndpointParamsInterceptor"
199 }
200
201 fn read_before_execution(
202 &self,
203 context: &::aws_smithy_runtime_api::client::interceptors::context::BeforeSerializationInterceptorContextRef<
204 '_,
205 ::aws_smithy_runtime_api::client::interceptors::context::Input,
206 ::aws_smithy_runtime_api::client::interceptors::context::Output,
207 ::aws_smithy_runtime_api::client::interceptors::context::Error,
208 >,
209 cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
210 ) -> ::std::result::Result<(), ::aws_smithy_runtime_api::box_error::BoxError> {
211 let _input = context.input().downcast_ref::<SearchInput>().ok_or("failed to downcast to SearchInput")?;
212
213 let params = crate::config::endpoint::Params::builder()
214 .set_region(cfg.load::<::aws_types::region::Region>().map(|r| r.as_ref().to_owned()))
215 .set_use_dual_stack(cfg.load::<::aws_types::endpoint_config::UseDualStack>().map(|ty| ty.0))
216 .set_use_fips(cfg.load::<::aws_types::endpoint_config::UseFips>().map(|ty| ty.0))
217 .set_endpoint(cfg.load::<::aws_types::endpoint_config::EndpointUrl>().map(|ty| ty.0.clone()))
218 .build()
219 .map_err(|err| {
220 ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
221 })?;
222 cfg.interceptor_state()
223 .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
224 ::std::result::Result::Ok(())
225 }
226}
227
228#[non_exhaustive]
233#[derive(::std::fmt::Debug)]
234pub enum SearchError {
235 #[deprecated(note = "Matching `Unhandled` directly is not forwards compatible. Instead, match using a \
237 variable wildcard pattern and check `.code()`:
238 \
239 `err if err.code() == Some(\"SpecificExceptionCode\") => { /* handle the error */ }`
240 \
241 See [`ProvideErrorMetadata`](#impl-ProvideErrorMetadata-for-SearchError) for what information is available for the error.")]
242 Unhandled(crate::error::sealed_unhandled::Unhandled),
243}
244impl SearchError {
245 pub fn unhandled(
247 err: impl ::std::convert::Into<::std::boxed::Box<dyn ::std::error::Error + ::std::marker::Send + ::std::marker::Sync + 'static>>,
248 ) -> Self {
249 Self::Unhandled(crate::error::sealed_unhandled::Unhandled {
250 source: err.into(),
251 meta: ::std::default::Default::default(),
252 })
253 }
254
255 pub fn generic(err: ::aws_smithy_types::error::ErrorMetadata) -> Self {
257 Self::Unhandled(crate::error::sealed_unhandled::Unhandled {
258 source: err.clone().into(),
259 meta: err,
260 })
261 }
262 pub fn meta(&self) -> &::aws_smithy_types::error::ErrorMetadata {
267 match self {
268 Self::Unhandled(e) => &e.meta,
269 }
270 }
271}
272impl ::std::error::Error for SearchError {
273 fn source(&self) -> ::std::option::Option<&(dyn ::std::error::Error + 'static)> {
274 match self {
275 Self::Unhandled(_inner) => ::std::option::Option::Some(&*_inner.source),
276 }
277 }
278}
279impl ::std::fmt::Display for SearchError {
280 fn fmt(&self, f: &mut ::std::fmt::Formatter<'_>) -> ::std::fmt::Result {
281 match self {
282 Self::Unhandled(_inner) => {
283 if let ::std::option::Option::Some(code) = ::aws_smithy_types::error::metadata::ProvideErrorMetadata::code(self) {
284 write!(f, "unhandled error ({code})")
285 } else {
286 f.write_str("unhandled error")
287 }
288 }
289 }
290 }
291}
292impl ::aws_smithy_types::retry::ProvideErrorKind for SearchError {
293 fn code(&self) -> ::std::option::Option<&str> {
294 ::aws_smithy_types::error::metadata::ProvideErrorMetadata::code(self)
295 }
296 fn retryable_error_kind(&self) -> ::std::option::Option<::aws_smithy_types::retry::ErrorKind> {
297 ::std::option::Option::None
298 }
299}
300impl ::aws_smithy_types::error::metadata::ProvideErrorMetadata for SearchError {
301 fn meta(&self) -> &::aws_smithy_types::error::ErrorMetadata {
302 match self {
303 Self::Unhandled(_inner) => &_inner.meta,
304 }
305 }
306}
307impl ::aws_smithy_runtime_api::client::result::CreateUnhandledError for SearchError {
308 fn create_unhandled_error(
309 source: ::std::boxed::Box<dyn ::std::error::Error + ::std::marker::Send + ::std::marker::Sync + 'static>,
310 meta: ::std::option::Option<::aws_smithy_types::error::ErrorMetadata>,
311 ) -> Self {
312 Self::Unhandled(crate::error::sealed_unhandled::Unhandled {
313 source,
314 meta: meta.unwrap_or_default(),
315 })
316 }
317}
318impl ::aws_types::request_id::RequestId for crate::operation::search::SearchError {
319 fn request_id(&self) -> Option<&str> {
320 self.meta().request_id()
321 }
322}
323
324pub use crate::operation::search::_search_output::SearchOutput;
325
326pub use crate::operation::search::_search_input::SearchInput;
327
328mod _search_input;
329
330mod _search_output;
331
332pub mod builders;
334
335pub mod paginator;