@@ -23,107 +23,26 @@ import (
2323
2424// ExporterConfig configures ports to scan to what filename to save it to.
2525// if path is set we will try to make a HTTP get and find # TYPE in the first 10 rows of the response to make sure we know its prometheus metrics.
26- // TODO make this runtime configurable.
2726type ExporterConfig []struct {
2827 port string
2928 filename string
3029 path string
3130}
3231
33- const ExporterExporterPort = "9999"
34-
3532var exporterConfig = ExporterConfig {
36- { // special exporter_exporter we scan this first to know if we can skip the other ports.
37- port : ExporterExporterPort ,
33+ // We only support exporter_exporter at the moment
34+ {
35+ port : "" ,
3836 filename : "" ,
3937 },
40- /*
41- disabled those since we only use exporter_exporter on 9999 for now.
42- TODO move this to a config
43- {
44- port: "8081",
45- filename: "php",
46- path: "http://%s/metrics",
47- },
48- {
49- port: "9100",
50- filename: "node",
51- },
52- {
53- port: "9108",
54- filename: "elasticsearch",
55- },
56- {
57- port: "9114",
58- filename: "elasticsearch",
59- },
60- {
61- port: "9216",
62- filename: "mongodb",
63- },
64- {
65- port: "9091",
66- filename: "minio",
67- path: "http://%s/minio/prometheus/metrics",
68- },
69- {
70- port: "9101",
71- filename: "haproxy",
72- },
73- {
74- port: "9104",
75- filename: "mysql",
76- },
77- {
78- port: "9113",
79- filename: "nginx",
80- },
81- {
82- port: "9121",
83- filename: "redis",
84- },
85- {
86- port: "9150",
87- filename: "memcached",
88- },
89- {
90- port: "9154",
91- filename: "postfix",
92- },
93- {
94- port: "9182",
95- filename: "wmi",
96- },
97- {
98- port: "9187",
99- filename: "postgres",
100- },
101- {
102- port: "9188",
103- filename: "pgbouncer",
104- },
105- {
106- port: "9189",
107- filename: "barman",
108- },
109- {
110- port: "9253",
111- filename: "php-fpm",
112- },
113- {
114- port: "9308",
115- filename: "kafka",
116- },
117- {
118- port: "9496",
119- filename: "389ds",
120- },
121- */
12238}
12339
40+ var mutex sync.Mutex
41+
12442func main () {
12543 config := & Config {}
12644 multiconfig .MustLoad (& config )
45+ exporterConfig [0 ].port = config .ExpoterExporterPort
12746
12847 fnxlogrus .Init (config .Log , logrus .StandardLogger ())
12948
@@ -138,14 +57,11 @@ func main() {
13857 logrus .Infof ("Running with interval %s" , interval )
13958
14059 ticker := time .NewTicker (interval )
141- interrupt := make (chan os.Signal , 1 )
142- signal .Notify (interrupt , os .Interrupt , syscall .SIGQUIT , syscall .SIGTERM , syscall .SIGINT )
143-
144- ctx , cancel := context .WithCancel (context .Background ())
60+ ctx , cancel := signal .NotifyContext (context .Background (), os .Interrupt , syscall .SIGQUIT , syscall .SIGTERM )
61+ defer cancel ()
14562
14663 go func () {
147- <- interrupt
148- cancel ()
64+ startWs (config , ctx )
14965 ticker .Stop ()
15066 log .Println ("Shutting down" )
15167 }()
@@ -226,6 +142,8 @@ func runDiscovery(parentCtx context.Context, config *Config, networks []string)
226142}
227143
228144func saveConfigs (ctx context.Context , config * Config , exporters Exporters ) {
145+ mutex .Lock ()
146+ defer mutex .Unlock ()
229147 for name , addresses := range exporters {
230148 if ctx .Err () != nil {
231149 return
@@ -260,16 +178,11 @@ func discoverNetwork(network string, queue chan func(context.Context), exporter
260178 port := data .port
261179 logrus .Debugf ("scanning port: %s:%s" , ip , port )
262180 var exporters []string
263- if port == ExporterExporterPort {
264- exporters , err = checkExporterExporter (ctx , ip , port )
265- if err != nil {
266- if ! errors .Is (err , context .DeadlineExceeded ) && ! IsTimeout (err ) {
267- logrus .Errorf ("error fetching from exporter_exporter: %s" , err )
268- }
269- logrus .Debugf ("%s:%s was not open" , ip , port )
270- continue
181+ exporters , err = checkExporterExporter (ctx , ip , port )
182+ if err != nil {
183+ if ! errors .Is (err , context .DeadlineExceeded ) && ! IsTimeout (err ) {
184+ logrus .Debugf ("error fetching from exporter_exporter: %s" , err )
271185 }
272- } else if ! alive (ctx , ip , port , data .path ) {
273186 logrus .Debugf ("%s:%s was not open" , ip , port )
274187 continue
275188 }
@@ -297,18 +210,7 @@ func discoverNetwork(network string, queue chan func(context.Context), exporter
297210 }
298211 exporter <- & a
299212 }
300- return // we found exporter_exporter dont scan other ports so return here
301213 }
302-
303- a := Address {
304- IP : strings .TrimSpace (ip ),
305- Hostname : strings .TrimSpace (hostname ),
306- Subnet : strings .TrimSpace (network ),
307- Exporter : data .filename ,
308- Port : port ,
309- }
310-
311- exporter <- & a
312214 }
313215 }
314216 }
@@ -335,9 +237,9 @@ func getOldGroups(path string) ([]Group, error) {
335237func writeFileSDConfig (config * Config , exporterName string , addresses []Address ) error {
336238 path := filepath .Join (config .FileSdPath , exporterName + ".json" )
337239
338- if _ , err := os .Stat (config .FileSdPath ); os .IsNotExist (err ) {
339- os .MkdirAll (config .FileSdPath , 0755 )
340- }
240+ if _ , err := os .Stat (config .FileSdPath ); os .IsNotExist (err ) {
241+ os .MkdirAll (config .FileSdPath , 0755 )
242+ }
341243
342244 groups := []Group {}
343245
@@ -349,7 +251,7 @@ func writeFileSDConfig(config *Config, exporterName string, addresses []Address)
349251 "host" : v .Hostname ,
350252 },
351253 }
352- if v .Port == ExporterExporterPort {
254+ if v .Port == config . ExpoterExporterPort {
353255 group .Labels ["__metrics_path__" ] = "/proxy"
354256 group .Labels ["__param_module" ] = exporterName
355257 }
0 commit comments