feox-server 0.5.2

High-performance Redis-compatible server for Feox DB
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
79
80
81
82
83
84
85
86
87
88
89

use super::ThreadId;
use bytes::Bytes;

#[derive(Debug, Clone)]
pub enum BroadcastMsg {
    Publish {
        channel: Vec<u8>,
        message: Bytes,
        exclude_thread: Option<ThreadId>,
    },
    PatternPublish {
        channel: Vec<u8>,
        message: Bytes,
        exclude_thread: Option<ThreadId>,
    },
}

#[derive(Debug, Clone)]
pub enum PubSubMessage {
    Message {
        channel: Vec<u8>,
        payload: Bytes,
    },
    PatternMessage {
        pattern: Vec<u8>,
        channel: Vec<u8>,
        payload: Bytes,
    },
    Subscribe {
        channel: Vec<u8>,
        count: usize,
    },
    Unsubscribe {
        channel: Option<Vec<u8>>,
        count: usize,
    },
    PSubscribe {
        pattern: Vec<u8>,
        count: usize,
    },
    PUnsubscribe {
        pattern: Option<Vec<u8>>,
        count: usize,
    },
}

impl PubSubMessage {
    pub fn to_resp(&self) -> crate::protocol::resp::RespValue {
        use crate::protocol::resp::RespValue;

        match self {
            PubSubMessage::Message { channel, payload } => RespValue::Array(Some(vec![
                RespValue::BulkString(Some(Bytes::from_static(b"message"))),
                RespValue::BulkString(Some(Bytes::from(channel.clone()))),
                RespValue::BulkString(Some(payload.clone())),
            ])),
            PubSubMessage::PatternMessage {
                pattern,
                channel,
                payload,
            } => RespValue::Array(Some(vec![
                RespValue::BulkString(Some(Bytes::from_static(b"pmessage"))),
                RespValue::BulkString(Some(Bytes::from(pattern.clone()))),
                RespValue::BulkString(Some(Bytes::from(channel.clone()))),
                RespValue::BulkString(Some(payload.clone())),
            ])),
            PubSubMessage::Subscribe { channel, count } => RespValue::Array(Some(vec![
                RespValue::BulkString(Some(Bytes::from_static(b"subscribe"))),
                RespValue::BulkString(Some(Bytes::from(channel.clone()))),
                RespValue::Integer(*count as i64),
            ])),
            PubSubMessage::Unsubscribe { channel, count } => RespValue::Array(Some(vec![
                RespValue::BulkString(Some(Bytes::from_static(b"unsubscribe"))),
                RespValue::BulkString(channel.as_ref().map(|c| Bytes::from(c.clone()))),
                RespValue::Integer(*count as i64),
            ])),
            PubSubMessage::PSubscribe { pattern, count } => RespValue::Array(Some(vec![
                RespValue::BulkString(Some(Bytes::from_static(b"psubscribe"))),
                RespValue::BulkString(Some(Bytes::from(pattern.clone()))),
                RespValue::Integer(*count as i64),
            ])),
            PubSubMessage::PUnsubscribe { pattern, count } => RespValue::Array(Some(vec![
                RespValue::BulkString(Some(Bytes::from_static(b"punsubscribe"))),
                RespValue::BulkString(pattern.as_ref().map(|p| Bytes::from(p.clone()))),
                RespValue::Integer(*count as i64),
            ])),
        }
    }
}