use crate::{
packet_route::link::{InfoData, InfoKind, InfoVrf},
LinkMessageBuilder,
};
#[derive(Debug)]
pub struct LinkVrf;
impl LinkVrf {
pub fn new(name: &str, table_id: u32) -> LinkMessageBuilder<Self> {
LinkMessageBuilder::<LinkVrf>::new(name).table_id(table_id)
}
}
impl LinkMessageBuilder<LinkVrf> {
pub fn new(name: &str) -> Self {
LinkMessageBuilder::<LinkVrf>::new_with_info_kind(InfoKind::Vrf)
.name(name.to_string())
}
pub fn append_info_data(mut self, info: InfoVrf) -> Self {
if let InfoData::Vrf(infos) = self
.info_data
.get_or_insert_with(|| InfoData::Vrf(Vec::new()))
{
infos.push(info);
}
self
}
pub fn table_id(self, table_id: u32) -> Self {
self.append_info_data(InfoVrf::TableId(table_id))
}
}