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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

use bytes::Bytes;

use super::{request_payload::RequestPayload, Error, FrameFlags, FrameHeader, FrameType, RSocketFrame};
use crate::{Frame, RawPayload};

#[cfg_attr(not(target = "wasm32-unknown-unknown"), derive(Debug))]
#[must_use]
#[derive(Clone)]
pub struct RequestChannel(pub RequestPayload);

impl RequestChannel {
  pub(crate) fn from_payload(stream_id: u32, payload: RawPayload, flags: FrameFlags, initial_n: u32) -> Self {
    Self(RequestPayload::from_payload(
      stream_id,
      payload,
      Self::FRAME_TYPE,
      flags,
      initial_n,
    ))
  }
}

impl RSocketFrame<RequestChannel> for RequestChannel {
  const FRAME_TYPE: FrameType = FrameType::RequestChannel;

  fn stream_id(&self) -> u32 {
    self.0.stream_id
  }

  fn decode_all(mut buffer: Bytes) -> Result<Self, Error> {
    let header = FrameHeader::from_bytes(buffer.split_to(Frame::LEN_HEADER));
    Self::decode_frame(&header, buffer)
  }

  fn decode_frame(header: &FrameHeader, buffer: Bytes) -> Result<Self, Error> {
    Self::check_type(header)?;
    Ok(Self(RequestPayload::decode(header, buffer)?))
  }

  fn encode(self) -> Bytes {
    self.0.encode()
  }

  fn gen_header(&self) -> FrameHeader {
    self.0.gen_header()
  }

  fn get_flag(&self) -> FrameFlags {
    self.0.get_flags()
  }
}

impl From<RequestChannel> for RawPayload {
  fn from(req: RequestChannel) -> Self {
    req.0.into()
  }
}

#[cfg(test)]
mod test {
  use anyhow::Result;

  use super::*;

  static BYTES: &[u8] = include_bytes!("../../testdata/frame.request_channel.bin");

  #[test]
  fn test_decode() -> Result<()> {
    println!("RAW: {:?}", BYTES);
    let p = RequestChannel::decode_all(BYTES.into())?;
    assert_eq!(p.0.stream_id, 1234);
    Ok(())
  }

  #[test]
  fn test_encode() -> Result<()> {
    let payload = RequestPayload {
      frame_type: FrameType::RequestChannel,
      stream_id: 1234,
      metadata: Bytes::from("hello"),
      data: Bytes::from("hello"),
      follows: true,
      complete: true,
      initial_n: 0,
    };
    let this = RequestChannel(payload);
    let encoded = this.encode();
    assert_eq!(encoded, Bytes::from(BYTES));
    Ok(())
  }
}