2021-07-03 07:03:52 +00:00
|
|
|
pub use crate::schemas::internal_protocol::*;
|
2019-10-16 15:47:56 +00:00
|
|
|
|
2021-07-03 07:03:52 +00:00
|
|
|
use bytes::{Buf, BufMut, BytesMut};
|
2019-10-16 15:47:56 +00:00
|
|
|
|
2021-07-03 07:03:52 +00:00
|
|
|
use std::convert::TryFrom;
|
2020-05-28 21:32:33 +00:00
|
|
|
|
2021-07-03 07:03:52 +00:00
|
|
|
use preserves::value::PackedReader;
|
|
|
|
use preserves::value::PackedWriter;
|
|
|
|
use preserves::value::Reader;
|
|
|
|
use preserves::value::Writer;
|
2019-10-16 15:47:56 +00:00
|
|
|
|
2021-07-03 07:03:52 +00:00
|
|
|
pub struct Codec;
|
2019-10-16 15:47:56 +00:00
|
|
|
|
2021-07-03 07:03:52 +00:00
|
|
|
impl tokio_util::codec::Decoder for Codec {
|
|
|
|
type Item = Packet;
|
2020-05-28 21:32:33 +00:00
|
|
|
type Error = Error;
|
2019-10-16 15:47:56 +00:00
|
|
|
fn decode(&mut self, bs: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
|
2020-07-10 15:07:49 +00:00
|
|
|
let mut r = PackedReader::decode_bytes(bs);
|
2021-07-03 07:03:52 +00:00
|
|
|
match r.next(false)? {
|
|
|
|
None => Ok(None),
|
|
|
|
Some(item) => {
|
|
|
|
let count = r.source.index;
|
2020-05-28 21:32:33 +00:00
|
|
|
bs.advance(count);
|
2021-07-03 07:03:52 +00:00
|
|
|
Ok(Some(Packet::try_from(&item)?))
|
2019-10-17 14:41:42 +00:00
|
|
|
}
|
2019-10-16 15:47:56 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-07-03 07:03:52 +00:00
|
|
|
impl tokio_util::codec::Encoder<&Packet> for Codec {
|
2020-05-28 21:32:33 +00:00
|
|
|
type Error = Error;
|
2021-07-03 07:03:52 +00:00
|
|
|
fn encode(&mut self, item: &Packet, bs: &mut BytesMut) -> Result<(), Self::Error> {
|
|
|
|
Ok(PackedWriter::new(&mut bs.writer()).write(&item.into())?)
|
2019-10-16 15:47:56 +00:00
|
|
|
}
|
|
|
|
}
|