mirror of
https://github.com/yusing/godoxy.git
synced 2026-04-23 08:48:32 +02:00
d56663d3f9
- Introduced addWithTime method for adding entries with specific timestamps. - Added validateInterval and fixInterval methods to ensure correct interval settings. - Updated JSON unmarshalling to respect entry timestamps and validate intervals post-load. - Refactored poller to use a constant PollInterval for consistency across the codebase.
86 lines
2.0 KiB
Go
86 lines
2.0 KiB
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
|
|
)
|
|
|
|
func NewPeriod[T any]() *Period[T] {
|
|
return &Period[T]{
|
|
Entries: map[Filter]*Entries[T]{
|
|
MetricsPeriod5m: newEntries[T](5 * time.Minute),
|
|
MetricsPeriod15m: newEntries[T](15 * time.Minute),
|
|
MetricsPeriod1h: newEntries[T](1 * time.Hour),
|
|
MetricsPeriod1d: newEntries[T](24 * time.Hour),
|
|
MetricsPeriod1mo: newEntries[T](30 * 24 * time.Hour),
|
|
},
|
|
}
|
|
}
|
|
|
|
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: 5 * time.Minute,
|
|
MetricsPeriod15m: 15 * time.Minute,
|
|
MetricsPeriod1h: 1 * time.Hour,
|
|
MetricsPeriod1d: 24 * time.Hour,
|
|
MetricsPeriod1mo: 30 * 24 * time.Hour,
|
|
}
|
|
|
|
for filter, entries := range p.Entries {
|
|
if expectedDuration, exists := durations[filter]; exists {
|
|
if !entries.validateInterval(expectedDuration) {
|
|
entries.fixInterval(expectedDuration)
|
|
}
|
|
}
|
|
}
|
|
}
|