Linq下有一个很是实用的SelectMany方法，不少人却不会用

在平时开发中常常会看到有些朋友或者同事在写代码时会充斥着各类for，foreach，这种程式代码太多的话阅读性特别差，并且还显得特别累赘，其实在FCL中有不少帮助咱们提升阅读感的方法，而现实中不少人不会用或者说不知道，这篇我就跟你们聊一聊。

一：SelectMany

这个方法绝对是提升开发速度的一大利器，有太多的业务场景须要使用这个函数，举一个我实际应用场景，商家按照年份和客户类型预先设置一些标签，而后让系统跑一下它的各自标签到底有多少人？

1. 定义Model

为了方便演示，这里作了一下简化代码，只有一个字典，key表示年份，value：就是该年份的多组标签。

public class EstimateModel
{
    public int ShopID { get; set; }

    //key：年份
    public Dictionary<string, List<TagCrowdFilterModel>> YearCrowdFilterDict { get; set; }
}

public class TagCrowdFilterModel
{
    /// <summary>
    /// 筛选条件
    /// </summary>
    public string CrowdFiter { get; set; }

    /// <summary>
    /// 获取人数
    /// </summary>
    public int TotalCustomerCount { get; set; }
}

为了更加清晰，我决定再填充一下数据

public static void Main(string[] args)
    {
        var estimateModel = new EstimateModel()
        {
            ShopID = 1,
            YearCrowdFilterDict = new Dictionary<string, List<TagCrowdFilterModel>>()
            {
                {
                    "17年",new List<TagCrowdFilterModel>()
                           {
                             new TagCrowdFilterModel(){ CrowdFiter="between 10 and 20" },
                             new TagCrowdFilterModel(){ CrowdFiter=" a<10  || a>30" },
                           }
                },
                {
                    "18年",new List<TagCrowdFilterModel>()
                           {
                             new TagCrowdFilterModel(){ CrowdFiter="between 100 and 200" },
                             new TagCrowdFilterModel(){ CrowdFiter=" a<100  || a>300" },
                           }
                },
                {
                    "19年",new List<TagCrowdFilterModel>()
                           {
                             new TagCrowdFilterModel(){ CrowdFiter="between 1000 and 2000" },
                             new TagCrowdFilterModel(){ CrowdFiter=" a<1000  || a>3000" },
                           }
                }
            }
        };
    }

    public static int GetCustomerID(string crowdfilter)
    {
        return BitConverter.ToInt32(Guid.NewGuid().ToByteArray(), 0);
    }

2. 实现需求

需求也很简单，就是依次获取 TagCrowdFilterModel 中的 CrowdFiter 字段再调用GetCustomerID方法把人数赋值给TotalCustomerCount便可，这么简单的需求，若是让你来搞定，你该怎么实现这个逻辑？没错，不少人可能就是两个foreach搞定。

foreach (var year in estimateModel.YearCrowdFilterDict.Keys)
    {
        var yearCrowdFitlerList = estimateModel.YearCrowdFilterDict[year];

        foreach (var crowdFitler in yearCrowdFitlerList)
        {
            crowdFitler.TotalCustomerCount = GetCustomerID(crowdFitler.CrowdFiter);
        }
    }

看似代码也很清爽，但现实哪有这么好的事情，真实状况是年份上可能还要套上一个客户类型，客户类型之上再套一个商品，商品之上再套一个商家，这样很深的层级你就须要多达3个foreach，4个foreach甚至5个foreach才能搞定，再放张图给你们看看，是否是看着头大...

3. 优化办法

若是你会selectMany，那只须要一个链式写法就能够搞定，是否是简单粗暴，虽然性能比不上命令式写法，但可读性和观赏性真的上了几个档次。

estimateModel.YearCrowdFilterDict.SelectMany(m => m.Value).ToList().ForEach(m =>
            m.TotalCustomerCount = GetCustomerID(m.CrowdFiter)
        );

二：原理探究

1. msdn解释

将序列的每一个元素投影到 IEnumerable，并将结果序列合并为一个序列，并对其中每一个元素调用结果选择器函数，连接：https://docs.microsoft.com/zh-cn/dotnet/api/system.linq.enumerable.selectmany?view=netframework-4.8 有了上面的案例解释，再看msdn的这句话，我想你应该不折不扣的明白了selectMany怎么使用。

2. 翻查源码

宏观上明白了，接下来用ILSpy去查下微观代码，到底这玩意是怎么实现的。

public static IEnumerable<TResult> SelectMany<TSource, TResult>(this IEnumerable<TSource> source, Func<TSource, IEnumerable<TResult>> selector)
{
    if (source == null)
    {
        throw Error.ArgumentNull("source");
    }
    if (selector == null)
    {
        throw Error.ArgumentNull("selector");
    }
    return SelectManyIterator(source, selector);
}

private static IEnumerable<TResult> SelectManyIterator<TSource, TResult>(IEnumerable<TSource> source, Func<TSource, IEnumerable<TResult>> selector)
{
    foreach (TSource item in source)
    {
        foreach (TResult item2 in selector(item))
        {
            yield return item2;
        }
    }
}

你们仔细体会下这两个foreach，尤为是第二个foreach，其中的selector(item)不就是年份下的标签集合吗？再遍历这个集合把每个item返回出去，返回值是IEnumerable，这得益于yield语法糖，它本质上就是一个编译器封装好的迭代类，作到了一个延迟，按需执行，后面我会专门分享一下yield，颇有意思

好了，本篇就说到这里，但愿对你有帮助。