67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package tool
|
||
|
||
import (
|
||
"github.com/pkg/errors"
|
||
"go.uber.org/zap"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// APIEntity API实体接口
|
||
type APIEntity interface {
|
||
GetPath() string
|
||
GetMethod() string
|
||
GetAPIGroup() string
|
||
}
|
||
|
||
// MenuEntity 菜单实体接口
|
||
type MenuEntity interface {
|
||
GetName() string
|
||
SetParentID(id int64)
|
||
}
|
||
|
||
// RegisterApis 注册API到数据库（通用版本）
|
||
func RegisterApis[T APIEntity](db *gorm.DB, model T, apis ...T) {
|
||
err := db.Transaction(func(tx *gorm.DB) error {
|
||
for _, api := range apis {
|
||
err := tx.Model(model).Where("path = ? AND method = ? AND api_group = ? ", api.GetPath(), api.GetMethod(), api.GetAPIGroup()).FirstOrCreate(&api).Error
|
||
if err != nil {
|
||
zap.L().Error("注册API失败", zap.Error(err), zap.String("api", api.GetPath()), zap.String("method", api.GetMethod()), zap.String("apiGroup", api.GetAPIGroup()))
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
})
|
||
if err != nil {
|
||
zap.L().Error("注册API失败", zap.Error(err))
|
||
}
|
||
}
|
||
|
||
// RegisterMenus 注册菜单到数据库（通用版本）
|
||
func RegisterMenus[T MenuEntity](db *gorm.DB, model T, menus ...T) {
|
||
if len(menus) == 0 {
|
||
return
|
||
}
|
||
parentMenu := menus[0]
|
||
otherMenus := menus[1:]
|
||
err := db.Transaction(func(tx *gorm.DB) error {
|
||
err := tx.Model(model).Where("name = ? ", parentMenu.GetName()).FirstOrCreate(&parentMenu).Error
|
||
if err != nil {
|
||
zap.L().Error("注册菜单失败", zap.Error(err))
|
||
return errors.Wrap(err, "注册菜单失败")
|
||
}
|
||
// 获取父菜单ID需要通过反射或其他方式
|
||
for i := range otherMenus {
|
||
err = tx.Model(model).Where("name = ? ", otherMenus[i].GetName()).FirstOrCreate(&otherMenus[i]).Error
|
||
if err != nil {
|
||
zap.L().Error("注册菜单失败", zap.Error(err))
|
||
return errors.Wrap(err, "注册菜单失败")
|
||
}
|
||
}
|
||
|
||
return nil
|
||
})
|
||
if err != nil {
|
||
zap.L().Error("注册菜单失败", zap.Error(err))
|
||
}
|
||
}
|