Skip to content

Commit fc90840

Browse files
committed
Table -> table
1 parent d189034 commit fc90840

File tree

2 files changed

+20
-22
lines changed

2 files changed

+20
-22
lines changed

niu/db.go

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,7 @@ var (
1313

1414
type tableHook func(gid int64) error
1515

16-
// Model ...
17-
type Model struct {
16+
type model struct {
1817
*gorm.DB
1918
}
2019

@@ -46,8 +45,7 @@ func registerTableHook(h ...tableHook) {
4645
tableHooks = append(tableHooks, h...)
4746
}
4847

49-
// TableFor 大写是为了防止数据操作哪里有问题留个保底可以在zbp的项目里直接改
50-
func TableFor(gid int64, prefix string) *Model {
48+
func tableFor(gid int64, prefix string) *model {
5149
// 先跑钩子
5250
for _, h := range tableHooks {
5351
if err := h(gid); err != nil {
@@ -56,35 +54,35 @@ func TableFor(gid int64, prefix string) *Model {
5654
}
5755

5856
tableName := fmt.Sprintf("group_%d_%s_info", gid, prefix)
59-
return &Model{db.Table(tableName)}
57+
return &model{db.Table(tableName)}
6058
}
6159

6260
func listUsers(gid int64) (users, error) {
6361
var us users
64-
err := TableFor(gid, ur).Find(&us).Error
62+
err := tableFor(gid, ur).Find(&us).Error
6563
return us, err
6664
}
6765

6866
func listAuction(gid int64) ([]AuctionInfo, error) {
6967
var as []AuctionInfo
70-
err := TableFor(gid, ac).Order("money DESC").Find(&as).Error
68+
err := tableFor(gid, ac).Order("money DESC").Find(&as).Error
7169
return as, err
7270
}
7371

7472
func createUser(gid int64, user *userInfo, fix string) error {
75-
return TableFor(gid, fix).Create(user).Error
73+
return tableFor(gid, fix).Create(user).Error
7674
}
7775

7876
func getUserByID(gid int64, uid int64) (*userInfo, error) {
7977
var user userInfo
80-
err := TableFor(gid, ur).Where("user_id = ?", uid).First(&user).Error
78+
err := tableFor(gid, ur).Where("user_id = ?", uid).First(&user).Error
8179
return &user, err
8280
}
8381

8482
func updatesUserByID(gid int64, id int64, fields map[string]interface{}) error {
85-
return TableFor(gid, ur).Where("user_id = ?", id).Updates(fields).Error
83+
return tableFor(gid, ur).Where("user_id = ?", id).Updates(fields).Error
8684
}
8785

8886
func deleteUserByID(gid int64, id int64) error {
89-
return TableFor(gid, ur).Where("user_id = ?", id).Delete(&userInfo{}).Error
87+
return tableFor(gid, ur).Where("user_id = ?", id).Delete(&userInfo{}).Error
9088
}

niu/main.go

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ const (
2222
)
2323

2424
var (
25-
db *Model
25+
db *model
2626
globalLock sync.Mutex
2727

2828
errCancelFail = errors.New("遇到不可抗力因素,注销失败!")
@@ -84,7 +84,7 @@ func init() {
8484
panic(err)
8585
}
8686

87-
db = &Model{sdb.LogMode(false)}
87+
db = &model{sdb.LogMode(false)}
8888

8989
registerTableHook(ensureUserInfo, ensureAuctionInfo)
9090
}
@@ -206,7 +206,7 @@ func HitGlue(gid, uid int64, prop string) (string, error) {
206206
return "", err
207207
}
208208

209-
if err = TableFor(gid, ur).Where("user_id = ?", uid).Save(niuniu).Error; err != nil {
209+
if err = tableFor(gid, ur).Where("user_id = ?", uid).Save(niuniu).Error; err != nil {
210210
return "", err
211211
}
212212

@@ -266,11 +266,11 @@ func JJ(gid, uid, adduser int64, prop string) (message string, adduserLength flo
266266
return "", 0, err
267267
}
268268

269-
if err = TableFor(gid, ur).Where("user_id =?", uid).Update("length", myniuniu.Length).Error; err != nil {
269+
if err = tableFor(gid, ur).Where("user_id =?", uid).Update("length", myniuniu.Length).Error; err != nil {
270270
return "", 0, err
271271
}
272272

273-
if err = TableFor(gid, ur).Where("user_id =?", adduser).Update("length", adduserniuniu.Length).Error; err != nil {
273+
if err = tableFor(gid, ur).Where("user_id =?", adduser).Update("length", adduserniuniu.Length).Error; err != nil {
274274
return "", 0, err
275275
}
276276

@@ -326,7 +326,7 @@ func Redeem(gid, uid int64, lastLength float64) error {
326326
return err
327327
}
328328

329-
return TableFor(gid, ur).Where("user_id = ?", uid).Update("length", lastLength).Error
329+
return tableFor(gid, ur).Where("user_id = ?", uid).Update("length", lastLength).Error
330330
}
331331

332332
// Store 牛牛商店
@@ -352,7 +352,7 @@ func Store(gid, uid int64, productID int, quantity int) error {
352352
return err
353353
}
354354

355-
return TableFor(gid, ur).Save(info).Error
355+
return tableFor(gid, ur).Save(info).Error
356356
}
357357

358358
// Sell 出售牛牛
@@ -388,7 +388,7 @@ func Sell(gid, uid int64) (string, error) {
388388
Money: money * 2,
389389
}
390390

391-
if err = TableFor(gid, ac).Create(&u).Error; err != nil {
391+
if err = tableFor(gid, ac).Create(&u).Error; err != nil {
392392
return "", err
393393
}
394394

@@ -413,7 +413,7 @@ func Auction(gid, uid int64, index int) (string, error) {
413413
}
414414

415415
var info AuctionInfo
416-
if err = TableFor(gid, ac).Where("id = ?", index).First(&info).Error; err != nil {
416+
if err = tableFor(gid, ac).Where("id = ?", index).First(&info).Error; err != nil {
417417
return "", err
418418
}
419419

@@ -437,11 +437,11 @@ func Auction(gid, uid int64, index int) (string, error) {
437437
niu.MeiYao += 2
438438
}
439439

440-
if err = TableFor(gid, ac).Delete(&info).Error; err != nil {
440+
if err = tableFor(gid, ac).Delete(&info).Error; err != nil {
441441
return "", err
442442
}
443443

444-
if err = TableFor(gid, ur).Save(&niu).Error; err != nil {
444+
if err = tableFor(gid, ur).Save(&niu).Error; err != nil {
445445
return "", err
446446
}
447447

0 commit comments

Comments
 (0)