递归linq查询

时间:2012-03-09 13:46:23

标签: c# .net linq entity-framework recursion

我有以下对象结构。

public class Study 
{
    public Guid? PreviousStudyVersionId { get; set; }
    public Guid StudyId { get; set; }
    //Other members left for brevity
}

首先使用实体​​框架代码保留它。

它产生一个像这样的表

PreviousStudyVersionId                  StudyId
EF90F9DC-C588-4136-8AAE-A00E010CE87B    E4315CFD-9638-4225-998E-A00E010CEEEC
NULL                                    1C965285-788A-4B67-9894-3D0D46949F11
1C965285-788A-4B67-9894-3D0D46949F11    7095B746-8D32-4CC5-80A9-A00E010CE0EA
7095B746-8D32-4CC5-80A9-A00E010CE0EA    EF90F9DC-C588-4136-8AAE-A00E010CE87B

现在我想查询所有studyId的递归。所以我提出了以下解决方案:

所以,当我在我的存储库GetAllStudyVersionIds(new Guid(“7095B746-8D32-4CC5-80A9-A00E010CE0EA”))中调用该方法时,它将返回我所有4个studyId。

    public IEnumerable<Guid> GetAllStudyVersionIds(Guid studyId)
    {
        return SearchPairsForward(studyId).Select(s => s.Item1)
            .Union(SearchPairsBackward(studyId).Select(s => s.Item1)).Distinct();
    }

    private IEnumerable<Tuple<Guid, Guid?>> SearchPairsForward(Guid studyId)
    {
        var result =
            GetAll().Where(s => s.PreviousStudyVersionId == studyId).ToList()
            .Select(s => new Tuple<Guid, Guid?>(s.StudyId, s.PreviousStudyVersionId));
        result = result.Traverse(a => SearchPairsForward(a.Item1));
        return result;
    }

    private IEnumerable<Tuple<Guid, Guid?>> SearchPairsBackward(Guid studyId)
    {
        var result = GetAll().Where(s => s.StudyId == studyId).ToList()
            .Select(s => new Tuple<Guid, Guid?>(s.StudyId, s.PreviousStudyVersionId));
        result = result.Traverse(a => a.Item2.HasValue ? SearchPairsBackward(a.Item2.Value) : Enumerable.Empty<Tuple<Guid, Guid?>>());
        return result;
    }

这是我的扩展方法的实现。

public static class MyExtensions
{
    public static IEnumerable<T> Traverse<T>(this IEnumerable<T> source, Func<T, IEnumerable<T>> fnRecurse)
    {
        foreach (var item in source)
        {
            yield return item;
            var seqRecurse = fnRecurse(item);
            if (seqRecurse == null) continue;
            foreach (var itemRecurse in Traverse(seqRecurse, fnRecurse))
            {
                yield return itemRecurse;
            }
        }
    }
}

有没有办法让它更接近数据库(IQueryable)并优化此代码。

1 个答案:

答案 0 :(得分:1)

我通过制作一个递归表函数来完成此操作,该函数根据id拉出所有父记录,然后创建一个视图以生成每个记录和所有父项的列表。然后我在EF模型中使用它并关联以便能够与LINQ一起使用。