init: 初始化仓库

service/scoresaber/hot.go

@@ -0,0 +1,74 @@
+package scoresaber
+
+import (
+	"log"
+	"sync"
+	"time"
+
+	"github.com/gorilla/websocket"
+)
+
+const wsURL = "wss://scoresaber.com/ws"
+
+var ScoresManager = scoresManager{}
+
+// scoresManager 管理最近的 5 条数据
+type scoresManager struct {
+	recentScores []Command
+	mu           sync.Mutex
+	conn         *websocket.Conn
+	retryTimes   int
+}
+
+func init() {
+	for err := ScoresManager.connect(); err != nil; err = ScoresManager.connect() {
+		log.Fatal("连接 WebSocket 失败:", err)
+		time.Sleep(time.Second)
+	}
+}
+
+func (sm *scoresManager) connect() error {
+	var err error
+	sm.conn, _, err = websocket.DefaultDialer.Dial(wsURL, nil)
+	if err != nil {
+		return err
+	}
+	sm.retryTimes = 0
+	sm.recentScores = make([]Command, 0, 50)
+	go sm.receiveData()
+	return nil
+}
+
+func (sm *scoresManager) receiveData() {
+	defer func() {
+		for err := ScoresManager.connect(); err != nil; err = ScoresManager.connect() {
+			log.Printf("连接 WebSocket 失败:%v", err)
+			time.Sleep(time.Second)
+		}
+	}()
+	for {
+		var cmd Command
+		err := sm.conn.ReadJSON(&cmd)
+		if err != nil {
+			log.Printf("读取数据失败:", err)
+			continue
+		}
+		sm.mu.Lock()
+		if len(sm.recentScores) >= 50 {
+			sm.recentScores = sm.recentScores[1:]
+		}
+		sm.recentScores = append(sm.recentScores, cmd)
+		sm.mu.Unlock()
+	}
+}
+
+func (sm *scoresManager) GetRecentScores(count int) []Command {
+	if count > len(sm.recentScores) {
+		count = len(sm.recentScores)
+	}
+	sm.mu.Lock()
+	defer sm.mu.Unlock()
+	scoresCopy := make([]Command, count)
+	copy(scoresCopy, sm.recentScores[len(sm.recentScores)-count:])
+	return scoresCopy
+}