blog-server/internal/service/sys_menu.go

package service

import (
	"context"
	"server/internal/db"
	"server/internal/db/sqlc"
	"server/internal/model/common"
	"server/internal/model/enum"
	"server/internal/model/request"
	"server/internal/pkg/cache"
	"server/internal/pkg/cache/cachekey"
	"server/internal/pkg/dberr"
	"server/internal/pkg/errs"
)

type SysMenuService struct {
	store *db.Store
	cache *cache.Caches
}

func NewSysMenuService(store *db.Store, cache *cache.Caches) *SysMenuService {
	return &SysMenuService{
		store: store,
		cache: cache,
	}
}

func (s *SysMenuService) Create(ctx context.Context, req request.CreateSysMenuRequest) error {
	err := s.store.WithTx(ctx, func(q *sqlc.Queries) error {
		menu := sqlc.CreateSysMenuParams{
			Name:      req.Name,
			Path:      req.Path,
			Component: req.Component,
			Type:      *req.Type,
			Status:    *req.Status,
			Hidden:    req.Hidden,
			Sort:      req.Sort,
			Icon:      req.Icon,
			ParentID:  req.ParentID,
		}

		// 先创建权限
		permissionId, err := q.CreateSysPermission(ctx, sqlc.CreateSysPermissionParams{
			Type: int16(enum.PermissionTypeMenu),
			Code: &req.PermissionCode,
		})

		// code重复性校验
		if err != nil {
			return dberr.MapUniqueViolation(err, dberr.SysPermissionsCodeKey, errs.ErrPermissionCodeAlreadyExists)
		}

		// 创建菜单
		menuId, err := q.CreateSysMenu(ctx, menu)
		if err != nil {
			return dberr.MapUniqueViolation(err, dberr.SysMenusPathUniqueIdx, errs.ErrSysMenusPathUniqueIdx)
		}

		// 绑定权限
		if err = q.CreateSysMenuPermission(ctx, sqlc.CreateSysMenuPermissionParams{
			MenuID:       menuId,
			PermissionID: permissionId,
		}); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return err
	}

	return nil
}

func (s *SysMenuService) Update(ctx context.Context, id int32, req request.UpdateSysMenuRequest) error {
	err := s.store.WithTx(ctx, func(q *sqlc.Queries) error {
		// 构造 menu 参数
		menu := sqlc.UpdateSysMenuParams{
			ID:        id,
			Name:      req.Name,
			Path:      req.Path,
			Component: req.Component,
			Hidden:    req.Hidden,
			Sort:      req.Sort,
			Type:      req.Type,
			Status:    req.Status,
		}

		if req.Icon.Set {
			menu.UpdateIcon = true
			if req.Icon.Valid {
				menu.Icon = &req.Icon.Value
			}
		}

		if req.ParentID.Set {
			menu.UpdateParentID = true
			if req.ParentID.Valid {
				menu.ParentID = &req.ParentID.Value
			}
		}

		// 执行更新
		rows, err := q.UpdateSysMenu(ctx, menu)

		// 先判断数据是否存在 在判断path是否重复
		if err = dberr.MapRowsAffected(rows, err, errs.ErrSysMenuNotFound); err != nil {
			return dberr.MapUniqueViolation(err, dberr.SysMenusPathUniqueIdx, errs.ErrSysMenusPathUniqueIdx)
		}

		permission := sqlc.UpdateSysMenuPermissionCodeParams{
			MenuID: id,
			Code:   req.PermissionCode,
		}

		if err = q.UpdateSysMenuPermissionCode(ctx, permission); err != nil {
			return dberr.MapUniqueViolation(err, dberr.SysPermissionsCodeKey, errs.ErrPermissionCodeAlreadyExists)
		}

		return nil
	})

	if err != nil {
		return err
	}

	// 清理缓存
	_ = s.cache.DelByPrefix(ctx, cachekey.UserInfoPattern)

	return nil
}

func (s *SysMenuService) ListPage(ctx context.Context, p *common.Pagination) ([]sqlc.ListSysMenusRow, int64, error) {
	params := sqlc.ListSysMenusParams{
		Limit:  p.PageSize,
		Offset: (p.Page - 1) * p.PageSize,
	}

	total, err := s.store.CountSysMenus(ctx)

	if err != nil {
		return nil, 0, err
	}

	list, err := s.store.ListSysMenus(ctx, params)
	if err != nil {
		return nil, 0, err
	}

	return list, total, nil
}

func (s *SysMenuService) GetMenus(ctx context.Context) ([]sqlc.GetAllSysMenusRow, error) {
	return s.store.GetAllSysMenus(ctx)
}

func (s *SysMenuService) Delete(ctx context.Context, id int32) error {
	err := s.store.WithTx(ctx, func(q *sqlc.Queries) error {
		// 1. 根据菜单id, 找出对应的权限id
		permissionID, err := q.GetSysPermissionIDBySysMenuID(ctx, id)
		if err != nil {
			return dberr.MapNoRows(err, errs.ErrSysMenuNotFound)
		}
		// 2. 根据权限id 删除角色权限关联
		if err = q.DeleteSysRolePermissionByPermissionID(ctx, permissionID); err != nil {
			return err
		}
		// 3. 根据菜单id 删除sys_menu_permission关联表数据
		if err = q.DeleteSysMenuPermission(ctx, id); err != nil {
			return err
		}
		// 4. 根据权限id删除权限表数据
		if err = q.DeleteSysPermission(ctx, permissionID); err != nil {
			return err
		}
		// 5. 删除菜单
		rows, err := q.DeleteSysMenu(ctx, id)
		if err = dberr.MapRowsAffected(rows, err, errs.ErrSysMenuNotFound); err != nil {
			return err
		}

		return nil
	})

	if err != nil {
		return err
	}

	// 清理缓存
	_ = s.cache.DelByPrefix(ctx, cachekey.UserInfoPattern)
	return nil
}