1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
/// Paginator for [`ListBatchLoadTasks`](crate::operation::list_batch_load_tasks::ListBatchLoadTasks)
pub struct ListBatchLoadTasksPaginator {
handle: std::sync::Arc<crate::client::Handle>,
builder: crate::operation::list_batch_load_tasks::builders::ListBatchLoadTasksInputBuilder,
stop_on_duplicate_token: bool,
}
impl ListBatchLoadTasksPaginator {
/// Create a new paginator-wrapper
pub(crate) fn new(
handle: std::sync::Arc<crate::client::Handle>,
builder: crate::operation::list_batch_load_tasks::builders::ListBatchLoadTasksInputBuilder,
) -> Self {
Self {
handle,
builder,
stop_on_duplicate_token: true,
}
}
/// Set the page size
///
/// _Note: this method will override any previously set value for `max_results`_
pub fn page_size(mut self, limit: i32) -> Self {
self.builder.max_results = ::std::option::Option::Some(limit);
self
}
/// Stop paginating when the service returns the same pagination token twice in a row.
///
/// Defaults to true.
///
/// For certain operations, it may be useful to continue on duplicate token. For example,
/// if an operation is for tailing a log file in real-time, then continuing may be desired.
/// This option can be set to `false` to accommodate these use cases.
pub fn stop_on_duplicate_token(mut self, stop_on_duplicate_token: bool) -> Self {
self.stop_on_duplicate_token = stop_on_duplicate_token;
self
}
/// Create the pagination stream
///
/// _Note:_ No requests will be dispatched until the stream is used
/// (e.g. with the [`.next().await`](aws_smithy_async::future::pagination_stream::PaginationStream::next) method).
pub fn send(
self,
) -> ::aws_smithy_async::future::pagination_stream::PaginationStream<
::std::result::Result<
crate::operation::list_batch_load_tasks::ListBatchLoadTasksOutput,
::aws_smithy_runtime_api::client::result::SdkError<
crate::operation::list_batch_load_tasks::ListBatchLoadTasksError,
::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
>,
>,
> {
// Move individual fields out of self for the borrow checker
let builder = self.builder;
let handle = self.handle;
let runtime_plugins = crate::operation::list_batch_load_tasks::ListBatchLoadTasks::operation_runtime_plugins(
handle.runtime_plugins.clone(),
&handle.conf,
::std::option::Option::None,
)
.with_operation_plugin(crate::sdk_feature_tracker::paginator::PaginatorFeatureTrackerRuntimePlugin::new());
::aws_smithy_async::future::pagination_stream::PaginationStream::new(::aws_smithy_async::future::pagination_stream::fn_stream::FnStream::new(
move |tx| {
::std::boxed::Box::pin(async move {
// Build the input for the first time. If required fields are missing, this is where we'll produce an early error.
let mut input = match builder
.build()
.map_err(::aws_smithy_runtime_api::client::result::SdkError::construction_failure)
{
::std::result::Result::Ok(input) => input,
::std::result::Result::Err(e) => {
let _ = tx.send(::std::result::Result::Err(e)).await;
return;
}
};
loop {
let resp = crate::operation::list_batch_load_tasks::ListBatchLoadTasks::orchestrate(&runtime_plugins, input.clone()).await;
// If the input member is None or it was an error
let done = match resp {
::std::result::Result::Ok(ref resp) => {
let new_token = crate::lens::reflens_list_batch_load_tasks_output_output_next_token(resp);
// Pagination is exhausted when the next token is an empty string
let is_empty = new_token.map(|token| token.is_empty()).unwrap_or(true);
if !is_empty && new_token == input.next_token.as_ref() && self.stop_on_duplicate_token {
true
} else {
input.next_token = new_token.cloned();
is_empty
}
}
::std::result::Result::Err(_) => true,
};
if tx.send(resp).await.is_err() {
// receiving end was dropped
return;
}
if done {
return;
}
}
})
},
))
}
}