2 * Copyright 2016 Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
22 #include <type_traits>
25 #include <folly/Preprocessor.h>
26 #include <folly/detail/UncaughtExceptionCounter.h>
31 * ScopeGuard is a general implementation of the "Initialization is
32 * Resource Acquisition" idiom. Basically, it guarantees that a function
33 * is executed upon leaving the currrent scope unless otherwise told.
35 * The makeGuard() function is used to create a new ScopeGuard object.
36 * It can be instantiated with a lambda function, a std::function<void()>,
37 * a functor, or a void(*)() function pointer.
40 * Usage example: Add a friend to memory iff it is also added to the db.
42 * void User::addFriend(User& newFriend) {
43 * // add the friend to memory
44 * friends_.push_back(&newFriend);
46 * // If the db insertion that follows fails, we should
47 * // remove it from memory.
48 * // (You could also declare this as "auto guard = makeGuard(...)")
49 * ScopeGuard guard = makeGuard([&] { friends_.pop_back(); });
51 * // this will throw an exception upon error, which
52 * // makes the ScopeGuard execute UserCont::pop_back()
53 * // once the Guard's destructor is called.
54 * db_->addFriend(GetName(), newFriend.GetName());
56 * // an exception was not thrown, so don't execute
61 * Examine ScopeGuardTest.cpp for some more sample usage.
64 * Andrei's and Petru Marginean's CUJ article:
65 * http://drdobbs.com/184403758
66 * and the loki library:
67 * http://loki-lib.sourceforge.net/index.php?n=Idioms.ScopeGuardPointer
68 * and triendl.kj article:
69 * http://www.codeproject.com/KB/cpp/scope_guard.aspx
71 class ScopeGuardImplBase {
73 void dismiss() noexcept {
78 ScopeGuardImplBase() noexcept : dismissed_(false) {}
80 static ScopeGuardImplBase makeEmptyScopeGuard() noexcept {
81 return ScopeGuardImplBase{};
85 static const T& asConst(const T& t) noexcept {
92 template <typename FunctionType>
93 class ScopeGuardImpl : public ScopeGuardImplBase {
95 explicit ScopeGuardImpl(FunctionType& fn) noexcept(
96 std::is_nothrow_copy_constructible<FunctionType>::value)
99 makeFailsafe(std::is_nothrow_copy_constructible<FunctionType>{},
102 explicit ScopeGuardImpl(const FunctionType& fn) noexcept(
103 std::is_nothrow_copy_constructible<FunctionType>::value)
106 makeFailsafe(std::is_nothrow_copy_constructible<FunctionType>{},
109 explicit ScopeGuardImpl(FunctionType&& fn) noexcept(
110 std::is_nothrow_move_constructible<FunctionType>::value)
112 std::move_if_noexcept(fn),
113 makeFailsafe(std::is_nothrow_move_constructible<FunctionType>{},
116 ScopeGuardImpl(ScopeGuardImpl&& other) noexcept(
117 std::is_nothrow_move_constructible<FunctionType>::value)
118 : function_(std::move_if_noexcept(other.function_)) {
119 // If the above line attempts a copy and the copy throws, other is
120 // left owning the cleanup action and will execute it (or not) depending
121 // on the value of other.dismissed_. The following lines only execute
122 // if the move/copy succeeded, in which case *this assumes ownership of
123 // the cleanup action and dismisses other.
124 dismissed_ = other.dismissed_;
125 other.dismissed_ = true;
128 ~ScopeGuardImpl() noexcept {
135 static ScopeGuardImplBase makeFailsafe(std::true_type, const void*) noexcept {
136 return makeEmptyScopeGuard();
139 template <typename Fn>
140 static auto makeFailsafe(std::false_type, Fn* fn) noexcept
141 -> ScopeGuardImpl<decltype(std::ref(*fn))> {
142 return ScopeGuardImpl<decltype(std::ref(*fn))>{std::ref(*fn)};
145 template <typename Fn>
146 explicit ScopeGuardImpl(Fn&& fn, ScopeGuardImplBase&& failsafe)
147 : ScopeGuardImplBase{}, function_(std::forward<Fn>(fn)) {
151 void* operator new(std::size_t) = delete;
153 void execute() noexcept { function_(); }
155 FunctionType function_;
158 template <typename FunctionType>
159 ScopeGuardImpl<typename std::decay<FunctionType>::type>
160 makeGuard(FunctionType&& fn) noexcept(
161 std::is_nothrow_constructible<typename std::decay<FunctionType>::type,
162 FunctionType>::value) {
163 return ScopeGuardImpl<typename std::decay<FunctionType>::type>(
164 std::forward<FunctionType>(fn));
168 * This is largely unneeded if you just use auto for your guards.
170 typedef ScopeGuardImplBase&& ScopeGuard;
174 #if defined(FOLLY_EXCEPTION_COUNT_USE_CXA_GET_GLOBALS) || \
175 defined(FOLLY_EXCEPTION_COUNT_USE_GETPTD) || \
176 defined(FOLLY_EXCEPTION_COUNT_USE_STD)
179 * ScopeGuard used for executing a function when leaving the current scope
180 * depending on the presence of a new uncaught exception.
182 * If the executeOnException template parameter is true, the function is
183 * executed if a new uncaught exception is present at the end of the scope.
184 * If the parameter is false, then the function is executed if no new uncaught
185 * exceptions are present at the end of the scope.
187 * Used to implement SCOPE_FAIL and SCOPE_SUCCES below.
189 template <typename FunctionType, bool executeOnException>
190 class ScopeGuardForNewException {
192 explicit ScopeGuardForNewException(const FunctionType& fn)
196 explicit ScopeGuardForNewException(FunctionType&& fn)
197 : function_(std::move(fn)) {
200 ScopeGuardForNewException(ScopeGuardForNewException&& other)
201 : function_(std::move(other.function_))
202 , exceptionCounter_(std::move(other.exceptionCounter_)) {
205 ~ScopeGuardForNewException() noexcept(executeOnException) {
206 if (executeOnException == exceptionCounter_.isNewUncaughtException()) {
212 ScopeGuardForNewException(const ScopeGuardForNewException& other) = delete;
214 void* operator new(std::size_t) = delete;
216 FunctionType function_;
217 UncaughtExceptionCounter exceptionCounter_;
221 * Internal use for the macro SCOPE_FAIL below
223 enum class ScopeGuardOnFail {};
225 template <typename FunctionType>
226 ScopeGuardForNewException<typename std::decay<FunctionType>::type, true>
227 operator+(detail::ScopeGuardOnFail, FunctionType&& fn) {
229 ScopeGuardForNewException<typename std::decay<FunctionType>::type, true>(
230 std::forward<FunctionType>(fn));
234 * Internal use for the macro SCOPE_SUCCESS below
236 enum class ScopeGuardOnSuccess {};
238 template <typename FunctionType>
239 ScopeGuardForNewException<typename std::decay<FunctionType>::type, false>
240 operator+(ScopeGuardOnSuccess, FunctionType&& fn) {
242 ScopeGuardForNewException<typename std::decay<FunctionType>::type, false>(
243 std::forward<FunctionType>(fn));
246 #endif // native uncaught_exception() supported
249 * Internal use for the macro SCOPE_EXIT below
251 enum class ScopeGuardOnExit {};
253 template <typename FunctionType>
254 ScopeGuardImpl<typename std::decay<FunctionType>::type>
255 operator+(detail::ScopeGuardOnExit, FunctionType&& fn) {
256 return ScopeGuardImpl<typename std::decay<FunctionType>::type>(
257 std::forward<FunctionType>(fn));
259 } // namespace detail
264 auto FB_ANONYMOUS_VARIABLE(SCOPE_EXIT_STATE) \
265 = ::folly::detail::ScopeGuardOnExit() + [&]() noexcept
267 #if defined(FOLLY_EXCEPTION_COUNT_USE_CXA_GET_GLOBALS) || \
268 defined(FOLLY_EXCEPTION_COUNT_USE_GETPTD) || \
269 defined(FOLLY_EXCEPTION_COUNT_USE_STD)
271 auto FB_ANONYMOUS_VARIABLE(SCOPE_FAIL_STATE) \
272 = ::folly::detail::ScopeGuardOnFail() + [&]() noexcept
274 #define SCOPE_SUCCESS \
275 auto FB_ANONYMOUS_VARIABLE(SCOPE_SUCCESS_STATE) \
276 = ::folly::detail::ScopeGuardOnSuccess() + [&]()
277 #endif // native uncaught_exception() supported