public/go-admin
1
0
Fork 0
mirror of https://github.com/go-admin-team/go-admin.git synced 2026-09-21 10:13:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

迁移方法set时增加锁

Browse Source
This commit is contained in:
linwenxiang
2020-09-04 14:33:46 +08:00
parent 77dfe418df
commit 3bd3656d15
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cmd/migrate/migration/init.go
+4
View File
@@ -4,6 +4,7 @@ import (
"gorm.io/gorm" "gorm.io/gorm"
"log" "log"
"sort" "sort"
"sync"
) )
var Migrate = &Migration{ var Migrate = &Migration{
@@ -13,6 +14,7 @@ var Migrate = &Migration{
type Migration struct { type Migration struct {
db *gorm.DB db *gorm.DB
version map[string]func(db *gorm.DB, version string) error version map[string]func(db *gorm.DB, version string) error
mutex sync.Mutex
} }
func (e *Migration) GetDb() *gorm.DB { func (e *Migration) GetDb() *gorm.DB {
@@ -24,6 +26,8 @@ func (e *Migration) SetDb(db *gorm.DB) {
} }
func (e *Migration) SetVersion(k string, f func(db *gorm.DB, version string) error) { func (e *Migration) SetVersion(k string, f func(db *gorm.DB, version string) error) {
e.mutex.Lock()
defer e.mutex.Unlock()
e.version[k] = f e.version[k] = f
} }