diff options
Diffstat (limited to 'crates/configuration/src/state')
| -rw-r--r-- | crates/configuration/src/state/cache_key.rs | 2 | ||||
| -rw-r--r-- | crates/configuration/src/state/typology.rs | 19 | ||||
| -rw-r--r-- | crates/configuration/src/state/typology/mutate_typology.rs | 170 | ||||
| -rw-r--r-- | crates/configuration/src/state/typology/query_typology.rs | 88 |
4 files changed, 279 insertions, 0 deletions
diff --git a/crates/configuration/src/state/cache_key.rs b/crates/configuration/src/state/cache_key.rs index a99700e..dd2ad12 100644 --- a/crates/configuration/src/state/cache_key.rs +++ b/crates/configuration/src/state/cache_key.rs @@ -5,6 +5,7 @@ pub enum CacheKey<'a> { ActiveRouting, Routing(&'a uuid::Uuid), Rule { id: &'a str, version: &'a str }, + Typology { id: &'a str, version: &'a str }, } impl ToRedisArgs for CacheKey<'_> { @@ -16,6 +17,7 @@ impl ToRedisArgs for CacheKey<'_> { CacheKey::ActiveRouting => "routing.active".into(), CacheKey::Routing(uuid) => format!("routing.{uuid}"), CacheKey::Rule { id, version } => format!("rule.{id}.{version}"), + CacheKey::Typology { id, version } => format!("typology.{id}.{version}"), }; out.write_arg(value.as_bytes()); diff --git a/crates/configuration/src/state/typology.rs b/crates/configuration/src/state/typology.rs new file mode 100644 index 0000000..1c7090a --- /dev/null +++ b/crates/configuration/src/state/typology.rs @@ -0,0 +1,19 @@ +use warden_core::configuration::typology::{TypologyConfiguration, TypologyConfigurationRequest}; + +use crate::state::cache_key::CacheKey; + +pub mod mutate_typology; +pub mod query_typology; + +pub struct TypologyRow { + pub configuration: sqlx::types::Json<TypologyConfiguration>, +} + +impl<'a> From<&'a TypologyConfigurationRequest> for CacheKey<'a> { + fn from(value: &'a TypologyConfigurationRequest) -> Self { + Self::Typology { + id: &value.id, + version: &value.version, + } + } +} diff --git a/crates/configuration/src/state/typology/mutate_typology.rs b/crates/configuration/src/state/typology/mutate_typology.rs new file mode 100644 index 0000000..f2ab2cc --- /dev/null +++ b/crates/configuration/src/state/typology/mutate_typology.rs @@ -0,0 +1,170 @@ +use opentelemetry_semantic_conventions::attribute; +use tonic::{Request, Response, Status, async_trait}; +use tracing::{Instrument, error, info_span}; +use tracing_opentelemetry::OpenTelemetrySpanExt; +use uuid::Uuid; +use warden_core::configuration::{ + ConfigKind, ReloadEvent, + typology::{ + DeleteTypologyConfigurationRequest, TypologyConfiguration, UpdateTypologyConfigRequest, + mutate_typologies_server::MutateTypologies, + }, +}; + +use crate::state::{ + AppHandle, cache_key::CacheKey, invalidate_cache, publish_reload, typology::TypologyRow, +}; + +#[async_trait] +impl MutateTypologies for AppHandle { + async fn create_typology_configuration( + &self, + request: Request<TypologyConfiguration>, + ) -> Result<Response<TypologyConfiguration>, Status> { + let request = request.into_inner(); + let span = info_span!("create.configuration.typology"); + span.set_attribute(attribute::DB_SYSTEM_NAME, "postgres"); + span.set_attribute(attribute::DB_OPERATION_NAME, "insert"); + span.set_attribute(attribute::DB_COLLECTION_NAME, "typology"); + span.set_attribute("otel.kind", "client"); + + sqlx::query!( + "insert into typology (uuid, configuration) values ($1, $2)", + Uuid::now_v7(), + sqlx::types::Json(&request) as _, + ) + .execute(&self.services.postgres) + .instrument(span) + .await + .map_err(|e| { + error!("{e}"); + tonic::Status::internal(e.to_string()) + })?; + + Ok(tonic::Response::new(request)) + } + + async fn update_typology_configuration( + &self, + request: Request<UpdateTypologyConfigRequest>, + ) -> Result<Response<TypologyConfiguration>, Status> { + let conf = self + .app_config + .nats + .subject + .split(".") + .next() + .expect("bad config"); + + let request = request.into_inner(); + + let config = request.configuration.expect("configuration to be provided"); + + let span = info_span!("update.configuration.typology"); + span.set_attribute(attribute::DB_SYSTEM_NAME, "postgres"); + span.set_attribute(attribute::DB_OPERATION_NAME, "update"); + span.set_attribute(attribute::DB_COLLECTION_NAME, "typology"); + span.set_attribute("otel.kind", "client"); + + sqlx::query!( + r#" + update typology + set configuration = $1 + where id = $2 and version = $3 + "#, + sqlx::types::Json(&config) as _, + config.id, + config.version, + ) + .execute(&self.services.postgres) + .instrument(span) + .await + .map_err(|e| { + error!("{e}"); + tonic::Status::internal(e.to_string()) + })?; + + let (_del_result, _publish_result) = tokio::try_join!( + invalidate_cache( + self, + CacheKey::Typology { + id: &config.id, + version: &config.version, + } + ), + publish_reload( + self, + conf, + ReloadEvent { + kind: ConfigKind::Typology.into(), + id: Some(config.id.to_owned()), + version: Some(config.version.to_owned()), + } + ) + )?; + + Ok(Response::new(config)) + } + + async fn delete_typology_configuration( + &self, + request: Request<DeleteTypologyConfigurationRequest>, + ) -> Result<Response<TypologyConfiguration>, Status> { + let conf = self + .app_config + .nats + .subject + .split(".") + .next() + .expect("bad config"); + + let request = request.into_inner(); + + let span = info_span!("delete.configuration.typology"); + span.set_attribute(attribute::DB_SYSTEM_NAME, "postgres"); + span.set_attribute(attribute::DB_OPERATION_NAME, "delete"); + span.set_attribute(attribute::DB_COLLECTION_NAME, "typology"); + span.set_attribute("otel.kind", "client"); + + let updated = sqlx::query_as!( + TypologyRow, + r#" + delete from typology + where id = $1 and version = $2 + returning configuration as "configuration: sqlx::types::Json<TypologyConfiguration>" + "#, + request.id, + request.version, + ) + .fetch_one(&self.services.postgres) + .instrument(span) + .await + .map_err(|e| { + error!("{e}"); + tonic::Status::internal(e.to_string()) + })?; + + let (_del_result, _publish_result) = tokio::try_join!( + invalidate_cache( + self, + CacheKey::Typology { + id: &request.id, + version: &request.version, + } + ), + publish_reload( + self, + conf, + ReloadEvent { + kind: ConfigKind::Typology.into(), + id: Some(request.id.to_owned()), + version: Some(request.version.to_owned()), + } + ) + )?; + + let res = updated.configuration.0; + + Ok(Response::new(res)) + } +} diff --git a/crates/configuration/src/state/typology/query_typology.rs b/crates/configuration/src/state/typology/query_typology.rs new file mode 100644 index 0000000..c71317b --- /dev/null +++ b/crates/configuration/src/state/typology/query_typology.rs @@ -0,0 +1,88 @@ +use opentelemetry_semantic_conventions::attribute; +use prost::Message; +use tonic::{Request, Response, Status, async_trait}; +use tracing::{Instrument, debug, info_span, instrument, warn}; +use tracing_opentelemetry::OpenTelemetrySpanExt; +use warden_core::configuration::typology::{ + GetTypologyConfigResponse, TypologyConfiguration, TypologyConfigurationRequest, + query_typologies_server::QueryTypologies, +}; +use warden_stack::redis::AsyncCommands; + +use crate::state::{AppHandle, cache_key::CacheKey, typology::TypologyRow}; + +#[async_trait] +impl QueryTypologies for AppHandle { + #[instrument(skip(self, request), Err(Debug))] + async fn get_typology_configuration( + &self, + request: Request<TypologyConfigurationRequest>, + ) -> Result<Response<GetTypologyConfigResponse>, Status> { + let data = request.into_inner(); + let mut cache = self + .services + .cache + .get() + .await + .map_err(|e| tonic::Status::internal(e.to_string()))?; + + let key = CacheKey::from(&data); + + let configuration = cache.get::<_, Vec<u8>>(&key).await.map(|value| { + if !value.is_empty() { + TypologyConfiguration::decode(value.as_ref()).ok() + } else { + None + } + }); + + if let Ok(Some(typology_config)) = configuration { + return Ok(tonic::Response::new(GetTypologyConfigResponse { + configuration: Some(typology_config), + })); + } + + let span = info_span!("get.typology"); + span.set_attribute(attribute::DB_SYSTEM_NAME, "postgres"); + span.set_attribute(attribute::DB_OPERATION_NAME, "select"); + span.set_attribute(attribute::DB_COLLECTION_NAME, "typology"); + span.set_attribute("otel.kind", "client"); + + let config = sqlx::query_as!( + TypologyRow, + r#"select configuration as "configuration: sqlx::types::Json<TypologyConfiguration>" from typology where + id = $1 and version = $2"#, + data.id, + data.version, + ) + .fetch_optional(&self.services.postgres) + .instrument(span) + .await.map_err(|e| tonic::Status::internal(e.to_string()))?; + + let config = config.map(|transaction| { + debug!(id = ?transaction.configuration.0.id, "found config"); + transaction.configuration.0 + }); + + match config { + Some(config) => { + let bytes = config.encode_to_vec(); + let span = info_span!("cache.set"); + span.set_attribute(attribute::DB_SYSTEM_NAME, "valkey"); + span.set_attribute(attribute::DB_OPERATION_NAME, "set"); + span.set_attribute("otel.kind", "client"); + + if let Err(e) = cache.set::<_, _, ()>(&key, bytes).instrument(span).await { + warn!("{e}"); + }; + + Ok(tonic::Response::new(GetTypologyConfigResponse { + configuration: Some(config), + })) + } + None => Ok(tonic::Response::new(GetTypologyConfigResponse { + configuration: None, + })), + } + } +} |