您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

scope_guard.h 2.9KB

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