opms_backend/opms_admin/app/service/sys_menu.go

// ============================================================================
// This is auto-generated by gf cli tool only once. Fill this file as you wish.
// ============================================================================

package service

import (
	"context"
	"dashoo.cn/common_definition/admin/menu_def"
	"dashoo.cn/micro/app/dao"
	"dashoo.cn/micro/app/model"
	"dashoo.cn/opms_libary/micro_srv"
	"errors"
	"github.com/gogf/gf/errors/gerror"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/util/gconv"
)

type MenuService struct {
	Dao *dao.SysMenuDao
}

func NewMenuService(ctx context.Context) (*MenuService, error) {
	menu := new(MenuService)
	// 获取租户码
	tenant, err := micro_srv.GetTenant(ctx)
	if err != nil {
		return nil, err
	}
	g.Log().Info("Received Menu.Login request @ " + tenant)
	menu.Dao = dao.NewSysMenuDao(tenant)
	return menu, err
}

// Add 添加菜单项
func (s MenuService) Add(ctx context.Context, param *model.SysMenuReq) (id int64, err error) {
	// 菜单类型 D目录 M菜单 F按钮或功能 I接口
	if param.MenuType == "F" && param.Perms == "" {
		return -1, errors.New("权限标识不能为空！")
	}
	if param.Perms != "" {
		count, err := s.Dao.M.Count("Perms", param.Perms)
		if err != nil {
			return -1, err
		}
		if count > 0 {
			return -1, errors.New("权限标识已存在。")
		}
	}
	if param.MenuType == "D" {
		param.Component = "Layout"
	}

	var entity *model.SysMenu
	if err := gconv.Struct(param, &entity); err != nil {
		return -1, err
	}
	// 用户基础信息
	//userInfo, err := micro_srv.GetUserInfo(ctx)
	//if err != nil {
	//	return 0, err
	//}
	SetCreatedInfo(entity, int(1), "userInfo.UserName")
	if result, err := s.Dao.M.Insert(entity); err != nil {
		return -1, err
	} else {
		id, err := result.LastInsertId()
		if err != nil {
			return -1, err
		}
		//err = s.initPaths(int(id), param.ParentId)
		//if err != nil {
		//	return -1, err
		//}
		return id, nil
	}
}

// UpdateById 根据ID更新菜单项
func (s MenuService) UpdateById(ctx context.Context, param *model.SysMenuReq) error {
	// 菜单类型 D目录 M菜单 F按钮或功能 I接口
	if param.MenuType == "F" && param.Perms == "" {
		return gerror.New("权限标识不能为空！")
	}
	if param.MenuType == "D" {
		param.Component = "Layout"
	}
	var entity *model.SysMenu
	if err := gconv.Struct(param, &entity); err != nil {
		g.Log().Error(err)
		return gerror.New("数据转换异常，请重新尝试")
	}

	db := s.Dao.M
	menu, err := db.FindOne("Id", entity.Id)
	if err != nil {
		return err
	}
	if menu.IsEmpty() {
		return gerror.New("菜单不存在")
	}

	if entity.Perms != "" {
		exist, err := db.Where("Id !=?", entity.Id).Where("perms", entity.Perms).FindOne()
		if err != nil {
			return err
		}
		if !exist.IsEmpty() {
			return gerror.New("权限标识已存在。")
		}
	}
	// 用户基础信息
	//userInfo, err := micro_srv.GetUserInfo(ctx)
	//if err != nil {
	//	return err
	//}
	SetUpdatedInfo(entity, int(1), "userInfo.UserName")
	_, err = db.FieldsEx(UpdateFieldEx).Where("Id", entity.Id).Update(entity)
	if err != nil {
		return err
	}
	//err = s.savePaths(ctx, entity.Id, entity.ParentId, menu["Paths"].String())
	return err
}

func (s MenuService) Delete(id int) (err error) {
	db := s.Dao.M
	record, err := db.FindOne("Id", id)
	if err != nil {
		return err
	}
	if record.IsEmpty() {
		return gerror.New("菜单不存在")
	}
	_, err = db.Delete("Id", id)
	//delParentId := record["ParentId"].String()
	//if err != nil {
	//	return err
	//}
	//res, _ := db.Data("parent_id", delParentId).Where("parent_id", id).Update()
	//
	//if res != nil {
	//	if delParentId == "0" {
	//		updatesql := "Paths = replace(Paths, '" + strconv.Itoa(id) + "/" + "', '')"
	//		_, err = db.Update(updatesql, " Paths like '"+strconv.Itoa(id)+"/%'")
	//	} else {
	//		updatesql := "Paths = replace(Paths, '" + record["Paths"].String() + "', '" + delParentId + "')"
	//		_, err = db.Update(updatesql, " Paths like '"+record["Paths"].String()+"/%'")
	//	}
	//}
	return err
}

// GetEntity 获取菜单项实体
func (s MenuService) GetEntity(id int64) (*model.SysMenu, error) {
	one, err := s.Dao.M.FindOne(id)
	if err != nil {
		return nil, err
	}
	if one.IsEmpty() {
		return nil, nil
	}
	var entity *model.SysMenu
	if err = one.Struct(&entity); err != nil {
		return nil, err
	}
	return entity, nil
}

// GetList 获取菜单列表
func (s MenuService) GetList(param *menu_def.SelectReq) ([]model.SysMenu, error) {
	db := s.Dao.M
	if param != nil {
		if param.Title != "" {
			db = db.Where("menu_name like ?", "%"+param.Title+"%")
		}
		if param.Visible != "" {
			db = db.Where("visible", param.Visible)
		}
		if param.Order != "" {
			sort := "ASC"
			if !param.IsAsc {
				sort = "DESC"
			}
			db = db.Order(param.Order + " " + sort)
		} else {
			db = db.Order("sort")
		}
	}

	result := make([]model.SysMenu, 0)
	if err := db.FindScan(&result); err != nil {
		return nil, err
	}
	return result, nil
}

// GetMenuTree 获取菜单树
func (s MenuService) GetMenuTree() ([]model.MenuTree, error) {
	db := s.Dao.M
	menuList := make([]model.SysMenu, 0)
	err := db.Where("visible = 10").Where("menu_type IN (?)", g.Slice{"M", "C"}).Order("sort").Scan(&menuList)

	treeMap := make(map[int][]model.MenuTree, 0)
	for _, v := range menuList {
		treeMap[v.ParentId] = append(treeMap[v.ParentId], v.ConvMenuTree())
	}
	menuTree := treeMap[0]
	for i := 0; i < len(menuTree); i++ {
		s.getChildrenList(&menuTree[i], treeMap)
	}
	return menuTree, err
}

// getChildrenList 获取菜单的子菜单
func (s MenuService) getChildrenList(menu *model.MenuTree, treeMap map[int][]model.MenuTree) {
	if value, ok := treeMap[menu.Id]; ok {
		menu.Children = value
		for i := 0; i < len(menu.Children); i++ {
			s.getChildrenList(&menu.Children[i], treeMap)
		}
	}
}