use serde::{Deserialize, Serialize};
use std::{io::Error, path::Path};
use tokio::{
fs::File,
io::{AsyncWriteExt, BufWriter},
sync::mpsc,
};
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct LinkDump {
pub present: bool,
pub link_id: u128,
pub unconfirmed: bool,
pub tx_idle: bool,
pub tx_pending: bool,
pub tx_flushing: bool,
pub tx_flushed: bool,
pub tx_ack_queue: usize,
pub txed_unacked_data: usize,
pub txed_unacked_data_limit: usize,
pub txed_unacked_data_limit_increased_consecutively: usize,
pub total_sent: u64,
pub total_recved: u64,
pub roundtrip: f32,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ConnDump {
pub conn_id: u128,
pub runtime: f32,
pub txed_unacked: usize,
pub txed_unconsumed: usize,
pub txed_unconsumable: usize,
pub send_buffer: u32,
pub remote_receive_buffer: u32,
pub resend_queue: usize,
pub rxed_reliable_size: usize,
pub rxed_reliable_consumed_since_last_ack: usize,
pub link0: LinkDump,
pub link1: LinkDump,
pub link2: LinkDump,
pub link3: LinkDump,
pub link4: LinkDump,
pub link5: LinkDump,
pub link6: LinkDump,
pub link7: LinkDump,
pub link8: LinkDump,
pub link9: LinkDump,
}
pub async fn dump_to_json_line_file(
path: impl AsRef<Path>, mut rx: mpsc::Receiver<ConnDump>,
) -> Result<(), Error> {
let file = File::create(path).await?;
let mut writer = BufWriter::new(file);
while let Some(dump) = rx.recv().await {
let line = serde_json::to_vec(&dump).unwrap();
writer.write_all(&line).await?;
writer.write_u8(b'\n').await?;
}
writer.flush().await?;
Ok(())
}