tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reduce DB's Use method complexity and make it easier to understand

Browse Source
This commit is contained in:
0x2d3c 2021-01-09 22:22:57 +08:00
parent d888c799d7
commit 1e0840ccd8
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
gorm.go
View File

@ -380,15 +380,14 @@ func (db *DB) SetupJoinTable(model interface{}, field string, joinTable interfac
return nil
}
func (db *DB) Use(plugin Plugin) (err error) {
func (db *DB) Use(plugin Plugin) error {
name := plugin.Name()
if _, ok := db.Plugins[name]; !ok {
if err = plugin.Initialize(db); err == nil {
db.Plugins[name] = plugin
}
} else {
if _, ok := db.Plugins[name]; ok {
return ErrRegistered
}
return err
if err := plugin.Initialize(db); err != nil {
return err
}
db.Plugins[name] = plugin
return nil
}