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
|
use std::{fmt, path::PathBuf};
use crate::{
az::mapper::AzureMapper, c8y::mapper::CumulocityMapper, collectd::mapper::CollectdMapper,
core::component::TEdgeComponent,
};
use clap::Parser;
use flockfile::check_another_instance_is_not_running;
use tedge_config::DEFAULT_TEDGE_CONFIG_PATH;
use tedge_config::*;
mod az;
mod c8y;
mod collectd;
mod core;
fn lookup_component(component_name: &MapperName) -> Box<dyn TEdgeComponent> {
match component_name {
MapperName::Az => Box::new(AzureMapper::new()),
MapperName::Collectd => Box::new(CollectdMapper::new()),
MapperName::C8y => Box::new(CumulocityMapper::new()),
}
}
#[derive(Debug, Parser)]
#[clap(
name = clap::crate_name!(),
version = clap::crate_version!(),
about = clap::crate_description!()
)]
pub struct MapperOpt {
#[clap(subcommand)]
pub name: MapperName,
/// Turn-on the debug log level.
///
/// If off only reports ERROR, WARN, and INFO
/// If on also reports DEBUG and TRACE
#[clap(long, global = true)]
pub debug: bool,
/// Start the mapper with clean session off, subscribe to the topics, so that no messages are lost
#[clap(short, long)]
pub init: bool,
/// Start the agent with clean session on, drop the previous session and subscriptions
///
/// WARNING: All pending messages will be lost.
#[clap(short, long)]
pub clear: bool,
/// Start the mapper from custom path
///
/// WARNING: This is mostly used in testing.
#[clap(long = "config-dir", default_value = DEFAULT_TEDGE_CONFIG_PATH)]
pub config_dir: PathBuf,
}
#[derive(Debug, clap::Subcommand)]
pub enum MapperName {
Az,
C8y,
Collectd,
}
impl fmt::Display for MapperName {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
MapperName::Az => write!(f, "tedge-mapper-az"),
MapperName::C8y => write!(f, "tedge-mapper-c8y"),
MapperName::Collectd => write!(f, "tedge-mapper-collectd"),
}
}
}
#[tokio::main]
async fn main() -> anyhow::Result<()> {
let mapper_opt = MapperOpt::parse();
tedge_utils::logging::initialise_tracing_subscriber(mapper_opt.debug);
let component = lookup_component(&mapper_opt.name);
let tedge_config_location =
tedge_config::TEdgeConfigLocation::from_custom_root(&mapper_opt.config_dir);
let config = tedge_config::TEdgeConfigRepository::new(tedge_config_location.clone()).load()?;
// Run only one instance of a mapper
let _flock = check_another_instance_is_not_running(
&mapper_opt.name.to_string(),
&config.query(RunPathDefaultSetting)?.into(),
)?;
if mapper_opt.init {
component.init().await
} else if mapper_opt.clear {
component.clear_session().await
} else {
component.start(config).await
}
}
|
