rsocket_rust/transport/
misc.rs

1use std::sync::atomic::{AtomicI64, AtomicU32, Ordering};
2use std::sync::Arc;
3
4use crate::frame::Frame;
5
6#[derive(Debug, Clone)]
7pub(crate) struct StreamID {
8    inner: Arc<AtomicU32>,
9}
10
11impl StreamID {
12    pub(crate) fn new(value: u32) -> StreamID {
13        let inner = Arc::new(AtomicU32::new(value));
14        StreamID { inner }
15    }
16
17    pub(crate) fn next(&self) -> u32 {
18        let counter = self.inner.clone();
19        counter.fetch_add(2, Ordering::SeqCst)
20    }
21}
22
23impl From<u32> for StreamID {
24    fn from(v: u32) -> StreamID {
25        StreamID::new(v)
26    }
27}
28
29#[derive(Debug, Clone)]
30pub(crate) struct Counter {
31    inner: Arc<AtomicI64>,
32}
33
34impl Counter {
35    pub(crate) fn new(value: i64) -> Counter {
36        Counter {
37            inner: Arc::new(AtomicI64::new(value)),
38        }
39    }
40
41    pub(crate) fn count_down(&self) -> i64 {
42        self.inner.fetch_add(-1, Ordering::SeqCst) - 1
43    }
44}
45
46#[inline]
47pub(crate) fn debug_frame(snd: bool, f: &Frame) {
48    if snd {
49        debug!("===> SND: {:?}", f);
50    } else {
51        debug!("<=== RCV: {:?}", f);
52    }
53}
rsocket_rust/transport/misc.rs

rsocket_rust/transport/
misc.rs
