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
use std::collections::HashMap;
use super::*;
use super::super::types::*;
use super::super::{IntoBytes};
pub struct BodyReqStartup {
pub map: HashMap<String, String>
}
impl BodyReqStartup {
pub fn new(compression: Option<String>) -> BodyReqStartup {
let mut map = HashMap::new();
map.insert("CQL_VERSION".to_string(), "3.0.0".to_string());
if let Some(c) = compression {
map.insert("COMPRESSION".to_string(), c);
}
return BodyReqStartup {
map: map
};
}
pub fn map_len(&self) -> u64 {
let mut l: usize = 0;
for (key, val) in self.map.iter() {
l += key.len() + SHORT_LEN + val.len() + SHORT_LEN;
}
return l as u64;
}
pub fn num(&self) -> Vec<u8> {
return to_n_bytes(self.map.len() as u64, SHORT_LEN);
}
}
impl IntoBytes for BodyReqStartup {
fn into_cbytes(&self) -> Vec<u8> {
let mut v = vec![];
v.extend_from_slice(&self.num().as_slice());
for (key, val) in self.map.iter() {
v.extend_from_slice(to_n_bytes(key.len() as u64, SHORT_LEN).as_slice());
v.extend_from_slice(key.as_bytes());
v.extend_from_slice(to_n_bytes(val.len() as u64, SHORT_LEN).as_slice());
v.extend_from_slice(val.as_bytes());
}
return v;
}
}
impl Frame {
pub fn new_req_startup(compression: Option<String>) -> Frame {
let version = Version::Request;
let flag = Flag::Ignore;
let stream: u64 = 0;
let opcode = Opcode::Startup;
let body = BodyReqStartup::new(compression);
return Frame {
version: version,
flags: vec![flag],
stream: stream,
opcode: opcode,
body: body.into_cbytes(),
tracing_id: None,
warnings: vec![]
};
}
}