[prometheus] add EnablePush flag

Signed-off-by: Leo Chen <leo@harmony.one>

api/service/prometheus/service.go

@@ -19,6 +19,7 @@ type PrometheusConfig struct {
 	Enabled    bool
 	IP         string
 	Port       int
+	EnablePush bool   // enable pushgateway support
 	Gateway    string // address of the pushgateway
 	Network    string // network type, used as job prefix
 	Shard      uint32 // shard id, used as job suffix
@@ -69,6 +70,10 @@ func NewService(additionalHandlers ...Handler) {
 	// start pusher to push metrics to prometheus pushgateway
 	// every minute
 	go func(config PrometheusConfig) {
+		if !config.EnablePush {
+			utils.Logger().Info().Msg("Prometheus pushgateway support is disabled...")
+			return
+		}
 		ticker := time.NewTicker(time.Minute)
 		defer ticker.Stop()
 		for {
@@ -130,6 +135,7 @@ func SetConfig(
 	enabled bool,
 	ip string,
 	port int,
+	enablepush bool,
 	gateway string,
 	network string,
 	shard uint32,
@@ -138,6 +144,7 @@ func SetConfig(
 	config.Enabled = enabled
 	config.IP = ip
 	config.Port = port
+	config.EnablePush = enablepush
 	config.Gateway = gateway
 	config.Network = network
 	config.Shard = shard

cmd/harmony/config.go

@@ -146,6 +146,7 @@ type prometheusConfig struct {
 	Enabled    bool
 	IP         string
 	Port       int
+	EnablePush bool
 	Gateway    string
 }
 

cmd/harmony/default.go

@@ -70,6 +70,7 @@ var defaultConfig = harmonyConfig{
 		IP:         "0.0.0.0",
 		Port:       nodeconfig.DefaultPrometheusPort,
 		Gateway:    "https://gateway.harmony.one",
+		EnablePush: true,
 	},
 }
 
@@ -103,6 +104,7 @@ var defaultPrometheusConfig = prometheusConfig{
 	Enabled:    true,
 	IP:         "0.0.0.0",
 	Port:       9900,
+	EnablePush: true,
 	Gateway:    "https://gateway.harmony.one",
 }
 

cmd/harmony/flags.go

@@ -171,6 +171,7 @@ var (
 		prometheusIPFlag,
 		prometheusPortFlag,
 		prometheusGatewayFlag,
+		prometheusEnablePushFlag,
 	}
 )
 
@@ -1211,6 +1212,11 @@ var (
 		Usage:    "prometheus pushgateway URL",
 		DefValue: defaultConfig.Prometheus.Gateway,
 	}
+	prometheusEnablePushFlag = cli.BoolFlag{
+		Name:     "prometheus.push",
+		Usage:    "enable prometheus pushgateway",
+		DefValue: defaultConfig.Prometheus.EnablePush,
+	}
 )
 
 func applyPrometheusFlags(cmd *cobra.Command, config *harmonyConfig) {
@@ -1234,4 +1240,7 @@ func applyPrometheusFlags(cmd *cobra.Command, config *harmonyConfig) {
 	if cli.IsFlagChanged(cmd, prometheusGatewayFlag) {
 		config.Prometheus.Gateway = cli.GetStringFlagValue(cmd, prometheusGatewayFlag)
 	}
+	if cli.IsFlagChanged(cmd, prometheusEnablePushFlag) {
+		config.Prometheus.EnablePush = cli.GetBoolFlagValue(cmd, prometheusEnablePushFlag)
+	}
 }

cmd/harmony/main.go

@@ -377,6 +377,7 @@ func setupNodeAndRun(hc harmonyConfig) {
 		hc.Prometheus.Enabled,
 		hc.Prometheus.IP,
 		hc.Prometheus.Port,
+		hc.Prometheus.EnablePush,
 		hc.Prometheus.Gateway,
 		hc.Network.NetworkType,
 		nodeConfig.ShardID,