插件功能开发完整度 +1

This commit is contained in:
pixel
2020-06-07 00:04:56 +08:00
parent c4a6b33f62
commit 8173ff6fc2
3 changed files with 33 additions and 1 deletions

25
server/initialize/plug.go Normal file
View 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
}