插件功能开发完整度 +1
This commit is contained in:
25
server/initialize/plug.go
Normal file
25
server/initialize/plug.go
Normal file
@@ -0,0 +1,25 @@
|
||||
package initialize
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/jinzhu/gorm"
|
||||
)
|
||||
|
||||
type Plug interface {
|
||||
InitRouter(*gin.Engine) error
|
||||
InitModel(*gorm.DB) error
|
||||
}
|
||||
|
||||
func InstallPlug(db *gorm.DB, router *gin.Engine, p ...Plug) (err error) {
|
||||
for _, v := range p {
|
||||
err = v.InitModel(db)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = v.InitRouter(router)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
Reference in New Issue
Block a user