后端重构

pkg/utils/plugin/tool/check.go

@@ -0,0 +1,66 @@
+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))
+	}
+}