AddRef和Release如何進行引用計數的實現

本篇文章給大家分享的是有關AddRef和Release如何進行引用計數的實現，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

#include <iostream><br/>#include <atlbase.h><br/>using namespace std;

interface IX : public IUnknown
{
virtual void _stdcall Fx() = 0;
};

interface IY : public IUnknown
{
virtual void _stdcall Fy() = 0;
};

interface IZ : public IUnknown
{
virtual void _stdcall Fz() = 0;
};

static const IID IID_IX = {0xec53e1de, 0x37d, 0x48f5, {0xa1, 0xee, 0x2d, 0xf5, 0xa5, 0xde, 0xf3, 0x49}}; // {EC53E1DE-037D-48f5-A1EE-2DF5A5DEF349}
static const IID IID_IY = {0x4c510b77, 0xc63a, 0x4e0f, {0xa2, 0x45, 0x7b, 0x21, 0x23, 0x12, 0x23, 0xe7}};// {4C510B77-C63A-4e0f-A245-7B21231223E7}
static const IID IID_IZ = {0xefb1f8d4, 0x674e, 0x4123, {0xa9, 0x19, 0x3c, 0x72, 0xee, 0xe8, 0x1d, 0x92}};// {EFB1F8D4-674E-4123-A919-3C72EEE81D92}

class CA : public IX,
public IY
{
public:
CA() : m_cRef(0) { }
virtual ~CA(){ }

public:
virtual HRESULT _stdcall QueryInterface(REFIID riid, __RPC__deref_out void **ppvObject);
virtual ULONG _stdcall AddRef();
virtual ULONG _stdcall Release();

virtual void _stdcall Fx() { cout<<"CA::Fx"<<endl; }

virtual void _stdcall Fy() { cout<<"CA::Fy"<<endl; }

private:
LONG m_cRef;
};

HRESULT _stdcall CA::QueryInterface(REFIID riid, __RPC__deref_out void **ppvObject)
{
if(IID_IX == riid)
{
*ppvObject = static_cast<IX*>(this);
}
else if(IID_IY == riid)
{
*ppvObject = static_cast<IY*>(this);
}
else if(IID_IUnknown == riid)
{
*ppvObject = static_cast<IX*>(this);
}
else
{
*ppvObject = NULL;
return E_NOINTERFACE;
}

reinterpret_cast<IUnknown*>(*ppvObject)->AddRef(); //對應 這五個地方需要對應起來
return S_OK;
}

ULONG _stdcall CA::AddRef()
{
InterlockedIncrement(&m_cRef);
return m_cRef;
}

ULONG _stdcall CA::Release()
{
if(InterlockedDecrement(&m_cRef) == 0)
{
delete this; //下面標記的3個Release 缺少一個則執行不到這
return 0;
}

return m_cRef;
}

IUnknown* CreateInstance()
{
IUnknown *pIUnk = static_cast<IX*>(new CA);
pIUnk->AddRef(); //對應 這五個地方需要對應起來
return pIUnk;
}

void _tmain(int argc,_TCHAR*argv[])
{
IUnknown *pIUnk = CreateInstance();

IX *pIX = NULL;
HRESULT hr = pIUnk->QueryInterface(IID_IX,(void**)&pIX);
if(SUCCEEDED(hr))
{
pIX->Fx();
pIX->Release(); //對應 這五個地方需要對應起來
}

IY *pIY = NULL;
hr = pIUnk->QueryInterface(IID_IY,(void**)&pIY);
if(SUCCEEDED(hr))
{
pIY->Fy();
pIY->Release(); //對應 這五個地方需要對應起來
}

IZ *pIZ = NULL;    //沒有執行到AddRef，故不需要釋放
hr = pIUnk->QueryInterface(IID_IZ,(void**)&pIZ);
if(SUCCEEDED(hr))
{
cout<<"found pIZ"<<endl;
pIZ->Release();
}

pIUnk->Release(); //對應 這五個地方需要對應起來
}

以上就是AddRef和Release如何進行引用計數的實現，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。