counters.go

package flows

import "sync"

type Var interface{}

type flagsInfo struct {
	i    int
	mask uint8
}

type Int struct {
	i int
	f flagsInfo
}

type Uint struct {
	i int
	f flagsInfo
}

type Float struct {
	i int
	f flagsInfo
}

type counterReg struct {
	mutex sync.Mutex

	ints   counterTypeReg
	uints  counterTypeReg
	floats counterTypeReg
}

type counterTypeReg struct {
	names []string
}

type flowStats struct {
	intFlags   []uint8
	uintFlags  []uint8
	floatFlags []uint8

	ints   []int64
	uints  []uint64
	floats []float64
}

func (c *Int) Add(f *Flow, delta int64) {
	ints := f.stats.ints
	if c.i < len(ints) {
		ints[c.i] += delta
		c.f.apply(f.stats.intFlags)
	}
}

func (c *Int) Set(f *Flow, value int64) {
	ints := f.stats.ints
	if c.i < len(ints) {
		ints[c.i] = value
		c.f.apply(f.stats.intFlags)
	}
}

func (c *Uint) Add(f *Flow, delta uint64) {
	uints := f.stats.uints
	if c.i < len(uints) {
		uints[c.i] += delta
		c.f.apply(f.stats.uintFlags)
	}
}

func (c *Uint) Set(f *Flow, value uint64) {
	uints := f.stats.uints
	if c.i < len(uints) {
		uints[c.i] = value
		c.f.apply(f.stats.uintFlags)
	}
}

func (c *Float) Add(f *Flow, delta float64) {
	floats := f.stats.floats
	if c.i < len(floats) {
		floats[c.i] += delta
		c.f.apply(f.stats.floatFlags)
	}
}

func (c *Float) Set(f *Flow, value float64) {
	floats := f.stats.floats
	if c.i < len(floats) {
		floats[c.i] = value
		c.f.apply(f.stats.floatFlags)
	}
}

func (c *counterReg) newInt(name string) (*Int, error) {
	c.mutex.Lock()
	defer c.mutex.Unlock()

	i, err := c.ints.reg(name)
	if err != nil {
		return nil, err
	}
	return &Int{i, makeFlagsInfo(i)}, nil
}

func (c *counterReg) newUint(name string) (*Uint, error) {
	c.mutex.Lock()
	defer c.mutex.Unlock()

	i, err := c.uints.reg(name)
	if err != nil {
		return nil, err
	}
	return &Uint{i, makeFlagsInfo(i)}, nil
}

func (c *counterReg) newFloat(name string) (*Float, error) {
	c.mutex.Lock()
	defer c.mutex.Unlock()

	i, err := c.floats.reg(name)
	if err != nil {
		return nil, err
	}
	return &Float{i, makeFlagsInfo(i)}, nil
}

// XXX:
//  - error on index > int max
//  - error if already in use
func (reg *counterTypeReg) reg(name string) (int, error) {
	debugf("register flow counter: %v", name)

	i := len(reg.names)
	reg.names = append(reg.names, name)
	return i, nil
}

func (reg *counterTypeReg) getNames() []string {
	return reg.names
}

func newFlowStats(reg *counterReg) *flowStats {
	s := &flowStats{}
	s.init(reg)
	return s
}

func (s *flowStats) init(reg *counterReg) {
	reg.mutex.Lock()
	defer reg.mutex.Unlock()

	nInts := len(reg.ints.names)
	nUints := len(reg.uints.names)
	nFloats := len(reg.floats.names)

	s.ints = make([]int64, nInts)
	s.uints = make([]uint64, nUints)
	s.floats = make([]float64, nFloats)

	s.intFlags = make([]uint8, (nInts+7)/8)
	s.uintFlags = make([]uint8, (nUints+7)/8)
	s.floatFlags = make([]uint8, (nFloats+7)/8)
}

func makeFlagsInfo(i int) flagsInfo {
	return flagsInfo{
		i:    i / 8,
		mask: 1 << uint(i%8),
	}
}

func (f *flagsInfo) apply(flags []uint8) {
	flags[f.i] |= f.mask
}