ClickHouse/src/IO/Expect404ResponseScope.cpp at master · StackRadiusForks/ClickHouse · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#include <IO/Expect404ResponseScope.h>

#include <base/defines.h>

namespace DB
{

thread_local size_t expected_404_scope_count = 0;

Expect404ResponseScope::Expect404ResponseScope()
    : initial_thread_id(std::this_thread::get_id())
{
    ++expected_404_scope_count;
}

Expect404ResponseScope::~Expect404ResponseScope()
{
    // check that instance is destroyed in the same thread
    chassert(initial_thread_id == std::this_thread::get_id());
    chassert(expected_404_scope_count);
    --expected_404_scope_count;
}

bool Expect404ResponseScope::is404Expected()
{
    return expected_404_scope_count != 0;
}

}