Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2026-02-12 10:10:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🔥 remove debug info in gorm

Browse Source
This commit is contained in:
源文雨
2022-04-16 00:04:07 +08:00
parent dfd184724b
commit c76d9d4461
6 changed files with 53 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
plugin/bilibili/model.go
View File

@@ -55,7 +55,7 @@ func initialize(dbpath string) (*vupdb, error) {
if err != nil {
return nil, err
}
gdb.Debug().AutoMigrate(&vup{}).AutoMigrate(&config{})
gdb.AutoMigrate(&vup{}).AutoMigrate(&config{})
return (*vupdb)(gdb), nil
}
@@ -66,9 +66,9 @@ func (vdb *vupdb) insertVupByMid(mid int64, uname string, roomid int64) (err err
Uname: uname,
Roomid: roomid,
}
if err = db.Debug().Model(&vup{}).First(&v, "mid = ? ", mid).Error; err != nil {
if err = db.Model(&vup{}).First(&v, "mid = ? ", mid).Error; err != nil {
if gorm.IsRecordNotFoundError(err) {
err = db.Debug().Model(&vup{}).Create(&v).Error
err = db.Model(&vup{}).Create(&v).Error
}
}
return
@@ -77,7 +77,7 @@ func (vdb *vupdb) insertVupByMid(mid int64, uname string, roomid int64) (err err
// filterVup 筛选vup
func (vdb *vupdb) filterVup(ids []int64) (vups []vup, err error) {
db := (*gorm.DB)(vdb)
if err = db.Debug().Model(&vup{}).Find(&vups, "mid in (?)", ids).Error; err != nil {
if err = db.Model(&vup{}).Find(&vups, "mid in (?)", ids).Error; err != nil {
return vups, err
}
return
@@ -112,13 +112,13 @@ func (vdb *vupdb) setBilibiliCookie(cookie string) (err error) {
Key: bilibiliCookie,
Value: cookie,
}
if err = db.Debug().Model(&config{}).First(&c, "key = ? ", bilibiliCookie).Error; err != nil {
if err = db.Model(&config{}).First(&c, "key = ? ", bilibiliCookie).Error; err != nil {
// error handling...
if gorm.IsRecordNotFoundError(err) {
err = db.Debug().Model(&config{}).Create(&c).Error
err = db.Model(&config{}).Create(&c).Error
}
} else {
err = db.Debug().Model(&config{}).Where("key = ? ", bilibiliCookie).Update(
err = db.Model(&config{}).Where("key = ? ", bilibiliCookie).Update(
map[string]interface{}{
"value": cookie,
}).Error
@@ -128,6 +128,6 @@ func (vdb *vupdb) setBilibiliCookie(cookie string) (err error) {
func (vdb *vupdb) getBilibiliCookie() (c config) {
db := (*gorm.DB)(vdb)
db.Debug().Model(&config{}).First(&c, "key = ?", bilibiliCookie)
db.Model(&config{}).First(&c, "key = ?", bilibiliCookie)
return
}