graphql-ws-client 0.13.0

A graphql over websockets client
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
90
91
92

//! Message definitions for the [graphql-transport-ws protocol][1]
//!
//! [1]: https://github.com/enisdenjo/graphql-ws/blob/HEAD/PROTOCOL.md

#[derive(Default, Debug)]
pub struct ConnectionInit<Payload = ()> {
    payload: Option<Payload>,
}

impl<Payload> ConnectionInit<Payload> {
    pub fn new(payload: Option<Payload>) -> Self {
        ConnectionInit { payload }
    }
}

impl<Payload> serde::Serialize for ConnectionInit<Payload>
where
    Payload: serde::Serialize,
{
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        use serde::ser::SerializeMap;

        let mut map = serializer.serialize_map(Some(2))?;
        map.serialize_entry("type", "connection_init")?;
        if self.payload.is_some() {
            map.serialize_entry("payload", &self.payload)?;
        }
        map.end()
    }
}

#[derive(serde::Serialize)]
#[serde(tag = "type")]
pub enum Message<'a, Operation> {
    #[serde(rename = "subscribe")]
    Subscribe { id: String, payload: &'a Operation },
    #[serde(rename = "complete")]
    Complete { id: String },
    #[serde(rename = "ping")]
    Ping,
    #[serde(rename = "pong")]
    Pong,
}

#[allow(dead_code)]
#[derive(serde::Deserialize, Debug)]
#[serde(tag = "type")]
pub enum Event {
    #[serde(rename = "next")]
    Next {
        id: String,
        payload: serde_json::Value,
    },
    #[serde(rename = "error")]
    Error {
        id: String,
        payload: Vec<serde_json::Value>,
    },
    #[serde(rename = "complete")]
    Complete { id: String },
    #[serde(rename = "connection_ack")]
    ConnectionAck { payload: Option<serde_json::Value> },
    #[serde(rename = "ping")]
    Ping { payload: Option<serde_json::Value> },
    #[serde(rename = "pong")]
    Pong { payload: Option<serde_json::Value> },
}

impl Event {
    pub fn id(&self) -> Option<&str> {
        match self {
            Event::Next { id, .. } | Event::Complete { id, .. } | Event::Error { id, .. } => {
                Some(id.as_ref())
            }
            Event::Ping { .. } | Event::Pong { .. } | Event::ConnectionAck { .. } => None,
        }
    }

    pub fn r#type(&self) -> &'static str {
        match self {
            Event::Next { .. } => "next",
            Event::Complete { .. } => "complete",
            Event::Error { .. } => "error",
            Event::Ping { .. } => "ping",
            Event::Pong { .. } => "pong",
            Event::ConnectionAck { .. } => "connection_ack",
        }
    }
}