fix(server): fixing profiler config (#3705)

* fix(server): fixing profiler config

* fix: profiler startup

server/cmd/serve.go

@@ -39,7 +39,7 @@ var serveCmd = &cobra.Command{
 
 		profilerConfig := cfg.ApplicationProfiler()
 		if profilerConfig.Enabled {
-			telemetry.StartProfiler(profilerConfig.Name, profilerConfig.Endpoint, profilerConfig.SamplingRate)
+			telemetry.StartProfiler(profilerConfig.Name, profilerConfig.Environment, profilerConfig.Endpoint, profilerConfig.SamplingRate)
 		}
 
 		wg.Add(1)

server/config/exporters.go

@@ -42,6 +42,7 @@ type (
 		Enabled      bool   `yaml:",omitempty" mapstructure:"enabled"`
 		Endpoint     string `yaml:",omitempty" mapstructure:"endpoint"`
 		SamplingRate int    `yaml:",omitempty" mapstructure:"samplingRate"`
+		Environment  string `yaml:",omitempty" mapstructure:"environment"`
 	}
 )
 

server/config/validate.go

@@ -37,7 +37,6 @@ func (c *AppConfig) Validate() error {
 			err,
 			fmt.Errorf("invalid config value for '%s': %s", opt.key, optErr.Error()),
 		)
-
 	}
 
 	return err

server/telemetry/profiler.go

@@ -1,6 +1,7 @@
 package telemetry
 
 import (
+	"fmt"
 	"os"
 	"runtime"
 
@@ -12,7 +13,10 @@ const (
 	BlockProfileFractionRate = 5
 )
 
-func StartProfiler(applicationName, telemetryServer string, samplingPercentage int) {
+func StartProfiler(applicationName, applicationEnvironment, telemetryServerAddress string, samplingPercentage int) {
+	fmt.Printf("Starting profiler for environment [%s] with sampling [%d]. Telemetry server address: %s\n",
+		applicationEnvironment, samplingPercentage, telemetryServerAddress)
+
 	samplingRate := 100 / samplingPercentage
 
 	runtime.SetMutexProfileFraction(samplingRate)
@@ -22,14 +26,16 @@ func StartProfiler(applicationName, telemetryServer string, samplingPercentage i
 		ApplicationName: applicationName,
 
 		// replace this with the address of pyroscope server
-		ServerAddress: telemetryServer,
+		ServerAddress: telemetryServerAddress,
 
 		// you can disable logging by setting this to nil
-		Logger: pyroscope.StandardLogger,
+		// Logger: pyroscope.StandardLogger,
+		Logger: nil,
 
 		// you can provide static tags via a map:
 		Tags: map[string]string{
-			"hostname": os.Getenv("HOSTNAME"),
+			"hostname":    os.Getenv("HOSTNAME"),
+			"environment": applicationEnvironment,
 		},
 
 		ProfileTypes: []pyroscope.ProfileType{