最优雅的方式来处理第一个IEnumerable项目不同

时间:2011-07-01 16:44:55

标签: linq ienumerable skip

以不同于其他项目的方式处理第一个IEnumerable项目的最优雅方式是什么,而不必在每次迭代时进行测试?

对每次迭代进行测试,它看起来像这样:

// "first item done" flag
bool firstDone = false;

// items is an IEnumerable<something>
foreach (var item in items)
{
    if (!firstDone)
    {
        // do this only once
        ProcessDifferently(item);
        firstDone = true;
        continue;
    }

    ProcessNormally(item);
}

如果我这样做:

ProcessDifferently(items.First());
ProcessNormally(items.Skip(1)); // this calls `items.GetEnumerator` again

它会调用GetEnumerator两次,我想避免(例如,对于Linq-to-Sql案例)。

如果您需要对代码进行多次操作,您会怎么做?

3 个答案:

答案 0 :(得分:12)

如果我需要在几个地方进行,我会提取一个方法:

public void Process<T>(IEnumerable<T> source,
                       Action<T> firstAction,
                       Action<T> remainderAction)
{
    // TODO: Argument validation
    using (var iterator = source.GetEnumerator())
    {
        if (iterator.MoveNext())
        {
            firstAction(iterator.Current);
        }
        while (iterator.MoveNext())
        {
            remainderAction(iterator.Current);
        }
    }
}

被称为:

Process(items, ProcessDifferently, ProcessNormally);

还有其他选择,但这实际上取决于具体情况。

答案 1 :(得分:1)

这是另一种方式:

    private static void Main(string[] args)
    {
        var testdata = new[] { "a", "b", "c", "d", "e" };

        var action = FirstThenRest<string>(
            s => Console.WriteLine("First: " + s),
            s => Console.WriteLine("Rest: " + s));

        foreach (var s in testdata)
            action(s);
    }

    public static Action<T> FirstThenRest<T>(Action<T> first, Action<T> rest)
    {
        Action<T> closure = t =>
                            {
                                first(t);
                                closure = rest;
                            };

        return t => closure(t);
    }

输出:

First: a
Rest: b
Rest: c
Rest: d
Rest: e

没有条件。 :d

编辑:“头”和“尾巴”可能是更好的术语,但我现在懒得去改变它。

答案 2 :(得分:0)

你可以用老式的方式做到:

var itemsList = items.ToList();
ProcessDifferently(itemsList[0]);

for(int i=1;i<itemsList.Count;i++)
{
   ProcessNormally(itemsList[i]);
}