From fb0518552cf97e2bcbfb00abee39e35dd6d612a3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Einar=20L=C3=B6nn?= <el@sunet.se>
Date: Thu, 12 Nov 2020 12:51:12 +0100
Subject: [PATCH] Another piggy back on the Telegraf class to test dual
 outputs.

---
 .../etc/puppet/modules/eid/manifests/telegraf.pp   | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/global/overlay/etc/puppet/modules/eid/manifests/telegraf.pp b/global/overlay/etc/puppet/modules/eid/manifests/telegraf.pp
index f3df245b..783d186b 100644
--- a/global/overlay/etc/puppet/modules/eid/manifests/telegraf.pp
+++ b/global/overlay/etc/puppet/modules/eid/manifests/telegraf.pp
@@ -35,12 +35,22 @@ class eid::telegraf(String $version = "stable", Array $volumes = [], Array $port
 
    $_volumes = flatten(["/dev/log:/dev/log","/var/log:/var/log","/etc/telegraf:/etc/telegraf","/var/run/utmp:/var/run/utmp:ro",$volumes])
 
-   sunet::docker_run {"telegraf":
+  if ($influx2_token != 'NOT_SET_IN_HIERA') {
+    sunet::docker_run {"telegraf":
+       hostname => "${::fqdn}",
+       image     => "docker.sunet.se/eduid/telegraf",
+       imagetag  => "latest",
+       volumes   => $_volumes,
+       ports     => $ports,
+       env       => ["INFLUXDB2_TOKEN=$influxdb2_token","INFLUXDB2_ORG=$influxdb2_org","INFLUXDB2_BUCKET=$influxdb2_bucket"]
+  } else {
+    sunet::docker_run {"telegraf":
       hostname => "${::fqdn}",
       image     => "docker.sunet.se/eduid/telegraf",
       imagetag  => $version,
       volumes   => $_volumes,
       ports     => $ports,
       env       => ["INFLUXDB2_TOKEN=$influxdb2_token","INFLUXDB2_ORG=$influxdb2_org","INFLUXDB2_BUCKET=$influxdb2_bucket"]
-   }
+    }
+  }
 }