mirrors/posthog
0
0
Fork 0
mirror of https://github.com/PostHog/posthog.git synced 2024-11-21 13:39:22 +01:00
Code Issues Releases Activity

fix: Fix TTL to be time.Second based (#24383)

Browse Source
This commit is contained in:
James Greenhill 2024-08-14 15:58:36 -07:00 committed by GitHub
parent 904c75050e
commit 190ecf82c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 28 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
livestream/live_stats.go
View File

@ -8,7 +8,7 @@ import (
)
const (
COUNTER_TTL = 60
COUNTER_TTL = time.Second * 60
)
type Stats struct {
@ -20,7 +20,7 @@ type Stats struct {
func newStatsKeeper() *Stats {
return &Stats{
Store: make(map[string]*expirable.LRU[string, string]),
GlobalStore: expirable.NewLRU[string, string](0, nil, time.Second*COUNTER_TTL),
GlobalStore: expirable.NewLRU[string, string](0, nil, COUNTER_TTL),
Counter: NewSlidingWindowCounter(COUNTER_TTL),
}
}
@ -28,16 +28,13 @@ func newStatsKeeper() *Stats {
func (ts *Stats) keepStats(statsChan chan PostHogEvent) {
log.Println("starting stats keeper...")
for { // ignore the range warning here - it's wrong
select {
case event := <-statsChan:
ts.Counter.Increment()
token := event.Token
if _, ok := ts.Store[token]; !ok {
ts.Store[token] = expirable.NewLRU[string, string](0, nil, time.Second*COUNTER_TTL)
}
ts.Store[token].Add(event.DistinctId, "1")
ts.GlobalStore.Add(event.DistinctId, "1")
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")
}
}

8
livestream/served.go
View File

@ -10,8 +10,8 @@ import (
)
type Counter struct {
EventCount uint32
UserCount uint32
EventCount int
UserCount int
}
func servedHandler(stats *Stats) func(c echo.Context) error {
@ -19,8 +19,8 @@ func servedHandler(stats *Stats) func(c echo.Context) error {
userCount := stats.GlobalStore.Len()
count := stats.Counter.Count()
resp := Counter{
EventCount: uint32(count),
UserCount: uint32(userCount),
EventCount: count,
UserCount: userCount,
}
return c.JSON(http.StatusOK, resp)
}

17
livestream/ttl_counter.go
View File

@ -12,10 +12,24 @@ type SlidingWindowCounter struct {
}
func NewSlidingWindowCounter(windowSize time.Duration) *SlidingWindowCounter {
return &SlidingWindowCounter{
swc := &SlidingWindowCounter{
events: make([]time.Time, 0),
windowSize: windowSize,
}
// Start a goroutine to periodically remove old events
go func() {
ticker := time.NewTicker(time.Second)
defer ticker.Stop()
for range ticker.C {
swc.mu.Lock()
swc.removeOldEvents(time.Now())
swc.mu.Unlock()
}
}()
return swc
}
func (swc *SlidingWindowCounter) Increment() {
@ -24,7 +38,6 @@ func (swc *SlidingWindowCounter) Increment() {
now := time.Now()
swc.events = append(swc.events, now)
swc.removeOldEvents(now)
}
func (swc *SlidingWindowCounter) Count() int {