98 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			98 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package admin_menu
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	"database/sql"
 | |
| 	"encoding/json"
 | |
| 	"time"
 | |
| 
 | |
| 	"aedata-server/app/main/api/internal/svc"
 | |
| 	"aedata-server/app/main/api/internal/types"
 | |
| 	"aedata-server/app/main/model"
 | |
| 	"aedata-server/common/xerr"
 | |
| 
 | |
| 	"github.com/pkg/errors"
 | |
| 	"github.com/zeromicro/go-zero/core/logx"
 | |
| )
 | |
| 
 | |
| type CreateMenuLogic struct {
 | |
| 	logx.Logger
 | |
| 	ctx    context.Context
 | |
| 	svcCtx *svc.ServiceContext
 | |
| }
 | |
| 
 | |
| func NewCreateMenuLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateMenuLogic {
 | |
| 	return &CreateMenuLogic{
 | |
| 		Logger: logx.WithContext(ctx),
 | |
| 		ctx:    ctx,
 | |
| 		svcCtx: svcCtx,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (l *CreateMenuLogic) CreateMenu(req *types.CreateMenuReq) (resp *types.CreateMenuResp, err error) {
 | |
| 	// 1. 参数验证
 | |
| 	if req.Name == "" {
 | |
| 		return nil, errors.Wrapf(xerr.NewErrMsg("菜单名称不能为空"), "菜单名称不能为空")
 | |
| 	}
 | |
| 	if req.Type == "menu" && req.Component == "" {
 | |
| 		return nil, errors.Wrapf(xerr.NewErrMsg("组件路径不能为空"), "组件路径不能为空")
 | |
| 	}
 | |
| 
 | |
| 	// 2. 检查名称和路径是否重复
 | |
| 	exists, err := l.svcCtx.AdminMenuModel.FindOneByNamePath(l.ctx, req.Name, req.Path)
 | |
| 	if err != nil && !errors.Is(err, model.ErrNotFound) {
 | |
| 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询菜单失败, err: %v", err)
 | |
| 	}
 | |
| 	if exists != nil {
 | |
| 		return nil, errors.Wrapf(xerr.NewErrMsg("菜单名称或路径已存在"), "菜单名称或路径已存在")
 | |
| 	}
 | |
| 
 | |
| 	// 3. 检查父菜单是否存在(如果不是根菜单)
 | |
| 	if req.Pid > 0 {
 | |
| 		parentMenu, err := l.svcCtx.AdminMenuModel.FindOne(l.ctx, req.Pid)
 | |
| 		if err != nil {
 | |
| 			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询父菜单失败, id: %d, err: %v", req.Pid, err)
 | |
| 		}
 | |
| 		if parentMenu == nil {
 | |
| 			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "父菜单不存在, id: %d", req.Pid)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	// 4. 将类型标签转换为值
 | |
| 	typeValue, err := l.svcCtx.DictService.GetDictValue(l.ctx, "admin_menu_type", req.Type)
 | |
| 	if err != nil {
 | |
| 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "菜单类型无效: %v", err)
 | |
| 	}
 | |
| 
 | |
| 	// 5. 创建菜单记录
 | |
| 	menu := &model.AdminMenu{
 | |
| 		Pid:        req.Pid,
 | |
| 		Name:       req.Name,
 | |
| 		Path:       req.Path,
 | |
| 		Component:  req.Component,
 | |
| 		Redirect:   sql.NullString{String: req.Redirect, Valid: req.Redirect != ""},
 | |
| 		Status:     req.Status,
 | |
| 		Type:       typeValue,
 | |
| 		Sort:       req.Sort,
 | |
| 		CreateTime: time.Now(),
 | |
| 		UpdateTime: time.Now(),
 | |
| 	}
 | |
| 
 | |
| 	// 将Meta转换为JSON字符串
 | |
| 	metaJson, err := json.Marshal(req.Meta)
 | |
| 	if err != nil {
 | |
| 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "Meta数据格式错误: %v", err)
 | |
| 	}
 | |
| 	menu.Meta = string(metaJson)
 | |
| 
 | |
| 	// 6. 保存到数据库
 | |
| 	_, err = l.svcCtx.AdminMenuModel.Insert(l.ctx, nil, menu)
 | |
| 	if err != nil {
 | |
| 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建菜单失败, err: %v", err)
 | |
| 	}
 | |
| 
 | |
| 	return &types.CreateMenuResp{
 | |
| 		Id: menu.Id,
 | |
| 	}, nil
 | |
| }
 |