System Initiative

use std::{ collections::HashMap, sync::Arc, time::Duration, }; use si_crypto::VeritechDecryptionKey; use si_data_nats::NatsClient; use si_pool_noodle::{ PoolNoodle, instance::cyclone::{ LocalUdsInstance, LocalUdsInstanceSpec, }, }; use tokio::sync::{ Mutex, oneshot, }; use veritech_core::ExecutionId; use crate::server::ServerMetadata; /// Application state. #[derive(Clone, Debug)] pub struct AppState { #[allow(unused)] pub metadata: Arc<ServerMetadata>, // NOTE(nick,fletcher,scott): this implements clone and the inner bits are wrapped in an Arc. // If that changes, then I hope you read this comment before that happens. pub cyclone_pool: PoolNoodle<LocalUdsInstance, LocalUdsInstanceSpec>, pub decryption_key: Arc<VeritechDecryptionKey>, // TODO(nick,fletcher,scott): make this mutable at runtime. pub cyclone_client_execution_timeout: Duration, pub nats: NatsClient, pub kill_senders: Arc<Mutex<HashMap<ExecutionId, oneshot::Sender<()>>>>, } impl AppState { /// Creates a new [`AppState`]. pub fn new( metadata: Arc<ServerMetadata>, cyclone_pool: PoolNoodle<LocalUdsInstance, LocalUdsInstanceSpec>, decryption_key: Arc<VeritechDecryptionKey>, cyclone_client_execution_timeout: Duration, nats: NatsClient, kill_senders: Arc<Mutex<HashMap<ExecutionId, oneshot::Sender<()>>>>, ) -> Self { Self { metadata, cyclone_pool, decryption_key, cyclone_client_execution_timeout, nats, kill_senders, } } /// Determine if the nats client is using a subject prefix. pub fn nats_subject_has_prefix(&self) -> bool { self.nats.metadata().subject_prefix().is_some() } } /// Application state for killing function executions. #[derive(Clone, Debug)] pub struct KillAppState { #[allow(unused)] pub metadata: Arc<ServerMetadata>, pub nats: NatsClient, pub kill_senders: Arc<Mutex<HashMap<ExecutionId, oneshot::Sender<()>>>>, } impl KillAppState { /// Creates a new [`KillAppState`]. pub fn new( metadata: Arc<ServerMetadata>, nats: NatsClient, kill_senders: Arc<Mutex<HashMap<ExecutionId, oneshot::Sender<()>>>>, ) -> Self { Self { metadata, nats, kill_senders, } } }