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
|
use std::path::Path;
use crate::{
az::converter::AzureConverter,
core::{component::TEdgeComponent, mapper::create_mapper, size_threshold::SizeThreshold},
};
use async_trait::async_trait;
use clock::WallClock;
use tedge_config::{AzureMapperTimestamp, MqttBindAddressSetting, TEdgeConfig};
use tedge_config::{ConfigSettingAccessor, MqttPortSetting};
use tedge_utils::file::create_directory_with_user_group;
use tracing::{info, info_span, Instrument};
const AZURE_MAPPER_NAME: &str = "tedge-mapper-az";
pub struct AzureMapper {}
impl AzureMapper {
pub fn new() -> AzureMapper {
AzureMapper {}
}
}
#[async_trait]
impl TEdgeComponent for AzureMapper {
fn session_name(&self) -> &str {
AZURE_MAPPER_NAME
}
async fn init(&self, cfg_dir: &Path) -> Result<(), anyhow::Error> {
info!("Initialize tedge mapper az");
let config_dir = cfg_dir.display().to_string();
create_directory_with_user_group(
&format!("{config_dir}/operations/az"),
"tedge",
"tedge",
0o775,
)?;
self.init_session(AzureConverter::in_topic_filter()).await?;
Ok(())
}
async fn start(
&self,
tedge_config: TEdgeConfig,
_config_dir: &Path,
) -> Result<(), anyhow::Error> {
let add_timestamp = tedge_config.query(AzureMapperTimestamp)?.is_set();
let mqtt_port = tedge_config.query(MqttPortSetting)?.into();
let mqtt_host = tedge_config.query(MqttBindAddressSetting)?.to_string();
let clock = Box::new(WallClock);
let size_threshold = SizeThreshold(255 * 1024);
let converter = Box::new(AzureConverter::new(add_timestamp, clock, size_threshold));
let mut mapper = create_mapper(AZURE_MAPPER_NAME, mqtt_host, mqtt_port, converter).await?;
mapper
.run()
.instrument(info_span!(AZURE_MAPPER_NAME))
.await?;
Ok(())
}
}
|