refactor: 重构ssid查询

This commit is contained in:
lixiangwuxian 2024-10-13 04:53:28 +08:00
parent d084a62815
commit c4ca758682
2 changed files with 29 additions and 8 deletions

View File

@ -291,15 +291,9 @@ func (ss *ssQuery) GetRecentScores(count int, qqId string) ([]RecordDataLite, er
} }
defer tx.Rollback() defer tx.Rollback()
// 查询绑定的 ssId ssId, err := getSSID(qqId)
var ssId string
err = tx.Get(&ssId, "SELECT ssid FROM ssBind WHERE qqid = ?", qqId)
if err != nil { if err != nil {
if err == sql.ErrNoRows { return nil, err
return nil, errors.New("未绑定ss账号输入\"绑定ss [ssId]\"绑定")
}
log.Println("查询 ssId 出错:", err)
return nil, errors.New("查询 ssId 失败")
} }
// 查询记录 // 查询记录

View File

@ -0,0 +1,27 @@
package scoresaber
import (
"database/sql"
"errors"
"log"
"git.lxtend.com/qqbot/sqlite3"
)
func getSSID(qqId string) (ssId string, err error) {
db := sqlite3.GetDB() // 假设 sqlite3.GetDB() 返回 *sqlx.DB
if err != nil {
log.Print(err)
return "", errors.New("数据库连接失败,请稍后重试")
}
err = db.Get(&ssId, "SELECT ssid FROM ssBind WHERE qqid = ?", qqId)
if err != nil {
if err == sql.ErrNoRows {
return "", errors.New("未绑定ss账号输入\"绑定ss [ssId]\"绑定")
}
log.Println("查询 ssId 出错:", err)
return "", errors.New("查询 ssId 失败")
}
return ssId, nil
}