Mercurial > templog
annotate rust/src/params.rs @ 633:490e9e15b98c rust
move some bits to riker
| author | Matt Johnston <matt@ucc.asn.au> |
|---|---|
| date | Wed, 04 Sep 2019 23:24:13 +0800 |
| parents | bde302def78e |
| children | a5721c02d3ee |
| 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 |
| 633 | 11 use serde::{Serialize,Deserialize}; |
| 592 | 12 |
| 633 | 13 use rand::rngs::{StdRng, OsRng}; |
| 14 use rand::{RngCore, SeedableRng}; | |
| 15 | |
| 626 | 16 use std::str::FromStr; |
| 17 use hyper; | |
| 18 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
|
19 |
| 633 | 20 use riker::actors::*; |
| 21 | |
| 22 use super::types::*; | |
| 23 use super::config::Config; | |
| 592 | 24 |
| 615 | 25 #[derive(Deserialize, Serialize, Debug, Clone)] |
| 26 pub struct Params { | |
| 27 pub fridge_setpoint: f32, | |
| 28 pub fridge_difference: f32, | |
| 621 | 29 pub overshoot_delay: u64, |
| 615 | 30 pub overshoot_factor: f32, |
| 31 pub disabled: bool, | |
| 32 pub nowort: bool, | |
| 33 pub fridge_range_lower: f32, | |
| 34 pub fridge_range_upper: f32, | |
| 35 } | |
| 36 | |
| 37 impl Params { | |
| 38 pub fn defaults() -> Params { | |
| 39 Params { | |
| 40 fridge_setpoint: 16.0, | |
| 41 fridge_difference: 0.2, | |
| 42 overshoot_delay: 720, // 12 minutes | |
| 43 overshoot_factor: 1.0, | |
| 44 disabled: false, | |
| 45 nowort: false, | |
| 46 fridge_range_lower: 3.0, | |
| 47 fridge_range_upper: 3.0, | |
| 48 } | |
| 49 } | |
| 50 | |
|
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
51 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
|
52 let mut s = String::new(); |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
53 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
|
54 Ok(serde_json::from_str(&s)?) |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
55 } |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
56 |
| 615 | 57 pub fn load(config: &Config) -> Params { |
|
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
58 Self::try_load(&config.PARAMS_FILE) |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
59 .unwrap_or_else(|_| Params::defaults()) |
| 615 | 60 } |
| 633 | 61 |
| 615 | 62 } |
| 63 | |
|
593
bf138339d20a
fiddling with timeouts and closures
Matt Johnston <matt@ucc.asn.au>
parents:
592
diff
changeset
|
64 pub struct ParamWaiter { |
|
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
65 limitlog: NotTooOften, |
| 616 | 66 // last_etag is used for long-polling. |
| 67 last_etag: RefCell<String>, | |
| 68 epoch: String, | |
| 633 | 69 chan: ChannelRef<Params>, |
|
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
70 |
|
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
71 config: Config, |
| 592 | 72 } |
| 73 | |
|
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
74 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
|
75 const MAX_RESPONSE_SIZE: usize = 10000; |
|
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
76 const TIMEOUT_MINUTES: u64 = 5; |
|
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
77 |
|
593
bf138339d20a
fiddling with timeouts and closures
Matt Johnston <matt@ucc.asn.au>
parents:
592
diff
changeset
|
78 impl ParamWaiter { |
| 633 | 79 |
| 80 fn new(config: Config) -> Self { | |
| 81 let mut b = [0u8; 15]; // 15 bytes -> 20 characters base64 | |
| 82 OsRng.fill_bytes(&mut b); | |
| 83 let epoch = base64::encode(&b); | |
| 84 | |
| 85 ParamWaiter { | |
| 86 limitlog: NotTooOften::new(LOG_MINUTES*60), | |
| 87 last_etag: RefCell::new(String::new()), | |
| 88 epoch: epoch, | |
| 89 config: config, | |
|
614
e1bab5b36352
using some refcells for the paramwaiter
Matt Johnston <matt@ucc.asn.au>
parents:
613
diff
changeset
|
90 } |
|
609
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
91 } |
|
7bda01659426
not building, paramwaiter work
Matt Johnston <matt@ucc.asn.au>
parents:
594
diff
changeset
|
92 |
| 626 | 93 |
| 633 | 94 async fn keep_waiting(&mut self) { |
| 95 loop { | |
| 96 self.wait_updates().await; | |
| 97 } | |
| 98 } | |
| 99 | |
| 100 async fn wait_updates(&mut self) { | |
| 101 | |
| 102 let uri = self.config.SETTINGS_URL.parse().expect("Bad SETTINGS_URL in config"); | |
| 103 let mut req = hyper::Request::new(hyper::Method::Get, uri); | |
| 104 req.headers_mut().insert(hyper::header::ETAG, self.last_etag.borrow()); | |
| 105 let resp = hyper::Client::new(&self.handle).request(req).await?; | |
| 106 let b = resp.body().concat2().await?; | |
| 107 let new_params = self.handle_response(b)?; | |
| 108 self.chan.tell(Publish{msg: new_params, topic: "params".into()}, None); | |
| 109 } | |
| 110 | |
| 626 | 111 fn handle_response(&self, buf : hyper::Chunk, status: hyper::StatusCode) -> Result<Params, TemplogError> { |
| 633 | 112 #[derive(Deserialize, Debug)] |
| 113 struct Response { | |
| 114 // sent as an opaque etag: header. Has format "epoch-nonce", | |
| 115 // responses where the epoch do not match ParamWaiter::epoch are dropped | |
| 116 etag: String, | |
| 117 params: Params, | |
| 118 } | |
| 119 | |
| 626 | 120 let text = String::from_utf8_lossy(buf.as_ref()); |
| 121 match status { | |
| 122 hyper::StatusCode::Ok => { | |
|
613
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
123 // new params |
|
5c2b0d47bb83
Response has epoch_tag and params
Matt Johnston <matt@ucc.asn.au>
parents:
611
diff
changeset
|
124 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
|
125 let mut le = self.last_etag.borrow_mut(); |
|
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
126 *le = r.etag; |
| 626 | 127 |
| 128 // update params if the epoch is correct | |
| 129 if let Some(e) = le.split('-').next() { | |
| 130 if e == &self.epoch { | |
| 131 self.write_params(&r.params); | |
| 132 return Ok(r.params); | |
| 133 } | |
|
618
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
134 } |
| 626 | 135 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
|
136 } |
| 626 | 137 hyper::StatusCode::NotModified => { |
| 629 | 138 // XXX this isn't really an error. Should handle_response() return |
| 139 // Result<Option<Params>, TemplogError> instead? | |
| 140 | |
|
618
2d65a9f0bed3
params returning stream of success
Matt Johnston <matt@ucc.asn.au>
parents:
616
diff
changeset
|
141 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
|
142 }, |
|
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
143 _ => { |
| 626 | 144 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
|
145 }, |
|
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
146 } |
|
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
147 } |
|
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
148 |
|
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
149 fn write_params(&self, params: &Params) { |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
150 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
|
151 p.write(|f| { |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
152 serde_json::to_writer(f, params) |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
153 }); |
|
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
154 } |
| 633 | 155 } |
|
624
2710649ab71e
read/write params local file. untested
Matt Johnston <matt@ucc.asn.au>
parents:
621
diff
changeset
|
156 |
| 633 | 157 impl Actor for ParamWaiter { |
|
611
f3e39e2107fd
still doesn't compile, improvements to TemplogError and tokio curl though
Matt Johnston <matt@ucc.asn.au>
parents:
609
diff
changeset
|
158 |
| 633 | 159 fn post_start(&mut self, ctx: &Context<Self::Msg>) { |
| 160 self.chan = channel("params", &ctx.system).unwrap(); | |
| 161 ctx.run(self.wait_updates()); | |
| 592 | 162 } |
| 163 } | |
| 164 | |
| 633 | 165 // pub fn stream(config: Config, handle: Handle) -> Result<(), TemplogError> { |
| 166 // let rcself = Rc::new(ParamWaiter::new(config, handle)); | |
| 167 | |
| 168 // let dur = Duration::from_millis(4000); | |
| 169 // for _ in Interval::new(dur, &rcself.handle).unwrap() { | |
| 170 // // fetch params | |
| 171 // // TODO - skip if inflight. | |
| 172 // let r = await!(rcself.make_request()).map_err(|e| TemplogError::new_hyper("response", e))?; | |
| 173 // let status = r.status(); | |
| 174 // let b = await!(r.body().concat2()).map_err(|e| TemplogError::new_hyper("body", e))?; | |
| 175 // if let Ok(params) = rcself.handle_response(b, status) { | |
| 176 // stream_yield!(params); | |
| 177 // } | |
| 178 // } | |
| 179 // Ok(()) | |
| 180 // } | |
| 181 |