在我前面的文章中介绍了如果在书写接口实现的时候,没有注意它的释构函可能会导致多次调用问题http://blog.sohu.com/members/lionwoo/1703919.html。那么我们在程序设计中如何避免这个问题,其实解决方法非常的简单:
class MyObject : public IUnknown
{
...
ULONG Release()
{
LONG cRef = InterlockedDecrement(&m_cRef);
if (cRef == 0) {
m_cRef = DESTRUCTOR_REFCOUNT;
delete this;
}
return cRef;
}
...
private:
}
enum { DESTRUCTOR_REFCOUNT = 42 };
~MyObject()
{
if (m_fNeedSave) Save();
assert(m_cRef == DESTRUCTOR_REFCOUNT);
}
};
上述代码就可以完成任务。
有一个问题,那就是在Release函数中,m_cRef = DESTRUCTOR_REFCOUNT就可以执行直接赋值,那么对于多线程这样应该是不安全的。其实你完全不用作这个担心,因为在调用这段代码的时候,就已经可以确保只会有一个线程在调用,如果还有多个线程拥有这个接口,那么m_cRef是不可能为0的。