godoxy/internal/metrics/period/period.go

package period

import (
	"sync"
	"time"
)

type Period[T any] struct {
	Entries map[Filter]*Entries[T] `json:"entries"`
	mu      sync.RWMutex
}

type Filter string // @name MetricsPeriod

const (
	MetricsPeriod5m  Filter = "5m"  // @name MetricsPeriod5m
	MetricsPeriod15m Filter = "15m" // @name MetricsPeriod15m
	MetricsPeriod1h  Filter = "1h"  // @name MetricsPeriod1h
	MetricsPeriod1d  Filter = "1d"  // @name MetricsPeriod1d
	MetricsPeriod1mo Filter = "1mo" // @name MetricsPeriod1mo

	Duration5m  = 5 * time.Minute
	Duration15m = 15 * time.Minute
	Duration1h  = 1 * time.Hour
	Duration1d  = 24 * time.Hour
	Duration30d = 30 * 24 * time.Hour
)

func NewPeriod[T any]() *Period[T] {
	return &Period[T]{
		Entries: map[Filter]*Entries[T]{
			MetricsPeriod5m:  newEntries[T](Duration5m),
			MetricsPeriod15m: newEntries[T](Duration15m),
			MetricsPeriod1h:  newEntries[T](Duration1h),
			MetricsPeriod1d:  newEntries[T](Duration1d),
			MetricsPeriod1mo: newEntries[T](Duration30d),
		},
	}
}

func (p *Period[T]) Add(info T) {
	p.mu.Lock()
	defer p.mu.Unlock()
	now := time.Now()
	for _, period := range p.Entries {
		period.Add(now, info)
	}
}

func (p *Period[T]) Get(filter Filter) ([]T, bool) {
	period, ok := p.Entries[filter]
	if !ok {
		return nil, false
	}
	p.mu.RLock()
	defer p.mu.RUnlock()
	return period.Get(), true
}

func (p *Period[T]) Total() int {
	p.mu.RLock()
	defer p.mu.RUnlock()
	total := 0
	for _, period := range p.Entries {
		total += period.count
	}
	return total
}

// ValidateAndFixIntervals checks all period intervals and fixes them if they're incorrect.
// This should be called after loading data from JSON to ensure data integrity.
func (p *Period[T]) ValidateAndFixIntervals() {
	p.mu.Lock()
	defer p.mu.Unlock()

	durations := map[Filter]time.Duration{
		MetricsPeriod5m:  Duration5m,
		MetricsPeriod15m: Duration15m,
		MetricsPeriod1h:  Duration1h,
		MetricsPeriod1d:  Duration1d,
		MetricsPeriod1mo: Duration30d,
	}

	for filter, entries := range p.Entries {
		if expectedDuration, exists := durations[filter]; exists {
			if !entries.validateInterval(expectedDuration) {
				entries.fixInterval(expectedDuration)
			}
		}
	}
}