rsocket_rust_transport_unix 0.5.3

Unix Domain Socket RSocket transport implementation.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use bytes::{Buf, BytesMut};
use rsocket_rust::frame::Frame;
use rsocket_rust::utils::{u24, Writeable};
use std::io::{Error, ErrorKind};
use tokio_util::codec::{Decoder, Encoder};

pub struct LengthBasedFrameCodec;

impl Decoder for LengthBasedFrameCodec {
    type Item = Frame;
    type Error = Error;

    fn decode(&mut self, buf: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
        let actual = buf.len();
        if actual < 3 {
            return Ok(None);
        }
        let l = u24::read(buf).into();
        if actual < 3 + l {
            return Ok(None);
        }
        buf.advance(3);
        let mut bb = buf.split_to(l);
        match Frame::decode(&mut bb) {
            Ok(v) => Ok(Some(v)),
            Err(_e) => Err(Error::from(ErrorKind::InvalidInput)),
        }
    }
}

impl Encoder<Frame> for LengthBasedFrameCodec {
    type Error = Error;
    fn encode(&mut self, item: Frame, buf: &mut BytesMut) -> Result<(), Self::Error> {
        let l = item.len();
        buf.reserve(3 + l);
        u24::from(l).write_to(buf);
        item.write_to(buf);
        Ok(())
    }
}