wsocket 0.3.1

Lighweight, fast and native WebSocket implementation for 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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;

use tokio::io::{split, AsyncRead, AsyncWrite, ReadHalf, WriteHalf};
use tokio::sync::broadcast;

use crate::Close;

mod read;
mod write;

pub struct WebSocket<IO> {
  io: IO,
  max_payload_len: usize,
  #[cfg(feature = "client")]
  masking: bool,
  closed: Arc<AtomicBool>,
  close: broadcast::Sender<Close>,
}

impl<IO> WebSocket<IO> {
  #[inline]
  pub fn server(io: IO, max_payload_len: usize) -> Self {
    Self {
      io,
      max_payload_len,
      #[cfg(feature = "client")]
      masking: false,
      closed: Arc::new(AtomicBool::new(false)),
      close: broadcast::Sender::new(1),
    }
  }

  #[inline]
  #[cfg(feature = "client")]
  pub fn client(io: IO, max_payload_len: usize, masking: bool) -> Self {
    Self {
      io,
      max_payload_len,
      masking,
      closed: Arc::new(AtomicBool::new(false)),
      close: broadcast::Sender::new(1),
    }
  }

  pub fn is_closed(&self) -> bool {
    self.closed.load(Ordering::SeqCst)
  }

  fn set_closed(&self, close: Close) {
    self.closed.store(true, Ordering::SeqCst);
    let _ = self.close.send(close);
  }
}

impl<IO: AsyncWrite + AsyncRead> WebSocket<IO> {
  pub fn split(self) -> (WebSocket<ReadHalf<IO>>, WebSocket<WriteHalf<IO>>) {
    let (read, write) = split(self.io);
    (
      WebSocket {
        io: read,
        max_payload_len: self.max_payload_len,
        #[cfg(feature = "client")]
        masking: self.masking,
        closed: self.closed.clone(),
        close: self.close.clone(),
      },
      WebSocket {
        io: write,
        max_payload_len: self.max_payload_len,
        #[cfg(feature = "client")]
        masking: self.masking,
        closed: self.closed,
        close: self.close,
      },
    )
  }
}