gestalt-sdk 0.0.1-alpha.19

// Code generated by sdkgen. DO NOT EDIT.

//! Generated native types and clients for cache.proto.

use crate::codec::cache::{
    from_wire_cache_delete_many_response, from_wire_cache_delete_response,
    from_wire_cache_get_many_response, from_wire_cache_get_response,
    from_wire_cache_touch_response, to_wire_cache_delete_many_request,
    to_wire_cache_delete_request, to_wire_cache_get_many_request, to_wire_cache_get_request,
    to_wire_cache_set_many_request, to_wire_cache_set_request, to_wire_cache_touch_request,
};
use crate::codec::host_service::{HostServiceChannel, connect_host_service, plain_channel};
use crate::generated::v1;
use crate::rpc_support::GestaltError;

/// CacheDeleteManyRequest removes multiple cache keys in one RPC.
///
/// Native message type for `gestalt.provider.v1.CacheDeleteManyRequest`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheDeleteManyRequest {
    /// The `keys` field.
    pub keys: Vec<String>,
}

/// CacheDeleteManyResponse reports how many keys were deleted.
///
/// Native message type for `gestalt.provider.v1.CacheDeleteManyResponse`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheDeleteManyResponse {
    /// The `deleted` field.
    pub deleted: i64,
}

/// CacheDeleteRequest removes one cache key.
///
/// Native message type for `gestalt.provider.v1.CacheDeleteRequest`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheDeleteRequest {
    /// The `key` field.
    pub key: String,
}

/// CacheDeleteResponse reports whether one key existed and was deleted.
///
/// Native message type for `gestalt.provider.v1.CacheDeleteResponse`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheDeleteResponse {
    /// The `deleted` field.
    pub deleted: bool,
}

/// CacheGetManyRequest loads multiple cache keys in one RPC.
///
/// Native message type for `gestalt.provider.v1.CacheGetManyRequest`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheGetManyRequest {
    /// The `keys` field.
    pub keys: Vec<String>,
}

/// CacheGetManyResponse returns every lookup result for GetMany.
///
/// Native message type for `gestalt.provider.v1.CacheGetManyResponse`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheGetManyResponse {
    /// The `entries` field.
    pub entries: Vec<CacheResult>,
}

/// CacheGetRequest loads one cache key.
///
/// Native message type for `gestalt.provider.v1.CacheGetRequest`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheGetRequest {
    /// The `key` field.
    pub key: String,
}

/// CacheGetResponse is the result of looking up one cache key.
///
/// Native message type for `gestalt.provider.v1.CacheGetResponse`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheGetResponse {
    /// The `found` field.
    pub found: bool,
    /// The `value` field.
    pub value: Vec<u8>,
}

/// CacheResult is one lookup result returned by GetMany.
///
/// Native message type for `gestalt.provider.v1.CacheResult`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheResult {
    /// The `key` field.
    pub key: String,
    /// The `found` field.
    pub found: bool,
    /// The `value` field.
    pub value: Vec<u8>,
}

/// CacheSetEntry is one key/value pair written by SetMany.
///
/// Native message type for `gestalt.provider.v1.CacheSetEntry`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheSetEntry {
    /// The `key` field.
    pub key: String,
    /// The `value` field.
    pub value: Vec<u8>,
}

/// CacheSetManyRequest writes multiple cache keys in one RPC.
///
/// Native message type for `gestalt.provider.v1.CacheSetManyRequest`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheSetManyRequest {
    /// The `entries` field.
    pub entries: Vec<CacheSetEntry>,
    /// The `ttl` field; None when unset.
    pub ttl: Option<std::time::Duration>,
}

/// CacheSetRequest writes one cache key.
///
/// Native message type for `gestalt.provider.v1.CacheSetRequest`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheSetRequest {
    /// The `key` field.
    pub key: String,
    /// The `value` field.
    pub value: Vec<u8>,
    /// ttl applies an optional expiration to the entry.
    ///
    /// The `ttl` field; None when unset.
    pub ttl: Option<std::time::Duration>,
}

