unitycatalog_client/codegen/
mod.rs

1// @generated — do not edit by hand.
2pub mod agent_skills;
3pub mod agents;
4pub mod catalogs;
5#[allow(clippy::too_many_arguments, clippy::doc_lazy_continuation)]
6pub mod client;
7pub mod credentials;
8pub mod entity_tag_assignments;
9pub mod external_locations;
10pub mod functions;
11pub mod model_versions;
12pub mod policies;
13pub mod providers;
14pub mod recipients;
15pub mod registered_models;
16pub mod schemas;
17pub mod shares;
18pub mod staging_tables;
19pub mod tables;
20pub mod tag_policies;
21pub mod temporary_credentials;
22pub mod volumes;
23#[allow(unused_imports)]
24pub use client::*;
25use futures::Future;
26pub(super) fn stream_paginated<F, Fut, S, T>(
27    state: S,
28    op: F,
29) -> impl futures::Stream<Item = crate::Result<T>>
30where
31    F: Fn(S, Option<String>) -> Fut + Copy,
32    Fut: Future<Output = crate::Result<(T, S, Option<String>)>>,
33{
34    enum PaginationState<T> {
35        Start(T),
36        HasMore(T, String),
37        Done,
38    }
39    futures::stream::unfold(PaginationState::Start(state), move |state| async move {
40        let (s, page_token) = match state {
41            PaginationState::Start(s) => (s, None),
42            PaginationState::HasMore(s, page_token) if !page_token.is_empty() => {
43                (s, Some(page_token))
44            }
45            _ => {
46                return None;
47            }
48        };
49        let (resp, s, continuation) = match op(s, page_token).await {
50            Ok(resp) => resp,
51            Err(e) => return Some((Err(e), PaginationState::Done)),
52        };
53        let next_state = match continuation {
54            Some(token) => PaginationState::HasMore(s, token),
55            None => PaginationState::Done,
56        };
57        Some((Ok(resp), next_state))
58    })
59}
unitycatalog_client/codegen/mod.rs

unitycatalog_client/codegen/
mod.rs
