36 lines
872 B
Go
36 lines
872 B
Go
package announcement
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"gorm.io/gorm"
|
||
"kra/internal/plugin/announcement/initialize"
|
||
interfaces "kra/pkg/utils/plugin/v2"
|
||
)
|
||
|
||
var _ interfaces.Plugin = (*plugin)(nil)
|
||
|
||
var Plugin = new(plugin)
|
||
|
||
type plugin struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// SetDB 设置数据库连接
|
||
func (p *plugin) SetDB(db *gorm.DB) {
|
||
p.db = db
|
||
}
|
||
|
||
func (p *plugin) Register(engine *gin.Engine) {
|
||
ctx := context.Background()
|
||
// 如果需要配置文件，请到config.Config中填充配置结构，且到下方发放中填入其在config.yaml中的key
|
||
// initialize.Viper()
|
||
// 安装插件时候自动注册的api数据请到下方法.Api方法中实现
|
||
initialize.Api(ctx, p.db)
|
||
// 安装插件时候自动注册的api数据请到下方法.Menu方法中实现
|
||
initialize.Menu(ctx, p.db)
|
||
initialize.Gorm(ctx, p.db)
|
||
initialize.Router(engine, p.db)
|
||
}
|