crunchy-rocks/clessy
Archived
2
0
Fork 0
Code Issues 1 Pull requests Releases Wiki Activity

stats: Added ByDay counters

Browse source
This commit is contained in:
Hamcha 2016-02-13 21:58:05 +01:00
parent 2083ce5e69
commit e04e45c969
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
stats/stats.go
View file

@ -28,6 +28,7 @@ type Stats struct {
ByWeekday [7]uint64 ByWeekday [7]uint64
ByHour [24]uint64 ByHour [24]uint64
ByType [MessageTypeMax]uint64 ByType [MessageTypeMax]uint64
ByDay map[string]uint64
TodayDate time.Time TodayDate time.Time
Today uint64 Today uint64
TotalCount uint64 TotalCount uint64
@ -97,12 +98,17 @@ func loadStats() {
stats.ByWeekday[i] = MakeUint(b.Get([]byte{byte(i)}), "weekday", strconv.Itoa(i)) stats.ByWeekday[i] = MakeUint(b.Get([]byte{byte(i)}), "weekday", strconv.Itoa(i))
} }
// Load today's message counter, if possible // Load day counters
b, err = tx.CreateBucketIfNotExists([]byte("date")) b, err = tx.CreateBucketIfNotExists([]byte("date"))
if err != nil { if err != nil {
return err return err
} }
b.ForEach(func(day, messages []byte) error {
stats.ByDay[string(day)] = MakeUint(messages, "date", string(day))
return nil
})
todayKey := stats.TodayDate.Format("2006-1-2") todayKey := stats.TodayDate.Format("2006-1-2")
stats.Today = MakeUint(b.Get([]byte(todayKey)), "date", todayKey) stats.Today = MakeUint(b.Get([]byte(todayKey)), "date", todayKey)
@ -132,9 +138,10 @@ func loadStats() {
} }
func updateDate() { func updateDate() {
dateKey := stats.TodayDate.Format("2006-1-2")
err := db.Update(func(tx *bolt.Tx) error { err := db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte("date")) b := tx.Bucket([]byte("date"))
err := b.Put([]byte(stats.TodayDate.Format("2006-1-2")), PutUint(stats.Today)) err := b.Put([]byte(dateKey), PutUint(stats.Today))
if err != nil { if err != nil {
return err return err
} }
@ -143,6 +150,7 @@ func updateDate() {
if err != nil { if err != nil {
log.Println("[updateDate] Couldn't save last day stats: " + err.Error()) log.Println("[updateDate] Couldn't save last day stats: " + err.Error())
} }
stats.ByDay[dateKey] = stats.Today
stats.TodayDate = time.Now() stats.TodayDate = time.Now()
stats.Today = 0 stats.Today = 0
} }