有没有办法阻止或断言C＃中的意外递归

Question

我正在使用大型且复杂的事件驱动的代码体，并且有大量的机会意外地创建递归条件。

有时，递归条件是暂时的，应用程序会自行追赶，但即使那样通常也会造成不必要的延迟。 其他时候，它会产生一个堆栈溢出，当它发生在客户端站点时，通常很难调试。

我想有一种方法可以将允许递归的代码部分列入黑名单或白名单。 如果递归条件在DEV期间发生，那么我希望它断言，以便我可以更正代码。

我正在考虑的是让应用程序检查其自己的堆栈，以确保刚输入的方法不在堆栈中。

任何指针将不胜感激。

注意：这是针对Web应用程序的，但是我已经在多个环境中遇到了这一挑战。

Answer 1

您可以像这样检查堆栈：

[MethodImpl(MethodImplOptions.NoInlining)]
// optionally decorate with Conditional to only be used in Debug configuration
[Conditional("DEBUG")]
public static void FailIfCallerIsRecursive() {
    var trace = new StackTrace();
    // previous frame is the caller
    var caller = trace.GetFrame(1).GetMethod();
    // inspect the rest
    for (int i = 2; i < trace.FrameCount; i++) {
        // if found caller somewhere up the stack - throw
        if (trace.GetFrame(i).GetMethod() == caller)
            throw new Exception("Recursion detected");
    }            
}

然后将其称为开始：

void MyPotentiallyRecursiveMethod() {
    FailIfCallerIsRecursive()
}

但是请注意，它非常昂贵。 但是，由于您仅在开发（调试）配置中使用它-为什么不这样做。 您还可以对其进行一些修改，以仅在检测到特定级别的递归时才抛出该异常（因此，调用方在堆栈中出现X倍）。

Answer 2

你可以把RuntimeHelpers.EnsureSufficientExecutionStack方法 ，然后赶上InsufficientExecutionStackException如果下一个方法调用会导致（不开捕）时引发StackOverflowException 。

您可以为其创建扩展方法：

public static T EnsureSafeRecursiveCall<T>(this Func<T> method)
{
    try
    {
        RuntimeHelpers.EnsureSufficientExecutionStack();
        return method();
    }
    catch (InsufficientExecutionStackException ex)
    {
        string msg = $"{method.Method.Name} would cause a {nameof(StackOverflowException)} on the next call";
        Debug.Fail(msg);
        // logging here is essential here because Debug.Fail works only with debug
        throw new StackOverflowException(msg, ex); // wrap in new exception to avoid that we get into this catch again and again(note we are in a recursive call)
    }
}

现在，您的原始方法几乎保持不变：

public static IEnumerable<T> YourRecursiveMethod<T>(IEnumerable<T> seq)
{
    var method = new Func<IEnumerable<T>>(() => YourRecursiveMethod(seq));
    return method.EnsureSafeRecursiveCall();
}