package selfstat

import (
	"sync"
)

type timingStat struct {
	measurement string
	field       string
	tags        map[string]string
	v           int64
	prev        int64
	count       int64
	mu          sync.Mutex
}

func (s *timingStat) Incr(v int64) {
	s.mu.Lock()
	s.v += v
	s.count++
	s.mu.Unlock()
}

func (s *timingStat) Set(v int64) {
	s.Incr(v)
}

func (s *timingStat) Get() int64 {
	var avg int64
	s.mu.Lock()
	if s.count > 0 {
		s.prev, avg = s.v/s.count, s.v/s.count
		s.v = 0
		s.count = 0
	} else {
		avg = s.prev
	}
	s.mu.Unlock()
	return avg
}

func (s *timingStat) Name() string {
	return s.measurement
}

func (s *timingStat) FieldName() string {
	return s.field
}

// Tags returns a copy of the timingStat's tags.
// NOTE this allocates a new map every time it is called.
func (s *timingStat) Tags() map[string]string {
	m := make(map[string]string, len(s.tags))
	for k, v := range s.tags {
		m[k] = v
	}
	return m
}

// Unregister removes this stat from the registry only
func (s *timingStat) Unregister() {
	registry.remove(s.measurement, s.field, s.tags)
}
