Add organization to influxdb configuration

Author: dmakarov

src/main.rs

@@ -4497,6 +4497,13 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                                 .required(true)
                                 .help("Access Token"),
                         )
+                        .arg(
+                            Arg::with_name("org")
+                                .value_name("ORG")
+                                .takes_value(true)
+                                .required(true)
+                                .help("Organization"),
+                        )
                         .arg(
                             Arg::with_name("bucket")
                                 .value_name("BUCKET")
@@ -6000,17 +6007,20 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
                 Some(MetricsConfig {
                     url,
                     token: _,
+                    org,
                     bucket,
                 }) => {
                     println!("Url: {url}");
                     println!("Token: ********");
+                    println!("Organization: {org}");
                     println!("Bucket: {bucket}");
                 }
             },
             ("set", Some(arg_matches)) => {
                 db.set_metrics_config(MetricsConfig {
                     url: value_t_or_exit!(arg_matches, "url", String),
                     token: value_t_or_exit!(arg_matches, "token", String),
+                    org: value_t_or_exit!(arg_matches, "org", String),
                     bucket: value_t_or_exit!(arg_matches, "bucket", String),
                 })?;
                 println!("InfluxDb configuration set");

src/metrics.rs

@@ -15,13 +15,15 @@ lazy_static::lazy_static! {
 pub struct MetricsConfig {
     pub url: String,
     pub token: String,
+    pub org: String,
     pub bucket: String,
 }
 
 pub fn env_config() -> Option<MetricsConfig> {
     Some(MetricsConfig {
         url: env::var("INFLUX_URL").ok()?,
         token: env::var("INFLUX_API_TOKEN").ok()?,
+        org: env::var("INFLUX_ORG").ok()?,
         bucket: env::var("INFLUX_BUCKET")
             .ok()
             .unwrap_or_else(|| "sys".into()),
@@ -35,6 +37,7 @@ pub async fn push(point: Point) {
 pub async fn send(config: Option<MetricsConfig>) {
     if let Some(config) = config {
         let client = Client::new(config.url, config.token)
+            .with_org(config.org)
             .with_bucket(config.bucket)
             .with_precision(Precision::MS);
         //let client = client.insert_to_stdout();