feat: set build version on upstream clients

flanksource · May 25, 2024 · 1cfe1af · 1cfe1af
1 parent 3e55225
commit 1cfe1af
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -5,6 +5,8 @@ import (
 	"os"
 	"strconv"
 
+	"github.com/flanksource/commons/http"
+
 	"github.com/flanksource/commons/logger"
 	"github.com/flanksource/config-db/api"
 	"github.com/flanksource/config-db/db"
@@ -67,6 +69,15 @@ var Root = &cobra.Command{
 		db.Schema = readFromEnv(db.Schema)
 		db.PGRSTLogLevel = readFromEnv(db.PGRSTLogLevel)
 
+		if api.UpstreamConfig.Valid() {
+			api.UpstreamConfig.Options = append(api.UpstreamConfig.Options, func(c *http.Client) {
+				c.UserAgent(fmt.Sprintf("config-db %s", version))
+			})
+			logger.Infof("pushing configs to %s", api.UpstreamConfig)
+		} else if partial, err := api.UpstreamConfig.IsPartiallyFilled(); partial && err != nil {
+			logger.Warnf("upstream is not fully configured: %s", api.UpstreamConfig)
+		}
+
 		if otelcollectorURL != "" {
 			logger.Infof("Sending traces to %s", otelcollectorURL)
 			_ = telemetry.InitTracer(otelServiceName, otelcollectorURL, true) // TODO: Setup runner