zmkgC/internal/app/system/controller/app_roles.go

// ==========================================================================
// GFast自动生成controller操作代码。
// 生成日期：2024-05-28 15:11:36
// 生成路径: internal/app/system/controller/app_roles.go
// 生成人：gfast
// desc:app角色
// company:云南奇讯科技有限公司
// ==========================================================================

package controller

import (
	"context"
	"database/sql"
	"errors"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/tiger1103/gfast/v3/api/v1/system"
	"github.com/tiger1103/gfast/v3/internal/app/system/dao"
	"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
	"github.com/tiger1103/gfast/v3/internal/app/system/model/entity"
	"github.com/tiger1103/gfast/v3/internal/app/system/service"
	"github.com/tiger1103/gfast/v3/library/liberr"
)

type appRolesController struct {
	BaseController
}

var AppRoles = new(appRolesController)

// List 列表
func (c *appRolesController) List(ctx context.Context, req *system.AppRolesSearchReq) (res *system.AppRolesSearchRes, err error) {
	res, err = service.AppRoles().List(ctx, req)
	return
}

// Get 获取app角色
func (c *appRolesController) Get(ctx context.Context, req *system.AppRolesGetReq) (res *system.AppRolesGetRes, err error) {
	res = new(system.AppRolesGetRes)
	res.AppRolesInfoRes, err = service.AppRoles().GetByRoleId(ctx, req.RoleId)
	return
}

// Add 添加app角色
func (c *appRolesController) Add(ctx context.Context, req *system.AppRolesAddReq) (res *system.AppRolesAddRes, err error) {
	err = service.AppRoles().Add(ctx, req)
	return
}

// Edit 修改app角色
func (c *appRolesController) Edit(ctx context.Context, req *system.AppRolesEditReq) (res *system.AppRolesEditRes, err error) {
	err = service.AppRoles().Edit(ctx, req)
	return
}

// Delete 删除app角色
func (c *appRolesController) Delete(ctx context.Context, req *system.AppRolesDeleteReq) (res *system.AppRolesDeleteRes, err error) {
	err = service.AppRoles().Delete(ctx, req.RoleIds)
	return
}

// AddMenu 添加角色时绑定菜单
func (c *appRolesController) AddMenu(ctx context.Context, req *system.AppRolesAddMenuReq) (*system.AppRolesAddMenuRes, error) {
	err := g.Try(ctx, func(ctx context.Context) {
		var roleID uint

		// 判断角色名是否存在
		var appRoles entity.AppRoles
		if err := dao.AppRoles.Ctx(ctx).Where(dao.AppRoles.Columns().RoleName, req.RoleName).Scan(&appRoles); errors.Is(err, sql.ErrNoRows) {
			roleId, err := dao.AppRoles.Ctx(ctx).Data(do.AppRoles{
				RoleName: req.RoleName,
			}).InsertAndGetId()
			liberr.ErrIsNil(ctx, err, "添加角色失败")
			roleID = uint(roleId)
		} else {
			roleID = appRoles.RoleId
		}

		if len(req.MenuIds) > 0 {
			insertData := make([]do.AppRoleMenus, 0, len(req.MenuIds))
			for _, menuId := range req.MenuIds {
				insertData = append(insertData, do.AppRoleMenus{
					RoleId: roleID,
					MenuId: menuId,
				})
			}

			// 删除原有的角色菜单
			_, err := dao.AppRoleMenus.Ctx(ctx).Where("role_id", roleID).Delete()
			liberr.ErrIsNil(ctx, err, "删除角色菜单失败")
			_, err = dao.AppRoleMenus.Ctx(ctx).Where("role_id", roleID).Data(insertData).Insert()
			liberr.ErrIsNil(ctx, err, "添加角色菜单失败")
		}
	})

	return &system.AppRolesAddMenuRes{}, err
}

// 获取一个角色的菜单
func (c *appRolesController) GetRoleMenus(ctx context.Context, req *system.AppRolesGetMenuReq) (*system.AppRolesGetMenuRes, error) {
	res := new(system.AppRolesGetMenuRes)
	err := g.Try(ctx, func(ctx context.Context) {
		err := dao.AppRoles.Ctx(ctx).WithAll().Where(dao.AppRoles.Columns().RoleId, req.RoleId).Scan(&res.List)

		// 		SELECT m.menu_id, m.menu_name
		// FROM app_menus m
		// JOIN app_role_menus rm ON m.menu_id = rm.menu_id
		// WHERE rm.role_id = 3;

		dao.AppMenus.Ctx(ctx).As("m").Fields("m.menu_id, m.menu_name").
			InnerJoin("app_role_menus rm", "m.menu_id = rm.menu_id").Where("rm.role_id", req.RoleId).Scan(&res.List.Menus)

		liberr.ErrIsNil(ctx, err, "获取信息失败")
	})
	return res, err
}