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
/// A Redis PING command.
use crate::cmd::Command;
use crate::frame::Frame;
use bytes::Bytes;
pub struct Ping {
msg: Option<Bytes>,
}
impl Ping {
/// Creates a new Ping command.
///
/// # Arguments
///
/// * `msg` - An optional message to send with ping
///
/// # Returns
///
/// A new Ping command
///
/// # Examples
///
/// ```ignore
/// let ping = Ping::new(Some("hello".into()));
/// ```
pub fn new(msg: Option<&[u8]>) -> Self {
Self {
msg: msg.map(|m| Bytes::from(m.to_vec())),
}
}
}
impl Command for Ping {
/// Converts the ping command into a Frame to be transimitted over the stream.
fn into_stream(self) -> Frame {
let mut frame: Frame = Frame::array();
frame
.push_frame_to_array(Frame::BulkString("PING".into()))
.unwrap();
// do not push the message if it is None
if let Some(msg) = self.msg {
frame.push_frame_to_array(Frame::BulkString(msg)).unwrap();
}
frame
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_ping() {
let ping = Ping::new(None);
let frame = ping.into_stream();
assert_eq!(frame, Frame::Array(vec![Frame::BulkString("PING".into())]));
let ping = Ping::new(Some("hello".as_bytes()));
let frame = ping.into_stream();
assert_eq!(
frame,
Frame::Array(vec![
Frame::BulkString("PING".into()),
Frame::BulkString("hello".into())
])
);
}
}