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
use bytes::{ Bytes, BytesMut };
use serde_json::Value;
use crate::Command;
#[derive(Debug)]
pub struct CommandParser { }
impl CommandParser {
pub fn parse_command(mut buffer: BytesMut) -> crate::Result<Option<Command>> {
let mut headers = [httparse::EMPTY_HEADER; 16];
let mut req = httparse::Request::new(&mut headers);
let status = req.parse(&buffer)?;
let amt = match status {
httparse::Status::Complete(amt) => amt,
httparse::Status::Partial => return Ok(None),
};
let _prev = buffer.split_to(amt);
let json_body: Value = serde_json::from_slice(&buffer)?;
match &json_body["command"] {
Value::String(command) => {
if command == "GET" {
let mut is_valid = true;
let key = match &json_body["key"] {
Value::String(val) => {
val.clone()
},
_ => {
is_valid = false;
String::from("")
}
};
if is_valid {
Ok(Command::Get { key: key }.into())
} else {
Err("Invalid command".into())
}
} else if command == "SET" {
let mut is_valid = true;
let key = match &json_body["key"] {
Value::String(val) => {
val.clone()
},
_ => {
is_valid = false;
String::from("")
}
};
let value = match &json_body["value"] {
Value::String(val) => {
val.clone()
},
_ => {
is_valid = false;
String::from("")
}
};
if is_valid {
Ok(Command::Set { key: key, value: Bytes::from(value) }.into())
} else {
Err("Invalid command".into())
}
} else if command == "INFO" {
Ok(Command::Info{}.into())
} else {
Err("Invalid command.".into())
}
},
_ => {
return Err("Invalid command.".into());
}
}
}
}