109 lines
2.7 KiB
Go
109 lines
2.7 KiB
Go
|
// package prometheuscollector allows to expose metrics for Prometheus.
|
||
|
|
//
|
||
|
|
// Using the provided collector, you can easily expose metrics for tusd in the
|
||
|
|
// Prometheus exposition format (https://prometheus.io/docs/instrumenting/exposition_formats/):
|
||
|
|
//
|
||
|
|
// handler, err := tusd.NewHandler(…)
|
||
|
|
// collector := prometheuscollector.New(handler.Metrics)
|
||
|
|
// prometheus.MustRegister(collector)
|
||
|
|
package prometheuscollector
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync/atomic"
|
||
|
|
|
||
|
|
"github.com/tus/tusd"
|
||
|
|
|
||
|
|
"github.com/prometheus/client_golang/prometheus"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
requestsTotalDesc = prometheus.NewDesc(
|
||
|
|
"tusd_requests_total",
|
||
|
|
"Total number of requests served by tusd per method.",
|
||
|
|
[]string{"method"}, nil)
|
||
|
|
errorsTotalDesc = prometheus.NewDesc(
|
||
|
|
"tusd_errors_total",
|
||
|
|
"Total number of erorrs per cause.",
|
||
|
|
[]string{"cause"}, nil)
|
||
|
|
bytesReceivedDesc = prometheus.NewDesc(
|
||
|
|
"tusd_bytes_received",
|
||
|
|
"Number of bytes received for uploads.",
|
||
|
|
nil, nil)
|
||
|
|
uploadsCreatedDesc = prometheus.NewDesc(
|
||
|
|
"tusd_uploads_created",
|
||
|
|
"Number of created uploads.",
|
||
|
|
nil, nil)
|
||
|
|
uploadsFinishedDesc = prometheus.NewDesc(
|
||
|
|
"tusd_uploads_finished",
|
||
|
|
"Number of finished uploads.",
|
||
|
|
nil, nil)
|
||
|
|
uploadsTerminatedDesc = prometheus.NewDesc(
|
||
|
|
"tusd_uploads_terminated",
|
||
|
|
"Number of terminted uploads.",
|
||
|
|
nil, nil)
|
||
|
|
)
|
||
|
|
|
||
|
|
type Collector struct {
|
||
|
|
metrics tusd.Metrics
|
||
|
|
}
|
||
|
|
|
||
|
|
// New creates a new collector which read froms the provided Metrics struct.
|
||
|
|
func New(metrics tusd.Metrics) Collector {
|
||
|
|
return Collector{
|
||
|
|
metrics: metrics,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (_ Collector) Describe(descs chan<- *prometheus.Desc) {
|
||
|
|
descs <- requestsTotalDesc
|
||
|
|
descs <- errorsTotalDesc
|
||
|
|
descs <- bytesReceivedDesc
|
||
|
|
descs <- uploadsCreatedDesc
|
||
|
|
descs <- uploadsFinishedDesc
|
||
|
|
descs <- uploadsTerminatedDesc
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c Collector) Collect(metrics chan<- prometheus.Metric) {
|
||
|
|
for method, valuePtr := range c.metrics.RequestsTotal {
|
||
|
|
metrics <- prometheus.MustNewConstMetric(
|
||
|
|
requestsTotalDesc,
|
||
|
|
prometheus.GaugeValue,
|
||
|
|
float64(atomic.LoadUint64(valuePtr)),
|
||
|
|
method,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
for error, valuePtr := range c.metrics.ErrorsTotal {
|
||
|
|
metrics <- prometheus.MustNewConstMetric(
|
||
|
|
errorsTotalDesc,
|
||
|
|
prometheus.GaugeValue,
|
||
|
|
float64(atomic.LoadUint64(valuePtr)),
|
||
|
|
error,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
metrics <- prometheus.MustNewConstMetric(
|
||
|
|
bytesReceivedDesc,
|
||
|
|
prometheus.CounterValue,
|
||
|
|
float64(atomic.LoadUint64(c.metrics.BytesReceived)),
|
||
|
|
)
|
||
|
|
|
||
|
|
metrics <- prometheus.MustNewConstMetric(
|
||
|
|
uploadsFinishedDesc,
|
||
|
|
prometheus.CounterValue,
|
||
|
|
float64(atomic.LoadUint64(c.metrics.UploadsFinished)),
|
||
|
|
)
|
||
|
|
|
||
|
|
metrics <- prometheus.MustNewConstMetric(
|
||
|
|
uploadsCreatedDesc,
|
||
|
|
prometheus.CounterValue,
|
||
|
|
float64(atomic.LoadUint64(c.metrics.UploadsCreated)),
|
||
|
|
)
|
||
|
|
|
||
|
|
metrics <- prometheus.MustNewConstMetric(
|
||
|
|
uploadsTerminatedDesc,
|
||
|
|
prometheus.CounterValue,
|
||
|
|
float64(atomic.LoadUint64(c.metrics.UploadsTerminated)),
|
||
|
|
)
|
||
|
|
}
|