将聚合函数作为参数传递

时间:2016-03-05 23:43:28

标签: c# linq lambda aggregate func

我有一个简单的对象:

public class Machine
{
    public string Name { get; set; }

    public int Power { get; set; }

    public int Type { get; set; }
}

然后是一个包含这些对象列表的类:

public class Aggregations
{
    private List<Machine> _machines;

    public Aggregations()
    {
        _machines = new List<Machine>
            {
                new Machine { Name = "XLR1", Power = 111, Type = 1 },
                new Machine { Name = "XLR2", Power = 222, Type = 1 },
                new Machine { Name = "XLR3", Power = 333, Type = 1 },
                new Machine { Name = "XLR4", Power = 444, Type = 1 },
                new Machine { Name = "XLR5", Power = 555, Type = 2 },
                new Machine { Name = "XLR6", Power = 666, Type = 2 }
            };           
    }
// ...
}

有两个函数可以返回具有特定条件的计算机列表:

    public IEnumerable<Machine> MaxPower(IEnumerable<Machine> machines)
    {
        var maxPowerMachinesPerType = new List<Machine>();

        var groups = machines.GroupBy(m => m.Type);
        foreach (var g in groups)
        {
            var max = g.Max(m => m.Power);
            var machine = g.First(m => m.Power == max);
            maxPowerMachinesPerType.Add(machine);
        }

        return maxPowerMachinesPerType;
    }


    public IEnumerable<Machine> MinPower(IEnumerable<Machine> machines)
    {
        var minPowerMachinesPerType = new List<Machine>();

        var groups = machines.GroupBy(m => m.Type);
        foreach (var g in groups)
        {
            var min = g.Min(m => m.Power);
            var machine = g.First(m => m.Power == min);
            minPowerMachinesPerType.Add(machine);
        }

        return minPowerMachinesPerType;
    }
}

如您所见,这两个功能几乎相同。只有&#34; max&#34;和&#34; min&#34;不同。

这些函数的调用如下:

IEnumerable<Machine> maxPowerMachines = MaxPower(_machines);
IEnumerable<Machine> minPowerMachines = MinPower(_machines);

因为我的实际程序稍微复杂一点,我想调用其他聚合函数,我想传递aggregate函数作为参数:(伪代码)

IEnumerable<Machine> maxPowerMachines = SuperFunction(_machines, m => m.Max);
IEnumerable<Machine> minPowerMachines = SuperFunction(_machines, m => m.Min);
IEnumerable<Machine> averagePowerMachines = SuperFunction(_machines, m => m.Average);

我希望你能得到这个意图。

2 个答案:

答案 0 :(得分:3)

由于MinMax具有相同的签名,即他们都使用IEnumerable<T>并生成T,您可以这样做:

public IEnumerable<Machine> SelectPower(
    IEnumerable<Machine> machines
,   Func<IEnumerable<int>,int> powerSelector
) {
    var res = new List<Machine>();
    var groups = machines.GroupBy(m => m.Type);
    foreach (var g in groups) {
        var targetPower =  powerSelector(g.Select(m => m.Power));
        var machine = g.First(m => m.Power == targetPower);
        res.Add(machine);
    }
    return res;
}

现在你可以像这样调用你的方法:

IEnumerable<Machine> maxPowerMachines = SuperFunction(_machines, m => m.Max());
IEnumerable<Machine> minPowerMachines = SuperFunction(_machines, m => m.Min());

答案 1 :(得分:0)

Func<T>听起来像你正在寻找的东西。主要类型<T>显示输入,最后<T>显示返回值,因此您正在寻找类似的内容:

    Func<IEnumerable<Machine>, IEnumerable<Machine>> aggregateFunction = MaxPower;


    //Now you can pass aggregateFunction around as a variable. You can call it like so:

    var machines = Aggregations();

    aggregateFunction.Invoke(machines);