69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package counters
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
c "github.com/Azareal/Gosora/common"
|
|
qgen "github.com/Azareal/Gosora/query_gen"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
var OSViewCounter *DefaultOSViewCounter
|
|
|
|
type DefaultOSViewCounter struct {
|
|
buckets []*RWMutexCounterBucket //[OSID]count
|
|
insert *sql.Stmt
|
|
}
|
|
|
|
func NewDefaultOSViewCounter(acc *qgen.Accumulator) (*DefaultOSViewCounter, error) {
|
|
var osBuckets = make([]*RWMutexCounterBucket, len(osMapEnum))
|
|
for bucketID, _ := range osBuckets {
|
|
osBuckets[bucketID] = &RWMutexCounterBucket{counter: 0}
|
|
}
|
|
co := &DefaultOSViewCounter{
|
|
buckets: osBuckets,
|
|
insert: acc.Insert("viewchunks_systems").Columns("count, createdAt, system").Fields("?,UTC_TIMESTAMP(),?").Prepare(),
|
|
}
|
|
c.AddScheduledFifteenMinuteTask(co.Tick)
|
|
//c.AddScheduledSecondTask(co.Tick)
|
|
c.AddShutdownTask(co.Tick)
|
|
return co, acc.FirstError()
|
|
}
|
|
|
|
func (co *DefaultOSViewCounter) Tick() error {
|
|
for id, bucket := range co.buckets {
|
|
var count int
|
|
bucket.RLock()
|
|
count = bucket.counter
|
|
bucket.counter = 0 // TODO: Add a SetZero method to reduce the amount of duplicate code between the OS and agent counters?
|
|
bucket.RUnlock()
|
|
|
|
err := co.insertChunk(count, id) // TODO: Bulk insert for speed?
|
|
if err != nil {
|
|
return errors.Wrap(errors.WithStack(err), "system counter")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (co *DefaultOSViewCounter) insertChunk(count int, os int) error {
|
|
if count == 0 {
|
|
return nil
|
|
}
|
|
osName := reverseOSMapEnum[os]
|
|
c.DebugLogf("Inserting a vchunk with a count of %d for OS %s (%d)", count, osName, os)
|
|
_, err := co.insert.Exec(count, osName)
|
|
return err
|
|
}
|
|
|
|
func (co *DefaultOSViewCounter) Bump(id int) {
|
|
// TODO: Test this check
|
|
c.DebugDetail("co.buckets[", id, "]: ", co.buckets[id])
|
|
if len(co.buckets) <= id || id < 0 {
|
|
return
|
|
}
|
|
co.buckets[id].Lock()
|
|
co.buckets[id].counter++
|
|
co.buckets[id].Unlock()
|
|
}
|