浏览代码

make metrics' source configuration (see compose-switch)

Signed-off-by: Nicolas De Loof <[email protected]>
Nicolas De Loof 4 年之前
父节点
当前提交
0cd080a786
共有 1 个文件被更改,包括 10 次插入2 次删除
  1. 10 2
      cli/metrics/client.go

+ 10 - 2
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