46 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			46 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// +build linux
 | 
						|
 | 
						|
package cgroups
 | 
						|
 | 
						|
import (
 | 
						|
	"github.com/containerd/cgroups"
 | 
						|
	metrics "github.com/docker/go-metrics"
 | 
						|
	"github.com/prometheus/client_golang/prometheus"
 | 
						|
)
 | 
						|
 | 
						|
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 *cgroups.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{"container_id", "namespace"}, m.labels...)...)
 | 
						|
}
 | 
						|
 | 
						|
func (m *metric) collect(id, namespace string, stats *cgroups.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:
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 |