feat: add daily challenge record storage and leaderboard

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-03-25 16:01:58 +09:00
parent cd0d6c1c4c
commit 253d1f80d3
3 changed files with 159 additions and 0 deletions

81
store/daily.go Normal file
View File

@@ -0,0 +1,81 @@
package store
import (
"encoding/json"
"sort"
"time"
bolt "go.etcd.io/bbolt"
)
var bucketDailyRuns = []byte("daily_runs")
type DailyRecord struct {
Date string `json:"date"`
Player string `json:"player"`
PlayerName string `json:"player_name"`
FloorReached int `json:"floor_reached"`
GoldEarned int `json:"gold_earned"`
}
func (d *DB) SaveDaily(record DailyRecord) error {
return d.db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket(bucketDailyRuns)
key := []byte(record.Date + ":" + record.Player)
data, err := json.Marshal(record)
if err != nil {
return err
}
return b.Put(key, data)
})
}
func (d *DB) GetDailyLeaderboard(date string, limit int) ([]DailyRecord, error) {
var records []DailyRecord
prefix := []byte(date + ":")
err := d.db.View(func(tx *bolt.Tx) error {
b := tx.Bucket(bucketDailyRuns)
c := b.Cursor()
for k, v := c.Seek(prefix); k != nil && len(k) >= len(prefix) && string(k[:len(prefix)]) == string(prefix); k, v = c.Next() {
var r DailyRecord
if json.Unmarshal(v, &r) == nil {
records = append(records, r)
}
}
return nil
})
if err != nil {
return nil, err
}
sort.Slice(records, func(i, j int) bool {
if records[i].FloorReached != records[j].FloorReached {
return records[i].FloorReached > records[j].FloorReached
}
return records[i].GoldEarned > records[j].GoldEarned
})
if len(records) > limit {
records = records[:limit]
}
return records, nil
}
func (d *DB) GetStreak(fingerprint, currentDate string) (int, error) {
streak := 0
date, err := time.Parse("2006-01-02", currentDate)
if err != nil {
return 0, err
}
err = d.db.View(func(tx *bolt.Tx) error {
b := tx.Bucket(bucketDailyRuns)
for {
key := []byte(date.Format("2006-01-02") + ":" + fingerprint)
if b.Get(key) == nil {
break
}
streak++
date = date.AddDate(0, 0, -1)
}
return nil
})
return streak, err
}

66
store/daily_test.go Normal file
View File

@@ -0,0 +1,66 @@
package store
import (
"testing"
)
func TestSaveAndGetDaily(t *testing.T) {
dir := t.TempDir()
db, err := Open(dir + "/test_daily.db")
if err != nil {
t.Fatal(err)
}
defer db.Close()
db.SaveDaily(DailyRecord{Date: "2026-03-25", Player: "fp1", PlayerName: "Alice", FloorReached: 10, GoldEarned: 200})
db.SaveDaily(DailyRecord{Date: "2026-03-25", Player: "fp2", PlayerName: "Bob", FloorReached: 15, GoldEarned: 100})
db.SaveDaily(DailyRecord{Date: "2026-03-25", Player: "fp3", PlayerName: "Charlie", FloorReached: 15, GoldEarned: 300})
records, err := db.GetDailyLeaderboard("2026-03-25", 10)
if err != nil {
t.Fatal(err)
}
if len(records) != 3 {
t.Fatalf("expected 3 records, got %d", len(records))
}
// Charlie and Bob both floor 15, Charlie has more gold so first
if records[0].PlayerName != "Charlie" {
t.Errorf("expected Charlie first, got %s", records[0].PlayerName)
}
if records[1].PlayerName != "Bob" {
t.Errorf("expected Bob second, got %s", records[1].PlayerName)
}
if records[2].PlayerName != "Alice" {
t.Errorf("expected Alice third, got %s", records[2].PlayerName)
}
}
func TestDailyStreak(t *testing.T) {
dir := t.TempDir()
db, err := Open(dir + "/test_streak.db")
if err != nil {
t.Fatal(err)
}
defer db.Close()
db.SaveDaily(DailyRecord{Date: "2026-03-23", Player: "fp1", PlayerName: "Alice", FloorReached: 5, GoldEarned: 50})
db.SaveDaily(DailyRecord{Date: "2026-03-24", Player: "fp1", PlayerName: "Alice", FloorReached: 8, GoldEarned: 100})
db.SaveDaily(DailyRecord{Date: "2026-03-25", Player: "fp1", PlayerName: "Alice", FloorReached: 10, GoldEarned: 200})
streak, err := db.GetStreak("fp1", "2026-03-25")
if err != nil {
t.Fatal(err)
}
if streak != 3 {
t.Errorf("expected streak 3, got %d", streak)
}
// Gap in streak
streak2, err := db.GetStreak("fp1", "2026-03-27")
if err != nil {
t.Fatal(err)
}
if streak2 != 0 {
t.Errorf("expected streak 0 after gap, got %d", streak2)
}
}

View File

@@ -39,6 +39,18 @@ func Open(path string) (*DB, error) {
if _, err := tx.CreateBucketIfNotExists(bucketAchievements); err != nil { if _, err := tx.CreateBucketIfNotExists(bucketAchievements); err != nil {
return err return err
} }
if _, err := tx.CreateBucketIfNotExists(bucketDailyRuns); err != nil {
return err
}
if _, err := tx.CreateBucketIfNotExists(bucketUnlocks); err != nil {
return err
}
if _, err := tx.CreateBucketIfNotExists(bucketTitles); err != nil {
return err
}
if _, err := tx.CreateBucketIfNotExists(bucketCodex); err != nil {
return err
}
return nil return nil
}) })
return &DB{db: db}, err return &DB{db: db}, err