Blame view

3rdparty/boost_1_81_0/boost/fiber/fss.hpp 2.53 KB
598bfd3f   Hu Chunming   提交_GLIBCXX_USE_CX...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
  
  //          Copyright Oliver Kowalke 2013.
  // Distributed under the Boost Software License, Version 1.0.
  //    (See accompanying file LICENSE_1_0.txt or copy at
  //          http://www.boost.org/LICENSE_1_0.txt)
  //
  //          based on tss.hpp from boost.thread
  
  #ifndef BOOST_FIBERS_FSS_H
  #define BOOST_FIBERS_FSS_H
  
  #include <boost/config.hpp>
  
  #include <boost/fiber/context.hpp>
  #include <boost/fiber/detail/fss.hpp>
  
  #ifdef BOOST_HAS_ABI_HEADERS
  #  include BOOST_ABI_PREFIX
  #endif
  
  namespace boost {
  namespace fibers {
  
  template< typename T >
  class fiber_specific_ptr {
  private:
      struct default_cleanup_function : public detail::fss_cleanup_function {
          void operator()( void * data) noexcept override {
              delete static_cast< T * >( data);
          }
      };
  
      struct custom_cleanup_function : public detail::fss_cleanup_function {
          void (*fn)(T*);
  
          explicit custom_cleanup_function( void(*fn_)(T*) ) noexcept :
              fn{ fn_ } {
          }
  
          void operator()( void * data) override {
              if ( BOOST_LIKELY( nullptr != fn) ) {
                  fn( static_cast< T * >( data) );
              }
          }
      };
  
      detail::fss_cleanup_function::ptr_t cleanup_fn_;
  
  public:
      using element_type = T;
  
      fiber_specific_ptr() :
          cleanup_fn_{ new default_cleanup_function() } {
      }
  
      explicit fiber_specific_ptr( void(*fn)(T*) ) :
          cleanup_fn_{ new custom_cleanup_function( fn) } {
      }
  
      ~fiber_specific_ptr() {
          context * active_ctx = context::active();
          if ( nullptr != active_ctx) {
              active_ctx->set_fss_data(
                  this, cleanup_fn_, nullptr, true);
          }
      }
  
      fiber_specific_ptr( fiber_specific_ptr const&) = delete;
      fiber_specific_ptr & operator=( fiber_specific_ptr const&) = delete;
  
      T * get() const noexcept {
          BOOST_ASSERT( context::active() );
          void * vp = context::active()->get_fss_data( this);
          return static_cast< T * >( vp);
      }
  
      T * operator->() const noexcept {
          return get();
      }
  
      T & operator*() const noexcept {
          return * get();
      }
  
      T * release() {
          T * tmp = get();
          context::active()->set_fss_data(
              this, cleanup_fn_, nullptr, false);
          return tmp;
      }
  
      void reset( T * t) {
          T * c = get();
          if ( BOOST_LIKELY( c != t) ) {
              context::active()->set_fss_data(
                  this, cleanup_fn_, t, true);
          }
      }
  };
  
  }}
  
  #ifdef BOOST_HAS_ABI_HEADERS
  #  include BOOST_ABI_SUFFIX
  #endif
  
  #endif //  BOOST_FIBERS_FSS_H