aws-sdk-bedrockagentcorecontrol 1.55.0

AWS SDK for Amazon Bedrock AgentCore Control
Documentation
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.

///
/// Fluent builder for the `policy_generation_completed` waiter.
///
/// This builder is intended to be used similar to the other fluent builders for
/// normal operations on the client. However, instead of a `send` method, it has
/// a `wait` method that takes a maximum amount of time to wait.
///
/// Construct this fluent builder using the client by importing the
/// [`Waiters`](crate::client::Waiters) trait and calling the methods
/// prefixed with `wait_until`.
///
#[derive(::std::clone::Clone, ::std::fmt::Debug)]
pub struct PolicyGenerationCompletedFluentBuilder {
    handle: ::std::sync::Arc<crate::client::Handle>,
    inner: crate::operation::get_policy_generation::builders::GetPolicyGenerationInputBuilder,
}
impl PolicyGenerationCompletedFluentBuilder {
    /// Creates a new `PolicyGenerationCompletedFluentBuilder`.
    pub(crate) fn new(handle: ::std::sync::Arc<crate::client::Handle>) -> Self {
        Self {
            handle,
            inner: ::std::default::Default::default(),
        }
    }
    /// Access the GetPolicyGeneration as a reference.
    pub fn as_input(&self) -> &crate::operation::get_policy_generation::builders::GetPolicyGenerationInputBuilder {
        &self.inner
    }
    /// Wait until policy generation is completed
    pub async fn wait(
        self,
        max_wait: ::std::time::Duration,
    ) -> ::std::result::Result<
        crate::waiters::policy_generation_completed::PolicyGenerationCompletedFinalPoll,
        crate::waiters::policy_generation_completed::WaitUntilPolicyGenerationCompletedError,
    > {
        let input = self
            .inner
            .build()
            .map_err(::aws_smithy_runtime_api::client::waiters::error::WaiterError::construction_failure)?;
        let runtime_plugins = crate::operation::get_policy_generation::GetPolicyGeneration::operation_runtime_plugins(
            self.handle.runtime_plugins.clone(),
            &self.handle.conf,
            ::std::option::Option::None,
        )
        .with_operation_plugin(crate::sdk_feature_tracker::waiter::WaiterFeatureTrackerRuntimePlugin::new());
        let mut cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
        let runtime_components_builder = runtime_plugins
            .apply_client_configuration(&mut cfg)
            .map_err(::aws_smithy_runtime_api::client::waiters::error::WaiterError::construction_failure)?;
        let time_components = runtime_components_builder.into_time_components();
        let sleep_impl = time_components.sleep_impl().expect("a sleep impl is required by waiters");
        let time_source = time_components.time_source().expect("a time source is required by waiters");

        let acceptor = move |result: ::std::result::Result<
            &crate::operation::get_policy_generation::GetPolicyGenerationOutput,
            &crate::operation::get_policy_generation::GetPolicyGenerationError,
        >| {
            // Matches: {"output":{"path":"status","expected":"GENERATED","comparator":"stringEquals"}}
            if crate::waiters::matchers::match_get_policy_generation_ba0115330a635f760(result) {
                return ::aws_smithy_runtime::client::waiters::AcceptorState::Success;
            }
            // Matches: {"output":{"path":"status","expected":"GENERATING","comparator":"stringEquals"}}
            if crate::waiters::matchers::match_get_policy_generation_ec5499a5e809693f8(result) {
                return ::aws_smithy_runtime::client::waiters::AcceptorState::Retry;
            }
            // Matches: {"output":{"path":"status","expected":"GENERATE_FAILED","comparator":"stringEquals"}}
            if crate::waiters::matchers::match_get_policy_generation_24b46858db5dce2c4(result) {
                return ::aws_smithy_runtime::client::waiters::AcceptorState::Failure;
            }
            // Matches: {"output":{"path":"status","expected":"DELETE_FAILED","comparator":"stringEquals"}}
            if crate::waiters::matchers::match_get_policy_generation_184758db81b6271a8(result) {
                return ::aws_smithy_runtime::client::waiters::AcceptorState::Failure;
            }
            ::aws_smithy_runtime::client::waiters::AcceptorState::NoAcceptorsMatched
        };
        let operation = move || {
            let input = input.clone();
            let runtime_plugins = runtime_plugins.clone();
            async move { crate::operation::get_policy_generation::GetPolicyGeneration::orchestrate(&runtime_plugins, input).await }
        };
        let orchestrator = ::aws_smithy_runtime::client::waiters::WaiterOrchestrator::builder()
            .min_delay(::std::time::Duration::from_secs(5))
            .max_delay(::std::time::Duration::from_secs(120))
            .max_wait(max_wait)
            .time_source(time_source)
            .sleep_impl(sleep_impl)
            .acceptor(acceptor)
            .operation(operation)
            .build();
        ::aws_smithy_runtime::client::waiters::attach_waiter_tracing_span(orchestrator.orchestrate()).await
    }
    /// <p>The unique identifier of the policy generation request to be retrieved. This must be a valid generation ID from a previous <a href="https://docs.aws.amazon.com/bedrock-agentcore-control/latest/APIReference/API_StartPolicyGeneration.html">StartPolicyGeneration</a> call.</p>
    pub fn policy_generation_id(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.policy_generation_id(input.into());
        self
    }
    /// <p>The unique identifier of the policy generation request to be retrieved. This must be a valid generation ID from a previous <a href="https://docs.aws.amazon.com/bedrock-agentcore-control/latest/APIReference/API_StartPolicyGeneration.html">StartPolicyGeneration</a> call.</p>
    pub fn set_policy_generation_id(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_policy_generation_id(input);
        self
    }
    /// <p>The unique identifier of the policy generation request to be retrieved. This must be a valid generation ID from a previous <a href="https://docs.aws.amazon.com/bedrock-agentcore-control/latest/APIReference/API_StartPolicyGeneration.html">StartPolicyGeneration</a> call.</p>
    pub fn get_policy_generation_id(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_policy_generation_id()
    }
    /// <p>The identifier of the policy engine associated with the policy generation request. This provides the context for the generation operation and schema validation.</p>
    pub fn policy_engine_id(mut self, input: impl ::std::convert::Into<::std::string::String>) -> Self {
        self.inner = self.inner.policy_engine_id(input.into());
        self
    }
    /// <p>The identifier of the policy engine associated with the policy generation request. This provides the context for the generation operation and schema validation.</p>
    pub fn set_policy_engine_id(mut self, input: ::std::option::Option<::std::string::String>) -> Self {
        self.inner = self.inner.set_policy_engine_id(input);
        self
    }
    /// <p>The identifier of the policy engine associated with the policy generation request. This provides the context for the generation operation and schema validation.</p>
    pub fn get_policy_engine_id(&self) -> &::std::option::Option<::std::string::String> {
        self.inner.get_policy_engine_id()
    }
}

/// Successful return type for the `policy_generation_completed` waiter.
pub type PolicyGenerationCompletedFinalPoll = ::aws_smithy_runtime_api::client::waiters::FinalPoll<
    crate::operation::get_policy_generation::GetPolicyGenerationOutput,
    ::aws_smithy_runtime_api::client::result::SdkError<
        crate::operation::get_policy_generation::GetPolicyGenerationError,
        ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
    >,
>;

/// Error type for the `policy_generation_completed` waiter.
pub type WaitUntilPolicyGenerationCompletedError = ::aws_smithy_runtime_api::client::waiters::error::WaiterError<
    crate::operation::get_policy_generation::GetPolicyGenerationOutput,
    crate::operation::get_policy_generation::GetPolicyGenerationError,
>;