我将使用非常简单的用户可以理解的shared_ptr实现来回答这个问题(2年后)。
首先,我将介绍一些辅助类,shared_ptr_base,sp_counted_base sp_counted_impl和checked_deleter,最后一个是模板。
class sp_counted_base
{
public:
sp_counted_base() : refCount( 1 )
{
}
virtual ~sp_deleter_base() {};
virtual void destruct() = 0;
void incref(); // increases reference count
void decref(); // decreases refCount atomically and calls destruct if it hits zero
private:
long refCount; // in a real implementation use an atomic int
};
template< typename T > class sp_counted_impl : public sp_counted_base
{
public:
typedef function< void( T* ) > func_type;
void destruct()
{
func(ptr); // or is it (*func)(ptr); ?
delete this; // self-destructs after destroying its pointer
}
template< typename F >
sp_counted_impl( T* t, F f ) :
ptr( t ), func( f )
private:
T* ptr;
func_type func;
};
template< typename T > struct checked_deleter
{
public:
template< typename T > operator()( T* t )
{
size_t z = sizeof( T );
delete t;
}
};
class shared_ptr_base
{
private:
sp_counted_base * counter;
protected:
shared_ptr_base() : counter( 0 ) {}
explicit shared_ptr_base( sp_counter_base * c ) : counter( c ) {}
~shared_ptr_base()
{
if( counter )
counter->decref();
}
shared_ptr_base( shared_ptr_base const& other )
: counter( other.counter )
{
if( counter )
counter->addref();
}
shared_ptr_base& operator=( shared_ptr_base& const other )
{
shared_ptr_base temp( other );
std::swap( counter, temp.counter );
}
// other methods such as reset
};
现在,我将创建两个名为make_sp_counted_impl的“免费”函数,该函数将返回指向新创建的指针的指针。
template< typename T, typename F >
sp_counted_impl<T> * make_sp_counted_impl( T* ptr, F func )
{
try
{
return new sp_counted_impl( ptr, func );
}
catch( ... ) // in case the new above fails
{
func( ptr ); // we have to clean up the pointer now and rethrow
throw;
}
}
template< typename T >
sp_counted_impl<T> * make_sp_counted_impl( T* ptr )
{
return make_sp_counted_impl( ptr, checked_deleter<T>() );
}
好的,这两个函数对于通过模板函数创建shared_ptr时接下来将发生的情况至关重要。
template< typename T >
class shared_ptr : public shared_ptr_base
{
public:
template < typename U >
explicit shared_ptr( U * ptr ) :
shared_ptr_base( make_sp_counted_impl( ptr ) )
{
}
// implement the rest of shared_ptr, e.g. operator*, operator->
};
请注意,如果T为空并且U是您的“测试”类,那么上面发生的事情。它将使用指向U的指针而不是指向T的指针来调用make_sp_counted_impl()。销毁的管理全部通过此处完成。shared_ptr_base类管理有关复制和赋值等的引用计数。shared_ptr类本身管理操作符重载(->,*等)的类型安全使用。
因此,尽管您有一个shared_ptr要作废,但是在下面,您正在管理传递给new的类型的指针。请注意,如果在将指针转换为void *并将其放入shared_ptr之前,它将无法在checked_delete上进行编译,因此实际上也很安全。