}
void clear_ops()
{
for( RefOpList::const_iterator it = _oplist.begin();
it != _oplist.end(); ++ it )
{
(*it)->be_null();
}
}
private:
RefOpList _oplist;
};
template <typename _Tp>
class auto_null : public ref_op
{
public:
void fetch( _Tp *t )
{
_t = t;
t->add_ref( this );
}
auto_null<_Tp> &operator = ( _Tp *t )
{
fetch( t );
return *this;
}
void be_null()
{
_t = 0;
}
operator _Tp*()
{
return _t;
}
private:
_Tp *_t;
};
//////////////////////////////////////////////////////////////////////////////
class CMonster : public ref_base
{
};
class CMonsterAI
{
public:
void SetOwner( CMonster *pOwner )
{
m_Owner = pOwner;
}
void Test()
{
if( (CMonster*)m_Owner == NULL )
{
printf( "The owner is null.\n" );
}
else
{
printf( "The owner is NOT null.\n" );
}
}
private:
auto_null<CMonster> m_Owner;
};
int main()
{
CMonster *pMonster = new CMonster();
CMonsterAI *pAI = new CMonsterAI();
pAI->SetOwner( pMonster );
pAI->Test();
delete pMonster;
pAI->Test();
delete pAI;
return 0;
}
CMonster內(nèi)部會保存一個(gè)CMonster的指針,當(dāng)CMonster被刪除掉時(shí),會自動(dòng)更新CMonsterAI內(nèi)部的CMonster“指針”為NULL。接口比較簡單:1)在會被引用(即被其他對象保存其指針)的類設(shè)計(jì)中派生(或組合)ref_base類,ref_base類簡單來說就是保存一個(gè)引用類對象列表,通過基類ref_op可以使得ref_base保存不同類型的引用類(例如CMonsterAI可以保存CMonster, CRegion也可以保存CMonster),ref_base在析構(gòu)時(shí)自動(dòng)回調(diào)引用類對象的be_null函數(shù),be_null函數(shù)在auto_null類中自動(dòng)把CMonster*設(shè)置為NULL。
通過重載operator=,使得SetOwner函數(shù)里不需要做其他操作(看起來)。