#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 |