Method GetMenuTree

internal/repository/dao/menu.go:278–322

GetMenuTree 获取菜单树形结构

(ctx context.Context)

// GetMenuTree 获取菜单树形结构
func (m *menuDAO) GetMenuTree(ctx context.Context) ([]*Menu, error) {
	// 预分配合适的初始容量
	menus := make([]*Menu, 0, 50)

	// 使用索引字段优化查询
	if err := m.db.WithContext(ctx).
		Select("id, name, parent_id, path, component, icon, sort_order, route_name, hidden, create_time, update_time").
		Where("is_deleted = ?", 0).
		Order("sort_order ASC, id ASC").
		Find(&menus).Error; err != nil {
		return nil, fmt.Errorf("查询菜单列表失败: %v", err)
	}

	// 预分配map容量
	menuMap := make(map[int]*Menu, len(menus))
	rootMenus := make([]*Menu, 0, len(menus)/3) // 假设大约1/3的菜单是根菜单

	// 第一次遍历,建立ID到菜单的映射
	for _, menu := range menus {
		if menu == nil {
			continue
		}
		menu.Children = make([]*Menu, 0, 4) // 预分配子菜单切片,假设平均4个子菜单
		menuMap[menu.ID] = menu
	}

	// 第二次遍历,构建树形结构
	for _, menu := range menus {
		if menu == nil {
			continue
		}
		if menu.ParentID == 0 {
			rootMenus = append(rootMenus, menu)
		} else {
			if parent, exists := menuMap[menu.ParentID]; exists {
				parent.Children = append(parent.Children, menu)
			} else {
				// 如果找不到父节点,作为根节点处理
				rootMenus = append(rootMenus, menu)
			}
		}
	}

	return rootMenus, nil
}