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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
use crate::{
error::RuarangoErr::Unreachable,
model::{
add_qp, AddHeaders, BuildUrl,
QueryParam::{KeepNull, ReturnNew, ReturnOld, WaitForSync},
},
Connection,
};
use anyhow::{Context, Result};
use derive_builder::Builder;
use getset::Getters;
use reqwest::{
header::{HeaderMap, HeaderName, HeaderValue},
Url,
};
use serde::{Deserialize, Serialize};
#[derive(Builder, Clone, Debug, Default, Deserialize, Getters, Serialize)]
#[getset(get = "pub(crate)")]
pub struct Config<T> {
#[builder(setter(into))]
graph: String,
#[builder(setter(into))]
collection: String,
#[builder(setter(into))]
key: String,
#[builder(setter(strip_option), default)]
wait_for_sync: Option<bool>,
#[builder(setter(strip_option), default)]
keep_null: Option<bool>,
#[builder(setter(strip_option), default)]
return_old: Option<bool>,
#[builder(setter(strip_option), default)]
return_new: Option<bool>,
#[builder(setter(strip_option, into), default)]
if_match: Option<String>,
edge: T,
}
impl<T> Config<T> {
fn build_suffix(&self, base: &str) -> String {
let mut url = format!(
"{}/{}/edge/{}/{}",
base, self.graph, self.collection, self.key
);
let mut has_qp = false;
add_qp(*self.wait_for_sync(), &mut url, &mut has_qp, WaitForSync);
add_qp(*self.keep_null(), &mut url, &mut has_qp, KeepNull);
add_qp(*self.return_old(), &mut url, &mut has_qp, ReturnOld);
add_qp(*self.return_new(), &mut url, &mut has_qp, ReturnNew);
url
}
}
impl<T> BuildUrl for Config<T> {
fn build_url(&self, base: &str, conn: &Connection) -> Result<Url> {
let suffix = self.build_suffix(base);
conn.db_url()
.join(&suffix)
.with_context(|| format!("Unable to build '{}' url", suffix))
}
}
impl<T> AddHeaders for Config<T> {
fn has_header(&self) -> bool {
self.if_match.is_some()
}
fn add_headers(&self) -> Result<Option<HeaderMap>> {
let mut headers = None;
if self.has_header() {
let mut headers_map = HeaderMap::new();
if let Some(rev) = self.if_match() {
let _ = headers_map.append(
HeaderName::from_static("if-match"),
HeaderValue::from_str(rev)?,
);
headers = Some(headers_map);
} else {
return Err(Unreachable {
msg: "'if_match' should be true!".to_string(),
}
.into());
}
}
Ok(headers)
}
}