You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

106 lines
2.7KB

  1. /*
  2. * Scope Guard
  3. * Copyright (C) 2017 offa
  4. *
  5. * This file is part of Scope Guard.
  6. *
  7. * Scope Guard is free software: you can redistribute it and/or modify
  8. * it under the terms of the GNU General Public License as published by
  9. * the Free Software Foundation, either version 3 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * Scope Guard is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with Scope Guard. If not, see <http://www.gnu.org/licenses/>.
  19. */
  20. #pragma once
  21. #include <utility>
  22. #include <type_traits>
  23. namespace sr
  24. {
  25. template<class Deleter>
  26. class scope_guard_t
  27. {
  28. public:
  29. template<class D,
  30. std::enable_if_t<std::is_constructible<Deleter, D>::value, int> = 0,
  31. std::enable_if_t<(!std::is_lvalue_reference<D>::value)
  32. && std::is_nothrow_constructible<Deleter, D>::value, int> = 0
  33. >
  34. explicit scope_guard_t(D&& deleter) : m_deleter(std::move(deleter)),
  35. m_execute_on_destruction(true)
  36. {
  37. }
  38. template<class D,
  39. std::enable_if_t<std::is_constructible<Deleter, D>::value, int> = 0,
  40. std::enable_if_t<std::is_lvalue_reference<D>::value, int> = 0
  41. >
  42. explicit scope_guard_t(D&& deleter) : m_deleter(deleter),
  43. m_execute_on_destruction(true)
  44. {
  45. }
  46. scope_guard_t(const scope_guard_t&) = delete;
  47. scope_guard_t(scope_guard_t&& other) : m_deleter(std::move(other.m_deleter)),
  48. m_execute_on_destruction(other.m_execute_on_destruction)
  49. {
  50. other.release();
  51. }
  52. ~scope_guard_t()
  53. {
  54. if( m_execute_on_destruction == true )
  55. {
  56. call_deleter_safe();
  57. }
  58. }
  59. void release() noexcept
  60. {
  61. m_execute_on_destruction = false;
  62. }
  63. scope_guard_t& operator=(const scope_guard_t&) = delete;
  64. scope_guard_t& operator=(scope_guard_t&&) = delete;
  65. private:
  66. void call_deleter_safe() noexcept
  67. {
  68. try
  69. {
  70. m_deleter();
  71. }
  72. catch( ... ) { /* Empty */ }
  73. }
  74. Deleter m_deleter;
  75. bool m_execute_on_destruction;
  76. };
  77. template<class Deleter>
  78. scope_guard_t<Deleter> scope_guard(Deleter&& deleter) noexcept
  79. {
  80. return scope_guard_t<Deleter>{std::move(deleter)};
  81. }
  82. }