mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-24 00:47:50 +01:00
41 lines
920 B
Go
41 lines
920 B
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/hashicorp/golang-lru/v2/expirable"
|
|
)
|
|
|
|
const (
|
|
COUNTER_TTL = time.Second * 60
|
|
)
|
|
|
|
type Stats struct {
|
|
Store map[string]*expirable.LRU[string, string]
|
|
GlobalStore *expirable.LRU[string, string]
|
|
Counter *SlidingWindowCounter
|
|
}
|
|
|
|
func newStatsKeeper() *Stats {
|
|
return &Stats{
|
|
Store: make(map[string]*expirable.LRU[string, string]),
|
|
GlobalStore: expirable.NewLRU[string, string](0, nil, COUNTER_TTL),
|
|
Counter: NewSlidingWindowCounter(COUNTER_TTL),
|
|
}
|
|
}
|
|
|
|
func (ts *Stats) keepStats(statsChan chan PostHogEvent) {
|
|
log.Println("starting stats keeper...")
|
|
|
|
for event := range statsChan {
|
|
ts.Counter.Increment()
|
|
token := event.Token
|
|
if _, ok := ts.Store[token]; !ok {
|
|
ts.Store[token] = expirable.NewLRU[string, string](0, nil, COUNTER_TTL)
|
|
}
|
|
ts.Store[token].Add(event.DistinctId, "1")
|
|
ts.GlobalStore.Add(event.DistinctId, "1")
|
|
}
|
|
}
|