Mercurial > templog
annotate rust/src/params.rs @ 632:bde302def78e rust
moving to riker, nowhere near yet
author | Matt Johnston <matt@ucc.asn.au> |
---|---|
date | Thu, 22 Aug 2019 23:59:50 +0800 |
parents | c57821a60e51 |
children | 490e9e15b98c |
rev | line source |
---|---|
592 | 1 use std::time::Duration; |
2 use std::io; | |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
3 use std::str; |
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
4 use std::rc::Rc; |
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
5 use std::sync::{Arc,Mutex}; |
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
6 use std::error::Error; |
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
7 use std::cell::{Cell,RefCell}; |
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
8 use std::fs::File; |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
9 use std::io::Read; |
627 | 10 |
592 | 11 |
632
bde302def78e
moving to riker, nowhere near yet
Matt Johnston <matt@ucc.asn.au>
parents:
631
diff
changeset
|
12 use rand::Rng; |
626 | 13 use std::str::FromStr; |
14 use hyper; | |
15 use hyper::header::{Headers, ETag, EntityTag}; | |
16 use hyper::client::Client; | |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
17 |
632
bde302def78e
moving to riker, nowhere near yet
Matt Johnston <matt@ucc.asn.au>
parents:
631
diff
changeset
|
18 use crate::types::*; |
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
19 use ::Config; |
592 | 20 |
615 | 21 #[derive(Deserialize, Serialize, Debug, Clone)] |
22 pub struct Params { | |
23 pub fridge_setpoint: f32, | |
24 pub fridge_difference: f32, | |
621 | 25 pub overshoot_delay: u64, |
615 | 26 pub overshoot_factor: f32, |
27 pub disabled: bool, | |
28 pub nowort: bool, | |
29 pub fridge_range_lower: f32, | |
30 pub fridge_range_upper: f32, | |
31 } | |
32 | |
33 impl Params { | |
34 pub fn defaults() -> Params { | |
35 Params { | |
36 fridge_setpoint: 16.0, | |
37 fridge_difference: 0.2, | |
38 overshoot_delay: 720, // 12 minutes | |
39 overshoot_factor: 1.0, | |
40 disabled: false, | |
41 nowort: false, | |
42 fridge_range_lower: 3.0, | |
43 fridge_range_upper: 3.0, | |
44 } | |
45 } | |
46 | |
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
47 fn try_load(filename: &str) -> Result<Params, TemplogError> { |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
48 let mut s = String::new(); |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
49 File::open(filename)?.read_to_string(&mut s)?; |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
50 Ok(serde_json::from_str(&s)?) |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
51 } |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
52 |
615 | 53 pub fn load(config: &Config) -> Params { |
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
54 Self::try_load(&config.PARAMS_FILE) |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
55 .unwrap_or_else(|_| Params::defaults()) |
615 | 56 } |
57 } | |
58 | |
613
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
59 #[derive(Deserialize, Debug)] |
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
60 struct Response { |
616 | 61 // sent as an opaque etag: header. Has format "epoch-nonce", |
62 // responses where the epoch do not match ParamWaiter::epoch are dropped | |
63 etag: String, | |
613
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
64 params: Params, |
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
65 } |
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
66 |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
67 #[derive(Clone)] |
593
bf138339d20a
fiddling with timeouts and closures
Matt Johnston <matt@ucc.asn.au>
parents:
592
diff
changeset
|
68 pub struct ParamWaiter { |
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
69 limitlog: NotTooOften, |
616 | 70 // last_etag is used for long-polling. |
71 last_etag: RefCell<String>, | |
72 epoch: String, | |
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
73 |
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
74 config: Config, |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
75 handle: Handle, |
592 | 76 } |
77 | |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
78 const LOG_MINUTES: u64 = 15; |
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
79 const MAX_RESPONSE_SIZE: usize = 10000; |
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
80 const TIMEOUT_MINUTES: u64 = 5; |
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
81 |
593
bf138339d20a
fiddling with timeouts and closures
Matt Johnston <matt@ucc.asn.au>
parents:
592
diff
changeset
|
82 impl ParamWaiter { |
626 | 83 fn make_request(&self) -> hyper::client::FutureResponse { |
84 let uri = self.config.SETTINGS_URL.parse().expect("Bad SETTINGS_URL in config"); | |
85 let mut req = hyper::client::Request::new(hyper::Method::Get, uri); | |
86 { | |
627 | 87 let headers = req.headers_mut(); |
626 | 88 headers.set(hyper::header::ETag(EntityTag::new(false, self.last_etag.borrow().clone()))); |
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
89 } |
626 | 90 hyper::client::Client::new(&self.handle).request(req) |
91 // XXX how do we do this? | |
92 // req.timeout(Duration::new(TIMEOUT_MINUTES * 60, 0)).unwrap(); | |
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
93 } |
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
94 |
626 | 95 |
96 fn handle_response(&self, buf : hyper::Chunk, status: hyper::StatusCode) -> Result<Params, TemplogError> { | |
97 let text = String::from_utf8_lossy(buf.as_ref()); | |
98 match status { | |
99 hyper::StatusCode::Ok => { | |
613
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
100 // new params |
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
101 let r: Response = serde_json::from_str(&text)?; |
618
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
102 let mut le = self.last_etag.borrow_mut(); |
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
103 *le = r.etag; |
626 | 104 |
105 // update params if the epoch is correct | |
106 if let Some(e) = le.split('-').next() { | |
107 if e == &self.epoch { | |
108 self.write_params(&r.params); | |
109 return Ok(r.params); | |
110 } | |
618
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
111 } |
626 | 112 Err(TemplogError::new(&format!("Bad epoch from server '{}' expected '{}'", *le, self.epoch))) |
613
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
113 } |
626 | 114 hyper::StatusCode::NotModified => { |
629 | 115 // XXX this isn't really an error. Should handle_response() return |
116 // Result<Option<Params>, TemplogError> instead? | |
117 | |
618
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
118 Err(TemplogError::new("304 unmodified (long polling timeout at the server)")) |
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
119 }, |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
120 _ => { |
626 | 121 Err(TemplogError::new(&format!("Wrong server response code {}: {}", status.as_u16(), text))) |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
122 }, |
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
123 } |
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
124 } |
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
125 |
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
126 fn write_params(&self, params: &Params) { |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
127 let p = atomicwrites::AtomicFile::new(&self.config.PARAMS_FILE, atomicwrites::AllowOverwrite); |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
128 p.write(|f| { |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
129 serde_json::to_writer(f, params) |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
130 }); |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
131 } |
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
132 |
627 | 133 pub fn stream(config: Config, handle: Handle) -> Result<(), TemplogError> { |
134 let rcself = Rc::new(ParamWaiter::new(config, handle)); | |
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
135 |
627 | 136 let dur = Duration::from_millis(4000); |
137 for _ in Interval::new(dur, &rcself.handle).unwrap() { | |
138 // fetch params | |
139 // TODO - skip if inflight. | |
140 let r = await!(rcself.make_request()).map_err(|e| TemplogError::new_hyper("response", e))?; | |
141 let status = r.status(); | |
142 let b = await!(r.body().concat2()).map_err(|e| TemplogError::new_hyper("body", e))?; | |
143 if let Ok(params) = rcself.handle_response(b, status) { | |
144 stream_yield!(params); | |
145 } | |
146 } | |
147 Ok(()) | |
592 | 148 } |
149 | |
627 | 150 fn new(config: Config, handle: Handle) -> Self { |
616 | 151 let mut b = [0u8; 15]; // 15 bytes -> 20 characters base64 |
626 | 152 rand::OsRng::new().expect("Opening system RNG failed").fill_bytes(&mut b); |
616 | 153 let epoch = base64::encode(&b); |
627 | 154 |
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
155 ParamWaiter { |
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
156 limitlog: NotTooOften::new(LOG_MINUTES*60), |
616 | 157 last_etag: RefCell::new(String::new()), |
158 epoch: epoch, | |
627 | 159 config: config, |
160 handle: handle, | |
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
161 } |
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
162 } |
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
163 |
592 | 164 } |
165 |