use std::marker::PhantomData;
use anyhow::Error;
use crate::bufext::BufExt;
use crate::framing::Framing;
use crate::framing::FramingDecoded;
use crate::framing::FramingEncodedFinal;
use crate::thrift_protocol::ProtocolID;
pub struct SerializedMessage<'a, Name: ?Sized, Buffer> {
pub protocol: ProtocolID,
pub buffer: Buffer,
pub method_name: &'a Name,
}
pub trait ContextStack {
type Name: ?Sized;
type Frame: Framing;
fn pre_read(&mut self) -> Result<(), Error>;
fn on_read_data(
&mut self,
msg: SerializedMessage<Self::Name, FramingDecoded<Self::Frame>>,
) -> Result<(), Error>;
fn post_read(&mut self, bytes: u32) -> Result<(), Error>;
fn pre_write(&mut self) -> Result<(), Error>;
fn on_write_data(
&mut self,
msg: SerializedMessage<Self::Name, FramingEncodedFinal<Self::Frame>>,
) -> Result<(), Error>;
fn post_write(&mut self, bytes: u32) -> Result<(), Error>;
}
pub struct DummyContextStack<Name: ?Sized, Frame> {
_phantom: PhantomData<(Frame, Name)>,
}
impl<Name: ?Sized, Frame> DummyContextStack<Name, Frame> {
pub fn new() -> Self {
Self {
_phantom: PhantomData,
}
}
}
impl<Name: ?Sized, Frame: Framing> ContextStack for DummyContextStack<Name, Frame>
where
FramingEncodedFinal<Frame>: BufExt,
{
type Name = Name;
type Frame = Frame;
fn pre_read(&mut self) -> Result<(), Error> {
Ok(())
}
fn on_read_data(
&mut self,
_msg: SerializedMessage<Self::Name, FramingDecoded<Self::Frame>>,
) -> Result<(), Error> {
Ok(())
}
fn post_read(&mut self, _bytes: u32) -> Result<(), Error> {
Ok(())
}
fn pre_write(&mut self) -> Result<(), Error> {
Ok(())
}
fn on_write_data(
&mut self,
_msg: SerializedMessage<Self::Name, FramingEncodedFinal<Self::Frame>>,
) -> Result<(), Error> {
Ok(())
}
fn post_write(&mut self, _bytes: u32) -> Result<(), Error> {
Ok(())
}
}
#[cfg(test)]
mod test {
use bytes::Bytes;
use super::*;
fn assert_context_stack(_: &impl ContextStack) {}
#[test]
fn check_unsized() {
assert_context_stack(&DummyContextStack::<std::ffi::CStr, Bytes>::new());
}
}