app.go

package main

import (
	"sort"
	"time"
)

type appRate struct {
	Name string
	D, U float64
}

func appDeltas(prev, cur map[string][2]int64, interval float64) (map[string][2]int64, []appRate) {
	ab := map[string][2]int64{}
	var rates []appRate
	for name, cv := range cur {
		pv, ok := prev[name]
		if !ok {
			pv = cv
		}
		db := cv[0] - pv[0]
		if db < 0 {
			db = 0
		}
		ub := cv[1] - pv[1]
		if ub < 0 {
			ub = 0
		}
		if db+ub > 0 {
			ab[name] = [2]int64{db, ub}
			rates = append(rates, appRate{name, float64(db) / interval, float64(ub) / interval})
		}
	}
	sort.Slice(rates, func(i, j int) bool { return rates[i].D+rates[i].U > rates[j].D+rates[j].U })
	return ab, rates
}

func runLogger(iface string, interval float64) {
	iface = pickIface(iface)
	pin, pout, ok := ifaceBytes(iface)
	if !ok {
		return
	}
	pproc := procBytes()
	log := NewUsageLog()
	cfg := LoadConfig()
	det := NewDetector(cfg.SpikeFloorKBps, cfg.SpikeMult)
	for {
		time.Sleep(time.Duration(interval * float64(time.Second)))
		cin, cout, ok := ifaceBytes(iface)
		if !ok {
			continue
		}
		din := cin - pin
		if din < 0 {
			din = 0
		}
		dout := cout - pout
		if dout < 0 {
			dout = 0
		}
		pin, pout = cin, cout
		log.Add(din, dout)
		cproc := procBytes()
		ab, rates := appDeltas(pproc, cproc, interval)
		pproc = cproc
		log.AddApps(ab)
		now := time.Now().Unix()
		for _, r := range rates {
			if det.Update(r.Name, r.D+r.U, now) {
				log.LogSpike(r.Name, r.D, r.U)
			}
		}
	}
}