rogueserver/api/stats.go

48 lines
756 B
Go
Raw Normal View History

2024-03-23 21:34:18 -04:00
package api
import (
"log"
2024-04-17 19:03:25 -04:00
"time"
2024-03-23 21:34:18 -04:00
2024-04-14 19:25:36 -04:00
"github.com/pagefaultgames/pokerogue-server/db"
2024-04-16 21:01:24 -04:00
"github.com/robfig/cron/v3"
2024-03-23 21:34:18 -04:00
)
var (
2024-04-17 19:03:25 -04:00
scheduler = cron.New(cron.WithLocation(time.UTC))
2024-04-06 18:15:47 -04:00
playerCount int
battleCount int
classicSessionCount int
)
2024-03-23 21:34:18 -04:00
func scheduleStatRefresh() {
2024-04-16 21:01:24 -04:00
scheduler.AddFunc("@every 10s", func() {
err := updateStats()
if err != nil {
log.Printf("failed to update stats: %s", err)
}
})
scheduler.Start()
}
2024-04-16 21:01:24 -04:00
func updateStats() error {
var err error
playerCount, err = db.FetchPlayerCount()
2024-03-23 21:34:18 -04:00
if err != nil {
2024-04-16 21:01:24 -04:00
return err
2024-03-23 21:34:18 -04:00
}
2024-04-08 20:44:36 -04:00
2024-04-06 18:15:47 -04:00
battleCount, err = db.FetchBattleCount()
if err != nil {
2024-04-16 21:01:24 -04:00
return err
2024-04-06 18:15:47 -04:00
}
2024-04-08 20:44:36 -04:00
2024-04-06 18:15:47 -04:00
classicSessionCount, err = db.FetchClassicSessionCount()
if err != nil {
2024-04-16 21:01:24 -04:00
return err
2024-04-06 18:15:47 -04:00
}
2024-04-16 21:01:24 -04:00
return nil
}