forked from FloatTech/ZeroBot-Plugin
/
store.go
106 lines (95 loc) · 2.98 KB
/
store.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package steam
import (
"strconv"
"sync"
"time"
fcext "github.com/FloatTech/floatbox/ctxext"
sql "github.com/FloatTech/sqlite"
ctrl "github.com/FloatTech/zbpctrl"
zero "github.com/wdvxdr1123/ZeroBot"
"github.com/wdvxdr1123/ZeroBot/message"
)
var (
database streamDB
// 开启并检查数据库链接
getDB = fcext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool {
database.db.DBPath = engine.DataFolder() + "steam.db"
err := database.db.Open(time.Hour)
if err != nil {
ctx.SendChain(message.Text("[steam] ERROR: ", err))
return false
}
if err = database.db.Create(tableListenPlayer, &player{}); err != nil {
ctx.SendChain(message.Text("[steam] ERROR: ", err))
return false
}
// 校验密钥是否初始化
m := ctx.State["manager"].(*ctrl.Control[*zero.Ctx])
apiKeyMu.Lock()
defer apiKeyMu.Unlock()
_ = m.GetExtra(&apiKey)
if apiKey == "" {
ctx.SendChain(message.Text("ERROR: 未设置steam apikey"))
return false
}
return true
})
)
// streamDB 继承方法的存储结构
type streamDB struct {
sync.RWMutex
db sql.Sqlite
}
const (
// tableListenPlayer 存储查询用户信息
tableListenPlayer = "listen_player"
)
// player 用户状态存储结构体
type player struct {
SteamID int64 `json:"steam_id"` // 绑定用户标识ID
PersonaName string `json:"persona_name"` // 用户昵称
Target string `json:"target"` // 信息推送群组
GameID int64 `json:"game_id"` // 游戏ID
GameExtraInfo string `json:"game_extra_info"` // 游戏信息
LastUpdate int64 `json:"last_update"` // 更新时间
}
// update 如果主键不存在则插入一条新的数据,如果主键存在直接复写
func (sdb *streamDB) update(dbInfo *player) error {
sdb.Lock()
defer sdb.Unlock()
return sdb.db.Insert(tableListenPlayer, dbInfo)
}
// find 根据主键查信息
func (sdb *streamDB) find(steamID int64) (dbInfo player, err error) {
sdb.Lock()
defer sdb.Unlock()
condition := "where steam_id = " + strconv.FormatInt(steamID, 10)
err = sdb.db.Find(tableListenPlayer, &dbInfo, condition)
if err == sql.ErrNullResult { // 规避没有该用户数据的报错
err = nil
}
return
}
// findWithGroupID 根据用户steamID和groupID查询信息
func (sdb *streamDB) findWithGroupID(steamID int64, groupID string) (dbInfo player, err error) {
sdb.Lock()
defer sdb.Unlock()
condition := "where steam_id = " + strconv.FormatInt(steamID, 10) + " AND target LIKE '%" + groupID + "%'"
err = sdb.db.Find(tableListenPlayer, &dbInfo, condition)
if err == sql.ErrNullResult { // 规避没有该用户数据的报错
err = nil
}
return
}
// findAll 查询所有库信息
func (sdb *streamDB) findAll() (dbInfos []*player, err error) {
sdb.Lock()
defer sdb.Unlock()
return sql.FindAll[player](&sdb.db, tableListenPlayer, "")
}
// del 删除指定数据
func (sdb *streamDB) del(steamID int64) error {
sdb.Lock()
defer sdb.Unlock()
return sdb.db.Del(tableListenPlayer, "where steam_id = "+strconv.FormatInt(steamID, 10))
}