Skip to content

Commit bab5d23

Browse files
committed
Add command line option to set metric server address
1 parent 033d3d2 commit bab5d23

File tree

3 files changed

+11
-4
lines changed

3 files changed

+11
-4
lines changed

Diff for: README.md

+4
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,10 @@ OPTIONS:
6565
-pprof.addr <ip:port>
6666
pprof server address. pprof will expose it's metrics on this address.
6767
68+
-metric.addr <ip:port>
69+
metric server address. prometheus metrics will be exposed on this address.
70+
Default: *:2112
71+
6872
-log.lvl {trace,debug,info,warn,error,fatal}
6973
Log level.
7074
Default: info

Diff for: main.go

+3-2
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,10 @@ func main() {
2929
}
3030

3131
func run(args []string, stdout io.Writer) error {
32-
var confFile, pprofAddr, loglvl string
32+
var confFile, pprofAddr, metricAddr, loglvl string
3333
flag.StringVar(&confFile, "config", "config.yaml", "path to config file")
3434
flag.StringVar(&pprofAddr, "pprof.addr", "", "pprof addr")
35+
flag.StringVar(&metricAddr, "metric.addr", ":2112", "metric server addr")
3536
flag.StringVar(&loglvl, "log.lvl", "debug", "log level. Can be {trace,debug,info,warn,error,fatal}")
3637
flag.Parse()
3738

@@ -67,7 +68,7 @@ func run(args []string, stdout io.Writer) error {
6768
}
6869

6970
// Create metrics server
70-
scanner.MetricsServ = *metrics.Init()
71+
scanner.MetricsServ = *metrics.Init(metricAddr)
7172

7273
// Start metrics server
7374
go func() {

Diff for: metrics/metrics.go

+4-2
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ type PingInfo struct {
3838
}
3939

4040
// Init initialize the metrics
41-
func Init() *Server {
41+
func Init(addr string) *Server {
4242
s := Server{
4343
NumOfTargets: prometheus.NewGauge(prometheus.GaugeOpts{
4444
Name: "scanexporter_targets_number_total",
@@ -96,6 +96,8 @@ func Init() *Server {
9696
s.Rtt,
9797
)
9898

99+
s.Addr = addr
100+
99101
// Initialize the map
100102
s.NotRespondingList = make(map[string]bool)
101103

@@ -108,7 +110,7 @@ func Init() *Server {
108110
// Start starts the prometheus server
109111
func (s *Server) Start() error {
110112
srv := &http.Server{
111-
Addr: ":2112",
113+
Addr: s.Addr,
112114
Handler: handlers.HandleFunc(),
113115
ReadTimeout: 5 * time.Second,
114116
WriteTimeout: 10 * time.Second,

0 commit comments

Comments
 (0)