From 91e0bf1349cd59be52f06d1105739cc560230926 Mon Sep 17 00:00:00 2001
From: nvnamsss <nvnam.c@gmail.com>
Date: Fri, 26 Aug 2022 13:54:15 +0700
Subject: [PATCH] Added flag to turn on / off prometheus

---
 adapters/config.go |  1 +
 controller.go      | 14 +++++++++-----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/adapters/config.go b/adapters/config.go
index ded16a6..1e71682 100644
--- a/adapters/config.go
+++ b/adapters/config.go
@@ -12,6 +12,7 @@ type Prometheus struct {
 	PushURL      string `default:"localhost:9091" envconfig:"PUSH_URL"`
 	PushJob      string `default:"sm-bridge-v2" envconfig:"PUSH_JOB"`
 	PushInterval int    `default:"15" envconfig:"PUSH_INTERVAl"`
+	TurnOn       bool   `envconfig:"PROMETHEUS_TURN_ON"`
 }
 
 func New() (*Config, error) {
diff --git a/controller.go b/controller.go
index e0b6d09..3cf294c 100644
--- a/controller.go
+++ b/controller.go
@@ -3,15 +3,17 @@ package internal
 import (
 	"context"
 	"errors"
-	"github.com/axieinfinity/bridge-contracts"
-	"github.com/axieinfinity/bridge-core/metrics"
-	"github.com/axieinfinity/bridge-core/stores"
-	"github.com/axieinfinity/bridge-core/utils"
 	"strings"
 	"sync"
 	"sync/atomic"
 	"time"
 
+	bridge_contracts "github.com/axieinfinity/bridge-contracts"
+	"github.com/axieinfinity/bridge-core/adapters"
+	"github.com/axieinfinity/bridge-core/metrics"
+	"github.com/axieinfinity/bridge-core/stores"
+	"github.com/axieinfinity/bridge-core/utils"
+
 	"github.com/ethereum/go-ethereum/accounts/abi"
 	"github.com/ethereum/go-ethereum/accounts/abi/bind"
 	"github.com/ethereum/go-ethereum/common"
@@ -101,7 +103,9 @@ func New(cfg *Config, db *gorm.DB, helpers utils.Utils) (*Controller, error) {
 		hasSubscriptionType: make(map[string]map[int]bool),
 	}
 
-	metrics.RunPusher(ctx)
+	if adapters.AppConfig.Prometheus.TurnOn {
+		metrics.RunPusher(ctx)
+	}
 
 	c.isClosed.Store(false)
 	if helpers != nil {