From 0cd080a78658cc57b24769987e5300901e3ebb99 Mon Sep 17 00:00:00 2001 From: Nicolas De Loof Date: Thu, 1 Apr 2021 14:42:05 +0200 Subject: [PATCH] make metrics' source configuration (see compose-switch) Signed-off-by: Nicolas De Loof --- cli/metrics/client.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/cli/metrics/client.go b/cli/metrics/client.go index d19f8762c..44954c790 100644 --- a/cli/metrics/client.go +++ b/cli/metrics/client.go @@ -22,6 +22,7 @@ import ( "encoding/json" "net" "net/http" + "os" "time" ) @@ -37,9 +38,16 @@ type Command struct { Status string `json:"status"` } +// CLISource is sent for cli metrics +var CLISource = "cli" + +func init() { + if v, ok := os.LookupEnv("DOCKER_METRICS_SOURCE"); ok { + CLISource = v + } +} + const ( - // CLISource is sent for cli metrics - CLISource = "cli" // APISource is sent for API metrics APISource = "api" // SuccessStatus is sent for API metrics