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
|
use c8y_smartrest::topic::C8yTopic;
use mqtt_channel::Message;
use serde::Deserialize;
use std::{borrow::Borrow, path::Path};
use std::{collections::HashSet, fs};
use tracing::warn;
#[derive(Deserialize, Debug, Eq, PartialEq, Default)]
#[serde(deny_unknown_fields)]
pub struct LogPluginConfig {
pub files: Vec<FileEntry>,
}
#[derive(Deserialize, Debug, Eq, Default, Clone)]
#[serde(deny_unknown_fields)]
pub struct FileEntry {
pub(crate) path: String,
#[serde(rename = "type")]
pub config_type: String,
}
impl PartialEq for FileEntry {
fn eq(&self, other: &Self) -> bool {
self.config_type == other.config_type
}
}
impl Borrow<String> for FileEntry {
fn borrow(&self) -> &String {
&self.config_type
}
}
impl LogPluginConfig {
pub fn new(config_file_path: &Path) -> Self {
Self::read_config(config_file_path)
}
pub fn read_config(path: &Path) -> Self {
let path_str = path.display().to_string();
match fs::read_to_string(path) {
Ok(contents) => match toml::from_str(contents.as_str()) {
Ok(config) => config,
_ => {
warn!("The config file {} is malformed.", path_str);
Self::default()
}
},
Err(_) => {
warn!(
"The config file {} does not exist or is not readable.",
path_str
);
Self::default()
}
}
}
pub fn to_supported_config_types_message(&self) -> Result<Message, anyhow::Error> {
let topic = C8yTopic::SmartRestResponse.to_topic()?;
Ok(Message::new(&topic, self.to_smartrest_payload()))
}
pub fn get_all_file_types(&self) -> Vec<String> {
self.files
.iter()
.map(|x| x.config_type.to_string())
.collect::<HashSet<_>>()
.iter()
.map(|x| x.to_string())
.collect::<Vec<_>>()
}
// 118,typeA,typeB,...
fn to_smartrest_payload(&self) -> String {
let mut config_types = self.get_all_file_types();
config_types.sort();
let supported_config_types = config_types.join(",");
format!("118,{supported_config_types}")
}
}
#[test]
fn test_no_duplicated_file_types() {
let files = vec![
FileEntry {
path: "a/path".to_string(),
config_type: "type_one".to_string(),
},
FileEntry {
path: "some/path".to_string(),
config_type: "type_one".to_string(),
},
];
let logs_config = LogPluginConfig { files: files };
assert_eq!(
logs_config.get_all_file_types(),
vec!["type_one".to_string()]
);
}
|
