-
Notifications
You must be signed in to change notification settings - Fork 1
/
AutoScope.h
77 lines (61 loc) · 1.61 KB
/
AutoScope.h
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#pragma once
#include <functional>
namespace zs
{
template <typename TLambda>
struct AutoScope
{
template <typename TLambdaRhs>
friend struct AutoScope;
AutoScope() noexcept = delete;
AutoScope(const AutoScope&) noexcept = delete;
AutoScope(AutoScope&& rhs) noexcept :
function_(std::move(rhs.function_))
{
rhs.function_ = {};
}
AutoScope(TLambda&& lambda) noexcept :
function_(std::move(lambda))
{
}
~AutoScope() noexcept(std::is_nothrow_invocable_v<TLambda>)
{
if (!function_)
return;
std::invoke(function_);
}
AutoScope& operator=(const AutoScope&) noexcept = delete;
AutoScope& operator=(AutoScope&& rhs) noexcept
{
if (this == &rhs)
return *this;
function_ = std::move(rhs.function_);
rhs.function_ = {};
return *this;
}
template <typename TLambdaRhs>
AutoScope& operator=(AutoScope<TLambdaRhs>&& rhs) noexcept
{
function_ = std::move(rhs.function_);
rhs.function_ = {};
return *this;
}
template <typename TLambdaRhs>
AutoScope& operator=(TLambdaRhs&& lambda) noexcept
{
function_ = std::move(lambda);
return *this;
}
void cancel() noexcept
{
function_ = {};
}
protected:
std::function<void()> function_;
};
template <typename TLambda>
[[nodiscard]] inline decltype(auto) on_scope_exit(TLambda&& lambda) noexcept(std::is_nothrow_invocable_v<TLambda>) //(std::is_nothrow_move_constructible<std::remove_const_t<TLambda>>::value)
{
return AutoScope{ std::move(lambda) };
}
} // namespace zs