/// CacheTouchRequest extends the TTL for one cache key.
///
/// Native message type for `gestalt.provider.v1.CacheTouchRequest`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheTouchRequest {
    /// The `key` field.
    pub key: String,
    /// The `ttl` field; None when unset.
    pub ttl: Option<std::time::Duration>,
}

/// CacheTouchResponse reports whether a key existed and had its TTL updated.
///
/// Native message type for `gestalt.provider.v1.CacheTouchResponse`.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct CacheTouchResponse {
    /// The `touched` field.
    pub touched: bool,
}

/// Cache models the shared Gestalt cache-provider protocol.
///
/// Client for the `gestalt.provider.v1.Cache` service.
pub struct Cache {
    inner: v1::cache_client::CacheClient<HostServiceChannel>,
    timeout: Option<std::time::Duration>,
}

impl Cache {
    /// Creates a client over an established channel.
    pub fn new(channel: tonic::transport::Channel) -> Self {
        Self {
            inner: v1::cache_client::CacheClient::new(plain_channel(channel)),
            timeout: None,
        }
    }

    /// Sets a deadline applied to every unary call; calls that run past it
    /// fail with DEADLINE_EXCEEDED. Streaming calls are unaffected.
    pub fn with_timeout(mut self, timeout: std::time::Duration) -> Self {
        self.timeout = Some(timeout);
        self
    }

    /// Connects to the `cache` host service described by the environment.
    pub async fn connect() -> Result<Self, GestaltError> {
        Self::connect_named("").await
    }

    /// Connects to the named `cache` host-service binding.
    pub async fn connect_named(name: &str) -> Result<Self, GestaltError> {
        Ok(Self {
            inner: v1::cache_client::CacheClient::new(connect_host_service("cache", name).await?),
            timeout: None,
        })
    }

