forked from openshift/origin
-
Notifications
You must be signed in to change notification settings - Fork 0
/
haproxy.go
654 lines (596 loc) · 23.6 KB
/
haproxy.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
// Package haproxy is inspired by https://github.com/prometheus/haproxy_exporter
package haproxy
import (
"encoding/csv"
"errors"
"fmt"
"io"
"io/ioutil"
"net"
"net/http"
_ "net/http/pprof"
"net/url"
"sort"
"strconv"
"strings"
"sync"
"time"
"github.com/golang/glog"
"github.com/prometheus/client_golang/prometheus"
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
)
const (
namespace = "haproxy" // For Prometheus metrics.
// HAProxy 1.4
// # pxname,svname,qcur,qmax,scur,smax,slim,stot,bin,bout,dreq,dresp,ereq,econ,eresp,wretr,wredis,status,weight,act,bck,chkfail,chkdown,lastchg,downtime,qlimit,pid,iid,sid,throttle,lbtot,tracked,type,rate,rate_lim,rate_max,check_status,check_code,check_duration,hrsp_1xx,hrsp_2xx,hrsp_3xx,hrsp_4xx,hrsp_5xx,hrsp_other,hanafail,req_rate,req_rate_max,req_tot,cli_abrt,srv_abrt,
// HAProxy 1.5
// These columns are part of the stable API for HAProxy and are documented here: https://cbonte.github.io/haproxy-dconv/1.5/configuration.html#9.1
// pxname,svname,qcur,qmax,scur,smax,slim,stot,bin,bout,dreq,dresp,ereq,econ,eresp,wretr,wredis,status,weight,act,bck,chkfail,chkdown,lastchg,downtime,qlimit,pid,iid,sid,throttle,lbtot,tracked,type,rate,rate_lim,rate_max,check_status,check_code,check_duration,hrsp_1xx,hrsp_2xx,hrsp_3xx,hrsp_4xx,hrsp_5xx,hrsp_other,hanafail,req_rate,req_rate_max,req_tot,cli_abrt,srv_abrt,comp_in,comp_out,comp_byp,comp_rsp,lastsess,last_chk,last_agt,qtime,ctime,rtime,ttime
expectedCsvFieldCount = 52
statusField = 17
frontendType = "0"
backendType = "1"
serverType = "2"
listenerType = "3"
)
var (
frontendLabelNames = []string{"frontend"}
backendLabelNames = []string{"backend", "namespace", "route"}
serverLabelNames = []string{"server", "namespace", "route", "pod", "service"}
)
func newFrontendMetric(metricName string, docString string, constLabels prometheus.Labels) *prometheus.GaugeVec {
return prometheus.NewGaugeVec(
prometheus.GaugeOpts{
Namespace: namespace,
Name: "frontend_" + metricName,
Help: docString,
ConstLabels: constLabels,
},
frontendLabelNames,
)
}
func newBackendMetric(metricName string, docString string, constLabels prometheus.Labels) *prometheus.GaugeVec {
return prometheus.NewGaugeVec(
prometheus.GaugeOpts{
Namespace: namespace,
Name: "backend_" + metricName,
Help: docString,
ConstLabels: constLabels,
},
backendLabelNames,
)
}
func newServerMetric(metricName string, docString string, constLabels prometheus.Labels) *prometheus.GaugeVec {
return prometheus.NewGaugeVec(
prometheus.GaugeOpts{
Namespace: namespace,
Name: "server_" + metricName,
Help: docString,
ConstLabels: constLabels,
},
serverLabelNames,
)
}
type metrics map[int]*prometheus.GaugeVec
func (m metrics) Names() []int {
keys := make([]int, 0, len(m))
for k := range m {
keys = append(keys, k)
}
sort.Ints(keys)
return keys
}
// defaultSelectedMetrics is the list of metrics included by default. These metrics are a subset
// of the metrics exposed by haproxy_exporter by default for performance reasons.
var defaultSelectedMetrics = []int{2, 4, 5, 7, 8, 9, 13, 14, 17, 21, 24, 33, 35, 40, 43, 60}
// Exporter collects HAProxy stats from the given URI and exports them using
// the prometheus metrics package.
type Exporter struct {
opts PrometheusOptions
mutex sync.RWMutex
fetch func() (io.ReadCloser, error)
// pendingScrape indicates that a scrape was triggered in the background, and so metrics should be
// reported without recollection
pendingScrape bool
// lastScrape is the time the last scrape was invoked if at all
lastScrape *time.Time
// scrapeInterval is a calculated value based on the number of rows returned by HAProxy
scrapeInterval time.Duration
// serverLimited is true when above opts.ServerThreshold, and no server metrics will be
// reported. Instead, the full set of backend metrics will be reported instead.
serverLimited bool
// reducedBackendExports is the list of metrics that are not redundant with servers - when
// server metrics are being reported only these backendExports are shown.
reducedBackendExports map[int]struct{}
up, nextScrapeInterval prometheus.Gauge
totalScrapes, csvParseFailures prometheus.Counter
serverThresholdCurrent, serverThresholdLimit prometheus.Gauge
frontendMetrics, backendMetrics, serverMetrics map[int]*prometheus.GaugeVec
}
// NewExporter returns an initialized Exporter. baseScrapeInterval is how often to scrape per 1000 entries
// (servers, backends, and frontends).
func NewExporter(opts PrometheusOptions) (*Exporter, error) {
u, err := url.Parse(opts.ScrapeURI)
if err != nil {
return nil, err
}
var fetch func() (io.ReadCloser, error)
switch u.Scheme {
case "http", "https", "file":
fetch = fetchHTTP(opts.ScrapeURI, opts.Timeout)
case "unix":
fetch = fetchUnix(u, opts.Timeout)
default:
return nil, fmt.Errorf("unsupported scheme: %q", u.Scheme)
}
return &Exporter{
opts: opts,
fetch: fetch,
up: prometheus.NewGauge(prometheus.GaugeOpts{
Namespace: namespace,
Name: "up",
Help: "Was the last scrape of haproxy successful.",
}),
totalScrapes: prometheus.NewCounter(prometheus.CounterOpts{
Namespace: namespace,
Name: "exporter_total_scrapes",
Help: "Current total HAProxy scrapes.",
}),
serverThresholdCurrent: prometheus.NewGauge(prometheus.GaugeOpts{
Namespace: namespace,
Name: "exporter_server_threshold",
Help: "Number of servers tracked and the current threshold value.",
ConstLabels: prometheus.Labels{"type": "current"},
}),
serverThresholdLimit: prometheus.NewGauge(prometheus.GaugeOpts{
Namespace: namespace,
Name: "exporter_server_threshold",
Help: "Number of servers tracked and the current threshold value.",
ConstLabels: prometheus.Labels{"type": "limit"},
}),
nextScrapeInterval: prometheus.NewGauge(prometheus.GaugeOpts{
Namespace: namespace,
Name: "exporter_scrape_interval",
Help: "The time in seconds before another scrape is allowed, proportional to size of data.",
}),
csvParseFailures: prometheus.NewCounter(prometheus.CounterOpts{
Namespace: namespace,
Name: "exporter_csv_parse_failures",
Help: "Number of errors while parsing CSV.",
}),
frontendMetrics: filterMetrics(opts.ExportedMetrics, metrics{
4: newFrontendMetric("current_sessions", "Current number of active sessions.", nil),
5: newFrontendMetric("max_sessions", "Maximum observed number of active sessions.", nil),
6: newFrontendMetric("limit_sessions", "Configured session limit.", nil),
7: newFrontendMetric("connections_total", "Total number of connections.", nil),
8: newFrontendMetric("bytes_in_total", "Current total of incoming bytes.", nil),
9: newFrontendMetric("bytes_out_total", "Current total of outgoing bytes.", nil),
10: newFrontendMetric("requests_denied_total", "Total of requests denied for security.", nil),
12: newFrontendMetric("request_errors_total", "Total of request errors.", nil),
33: newFrontendMetric("current_session_rate", "Current number of sessions per second over last elapsed second.", nil),
34: newFrontendMetric("limit_session_rate", "Configured limit on new sessions per second.", nil),
35: newFrontendMetric("max_session_rate", "Maximum observed number of sessions per second.", nil),
39: newFrontendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "1xx"}),
40: newFrontendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "2xx"}),
41: newFrontendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "3xx"}),
42: newFrontendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "4xx"}),
43: newFrontendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "5xx"}),
44: newFrontendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "other"}),
48: newFrontendMetric("http_requests_total", "Total HTTP requests.", nil),
60: newFrontendMetric("http_average_response_latency_milliseconds", "Average response latency of the last 1024 requests in milliseconds.", nil),
}),
reducedBackendExports: map[int]struct{}{2: {}, 3: {}, 7: {}, 17: {}},
backendMetrics: filterMetrics(opts.ExportedMetrics, metrics{
2: newBackendMetric("current_queue", "Current number of queued requests not assigned to any server.", nil),
3: newBackendMetric("max_queue", "Maximum observed number of queued requests not assigned to any server.", nil),
4: newBackendMetric("current_sessions", "Current number of active sessions.", nil),
5: newBackendMetric("max_sessions", "Maximum observed number of active sessions.", nil),
6: newBackendMetric("limit_sessions", "Configured session limit.", nil),
7: newBackendMetric("connections_total", "Total number of connections.", nil),
8: newBackendMetric("bytes_in_total", "Current total of incoming bytes.", nil),
9: newBackendMetric("bytes_out_total", "Current total of outgoing bytes.", nil),
13: newBackendMetric("connection_errors_total", "Total of connection errors.", nil),
14: newBackendMetric("response_errors_total", "Total of response errors.", nil),
15: newBackendMetric("retry_warnings_total", "Total of retry warnings.", nil),
16: newBackendMetric("redispatch_warnings_total", "Total of redispatch warnings.", nil),
17: newBackendMetric("up", "Current health status of the backend (1 = UP, 0 = DOWN).", nil),
18: newBackendMetric("weight", "Total weight of the servers in the backend.", nil),
33: newBackendMetric("current_session_rate", "Current number of sessions per second over last elapsed second.", nil),
35: newBackendMetric("max_session_rate", "Maximum number of sessions per second.", nil),
39: newBackendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "1xx"}),
40: newBackendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "2xx"}),
41: newBackendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "3xx"}),
42: newBackendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "4xx"}),
43: newBackendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "5xx"}),
44: newBackendMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "other"}),
60: newBackendMetric("http_average_response_latency_milliseconds", "Average response latency of the last 1024 requests in milliseconds.", nil),
}),
serverMetrics: filterMetrics(opts.ExportedMetrics, metrics{
2: newServerMetric("current_queue", "Current number of queued requests assigned to this server.", nil),
3: newServerMetric("max_queue", "Maximum observed number of queued requests assigned to this server.", nil),
4: newServerMetric("current_sessions", "Current number of active sessions.", nil),
5: newServerMetric("max_sessions", "Maximum observed number of active sessions.", nil),
6: newServerMetric("limit_sessions", "Configured session limit.", nil),
7: newServerMetric("connections_total", "Total number of connections.", nil),
8: newServerMetric("bytes_in_total", "Current total of incoming bytes.", nil),
9: newServerMetric("bytes_out_total", "Current total of outgoing bytes.", nil),
13: newServerMetric("connection_errors_total", "Total of connection errors.", nil),
14: newServerMetric("response_errors_total", "Total of response errors.", nil),
15: newServerMetric("retry_warnings_total", "Total of retry warnings.", nil),
16: newServerMetric("redispatch_warnings_total", "Total of redispatch warnings.", nil),
17: newServerMetric("up", "Current health status of the server (1 = UP, 0 = DOWN).", nil),
18: newServerMetric("weight", "Current weight of the server.", nil),
21: newServerMetric("check_failures_total", "Total number of failed health checks.", nil),
24: newServerMetric("downtime_seconds_total", "Total downtime in seconds.", nil),
33: newServerMetric("current_session_rate", "Current number of sessions per second over last elapsed second.", nil),
35: newServerMetric("max_session_rate", "Maximum observed number of sessions per second.", nil),
38: newServerMetric("check_duration_milliseconds", "Previously run health check duration, in milliseconds", nil),
39: newServerMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "1xx"}),
40: newServerMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "2xx"}),
41: newServerMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "3xx"}),
42: newServerMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "4xx"}),
43: newServerMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "5xx"}),
44: newServerMetric("http_responses_total", "Total of HTTP responses.", prometheus.Labels{"code": "other"}),
60: newServerMetric("http_average_response_latency_milliseconds", "Average response latency of the last 1024 requests in milliseconds.", nil),
}),
}, nil
}
// Describe describes all the metrics ever exported by the HAProxy exporter. It
// implements prometheus.Collector.
func (e *Exporter) Describe(ch chan<- *prometheus.Desc) {
for _, m := range e.frontendMetrics {
m.Describe(ch)
}
for _, m := range e.backendMetrics {
m.Describe(ch)
}
for _, m := range e.serverMetrics {
m.Describe(ch)
}
ch <- e.up.Desc()
ch <- e.totalScrapes.Desc()
ch <- e.nextScrapeInterval.Desc()
ch <- e.serverThresholdCurrent.Desc()
ch <- e.serverThresholdLimit.Desc()
ch <- e.csvParseFailures.Desc()
}
// Collect fetches the stats from configured HAProxy location and delivers them
// as Prometheus metrics. It implements prometheus.Collector. It will not collect
// metrics more often than interval.
func (e *Exporter) Collect(ch chan<- prometheus.Metric) {
e.mutex.Lock()
defer e.mutex.Unlock()
now := time.Now()
switch {
case e.pendingScrape:
// CollectNow was called before
e.pendingScrape = false
case e.lastScrape != nil && e.lastScrape.Add(e.scrapeInterval).After(now):
// do nothing, return the most recently scraped metrics
glog.V(6).Infof("Will not scrape HAProxy metrics more often than every %s", e.scrapeInterval)
default:
e.lastScrape = &now
e.resetMetrics()
e.scrape()
}
ch <- e.up
ch <- e.totalScrapes
ch <- e.nextScrapeInterval
ch <- e.serverThresholdCurrent
ch <- e.serverThresholdLimit
ch <- e.csvParseFailures
e.collectMetrics(ch)
}
// CollectNow performs a collection immediately. The next Collect() will report the metrics.
func (e *Exporter) CollectNow() {
e.mutex.Lock()
defer e.mutex.Unlock()
e.resetMetrics()
e.scrape()
e.lastScrape = nil
e.pendingScrape = true
}
func fetchHTTP(uri string, timeout time.Duration) func() (io.ReadCloser, error) {
client := http.Client{
Timeout: timeout,
}
return func() (io.ReadCloser, error) {
resp, err := client.Get(uri)
if err != nil {
return nil, err
}
if !(resp.StatusCode >= 200 && resp.StatusCode < 300) {
resp.Body.Close()
return nil, fmt.Errorf("HTTP status %d", resp.StatusCode)
}
return resp.Body, nil
}
}
func fetchUnix(u *url.URL, timeout time.Duration) func() (io.ReadCloser, error) {
return func() (io.ReadCloser, error) {
f, err := net.DialTimeout("unix", u.Path, timeout)
if err != nil {
return nil, err
}
if err := f.SetDeadline(time.Now().Add(timeout)); err != nil {
f.Close()
return nil, err
}
cmd := "show stat\n"
n, err := io.WriteString(f, cmd)
if err != nil {
f.Close()
return nil, err
}
if n != len(cmd) {
f.Close()
return nil, errors.New("write error")
}
return f, nil
}
}
func (e *Exporter) scrape() {
e.totalScrapes.Inc()
body, err := e.fetch()
if err != nil {
e.up.Set(0)
utilruntime.HandleError(fmt.Errorf("can't scrape HAProxy: %v", err))
return
}
defer body.Close()
e.up.Set(1)
reader := csv.NewReader(body)
reader.TrailingComma = true
reader.Comment = '#'
rows, servers := 0, 0
loop:
for {
row, err := reader.Read()
switch err {
case nil:
case io.EOF:
break loop
default:
if _, ok := err.(*csv.ParseError); ok {
utilruntime.HandleError(fmt.Errorf("can't read CSV: %v", err))
e.csvParseFailures.Inc()
continue loop
}
utilruntime.HandleError(fmt.Errorf("unexpected error while reading CSV: %v", err))
e.up.Set(0)
break loop
}
// consider ourselves broken, and refuse to parse anything else
if len(row) < expectedCsvFieldCount {
utilruntime.HandleError(fmt.Errorf("wrong CSV field count in metrics row %d: %d vs. %d", rows, len(row), expectedCsvFieldCount))
e.csvParseFailures.Inc()
return
}
// If we exceed the server threshold, ignore the rest of the servers because we will be
// displaying only backends and frontends.
if row[32] == serverType {
servers++
}
if servers > e.opts.ServerThreshold {
continue
}
rows++
e.parseRow(row)
}
e.serverLimited = servers > e.opts.ServerThreshold
e.serverThresholdCurrent.Set(float64(servers))
e.serverThresholdLimit.Set(float64(e.opts.ServerThreshold))
e.scrapeInterval = time.Duration(((float32(rows) / 1000) + 1) * float32(e.opts.BaseScrapeInterval))
e.nextScrapeInterval.Set(float64(e.scrapeInterval / time.Second))
}
func (e *Exporter) resetMetrics() {
for _, m := range e.frontendMetrics {
m.Reset()
}
for _, m := range e.backendMetrics {
m.Reset()
}
for _, m := range e.serverMetrics {
m.Reset()
}
}
func (e *Exporter) collectMetrics(metrics chan<- prometheus.Metric) {
for _, m := range e.frontendMetrics {
m.Collect(metrics)
}
for i, m := range e.backendMetrics {
if _, ok := e.reducedBackendExports[i]; !e.serverLimited && !ok {
continue
}
m.Collect(metrics)
}
if !e.serverLimited {
for _, m := range e.serverMetrics {
m.Collect(metrics)
}
}
}
func (e *Exporter) parseRow(csvRow []string) {
pxname, svname, typ := csvRow[0], csvRow[1], csvRow[32]
switch typ {
case frontendType:
e.exportCsvFields(e.frontendMetrics, csvRow, pxname)
case backendType:
if mode, value, ok := knownBackendSegment(pxname); ok {
if namespace, name, ok := parseNameSegment(value); ok {
e.exportCsvFields(e.backendMetrics, csvRow, mode, namespace, name)
return
}
}
e.exportCsvFields(e.backendMetrics, csvRow, "other/"+pxname, "", "")
case serverType:
pod, service, server, _ := knownServerSegment(svname)
if _, value, ok := knownBackendSegment(pxname); ok {
if namespace, name, ok := parseNameSegment(value); ok {
e.exportCsvFields(e.serverMetrics, csvRow, server, namespace, name, pod, service)
return
}
}
e.exportCsvFields(e.serverMetrics, csvRow, server, "", "", pod, service)
}
}
// knownServerSegment takes a server name that has a known prefix and returns
// the pod, service, and simpler service name label for that type. If the prefix does not
// match false is returned.
func knownServerSegment(value string) (string, string, string, bool) {
if i := strings.Index(value, ":"); i != -1 {
switch value[:i] {
case "ept":
if service, server, ok := parseNameSegment(value[i+1:]); ok {
return "", service, server, true
}
case "pod":
if pod, remainder, ok := parseNameSegment(value[i+1:]); ok {
if service, server, ok := parseNameSegment(remainder); ok {
return pod, service, server, true
}
}
}
}
return "", "", value, false
}
// knownBackendSegment takes a backend name that has a known prefix and returns
// the preferred metric label for that type and the remainder. If the prefix does not
// match false is returned.
func knownBackendSegment(value string) (string, string, bool) {
if i := strings.Index(value, ":"); i != -1 {
switch value[:i] {
case "be_http":
return "http", value[i+1:], true
case "be_edge_http":
return "https-edge", value[i+1:], true
case "be_secure":
return "https", value[i+1:], true
case "be_tcp":
return "tcp", value[i+1:], true
}
}
return "", "", false
}
// parseNameSegment splits a name by the first colon and returns both segments. If
// no colon is found it returns false.
func parseNameSegment(value string) (string, string, bool) {
i := strings.Index(value, ":")
if i == -1 {
return "", "", false
}
return value[:i], value[i+1:], true
}
func parseStatusField(value string) int64 {
switch value {
case "UP", "UP 1/3", "UP 2/3", "OPEN", "no check":
return 1
case "DOWN", "DOWN 1/2", "NOLB", "MAINT":
return 0
}
return 0
}
func (e *Exporter) exportCsvFields(metrics metrics, csvRow []string, labels ...string) {
for fieldIdx, metric := range metrics {
valueStr := csvRow[fieldIdx]
if valueStr == "" {
continue
}
var value int64
switch fieldIdx {
case statusField:
value = parseStatusField(valueStr)
default:
var err error
value, err = strconv.ParseInt(valueStr, 10, 64)
if err != nil {
utilruntime.HandleError(fmt.Errorf("can't parse CSV field value %s: %v", valueStr, err))
e.csvParseFailures.Inc()
continue
}
}
metric.WithLabelValues(labels...).Set(float64(value))
}
}
// filterMetrics returns the set of server metrics specified by the export array.
func filterMetrics(export []int, available metrics) metrics {
metrics := map[int]*prometheus.GaugeVec{}
if len(export) == 0 {
return metrics
}
selected := map[int]struct{}{}
for _, f := range export {
selected[f] = struct{}{}
}
for field, metric := range available {
if _, ok := selected[field]; ok {
metrics[field] = metric
}
}
return metrics
}
type PrometheusOptions struct {
// ScrapeURI is the URI to access HAProxy under. Defaults to the unix domain socket.
ScrapeURI string
// PidFile is optional and will collect process metrics.
PidFile string
// Timeout is the maximum interval to wait for stats.
Timeout time.Duration
// BaseScrapeInterval is the minimum time to wait between stat calls per 1000 servers.
BaseScrapeInterval time.Duration
// ServerThreshold is the maximum number of servers that can be reported before switching
// to only using backend metrics. This reduces metrics load when there is a very large set
// of endpoints.
ServerThreshold int
// ExportedMetrics is a list of HAProxy stats to export.
ExportedMetrics []int
}
// NewPrometheusCollector starts collectors for prometheus metrics from the
// provided HAProxy stats port. It will start goroutines. Use the default
// prometheus handler to access these metrics.
func NewPrometheusCollector(opts PrometheusOptions) (*Exporter, error) {
if len(opts.ScrapeURI) == 0 {
opts.ScrapeURI = "unix:///var/lib/haproxy/run/haproxy.sock"
}
if len(opts.PidFile) == 0 {
opts.PidFile = "/var/lib/haproxy/run/haproxy.pid"
}
if opts.Timeout == 0 {
opts.Timeout = 5 * time.Second
}
if opts.BaseScrapeInterval == 0 {
opts.BaseScrapeInterval = 5 * time.Second
}
if len(opts.ExportedMetrics) == 0 {
opts.ExportedMetrics = defaultSelectedMetrics
}
if opts.ServerThreshold == 0 {
opts.ServerThreshold = 500
}
exporter, err := NewExporter(opts)
if err != nil {
return nil, err
}
if err := prometheus.Register(exporter); err != nil {
return nil, err
}
// TODO: register a version collector?
if len(opts.PidFile) > 0 {
procExporter := prometheus.NewProcessCollectorPIDFn(
func() (int, error) {
content, err := ioutil.ReadFile(opts.PidFile)
if err != nil {
return 0, fmt.Errorf("can't read haproxy pid file: %s", err)
}
value, err := strconv.Atoi(strings.TrimSpace(string(content)))
if err != nil {
return 0, fmt.Errorf("can't parse haproxy pid file: %s", err)
}
return value, nil
}, namespace)
if err := prometheus.Register(procExporter); err != nil {
return nil, err
}
}
return exporter, nil
}