go-metrics/gauge.go

package metrics

import "sync/atomic"

// Gauges hold an int64 value that can be set arbitrarily.
//
// This is an interface so as to encourage other structs to implement
// the Gauge API as appropriate.
type Gauge interface {
	Update(int64)
	Value() int64
}

// Create a new Gauge.
func NewGauge() Gauge {
	if !ObserverEffect {
		return NilGauge{}
	}
	return &StandardGauge{0}
}

// No-op Gauge.
type NilGauge struct{}

// Force the compiler to check that NilGauge implements Gauge.
var _ Gauge = NilGauge{}

// No-op.
func (g NilGauge) Update(v int64) {}

// No-op.
func (g NilGauge) Value() int64 { return 0 }

// The standard implementation of a Gauge uses the sync/atomic package
// to manage a single int64 value.
type StandardGauge struct {
	value int64
}

// Force the compiler to check that StandardGauge implements Gauge.
var _ Gauge = &StandardGauge{}

// Update the gauge's value.
func (g *StandardGauge) Update(v int64) {
	atomic.StoreInt64(&g.value, v)
}

// Return the gauge's current value.
func (g *StandardGauge) Value() int64 {
	return atomic.LoadInt64(&g.value)
}