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
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
use crate::error::{Error, ErrorResponse};
use crate::telemetry::SyncTelemetryPing;
use std::collections::HashMap;
use std::time::{Duration, SystemTime};
/// The general status of sync - should probably be moved to the "sync manager"
/// once we have one!
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum ServiceStatus {
/// Everything is fine.
Ok,
/// Some general network issue.
NetworkError,
/// Some apparent issue with the servers.
ServiceError,
/// Some external FxA action needs to be taken.
AuthenticationError,
/// We declined to do anything for backoff or rate-limiting reasons.
BackedOff,
/// We were interrupted.
Interrupted,
/// Something else - you need to check the logs for more details. May
/// or may not be transient, we really don't know.
OtherError,
}
impl ServiceStatus {
// This is a bit naive and probably will not survive in this form in the
// SyncManager - eg, we'll want to handle backoff etc.
pub fn from_err(err: &Error) -> ServiceStatus {
match err {
// HTTP based errors.
Error::TokenserverHttpError(status) => {
// bit of a shame the tokenserver is different to storage...
if *status == 401 {
ServiceStatus::AuthenticationError
} else {
ServiceStatus::ServiceError
}
}
// BackoffError is also from the tokenserver.
Error::BackoffError(_) => ServiceStatus::ServiceError,
Error::StorageHttpError(ref e) => match e {
ErrorResponse::Unauthorized { .. } => ServiceStatus::AuthenticationError,
_ => ServiceStatus::ServiceError,
},
// Network errors.
Error::RequestError(_) | Error::UnexpectedStatus(_) | Error::HawkError(_) => {
ServiceStatus::NetworkError
}
Error::Interrupted(_) => ServiceStatus::Interrupted,
_ => ServiceStatus::OtherError,
}
}
}
/// The result of a sync request. This too is from the "sync manager", but only
/// has a fraction of the things it will have when we actually build that.
#[derive(Debug)]
pub struct SyncResult {
/// The general health.
pub service_status: ServiceStatus,
/// The set of declined engines, if we know them.
pub declined: Option<Vec<String>>,
/// The result of the sync.
pub result: Result<(), Error>,
/// The result for each engine.
/// Note that we expect the `String` to be replaced with an enum later.
pub engine_results: HashMap<String, Result<(), Error>>,
pub telemetry: SyncTelemetryPing,
pub next_sync_after: Option<std::time::SystemTime>,
}
// If `r` has a BackoffError, then returns the later backoff value.
fn advance_backoff(cur_best: SystemTime, r: &Result<(), Error>) -> SystemTime {
if let Err(e) = r {
if let Some(time) = e.get_backoff() {
return std::cmp::max(time, cur_best);
}
}
cur_best
}
impl SyncResult {
pub(crate) fn set_sync_after(&mut self, backoff_duration: Duration) {
let now = SystemTime::now();
let toplevel = advance_backoff(now + backoff_duration, &self.result);
let sync_after = self.engine_results.values().fold(toplevel, advance_backoff);
if sync_after <= now {
self.next_sync_after = None;
} else {
self.next_sync_after = Some(sync_after);
}
}
}
|