use serde::{Deserialize, Serialize};
use zenoh_keyexpr::keyexpr_tree::{IKeyExprTreeMut, KeBoxTree};
use zenoh_protocol::core::{key_expr::OwnedKeyExpr, CongestionControl, Priority, Reliability};
#[derive(Debug, Deserialize, Default, Serialize, Clone)]
pub struct PublisherQoSConfList(pub(crate) Vec<PublisherQoSConf>);
impl From<PublisherQoSConfList> for KeBoxTree<PublisherQoSConfig> {
fn from(value: PublisherQoSConfList) -> KeBoxTree<PublisherQoSConfig> {
let mut tree = KeBoxTree::new();
for conf in value.0 {
for key_expr in conf.key_exprs {
tree.insert(&key_expr, conf.config.clone());
}
}
tree
}
}
#[derive(Debug, Deserialize, Serialize, Clone)]
pub(crate) struct PublisherQoSConf {
pub key_exprs: Vec<OwnedKeyExpr>,
pub config: PublisherQoSConfig,
}
#[derive(Debug, Default, Deserialize, Serialize, Clone)]
pub struct PublisherQoSConfig {
pub congestion_control: Option<CongestionControlConf>,
pub priority: Option<PriorityConf>,
pub express: Option<bool>,
#[cfg(feature = "unstable")]
pub reliability: Option<ReliabilityConf>,
#[cfg(feature = "unstable")]
pub allowed_destination: Option<PublisherLocalityConf>,
}
#[derive(Debug, Deserialize, Serialize, Clone, Copy)]
#[serde(rename_all = "lowercase")]
pub enum CongestionControlConf {
Drop,
Block,
}
impl From<CongestionControlConf> for CongestionControl {
fn from(value: CongestionControlConf) -> Self {
match value {
CongestionControlConf::Drop => Self::Drop,
CongestionControlConf::Block => Self::Block,
}
}
}
impl From<CongestionControl> for CongestionControlConf {
fn from(value: CongestionControl) -> Self {
match value {
CongestionControl::Drop => Self::Drop,
CongestionControl::Block => Self::Block,
}
}
}
#[derive(Debug, Deserialize, Serialize, Clone, Copy)]
#[serde(rename_all = "snake_case")]
pub enum PriorityConf {
RealTime = 1,
InteractiveHigh = 2,
InteractiveLow = 3,
DataHigh = 4,
Data = 5,
DataLow = 6,
Background = 7,
}
impl From<PriorityConf> for Priority {
fn from(value: PriorityConf) -> Self {
match value {
PriorityConf::RealTime => Self::RealTime,
PriorityConf::InteractiveHigh => Self::InteractiveHigh,
PriorityConf::InteractiveLow => Self::InteractiveLow,
PriorityConf::DataHigh => Self::DataHigh,
PriorityConf::Data => Self::Data,
PriorityConf::DataLow => Self::DataLow,
PriorityConf::Background => Self::Background,
}
}
}
#[derive(Debug, Deserialize, Serialize, Clone, Copy)]
#[serde(rename_all = "snake_case")]
pub enum ReliabilityConf {
BestEffort,
Reliable,
}
impl From<ReliabilityConf> for Reliability {
fn from(value: ReliabilityConf) -> Self {
match value {
ReliabilityConf::BestEffort => Self::BestEffort,
ReliabilityConf::Reliable => Self::Reliable,
}
}
}
impl From<Reliability> for ReliabilityConf {
fn from(value: Reliability) -> Self {
match value {
Reliability::BestEffort => Self::BestEffort,
Reliability::Reliable => Self::Reliable,
}
}
}
#[derive(Debug, Deserialize, Serialize, Clone, Copy)]
#[serde(rename_all = "snake_case")]
pub enum PublisherLocalityConf {
SessionLocal,
Remote,
Any,
}
#[derive(Clone, Copy, Debug, Serialize, Deserialize, Eq, Hash, PartialEq)]
#[serde(rename_all = "snake_case")]
pub enum QosOverwriteMessage {
Put,
Delete,
Query,
Reply,
}
#[derive(Debug, Default, Deserialize, Serialize, Clone)]
pub struct QosOverwrites {
pub congestion_control: Option<CongestionControlConf>,
pub priority: Option<PriorityConf>,
pub express: Option<bool>,
}