d654dbafac
This makes the metrics package more extensible by allowing the default name of `container_id` to be changed by the package caller. Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
// +build linux
|
|
|
|
/*
|
|
Copyright The containerd Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package v1
|
|
|
|
import (
|
|
v1 "github.com/containerd/containerd/metrics/types/v1"
|
|
metrics "github.com/docker/go-metrics"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
// IDName is the name that is used to identify the id being collected in the metric
|
|
var IDName = "container_id"
|
|
|
|
type value struct {
|
|
v float64
|
|
l []string
|
|
}
|
|
|
|
type metric struct {
|
|
name string
|
|
help string
|
|
unit metrics.Unit
|
|
vt prometheus.ValueType
|
|
labels []string
|
|
// getValues returns the value and labels for the data
|
|
getValues func(stats *v1.Metrics) []value
|
|
}
|
|
|
|
func (m *metric) desc(ns *metrics.Namespace) *prometheus.Desc {
|
|
// the namespace label is for containerd namespaces
|
|
return ns.NewDesc(m.name, m.help, m.unit, append([]string{IDName, "namespace"}, m.labels...)...)
|
|
}
|
|
|
|
func (m *metric) collect(id, namespace string, stats *v1.Metrics, ns *metrics.Namespace, ch chan<- prometheus.Metric, block bool) {
|
|
values := m.getValues(stats)
|
|
for _, v := range values {
|
|
// block signals to block on the sending the metrics so none are missed
|
|
if block {
|
|
ch <- prometheus.MustNewConstMetric(m.desc(ns), m.vt, v.v, append([]string{id, namespace}, v.l...)...)
|
|
continue
|
|
}
|
|
// non-blocking metrics can be dropped if the chan is full
|
|
select {
|
|
case ch <- prometheus.MustNewConstMetric(m.desc(ns), m.vt, v.v, append([]string{id, namespace}, v.l...)...):
|
|
default:
|
|
}
|
|
}
|
|
}
|