couchbase-core 1.0.1

Couchbase SDK core networking and protocol implementation, not intended for direct use
Documentation
/*
 *
 *  * Copyright (c) 2025 Couchbase, Inc.
 *  *
 *  * Licensed under the Apache License, Version 2.0 (the "License");
 *  * you may not use this file except in compliance with the License.
 *  * You may obtain a copy of the License at
 *  *
 *  *    http://www.apache.org/licenses/LICENSE-2.0
 *  *
 *  * Unless required by applicable law or agreed to in writing, software
 *  * distributed under the License is distributed on an "AS IS" BASIS,
 *  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  * See the License for the specific language governing permissions and
 *  * limitations under the License.
 *
 */

use crate::httpx::request::OnBehalfOfInfo;
use crate::retry::{RetryStrategy, DEFAULT_RETRY_STRATEGY};
use crate::service_type::ServiceType;
use std::sync::Arc;

#[derive(Copy, Debug, Default, Clone, Eq, PartialEq)]
pub enum ClusterState {
    #[default]
    Online,
    Degraded,
    Offline,
}

#[derive(Debug, Clone)]
#[non_exhaustive]
pub struct WaitUntilReadyOptions {
    pub desired_state: Option<ClusterState>,
    pub service_types: Option<Vec<ServiceType>>,
    pub retry_strategy: Arc<dyn RetryStrategy>,

    pub on_behalf_of: Option<OnBehalfOfInfo>,
}

impl Default for WaitUntilReadyOptions {
    fn default() -> Self {
        Self::new()
    }
}

impl WaitUntilReadyOptions {
    pub fn new() -> Self {
        Self {
            desired_state: None,
            service_types: None,
            retry_strategy: DEFAULT_RETRY_STRATEGY.clone(),
            on_behalf_of: None,
        }
    }

    pub fn desired_state(mut self, state: ClusterState) -> Self {
        self.desired_state = Some(state);
        self
    }

    pub fn service_types(mut self, service_types: Vec<ServiceType>) -> Self {
        self.service_types = Some(service_types);
        self
    }

    pub fn retry_strategy(mut self, retry_strategy: Arc<dyn RetryStrategy>) -> Self {
        self.retry_strategy = retry_strategy;
        self
    }

    pub fn on_behalf_of(mut self, info: OnBehalfOfInfo) -> Self {
        self.on_behalf_of = Some(info);
        self
    }
}