修改model -> Model

Author: xyy0411

niu/db.go

@@ -15,7 +15,7 @@ var (
 
 type tableHook func(gid int64) error
 
-type model struct {
+type Model struct {
 	*gorm.DB
 }
 
@@ -50,7 +50,7 @@ func registerTableHook(h ...tableHook) {
 }
 
 // TableFor 大写是为了防止数据操作哪里有问题留个保底可以在zbp的项目里直接改
-func TableFor(gid int64, prefix string) *model {
+func TableFor(gid int64, prefix string) *Model {
 	// 先执行钩子
 	hooksMtx.RLock()
 	for _, h := range tableHooks {
@@ -61,7 +61,7 @@ func TableFor(gid int64, prefix string) *model {
 	hooksMtx.RUnlock()
 
 	tableName := fmt.Sprintf("group_%d_%s_info", gid, prefix)
-	return &model{db.Table(tableName)}
+	return &Model{db.Table(tableName)}
 }
 
 func listUsers(gid int64) (users, error) {

niu/main.go

@@ -22,7 +22,7 @@ const (
 )
 
 var (
-	db         *model
+	db         *Model
 	globalLock sync.Mutex
 
 	errCancelFail = errors.New("遇到不可抗力因素，注销失败！")
@@ -84,7 +84,7 @@ func init() {
 		panic(err)
 	}
 
-	db = &model{sdb.LogMode(false)}
+	db = &Model{sdb.LogMode(false)}
 
 	registerTableHook(ensureUserInfo, ensureAuctionInfo)
 }