You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and dots ('.'), can be up to 35 characters long. Letters must be lowercase.
113 lines
3.1 KiB
113 lines
3.1 KiB
#ifndef BOOST_THREAD_TSS_HPP |
|
#define BOOST_THREAD_TSS_HPP |
|
// 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) |
|
// (C) Copyright 2007-8 Anthony Williams |
|
|
|
#include <boost/thread/detail/config.hpp> |
|
#include <boost/shared_ptr.hpp> |
|
#include <boost/thread/detail/thread_heap_alloc.hpp> |
|
|
|
#include <boost/config/abi_prefix.hpp> |
|
|
|
namespace boost |
|
{ |
|
namespace detail |
|
{ |
|
struct tss_cleanup_function |
|
{ |
|
virtual ~tss_cleanup_function() |
|
{} |
|
|
|
virtual void operator()(void* data)=0; |
|
}; |
|
|
|
BOOST_THREAD_DECL void set_tss_data(void const* key,boost::shared_ptr<tss_cleanup_function> func,void* tss_data,bool cleanup_existing); |
|
BOOST_THREAD_DECL void* get_tss_data(void const* key); |
|
} |
|
|
|
template <typename T> |
|
class thread_specific_ptr |
|
{ |
|
private: |
|
thread_specific_ptr(thread_specific_ptr&); |
|
thread_specific_ptr& operator=(thread_specific_ptr&); |
|
|
|
struct delete_data: |
|
detail::tss_cleanup_function |
|
{ |
|
void operator()(void* data) |
|
{ |
|
delete static_cast<T*>(data); |
|
} |
|
}; |
|
|
|
struct run_custom_cleanup_function: |
|
detail::tss_cleanup_function |
|
{ |
|
void (*cleanup_function)(T*); |
|
|
|
explicit run_custom_cleanup_function(void (*cleanup_function_)(T*)): |
|
cleanup_function(cleanup_function_) |
|
{} |
|
|
|
void operator()(void* data) |
|
{ |
|
cleanup_function(static_cast<T*>(data)); |
|
} |
|
}; |
|
|
|
|
|
boost::shared_ptr<detail::tss_cleanup_function> cleanup; |
|
|
|
public: |
|
typedef T element_type; |
|
|
|
thread_specific_ptr(): |
|
cleanup(detail::heap_new<delete_data>(),detail::do_heap_delete<delete_data>()) |
|
{} |
|
explicit thread_specific_ptr(void (*func_)(T*)) |
|
{ |
|
if(func_) |
|
{ |
|
cleanup.reset(detail::heap_new<run_custom_cleanup_function>(func_),detail::do_heap_delete<run_custom_cleanup_function>()); |
|
} |
|
} |
|
~thread_specific_ptr() |
|
{ |
|
detail::set_tss_data(this,boost::shared_ptr<detail::tss_cleanup_function>(),0,true); |
|
} |
|
|
|
T* get() const |
|
{ |
|
return static_cast<T*>(detail::get_tss_data(this)); |
|
} |
|
T* operator->() const |
|
{ |
|
return get(); |
|
} |
|
T& operator*() const |
|
{ |
|
return *get(); |
|
} |
|
T* release() |
|
{ |
|
T* const temp=get(); |
|
detail::set_tss_data(this,boost::shared_ptr<detail::tss_cleanup_function>(),0,false); |
|
return temp; |
|
} |
|
void reset(T* new_value=0) |
|
{ |
|
T* const current_value=get(); |
|
if(current_value!=new_value) |
|
{ |
|
detail::set_tss_data(this,cleanup,new_value,true); |
|
} |
|
} |
|
}; |
|
} |
|
|
|
#include <boost/config/abi_suffix.hpp> |
|
|
|
#endif
|
|
|