属性随产量而丢失

Question

我正在尝试将一些代码从 IList 转换为 IEnumerable：

[Something(123)]
public IEnumerable<Foo> GetAllFoos()
{
  SetupSomething();

  DataReader dr = RunSomething();
  while (dr.Read())
  {
    yield return Factory.Create(dr);
  }
}

问题是，SetupSomething() 来自基类并使用：

Attribute.GetCustomAttribute(
    new StackTrace().GetFrame(1).GetMethod(), typeof(Something))

yield 最终创建 MoveNext()，MoveNext() 调用 SetupSomething()，而 MoveNext() 没有 [Something(123)] 属性。

我无法更改基类，所以看来我不得不继续使用IList 或手动实现IEnumerable（并将属性添加到MoveNext()）。

在这种情况下，还有其他方法可以让 yield 工作吗？

Answer 1

如果您需要堆栈框架功能，则不能使用迭代器（yield）。正如您所发现的，这会将您的方法重写为实现IEnumerable<T> 的自定义类。

但是，您可以轻松地将其修改为：

[Something(123)]
public IEnumerable<Foo> GetAllFoos()
{
  SetupSomething();

  List<Foo> results = new List<Foo>();
  DataReader dr = RunSomething();
  while (dr.Read())
  {
    results.Add(Factory.Create(dr));
  }
  return results;
}

你失去了迭代器的延迟执行，但它会正常工作。

Answer 2

您可以将该方法包装在另一个方法中，该方法执行所有必需的预处理：

[Something(123)]
public IEnumerable<Foo> GetAllFoos()
{
    SetupSomething();
    return GetAllFoosInternal();
}

private IEnumerable<Foo> GetAllFoosInternal()
{
    DataReader dr = RunSomething();
    while (dr.Read())
    {
        yield return Factory.Create(dr);
    }
}

Answer 3

你能像这样拆分你的方法吗？

[Something(123)]
public void GetAllFoosHelper()
{
  SetupSomething(); 
}

public IEnumerable<Foo> GetAllFoos() 
{ 
  GetAllFoosHelper();

  DataReader dr = RunSomething(); 
  while (dr.Read()) 
  { 
    yield return Factory.Create(dr); 
  } 
} 

Answer 4

根据您的描述，听起来问题在于 SetupSomething 仅查看堆栈跟踪上的直接调用者。如果它再往上看一点（调用者的调用者），它会找到您的 GetAllFocus 调用和所需的属性。

我不记得了，但是如果 yield 只是因为你的类还没有实现它而创建了一个 MoveNext() 实现，也许你可以实现你自己的 MoveNext，把属性放在它上面，并且 yield 会找到并使用您的 MoveNext()？只是一个疯狂的猜测。

Answer 5

我可能遗漏了一些东西，但我无法理解在这里使用属性。你不妨这样写：

public IEnumerable<Foo> GetAllFoos()
{
  SetupSomething(123);
  // etc..
}

也快了一大截。更安全的是，当 JIT 编译器内联 SetupSomething() 时，你就死定了。