rsocket_rust 0.6.0

rsocket-rust is an implementation of the RSocket protocol in Rust.
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use super::{utils, Body, Frame, REQUEST_MAX};
use crate::error::RSocketError;
use crate::utils::Writeable;
use bytes::{Buf, BufMut, Bytes, BytesMut};

#[derive(Debug, PartialEq)]
pub struct RequestN {
    n: u32,
}

pub struct RequestNBuilder {
    stream_id: u32,
    flag: u16,
    value: RequestN,
}

impl RequestNBuilder {
    fn new(stream_id: u32, flag: u16) -> RequestNBuilder {
        RequestNBuilder {
            stream_id,
            flag,
            value: RequestN { n: REQUEST_MAX },
        }
    }

    pub fn set_n(mut self, n: u32) -> Self {
        self.value.n = n;
        self
    }

    pub fn build(self) -> Frame {
        Frame::new(self.stream_id, Body::RequestN(self.value), self.flag)
    }
}

impl RequestN {
    pub(crate) fn decode(flag: u16, bf: &mut BytesMut) -> crate::Result<RequestN> {
        if bf.len() < 4 {
            Err(RSocketError::InCompleteFrame.into())
        } else {
            let n = bf.get_u32();
            Ok(RequestN { n })
        }
    }

    pub fn builder(stream_id: u32, flag: u16) -> RequestNBuilder {
        RequestNBuilder::new(stream_id, flag)
    }

    pub fn get_n(&self) -> u32 {
        self.n
    }
}

impl Writeable for RequestN {
    fn write_to(&self, bf: &mut BytesMut) {
        bf.put_u32(self.get_n())
    }

    fn len(&self) -> usize {
        4
    }
}