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.

107 lines
2.8KB

  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. // TODO: Handle copying correctly (#41)
  46. }
  47. scope_guard_t(const scope_guard_t&) = delete;
  48. scope_guard_t(scope_guard_t&& other) noexcept : m_deleter(std::move(other.m_deleter)),
  49. m_execute_on_destruction(other.m_execute_on_destruction)
  50. {
  51. other.release();
  52. }
  53. ~scope_guard_t()
  54. {
  55. if( m_execute_on_destruction == true )
  56. {
  57. call_deleter_safe();
  58. }
  59. }
  60. void release() noexcept
  61. {
  62. m_execute_on_destruction = false;
  63. }
  64. scope_guard_t& operator=(const scope_guard_t&) = delete;
  65. scope_guard_t& operator=(scope_guard_t&&) = delete;
  66. private:
  67. void call_deleter_safe() noexcept
  68. {
  69. try
  70. {
  71. m_deleter();
  72. }
  73. catch( ... ) { /* Empty */ }
  74. }
  75. Deleter m_deleter;
  76. bool m_execute_on_destruction;
  77. };
  78. template<class Deleter>
  79. scope_guard_t<Deleter> scope_guard(Deleter&& deleter) noexcept
  80. {
  81. return scope_guard_t<Deleter>{std::move(deleter)};
  82. }
  83. }