cost-model/modules/collector-source/pkg/metric/aggregator/uptime.go

package aggregator

import (
	"sync"
	"time"
)

// uptimeAggregator is a MetricAggregator which records the first and last timestamp of updates called on it
type uptimeAggregator struct {
	lock        sync.Mutex
	labelValues []string
	start       *time.Time
	end         *time.Time
}

func Uptime(labelValues []string) MetricAggregator {
	return &uptimeAggregator{
		labelValues: labelValues,
	}
}

func (a *uptimeAggregator) AdditionInfo() map[string]string {
	return nil
}

func (a *uptimeAggregator) LabelValues() []string {
	return a.labelValues
}

func (a *uptimeAggregator) Update(value float64, timestamp time.Time, additionalInfo map[string]string) {
	a.lock.Lock()
	defer a.lock.Unlock()
	if a.start == nil {
		a.start = &timestamp
		return
	}

	if timestamp.After(*a.start) {
		a.end = &timestamp
	}
}

func (a *uptimeAggregator) Value() []MetricValue {
	a.lock.Lock()
	defer a.lock.Unlock()
	metricValues := make([]MetricValue, 0)
	if a.start != nil {
		metricValues = append(metricValues, MetricValue{Value: 1, Timestamp: a.start})
	}
	if a.end != nil {
		metricValues = append(metricValues, MetricValue{Value: 1, Timestamp: a.end})
	}
	return metricValues
}