@@ -2,13 +2,14 @@ package minecraftobserver
2
2
3
3
import (
4
4
"errors"
5
+ "os"
6
+ "sync"
7
+ "time"
8
+
5
9
fcext "github.com/FloatTech/floatbox/ctxext"
6
10
"github.com/jinzhu/gorm"
7
11
zero "github.com/wdvxdr1123/ZeroBot"
8
12
"github.com/wdvxdr1123/ZeroBot/message"
9
- "os"
10
- "sync"
11
- "time"
12
13
)
13
14
14
15
const (
@@ -43,7 +44,7 @@ func initializeDB(dbpath string) error {
43
44
}
44
45
gdb , err := gorm .Open ("sqlite3" , dbpath )
45
46
if err != nil {
46
- //logrus.Errorln(logPrefix+"initializeDB ERROR: ", err)
47
+ // logrus.Errorln(logPrefix+"initializeDB ERROR: ", err)
47
48
return err
48
49
}
49
50
gdb .AutoMigrate (& serverStatus {}, & serverSubscribe {})
60
61
dbInstance * db
61
62
// 开启并检查数据库链接
62
63
getDB = fcext .DoOnceOnSuccess (func (ctx * zero.Ctx ) bool {
63
- var err error
64
- err = initializeDB (engine .DataFolder () + dbPath )
64
+ var err error = initializeDB (engine .DataFolder () + dbPath )
65
65
if err != nil {
66
- //logrus.Errorln(logPrefix+"initializeDB ERROR: ", err)
66
+ // logrus.Errorln(logPrefix+"initializeDB ERROR: ", err)
67
67
ctx .SendChain (message .Text ("[mc-ob] ERROR: " , err ))
68
68
return false
69
69
}
@@ -81,7 +81,7 @@ func (d *db) getServerStatus(addr string) (*serverStatus, error) {
81
81
}
82
82
var ss serverStatus
83
83
if err := d .sdb .Model (& ss ).Where ("server_addr = ?" , addr ).First (& ss ).Error ; err != nil {
84
- //logrus.Errorln(logPrefix+"getServerStatus ERROR: ", err)
84
+ // logrus.Errorln(logPrefix+"getServerStatus ERROR: ", err)
85
85
return nil , err
86
86
}
87
87
return & ss , nil
@@ -100,7 +100,7 @@ func (d *db) updateServerStatus(ss *serverStatus) (err error) {
100
100
ss .LastUpdate = time .Now ().Unix ()
101
101
ss2 := ss .deepCopy ()
102
102
if err = d .sdb .Where (& serverStatus {ServerAddr : ss .ServerAddr }).Assign (ss2 ).FirstOrCreate (ss ).Debug ().Error ; err != nil {
103
- //logrus.Errorln(logPrefix, fmt.Sprintf("updateServerStatus %v ERROR: %v", ss, err))
103
+ // logrus.Errorln(logPrefix, fmt.Sprintf("updateServerStatus %v ERROR: %v", ss, err))
104
104
return
105
105
}
106
106
return
@@ -116,7 +116,7 @@ func (d *db) delServerStatus(addr string) (err error) {
116
116
d .statusLock .Lock ()
117
117
defer d .statusLock .Unlock ()
118
118
if err = d .sdb .Where ("server_addr = ?" , addr ).Delete (& serverStatus {}).Error ; err != nil {
119
- //logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
119
+ // logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
120
120
return
121
121
}
122
122
return
@@ -128,7 +128,7 @@ func (d *db) newSubscribe(addr string, targetID, targetType int64) (err error) {
128
128
return errDBConn
129
129
}
130
130
if targetID == 0 || (targetType != 1 && targetType != 2 ) {
131
- //logrus.Errorln(logPrefix+"newSubscribe ERROR: 参数错误 ", targetID, " ", targetType)
131
+ // logrus.Errorln(logPrefix+"newSubscribe ERROR: 参数错误 ", targetID, " ", targetType)
132
132
return errParam
133
133
}
134
134
d .subscribeLock .Lock ()
@@ -137,7 +137,7 @@ func (d *db) newSubscribe(addr string, targetID, targetType int64) (err error) {
137
137
existedRec := & serverSubscribe {}
138
138
err = d .sdb .Model (& serverSubscribe {}).Where ("server_addr = ? and target_id = ? and target_type = ?" , addr , targetID , targetType ).First (existedRec ).Error
139
139
if err != nil && ! gorm .IsRecordNotFoundError (err ) {
140
- //logrus.Errorln(logPrefix+"newSubscribe ERROR: ", err)
140
+ // logrus.Errorln(logPrefix+"newSubscribe ERROR: ", err)
141
141
return
142
142
}
143
143
if existedRec .ID != 0 {
@@ -150,7 +150,7 @@ func (d *db) newSubscribe(addr string, targetID, targetType int64) (err error) {
150
150
LastUpdate : time .Now ().Unix (),
151
151
}
152
152
if err = d .sdb .Model (& ss ).Create (ss ).Error ; err != nil {
153
- //logrus.Errorln(logPrefix+"newSubscribe ERROR: ", err)
153
+ // logrus.Errorln(logPrefix+"newSubscribe ERROR: ", err)
154
154
return
155
155
}
156
156
return
@@ -171,20 +171,20 @@ func (d *db) deleteSubscribe(addr string, targetID int64, targetType int64) (err
171
171
if gorm .IsRecordNotFoundError (err ) {
172
172
return errors .New ("未找到订阅" )
173
173
}
174
- //logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
174
+ // logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
175
175
return
176
176
}
177
177
178
178
if err = d .sdb .Where ("server_addr = ? and target_id = ? and target_type = ?" , addr , targetID , targetType ).Delete (& serverSubscribe {}).Error ; err != nil {
179
- //logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
179
+ // logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
180
180
return
181
181
}
182
182
183
183
// 扫描是否还有订阅,如果没有则删除服务器状态
184
184
var cnt int
185
185
err = d .sdb .Model (& serverSubscribe {}).Where ("server_addr = ?" , addr ).Count (& cnt ).Error
186
186
if err != nil {
187
- //logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
187
+ // logrus.Errorln(logPrefix+"deleteSubscribe ERROR: ", err)
188
188
return
189
189
}
190
190
if cnt == 0 {
@@ -200,7 +200,7 @@ func (d *db) getAllSubscribes() (subs []serverSubscribe, err error) {
200
200
}
201
201
subs = []serverSubscribe {}
202
202
if err = d .sdb .Find (& subs ).Error ; err != nil {
203
- //logrus.Errorln(logPrefix+"getAllSubscribes ERROR: ", err)
203
+ // logrus.Errorln(logPrefix+"getAllSubscribes ERROR: ", err)
204
204
return
205
205
}
206
206
return
@@ -213,7 +213,7 @@ func (d *db) getSubscribesByTarget(targetID, targetType int64) (subs []serverSub
213
213
}
214
214
subs = []serverSubscribe {}
215
215
if err = d .sdb .Model (& serverSubscribe {}).Where ("target_id = ? and target_type = ?" , targetID , targetType ).Find (& subs ).Error ; err != nil {
216
- //logrus.Errorln(logPrefix+"getSubscribesByTarget ERROR: ", err)
216
+ // logrus.Errorln(logPrefix+"getSubscribesByTarget ERROR: ", err)
217
217
return
218
218
}
219
219
return
0 commit comments