recursionguard.h

// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
 
#pragma once
 
#include <bond/core/config.h>
#include <bond/core/exception.h>
#include <stdint.h>
 
namespace bond
{
namespace detail
{
 
template<class T>
struct RecursionGuardStaticsHolder
{
    static uint32_t maxDepth;

    thread_local static uint32_t currentDepth;
};
 
template<class T>
uint32_t RecursionGuardStaticsHolder<T>::maxDepth = 64;
 
template<class T>
thread_local uint32_t RecursionGuardStaticsHolder<T>::currentDepth;

class RecursionGuard : RecursionGuardStaticsHolder<void>
{
public:
    RecursionGuard()
    {
        uint32_t depth = currentDepth;
        if (depth >= maxDepth)
        {
            bond::ExceededMaxRecursionDepthException();
        }
 
        currentDepth = depth + 1;
    }
 
    ~RecursionGuard()
    {
        currentDepth--;
    }

    static void SetMaxDepth(uint32_t value)
    {
        maxDepth = value;
    }
};
 
} // namespace detail
} // namespace bond