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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
use log::error;
use serde::{Deserialize, Serialize};
use std::io;
use rd_util::*;
pub mod args;
pub mod bandit_report;
pub mod bench;
pub mod cmd;
pub mod cmd_ack;
pub mod index;
pub mod oomd;
pub mod report;
pub mod side_defs;
pub mod slices;
pub mod sysreqs;
pub use args::{Args, Bandit, BanditMemHogArgs, EnforceConfig};
pub use bandit_report::BanditMemHogReport;
pub use bench::{BenchKnobs, HashdKnobs, IoCostKnobs, BENCH_FILENAME};
pub use cmd::{Cmd, HashdCmd, SideloaderCmd};
pub use cmd_ack::CmdAck;
pub use index::Index;
pub use oomd::{OomdKnobs, OomdSliceMemPressureKnobs, OomdSliceSenpaiKnobs};
pub use report::{
BenchHashdReport, BenchIoCostReport, HashdReport, IoCostModelReport, IoCostQoSReport,
IoCostReport, IoLatReport, OomdReport, Report, ReportIter, ReportPathIter, ResCtlReport,
SideloadReport, SideloaderReport, StatMap, SvcReport, SvcStateReport, SysloadReport,
UsageReport,
};
pub use side_defs::{SideloadDefs, SideloadSpec};
pub use slices::{DisableSeqKnobs, MemoryKnob, Slice, SliceConfig, SliceKnobs, ROOT_SLICE};
pub use sysreqs::{MissedSysReqs, SysReq, SysReqsReport, ALL_SYSREQS_SET};
lazy_static::lazy_static! {
pub static ref VERSION: &'static str = env!("CARGO_PKG_VERSION");
pub static ref FULL_VERSION: String = full_version(*VERSION);
}
#[derive(Clone, Copy, Debug, PartialEq, Eq, Serialize, Deserialize)]
pub enum RunnerState {
Idle,
Running,
BenchHashd,
BenchIoCost,
}
pub const AGENT_SVC_NAME: &str = "rd-agent.service";
pub const HASHD_BENCH_SVC_NAME: &str = "rd-hashd-bench.service";
pub const IOCOST_BENCH_SVC_NAME: &str = "rd-iocost-bench.service";
pub const HASHD_A_SVC_NAME: &str = "rd-hashd-A.service";
pub const HASHD_B_SVC_NAME: &str = "rd-hashd-B.service";
pub const OOMD_SVC_NAME: &str = "rd-oomd.service";
pub const SIDELOADER_SVC_NAME: &str = "rd-sideloader.service";
pub const SIDELOAD_SVC_PREFIX: &str = "rd-sideload-";
pub const SYSLOAD_SVC_PREFIX: &str = "rd-sysload-";
pub fn sysload_svc_name(name: &str) -> String {
format!("{}{}.service", SYSLOAD_SVC_PREFIX, name)
}
pub fn sideload_svc_name(name: &str) -> String {
format!("{}{}.service", SIDELOAD_SVC_PREFIX, name)
}
#[derive(Default)]
pub struct AgentFiles {
pub args_path: String,
pub index_path: String,
pub args: JsonConfigFile<Args>,
pub index: JsonConfigFile<Index>,
pub cmd: JsonConfigFile<Cmd>,
pub cmd_ack: JsonConfigFile<CmdAck>,
pub sysreqs: JsonConfigFile<SysReqsReport>,
pub report: JsonConfigFile<Report>,
pub bench: JsonConfigFile<BenchKnobs>,
pub slices: JsonConfigFile<SliceKnobs>,
pub oomd: JsonConfigFile<OomdKnobs>,
}
impl AgentFiles {
pub fn new(dir: &str) -> Self {
Self {
args_path: dir.to_string() + "/args.json",
index_path: dir.to_string() + "/index.json",
..Default::default()
}
}
fn refresh_one<T>(file: &mut JsonConfigFile<T>, path: &str) -> bool
where
T: JsonLoad + JsonSave,
{
match &file.path {
None => match JsonConfigFile::<T>::load(path) {
Ok(v) => {
*file = v;
true
}
Err(e) => {
match e.downcast_ref::<io::Error>() {
Some(e) if e.raw_os_error() == Some(libc::ENOENT) => (),
_ => error!("Failed to read {:?} ({:?})", path, &e),
}
false
}
},
Some(_) => match file.maybe_reload() {
Ok(v) => v,
Err(e) => {
match e.downcast_ref::<io::Error>() {
Some(e) if e.raw_os_error() == Some(libc::ENOENT) => (),
_ => error!("Failed to reload {:?} ({:?})", path, &e),
}
false
}
},
}
}
pub fn refresh(&mut self) {
Self::refresh_one(&mut self.args, &self.args_path);
if Self::refresh_one(&mut self.index, &self.index_path) {
self.cmd = Default::default();
self.cmd_ack = Default::default();
self.sysreqs = Default::default();
self.report = Default::default();
self.bench = Default::default();
self.slices = Default::default();
self.oomd = Default::default();
}
if let None = self.index.path {
return;
}
let index = &self.index.data;
Self::refresh_one(&mut self.cmd, &index.cmd);
Self::refresh_one(&mut self.cmd_ack, &index.cmd_ack);
Self::refresh_one(&mut self.sysreqs, &index.sysreqs);
Self::refresh_one(&mut self.report, &index.report);
Self::refresh_one(&mut self.bench, &index.bench);
Self::refresh_one(&mut self.slices, &index.slices);
Self::refresh_one(&mut self.oomd, &index.oomd);
}
}