package counters

import "database/sql"
import ".."
import "../../query_gen/lib"

var OSViewCounter *DefaultOSViewCounter

type DefaultOSViewCounter struct {
	osBuckets []*RWMutexCounterBucket //[OSID]count
	insert    *sql.Stmt
}

func NewDefaultOSViewCounter() (*DefaultOSViewCounter, error) {
	acc := qgen.Builder.Accumulator()
	var osBuckets = make([]*RWMutexCounterBucket, len(osMapEnum))
	for bucketID, _ := range osBuckets {
		osBuckets[bucketID] = &RWMutexCounterBucket{counter: 0}
	}
	counter := &DefaultOSViewCounter{
		osBuckets: osBuckets,
		insert:    acc.Insert("viewchunks_systems").Columns("count, createdAt, system").Fields("?,UTC_TIMESTAMP(),?").Prepare(),
	}
	common.AddScheduledFifteenMinuteTask(counter.Tick)
	//common.AddScheduledSecondTask(counter.Tick)
	common.AddShutdownTask(counter.Tick)
	return counter, acc.FirstError()
}

func (counter *DefaultOSViewCounter) Tick() error {
	for osID, osBucket := range counter.osBuckets {
		var count int
		osBucket.RLock()
		count = osBucket.counter
		osBucket.counter = 0 // TODO: Add a SetZero method to reduce the amount of duplicate code between the OS and agent counters?
		osBucket.RUnlock()

		err := counter.insertChunk(count, osID) // TODO: Bulk insert for speed?
		if err != nil {
			return err
		}
	}
	return nil
}

func (counter *DefaultOSViewCounter) insertChunk(count int, os int) error {
	if count == 0 {
		return nil
	}
	var osName = reverseOSMapEnum[os]
	common.DebugLogf("Inserting a viewchunk with a count of %d for OS %s (%d)", count, osName, os)
	_, err := counter.insert.Exec(count, osName)
	return err
}

func (counter *DefaultOSViewCounter) Bump(os int) {
	// TODO: Test this check
	common.DebugDetail("counter.osBuckets[", os, "]: ", counter.osBuckets[os])
	if len(counter.osBuckets) <= os || os < 0 {
		return
	}
	counter.osBuckets[os].Lock()
	counter.osBuckets[os].counter++
	counter.osBuckets[os].Unlock()
}