eventlogger.go

package bot

import (
	"strconv"
	"sync"
	"sync/atomic"
	"time"

	"github.com/botlabs-gg/yagpdb/v2/bot/eventsystem"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promauto"
)

const EventLoggerPeriodDuration = time.Second * 10

var (
	EventLogger = &eventLogger{}
)

type eventLogger struct {
	sync.Mutex

	// these slices are 2d of shard id and event type

	// total stats is the cumulative number of events produced
	totalStats [][]*int64

	// lastPeriod is a snapshot of totalstats at the last logging period
	lastPeriod [][]int64

	// perPeriod is the number of events proccessed the last period alone (the rate of events)
	perPeriod [][]int64
}

func (e *eventLogger) init(numShards int) {
	e.totalStats = make([][]*int64, numShards)
	e.lastPeriod = make([][]int64, numShards)
	e.perPeriod = make([][]int64, numShards)

	// Initialize these
	for i := range e.totalStats {
		e.totalStats[i] = make([]*int64, len(eventsystem.AllEvents))
		e.lastPeriod[i] = make([]int64, len(eventsystem.AllEvents))
		e.perPeriod[i] = make([]int64, len(eventsystem.AllEvents))

		for j := range e.totalStats[i] {
			e.totalStats[i][j] = new(int64)
		}
	}

}

func (e *eventLogger) run() {

	ticker := time.NewTicker(EventLoggerPeriodDuration)
	for {
		<-ticker.C
		e.flushStats()
	}
}

func (e *eventLogger) GetStats() (total [][]int64, perPeriod [][]int64) {
	e.Lock()

	total = make([][]int64, len(e.totalStats))
	perPeriod = make([][]int64, len(e.totalStats))

	for i := 0; i < len(e.totalStats); i++ {

		perPeriod[i] = make([]int64, len(e.totalStats[i]))
		total[i] = make([]int64, len(e.totalStats[i]))

		for j := 0; j < len(e.totalStats[i]); j++ {
			perPeriod[i][j] = e.perPeriod[i][j]
			total[i][j] = atomic.LoadInt64(e.totalStats[i][j])
		}
	}

	e.Unlock()

	return
}

var metricsHandledEventsHandledShards = promauto.NewCounterVec(prometheus.CounterOpts{
	Name: "yagpdb_discord_events_shards_total",
	Help: "The total number of processed events, with a shard label",
}, []string{"shard"})

var metricsHandledEventsHandledTypes = promauto.NewCounterVec(prometheus.CounterOpts{
	Name: "yagpdb_discord_events_types_total",
	Help: "The total number of processed events, with a type label",
}, []string{"type"})

func (e *eventLogger) flushStats() {
	shardTotals := make([]int64, len(e.totalStats))
	typeTotals := make([]int64, len(eventsystem.AllEvents))

	e.Lock()
	for i := 0; i < len(e.totalStats); i++ {
		for j := 0; j < len(e.totalStats[i]); j++ {
			currentVal := atomic.LoadInt64(e.totalStats[i][j])
			e.perPeriod[i][j] = currentVal - e.lastPeriod[i][j]
			e.lastPeriod[i][j] = currentVal

			shardTotals[i] += e.perPeriod[i][j]
			typeTotals[j] += e.perPeriod[i][j]
			// totalPerPeriod += e.perPeriod[i][j]
		}
	}
	e.Unlock()

	pShards := ReadyTracker.GetProcessShards()

	for _, shard := range pShards {
		metricsHandledEventsHandledShards.With(prometheus.Labels{"shard": strconv.Itoa(shard)}).Add(float64(shardTotals[shard]))
		// common.Statsd.Count("discord.processed.events", shardTotals[shard], []string{"shard:" + strconv.Itoa(shard)}, EventLoggerPeriodDuration.Seconds())
	}

	for typ, count := range typeTotals {
		if count < 1 {
			continue
		}

		metricsHandledEventsHandledTypes.With(prometheus.Labels{"type": eventsystem.Event(typ).String()}).Add(float64(count))
	}

}

func (e *eventLogger) handleEvent(evt *eventsystem.EventData) {
	if evt.Session == nil {
		return
	}

	if int(evt.Type) >= len(eventsystem.AllEvents) {
		return
	}

	atomic.AddInt64(e.totalStats[evt.Session.ShardID][evt.Type], 1)
}