LINQ:遍历层次结构以检索父层次结构

时间:2019-12-17 08:39:02

标签: linq parent-child hierarchical-data ef-core-3.0

我有一些数据表示同一实体上的父子关系。给定一个节点,我需要找到它的整个上级层次结构(父级,祖父母级,曾祖父母级等等)。

我的实体是这样的:

 public partial class Location{
        public int LocationId { get; set; }
        public int? FkParentLocationId { get; set; }
        ..... more properties here.......
        public virtual Location FkParentLocation { get; set; }
        public virtual ICollection<Location> InverseFkParentLocation { get; set; }
}

我指的是建议here的“遍历遍历”实现,但是当您沿层次结构向下浏览时,它可以工作。如何使用LINQ检索较高的层次结构?

样本数据:

            List<Location> locations = new List<Location> {
            new Location { LocationId = 5, FkParentLocationId = 3, LocationName = "Windsor", LocationDisplayName = "Windsor"},
            new Location { LocationId = 15, FkParentLocationId = 3, LocationName = "Hampshire", LocationDisplayName = "Hampshire" },
            new Location { LocationId = 12, FkParentLocationId = 3, LocationName = "Sussex", LocationDisplayName = "Sussex"},
            new Location { LocationId = 13, FkParentLocationId = 3, LocationName = "Willowood", LocationDisplayName = "Willowood"},
            new Location { LocationId = 1, FkParentLocationId = 3, LocationName = "Gerbshire", LocationDisplayName = "Gerbshire"},
            new Location { LocationId = 3, FkParentLocationId = 2, LocationName = "Lincoln", LocationDisplayName = "Lincoln"},
            new Location { LocationId = 2, LocationName = "Mains", LocationDisplayName = "Mains" }            };

预期的输出:给定位置ID:5,我应该得到一个包含位置3和2的列表(因为它们是父级)。

1 个答案:

答案 0 :(得分:0)

这是您可以使用的一种方法,通过控制台应用程序进行了演示。从Jon Skeet大量借用。

using System;
using System.Collections.Generic;
using System.Linq;

namespace Locations
{
    public partial class Location 
    {
        public int LocationId { get; set; }
        public int? FkParentLocationId { get; set; }
        public virtual Location FkParentLocation { get; set; }
        public virtual ICollection<Location> InverseFkParentLocation { get; set; }
        public string LocationName { get; set; }
        public string LocationDisplayName { get; set; }      
    }  

    class Program
    {
        static void Main(string[] args)
        {
            List<Location> locations = new List<Location> {
            new Location { LocationId = 5, FkParentLocationId = 3, LocationName = "Windsor", LocationDisplayName = "Windsor"},
            new Location { LocationId = 15, FkParentLocationId = 3, LocationName = "Hampshire", LocationDisplayName = "Hampshire" },
            new Location { LocationId = 12, FkParentLocationId = 3, LocationName = "Sussex", LocationDisplayName = "Sussex"},
            new Location { LocationId = 13, FkParentLocationId = 3, LocationName = "Willowood", LocationDisplayName = "Willowood"},
            new Location { LocationId = 1, FkParentLocationId = 3, LocationName = "Gerbshire", LocationDisplayName = "Gerbshire"},
            new Location { LocationId = 3, FkParentLocationId = 2, LocationName = "Lincoln", LocationDisplayName = "Lincoln"},
            new Location { LocationId = 2, LocationName = "Mains", LocationDisplayName = "Mains" }            };


            var result  = GetAncestorsIds(locations, 5);
            foreach (var id in result)
            {
                System.Console.WriteLine(id);
            }          
        }

        private static IEnumerable<int> GetAncestorsIds(List<Location> locations, int id)
        {
            Location location = locations.SingleOrDefault(l => l.LocationId == id);
            if(location != null)
            {   
                while(location != null && location.FkParentLocationId != null)
                {
                    location = locations.SingleOrDefault(l => l.LocationId == location.FkParentLocationId);
                    if(location != null)
                    {
                        yield return location.LocationId;
                    }
                }
            }   
        }
    }
}

这种方法可以变成您自己的Linq扩展。这是它的外观。

public static class MyExtensions
{
    public static IEnumerable<int> GetAncestorIds<TSource>(this IEnumerable<TSource> source, Func<TSource, int> pk, Func<TSource, int?> fk, int id)
    {
        TSource currentObj = source.SingleOrDefault(s => pk(s) == id);

        while(currentObj != null && fk(currentObj) != null)
        {
            currentObj = source.SingleOrDefault(s => pk(s) == fk(currentObj));
            if(currentObj != null)
            {
                yield return pk(currentObj);
            }
        }
    }
}

,然后为您的情况调用此方法,您将执行此操作

var result = locations.GetAncestorIds(l => l.LocationId, l => l.FkParentLocationId, 5);