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 |