sequential-exchange/src/single_thread.rs

use crate::{Error, SeqEx, SeqNo, TransportLayer};

pub struct ReplyGuard<'a, TL: TransportLayer>(&'a mut SeqEx<TL>, TL, SeqNo);
impl<'a, TL: TransportLayer> ReplyGuard<'a, TL> {
    pub fn seq_no(&self) -> SeqNo {
        self.0.seq_no()
    }
    pub fn reply_no(&self) -> SeqNo {
        self.2
    }
    pub fn reply(self, packet_data: TL::SendData) {
        self.0.reply_raw(self.1.clone(), self.2, packet_data);
        core::mem::forget(self);
    }
}
impl<'a, TL: TransportLayer> Drop for ReplyGuard<'a, TL> {
    fn drop(&mut self) {
        self.0.reply_empty_raw(self.1.clone(), self.2)
    }
}

impl<TL: TransportLayer> SeqEx<TL> {
    pub fn receive<P: Into<TL::RecvData>>(
        &mut self,
        app: TL,
        seq_no: SeqNo,
        reply_no: Option<SeqNo>,
        packet: P,
    ) -> Result<(ReplyGuard<'_, TL>, P, Option<TL::SendData>), Error> {
        self.receive_raw(app.clone(), seq_no, reply_no, packet)
            .map(|(reply_no, packet, data)| (ReplyGuard(self, app, reply_no), packet, data))
    }
    pub fn pump(&mut self, app: TL) -> Result<(ReplyGuard<'_, TL>, TL::RecvData, Option<TL::SendData>), Error> {
        self.pump_raw()
            .map(|(reply_no, packet, data)| (ReplyGuard(self, app, reply_no), packet, data))
    }
}