Quellcode durchsuchen

actually add token

Ajay Tripathy vor 5 Jahren
Ursprung
Commit
96eafb2443
2 geänderte Dateien mit 5 neuen und 3 gelöschten Zeilen
  1. 4 2
      pkg/costmodel/router.go
  2. 1 1
      pkg/env/costmodelenv.go

+ 4 - 2
pkg/costmodel/router.go

@@ -48,8 +48,10 @@ var (
 	gitCommit                       string
 	dbBasicAuthUsername             string = env.GetDBBasicAuthUsername()
 	dbBasicAuthPW                   string = env.GetDBBasicAuthUserPassword()
+	dbBearerToken                   string = env.GetDBBearerToken()
 	multiclusterDBBasicAuthUsername string = env.GetMultiClusterBasicAuthUsername()
 	multiclusterDBBasicAuthPW       string = env.GetMultiClusterBasicAuthPassword()
+	multiClusterBearerToken         string = env.GetMultiClusterBearerToken()
 )
 
 var Router = httprouter.New()
@@ -741,7 +743,7 @@ func Initialize(additionalConfigWatchers ...ConfigWatchers) {
 		Address:      address,
 		RoundTripper: LongTimeoutRoundTripper,
 	}
-	promCli, _ := prom.NewRateLimitedClient(pc, queryConcurrency, dbBasicAuthUsername, dbBasicAuthPW)
+	promCli, _ := prom.NewRateLimitedClient(pc, queryConcurrency, dbBasicAuthUsername, dbBasicAuthPW, dbBearerToken)
 
 	m, err := ValidatePrometheus(promCli, false)
 	if err != nil || m.Running == false {
@@ -975,7 +977,7 @@ func Initialize(additionalConfigWatchers ...ConfigWatchers) {
 				RoundTripper: thanosRT,
 			}
 
-			thanosCli, _ := prom.NewRateLimitedClient(thanosConfig, queryConcurrency, multiclusterDBBasicAuthUsername, multiclusterDBBasicAuthPW)
+			thanosCli, _ := prom.NewRateLimitedClient(thanosConfig, queryConcurrency, multiclusterDBBasicAuthUsername, multiclusterDBBasicAuthPW, multiClusterBearerToken)
 
 			_, err = ValidatePrometheus(thanosCli, true)
 			if err != nil {

+ 1 - 1
pkg/env/costmodelenv.go

@@ -206,6 +206,6 @@ func GetMultiClusterBasicAuthPassword() string {
 	return Get(MultiClusterBasicAuthPassword, "")
 }
 
-func GetMultiClusterToken() string {
+func GetMultiClusterBearerToken() string {
 	return Get(MultiClusterBearerToken, "")
 }