aws_sdk_sagemaker/operation/list_inference_experiments/
paginator.rs

1// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
2/// Paginator for [`ListInferenceExperiments`](crate::operation::list_inference_experiments::ListInferenceExperiments)
3pub struct ListInferenceExperimentsPaginator {
4    handle: std::sync::Arc<crate::client::Handle>,
5    builder: crate::operation::list_inference_experiments::builders::ListInferenceExperimentsInputBuilder,
6    stop_on_duplicate_token: bool,
7}
8
9impl ListInferenceExperimentsPaginator {
10    /// Create a new paginator-wrapper
11    pub(crate) fn new(
12        handle: std::sync::Arc<crate::client::Handle>,
13        builder: crate::operation::list_inference_experiments::builders::ListInferenceExperimentsInputBuilder,
14    ) -> Self {
15        Self {
16            handle,
17            builder,
18            stop_on_duplicate_token: true,
19        }
20    }
21
22    /// Set the page size
23    ///
24    /// _Note: this method will override any previously set value for `max_results`_
25    pub fn page_size(mut self, limit: i32) -> Self {
26        self.builder.max_results = ::std::option::Option::Some(limit);
27        self
28    }
29
30    /// Create a flattened paginator
31    ///
32    /// This paginator automatically flattens results using `inference_experiments`. Queries to the underlying service
33    /// are dispatched lazily.
34    pub fn items(self) -> crate::operation::list_inference_experiments::paginator::ListInferenceExperimentsPaginatorItems {
35        crate::operation::list_inference_experiments::paginator::ListInferenceExperimentsPaginatorItems(self)
36    }
37
38    /// Stop paginating when the service returns the same pagination token twice in a row.
39    ///
40    /// Defaults to true.
41    ///
42    /// For certain operations, it may be useful to continue on duplicate token. For example,
43    /// if an operation is for tailing a log file in real-time, then continuing may be desired.
44    /// This option can be set to `false` to accommodate these use cases.
45    pub fn stop_on_duplicate_token(mut self, stop_on_duplicate_token: bool) -> Self {
46        self.stop_on_duplicate_token = stop_on_duplicate_token;
47        self
48    }
49
50    /// Create the pagination stream
51    ///
52    /// _Note:_ No requests will be dispatched until the stream is used
53    /// (e.g. with the [`.next().await`](aws_smithy_async::future::pagination_stream::PaginationStream::next) method).
54    pub fn send(
55        self,
56    ) -> ::aws_smithy_async::future::pagination_stream::PaginationStream<
57        ::std::result::Result<
58            crate::operation::list_inference_experiments::ListInferenceExperimentsOutput,
59            ::aws_smithy_runtime_api::client::result::SdkError<
60                crate::operation::list_inference_experiments::ListInferenceExperimentsError,
61                ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
62            >,
63        >,
64    > {
65        // Move individual fields out of self for the borrow checker
66        let builder = self.builder;
67        let handle = self.handle;
68        let runtime_plugins = crate::operation::list_inference_experiments::ListInferenceExperiments::operation_runtime_plugins(
69            handle.runtime_plugins.clone(),
70            &handle.conf,
71            ::std::option::Option::None,
72        )
73        .with_operation_plugin(crate::sdk_feature_tracker::paginator::PaginatorFeatureTrackerRuntimePlugin::new());
74        ::aws_smithy_async::future::pagination_stream::PaginationStream::new(::aws_smithy_async::future::pagination_stream::fn_stream::FnStream::new(
75            move |tx| {
76                ::std::boxed::Box::pin(async move {
77                    // Build the input for the first time. If required fields are missing, this is where we'll produce an early error.
78                    let mut input = match builder
79                        .build()
80                        .map_err(::aws_smithy_runtime_api::client::result::SdkError::construction_failure)
81                    {
82                        ::std::result::Result::Ok(input) => input,
83                        ::std::result::Result::Err(e) => {
84                            let _ = tx.send(::std::result::Result::Err(e)).await;
85                            return;
86                        }
87                    };
88                    loop {
89                        let resp =
90                            crate::operation::list_inference_experiments::ListInferenceExperiments::orchestrate(&runtime_plugins, input.clone())
91                                .await;
92                        // If the input member is None or it was an error
93                        let done = match resp {
94                            ::std::result::Result::Ok(ref resp) => {
95                                let new_token = crate::lens::reflens_list_inference_experiments_output_output_next_token(resp);
96                                // Pagination is exhausted when the next token is an empty string
97                                let is_empty = new_token.map(|token| token.is_empty()).unwrap_or(true);
98                                if !is_empty && new_token == input.next_token.as_ref() && self.stop_on_duplicate_token {
99                                    true
100                                } else {
101                                    input.next_token = new_token.cloned();
102                                    is_empty
103                                }
104                            }
105                            ::std::result::Result::Err(_) => true,
106                        };
107                        if tx.send(resp).await.is_err() {
108                            // receiving end was dropped
109                            return;
110                        }
111                        if done {
112                            return;
113                        }
114                    }
115                })
116            },
117        ))
118    }
119}
120
121/// Flattened paginator for `ListInferenceExperimentsPaginator`
122///
123/// This is created with [`.items()`](ListInferenceExperimentsPaginator::items)
124pub struct ListInferenceExperimentsPaginatorItems(ListInferenceExperimentsPaginator);
125
126impl ListInferenceExperimentsPaginatorItems {
127    /// Create the pagination stream
128    ///
129    /// _Note_: No requests will be dispatched until the stream is used
130    /// (e.g. with the [`.next().await`](aws_smithy_async::future::pagination_stream::PaginationStream::next) method).
131    ///
132    /// To read the entirety of the paginator, use [`.collect::<Result<Vec<_>, _>()`](aws_smithy_async::future::pagination_stream::PaginationStream::collect).
133    pub fn send(
134        self,
135    ) -> ::aws_smithy_async::future::pagination_stream::PaginationStream<
136        ::std::result::Result<
137            crate::types::InferenceExperimentSummary,
138            ::aws_smithy_runtime_api::client::result::SdkError<
139                crate::operation::list_inference_experiments::ListInferenceExperimentsError,
140                ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
141            >,
142        >,
143    > {
144        ::aws_smithy_async::future::pagination_stream::TryFlatMap::new(self.0.send()).flat_map(|page| {
145            crate::lens::lens_list_inference_experiments_output_output_inference_experiments(page)
146                .unwrap_or_default()
147                .into_iter()
148        })
149    }
150}