lixiangwuxian/qq_bot
0
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加喜报功能

Browse Source
This commit is contained in:
lixiangwuxian
2024-10-09 01:30:50 +08:00
parent f5e9b74c5a
commit c13b7f2068
12 changed files with 95 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
service/scoresaber/bind_ss.go
View File

@@ -13,7 +13,7 @@ import (
func initDB() {
db, err := sql.Open("sqlite3", "./bindss.db")
if err != nil {
log.Fatal(err)
log.Print(err)
}
defer db.Close()
@@ -69,15 +69,15 @@ func initDB() {
_, err = db.Exec(createBindTableSQL)
if err != nil {
log.Fatal(err)
log.Print(err)
}
_, err = db.Exec(createScoreTableSQL)
if err != nil {
log.Fatal(err)
log.Print(err)
}
_, err = db.Exec(createRecordTableSQL)
if err != nil {
log.Fatal(err)
log.Print(err)
}
}
@@ -91,7 +91,7 @@ func init() {
initDB()
db, err := sql.Open("sqlite3", "./bindss.db")
if err != nil {
log.Fatal(err)
log.Print(err)
}
SSQuery = &ssQuery{db: db}
}
@@ -99,7 +99,7 @@ func init() {
func (ss *ssQuery) BindSS(qqId string, ssId string) (reply string) {
tx, err := ss.db.Begin()
if err != nil {
log.Fatal(err)
log.Print(err)
}
defer tx.Rollback()
// ssId为数字
@@ -131,7 +131,7 @@ func (ss *ssQuery) BindSS(qqId string, ssId string) (reply string) {
func (ss *ssQuery) UnbindSS(qqId string) (reply string) {
tx, err := ss.db.Begin()
if err != nil {
log.Fatal(err)
log.Print(err)
}
defer tx.Rollback()
//是否已绑定
@@ -155,7 +155,7 @@ func (ss *ssQuery) UnbindSS(qqId string) (reply string) {
func (ss *ssQuery) GetScore(qqId string) (reply string) {
tx, err := ss.db.Begin()
if err != nil {
log.Fatal(err)
log.Print(err)
}
defer tx.Rollback()
//是否已绑定
@@ -205,7 +205,7 @@ func (ss *ssQuery) GetScore(qqId string) (reply string) {
func (ss *ssQuery) SaveRecord(cmdData CommandData) {
tx, err := ss.db.Begin()
if err != nil {
log.Fatal(err)
log.Print(err)
}
defer tx.Rollback()
dataLite := RecordDataLite{
@@ -244,18 +244,18 @@ func (ss *ssQuery) SaveRecord(cmdData CommandData) {
dataLite.DeviceControllerRight = *cmdData.Score.DeviceControllerRight
}
if _, err = tx.Exec("INSERT INTO ssRecordData(score_id, ss_id, name, country, song_name, song_sub_name, song_author_name, song_hash, cover_image, difficulty_raw, pp, stars, weight, modifiers, multiplier, bad_cuts, missed_notes, max_combo,score, full_combo, device_hmd, device_controller_left, device_controller_right, generated_time) VALUES(?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12, ?13, ?14, ?15, ?16, ?17, ?18, ?19, ?20, ?21, ?22, ?23,?24)", dataLite.ScoreID, dataLite.SsID, dataLite.Name, dataLite.Country, dataLite.SongName, dataLite.SongSubName, dataLite.SongAuthorName, dataLite.SongHash, dataLite.CoverImage, dataLite.DifficultyRaw, dataLite.PP, dataLite.Stars, dataLite.Weight, dataLite.Modifiers, dataLite.Multiplier, dataLite.BadCuts, dataLite.MissedNotes, dataLite.MaxCombo, dataLite.Score, dataLite.FullCombo, dataLite.DeviceHmd, dataLite.DeviceControllerLeft, dataLite.DeviceControllerRight, dataLite.GeneratedTime); err != nil {
log.Fatal(err)
log.Print(err)
}
err = tx.Commit()
if err != nil {
log.Fatal(err)
log.Print(err)
}
}
func (ss *ssQuery) GetRecentScores(count int, qqId string) ([]RecordDataLite, error) {
tx, err := ss.db.Begin()
if err != nil {
log.Fatal(err)
log.Print(err)
}
defer tx.Rollback()
var ssId string
@@ -283,7 +283,7 @@ func (ss *ssQuery) GetRecentScores(count int, qqId string) ([]RecordDataLite, er
}
err = tx.Commit()
if err != nil {
log.Fatal(err)
log.Print(err)
}
return records, nil
}

2
service/scoresaber/hot.go
View File

@@ -21,7 +21,7 @@ type scoresManager struct {
func init() {
for err := ScoresManager.connect(); err != nil; err = ScoresManager.connect() {
log.Fatal("连接 WebSocket 失败:", err)
log.Print("连接 WebSocket 失败:", err)
time.Sleep(time.Second)
}
}