42 lines
920 B
C++
42 lines
920 B
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include "common.hpp"
|
||
|
|
|
||
|
|
#include "components/logger.hpp"
|
||
|
|
|
||
|
|
LEMONBUDDY_NS
|
||
|
|
|
||
|
|
namespace scope_util {
|
||
|
|
template <typename... Args>
|
||
|
|
class on_exit {
|
||
|
|
public:
|
||
|
|
on_exit(function<void(Args...)>&& fn, Args... args) : m_callback(bind(fn, args...)) {}
|
||
|
|
|
||
|
|
virtual ~on_exit() {
|
||
|
|
m_callback();
|
||
|
|
}
|
||
|
|
|
||
|
|
protected:
|
||
|
|
function<void()> m_callback;
|
||
|
|
};
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Creates a wrapper that will trigger given callback when
|
||
|
|
* leaving the object's scope (i.e, when it gets destroyed)
|
||
|
|
*
|
||
|
|
* Example usage:
|
||
|
|
* @code cpp
|
||
|
|
* {
|
||
|
|
* auto handler = scope_util::make_exit_handler([]{ ... })
|
||
|
|
* ...
|
||
|
|
* }
|
||
|
|
* @endcode
|
||
|
|
*/
|
||
|
|
template <typename Fn = function<void()>, typename... Args>
|
||
|
|
decltype(auto) make_exit_handler = [](Fn&& fn, Args&&... args) -> unique_ptr<on_exit<Args...>> {
|
||
|
|
return make_unique<on_exit<Args...>>(forward<Fn>(fn), forward<Args>(args)...);
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
LEMONBUDDY_NS_END
|