    /// Calls `gestalt.provider.v1.Cache.Get`.
    pub async fn get(&mut self, key: String) -> Result<Option<Vec<u8>>, GestaltError> {
        let request = CacheGetRequest { key };
        let mut tonic_request = tonic::Request::new(to_wire_cache_get_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response =
            from_wire_cache_get_response(self.inner.get(tonic_request).await?.into_inner());
        if !response.found {
            return Ok(None);
        }
        Ok(Some(response.value))
    }

    /// Calls `gestalt.provider.v1.Cache.Get` with the full request and response messages.
    pub async fn get_raw(
        &mut self,
        request: CacheGetRequest,
    ) -> Result<CacheGetResponse, GestaltError> {
        let mut tonic_request = tonic::Request::new(to_wire_cache_get_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response = self.inner.get(tonic_request).await?;
        Ok(from_wire_cache_get_response(response.into_inner()))
    }

    /// Calls `gestalt.provider.v1.Cache.GetMany`.
    pub async fn get_many(
        &mut self,
        keys: Vec<String>,
    ) -> Result<std::collections::BTreeMap<String, Vec<u8>>, GestaltError> {
        let request = CacheGetManyRequest { keys };
        let mut tonic_request = tonic::Request::new(to_wire_cache_get_many_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response = from_wire_cache_get_many_response(
            self.inner.get_many(tonic_request).await?.into_inner(),
        );
        let mut out = std::collections::BTreeMap::new();
        for entry in response.entries {
            if entry.found {
                out.insert(entry.key, entry.value);
            }
        }
        Ok(out)
    }

    /// Calls `gestalt.provider.v1.Cache.GetMany` with the full request and response messages.
    pub async fn get_many_raw(
        &mut self,
        request: CacheGetManyRequest,
    ) -> Result<CacheGetManyResponse, GestaltError> {
        let mut tonic_request = tonic::Request::new(to_wire_cache_get_many_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response = self.inner.get_many(tonic_request).await?;
        Ok(from_wire_cache_get_many_response(response.into_inner()))
    }

    /// Calls `gestalt.provider.v1.Cache.Set`.
    pub async fn set(
        &mut self,
        key: String,
        value: Vec<u8>,
        ttl: Option<std::time::Duration>,
    ) -> Result<(), GestaltError> {
        let request = CacheSetRequest { key, value, ttl };
        let mut tonic_request = tonic::Request::new(to_wire_cache_set_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        self.inner.set(tonic_request).await?;
        Ok(())
    }

    /// Calls `gestalt.provider.v1.Cache.Set` with the full request and response messages.
    pub async fn set_raw(&mut self, request: CacheSetRequest) -> Result<(), GestaltError> {
        let mut tonic_request = tonic::Request::new(to_wire_cache_set_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        self.inner.set(tonic_request).await?;
        Ok(())
    }

    /// Calls `gestalt.provider.v1.Cache.SetMany`.
    pub async fn set_many(
        &mut self,
        entries: Vec<CacheSetEntry>,
        ttl: Option<std::time::Duration>,
    ) -> Result<(), GestaltError> {
        let request = CacheSetManyRequest { entries, ttl };
        let mut tonic_request = tonic::Request::new(to_wire_cache_set_many_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        self.inner.set_many(tonic_request).await?;
        Ok(())
    }

    /// Calls `gestalt.provider.v1.Cache.SetMany` with the full request and response messages.
    pub async fn set_many_raw(&mut self, request: CacheSetManyRequest) -> Result<(), GestaltError> {
        let mut tonic_request = tonic::Request::new(to_wire_cache_set_many_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        self.inner.set_many(tonic_request).await?;
        Ok(())
    }

    /// Calls `gestalt.provider.v1.Cache.Delete`.
    pub async fn delete(&mut self, key: String) -> Result<bool, GestaltError> {
        let request = CacheDeleteRequest { key };
        let mut tonic_request = tonic::Request::new(to_wire_cache_delete_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response =
            from_wire_cache_delete_response(self.inner.delete(tonic_request).await?.into_inner());
        Ok(response.deleted)
    }

    /// Calls `gestalt.provider.v1.Cache.Delete` with the full request and response messages.
    pub async fn delete_raw(
        &mut self,
        request: CacheDeleteRequest,
    ) -> Result<CacheDeleteResponse, GestaltError> {
        let mut tonic_request = tonic::Request::new(to_wire_cache_delete_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response = self.inner.delete(tonic_request).await?;
        Ok(from_wire_cache_delete_response(response.into_inner()))
    }

    /// Calls `gestalt.provider.v1.Cache.DeleteMany`.
    pub async fn delete_many(&mut self, keys: Vec<String>) -> Result<i64, GestaltError> {
        let request = CacheDeleteManyRequest { keys };
        let mut tonic_request = tonic::Request::new(to_wire_cache_delete_many_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response = from_wire_cache_delete_many_response(
            self.inner.delete_many(tonic_request).await?.into_inner(),
        );
        Ok(response.deleted)
    }

    /// Calls `gestalt.provider.v1.Cache.DeleteMany` with the full request and response messages.
    pub async fn delete_many_raw(
        &mut self,
        request: CacheDeleteManyRequest,
    ) -> Result<CacheDeleteManyResponse, GestaltError> {
        let mut tonic_request = tonic::Request::new(to_wire_cache_delete_many_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response = self.inner.delete_many(tonic_request).await?;
        Ok(from_wire_cache_delete_many_response(response.into_inner()))
    }

    /// Calls `gestalt.provider.v1.Cache.Touch`.
    pub async fn touch(
        &mut self,
        key: String,
        ttl: Option<std::time::Duration>,
    ) -> Result<bool, GestaltError> {
        let request = CacheTouchRequest { key, ttl };
        let mut tonic_request = tonic::Request::new(to_wire_cache_touch_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response =
            from_wire_cache_touch_response(self.inner.touch(tonic_request).await?.into_inner());
        Ok(response.touched)
    }

    /// Calls `gestalt.provider.v1.Cache.Touch` with the full request and response messages.
    pub async fn touch_raw(
        &mut self,
        request: CacheTouchRequest,
    ) -> Result<CacheTouchResponse, GestaltError> {
        let mut tonic_request = tonic::Request::new(to_wire_cache_touch_request(request));
        if let Some(timeout) = self.timeout {
            tonic_request.set_timeout(timeout);
        }
        let response = self.inner.touch(tonic_request).await?;
        Ok(from_wire_cache_touch_response(response.into_inner()))
    }
}