go-server/internal/services/admin/service_mymenu.go

package admin

import (
	"github.com/xinliangnote/go-gin-api/internal/pkg/core"
	"github.com/xinliangnote/go-gin-api/internal/repository/mysql"
	"github.com/xinliangnote/go-gin-api/internal/repository/mysql/admin_menu"
	"github.com/xinliangnote/go-gin-api/internal/repository/mysql/menu"
)

type SearchMyMenuData struct {
	AdminId int32 `json:"admin_id"` // 管理员ID
}

type ListMyMenuData struct {
	Id   int32  `json:"id"`   // ID
	Pid  int32  `json:"pid"`  // 父类ID
	Name string `json:"name"` // 菜单名称
	Link string `json:"link"` // 链接地址
	Icon string `json:"icon"` // 图标
}

func (s *service) MyMenu(ctx core.Context, searchData *SearchMyMenuData) (menuData []ListMyMenuData, err error) {
	adminMenuQb := admin_menu.NewQueryBuilder()
	if searchData.AdminId != 0 {
		adminMenuQb.WhereAdminId(mysql.EqualPredicate, searchData.AdminId)
	}

	adminMenuListData, err := adminMenuQb.
		OrderById(false).
		QueryAll(s.db.GetDbR().WithContext(ctx.RequestContext()))
	if err != nil {
		return nil, err
	}

	if len(adminMenuListData) <= 0 {
		return
	}

	menuQb := menu.NewQueryBuilder()
	menuQb.WhereIsDeleted(mysql.EqualPredicate, -1)
	menuListData, err := menuQb.
		OrderBySort(true).
		QueryAll(s.db.GetDbR().WithContext(ctx.RequestContext()))
	if err != nil {
		return nil, err
	}

	if len(menuListData) <= 0 {
		return
	}

	for _, menuAllV := range menuListData {
		for _, v := range adminMenuListData {
			if menuAllV.Id == v.MenuId {
				data := ListMyMenuData{
					Id:   menuAllV.Id,
					Pid:  menuAllV.Pid,
					Name: menuAllV.Name,
					Link: menuAllV.Link,
					Icon: menuAllV.Icon,
				}

				menuData = append(menuData, data)
			}
		}
	}

	return
}