/*
- * Copyright 2016 Facebook, Inc.
+ * Copyright 2017 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* limitations under the License.
*/
-#ifndef FOLLY_SCOPEGUARD_H_
-#define FOLLY_SCOPEGUARD_H_
+#pragma once
#include <cstddef>
#include <functional>
#include <type_traits>
#include <utility>
+#include <folly/Portability.h>
#include <folly/Preprocessor.h>
#include <folly/detail/UncaughtExceptionCounter.h>
* a functor, or a void(*)() function pointer.
*
*
- * Usage example: Add a friend to memory iff it is also added to the db.
+ * Usage example: Add a friend to memory if and only if it is also added
+ * to the db.
*
* void User::addFriend(User& newFriend) {
* // add the friend to memory
dismissed_ = true;
}
+ template <typename T>
+ FOLLY_ALWAYS_INLINE static void runAndWarnAboutToCrashOnException(
+ T& function) noexcept {
+ try {
+ function();
+ } catch (...) {
+ warnAboutToCrash();
+ std::terminate();
+ }
+ }
+
protected:
ScopeGuardImplBase() noexcept : dismissed_(false) {}
}
bool dismissed_;
+
+ private:
+ static void warnAboutToCrash() noexcept;
};
template <typename FunctionType>
void* operator new(std::size_t) = delete;
- void execute() noexcept { function_(); }
+ void execute() noexcept {
+ runAndWarnAboutToCrashOnException(function_);
+ }
FunctionType function_;
};
namespace detail {
#if defined(FOLLY_EXCEPTION_COUNT_USE_CXA_GET_GLOBALS) || \
- defined(FOLLY_EXCEPTION_COUNT_USE_GETPTD)
+ defined(FOLLY_EXCEPTION_COUNT_USE_GETPTD) || \
+ defined(FOLLY_EXCEPTION_COUNT_USE_STD)
/**
* ScopeGuard used for executing a function when leaving the current scope
* If the parameter is false, then the function is executed if no new uncaught
* exceptions are present at the end of the scope.
*
- * Used to implement SCOPE_FAIL and SCOPE_SUCCES below.
+ * Used to implement SCOPE_FAIL and SCOPE_SUCCESS below.
*/
template <typename FunctionType, bool executeOnException>
class ScopeGuardForNewException {
~ScopeGuardForNewException() noexcept(executeOnException) {
if (executeOnException == exceptionCounter_.isNewUncaughtException()) {
- function_();
+ if (executeOnException) {
+ ScopeGuardImplBase::runAndWarnAboutToCrashOnException(function_);
+ } else {
+ function_();
+ }
}
}
}
} // namespace detail
-} // folly
+} // namespace folly
#define SCOPE_EXIT \
auto FB_ANONYMOUS_VARIABLE(SCOPE_EXIT_STATE) \
= ::folly::detail::ScopeGuardOnExit() + [&]() noexcept
#if defined(FOLLY_EXCEPTION_COUNT_USE_CXA_GET_GLOBALS) || \
- defined(FOLLY_EXCEPTION_COUNT_USE_GETPTD)
+ defined(FOLLY_EXCEPTION_COUNT_USE_GETPTD) || \
+ defined(FOLLY_EXCEPTION_COUNT_USE_STD)
#define SCOPE_FAIL \
auto FB_ANONYMOUS_VARIABLE(SCOPE_FAIL_STATE) \
= ::folly::detail::ScopeGuardOnFail() + [&]() noexcept
auto FB_ANONYMOUS_VARIABLE(SCOPE_SUCCESS_STATE) \
= ::folly::detail::ScopeGuardOnSuccess() + [&]()
#endif // native uncaught_exception() supported
-
-#endif // FOLLY_SCOPEGUARD_H_