s3rw_exporter/export.go at master · orange-cloudfoundry/s3rw_exporter · GitHub

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
package main

import (
	// "fmt"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promauto"
	"time"
)

var (
	uploadDuration   prometheus.Gauge
	uploadStatus     prometheus.Gauge
	uploadError      *prometheus.GaugeVec
	downloadDuration prometheus.Gauge
	downloadStatus   prometheus.Gauge
	downloadError    *prometheus.GaugeVec
)

func loadMetricsReporter(namespace string) {
	downloadDuration = promauto.NewGauge(
		prometheus.GaugeOpts{
			Namespace: namespace,
			Name:      "download_duration_seconds",
			Help:      "Last download duration in seconds",
		})
	uploadDuration = promauto.NewGauge(
		prometheus.GaugeOpts{
			Namespace: namespace,
			Name:      "upload_duration_seconds",
			Help:      "Last upload duration in seconds",
		})

	downloadStatus = promauto.NewGauge(
		prometheus.GaugeOpts{
			Namespace: namespace,
			Name:      "download_status",
			Help:      "Last download status, 1 is ok",
		})
	uploadStatus = promauto.NewGauge(
		prometheus.GaugeOpts{
			Namespace: namespace,
			Name:      "upload_status",
			Help:      "Last upload status, 1 is ok",
		})

	downloadError = promauto.NewGaugeVec(
		prometheus.GaugeOpts{
			Namespace: namespace,
			Name:      "download_errors",
			Help:      "Active download errors",
		}, []string{"error"},
	)
	uploadError = promauto.NewGaugeVec(
		prometheus.GaugeOpts{
			Namespace: namespace,
			Name:      "upload_errors",
			Help:      "Active upload errors",
		}, []string{"error"},
	)
}

// RecordMetrics -
func RecordMetrics(manager *Manager) {
	go func() {
		for {
			downloadError.Reset()
			start := time.Now()
			if err := manager.Download(); err != nil {
				downloadError.With(prometheus.Labels{
					"error": err.Error(),
				}).Set(1)
				downloadStatus.Set(0)
			} else {
				downloadStatus.Set(1)
				downloadDuration.Set(time.Since(start).Seconds())
			}

			uploadError.Reset()
			start = time.Now()
			if err := manager.Upload(); err != nil {
				uploadError.With(prometheus.Labels{
					"error": err.Error(),
				}).Set(1)
				uploadStatus.Set(0)
			} else {
				uploadStatus.Set(1)
				uploadDuration.Set(time.Since(start).Seconds())
			}
			time.Sleep(manager.config.Exporter.IntervalDuration)
		}
	}()
}

// Local Variables:
// ispell-local-dictionary: "american"
// End: