ATL---Collection & Enumerater（2）

练习一： 使用CComEnumOnSTL实现，使用的IEnumVARIANT枚举接口，自定义复制策略_CopyVariantFromLong，实现使用vector<long>存储数据。

 

1.IDL文件

import "oaidl.idl";import "ocidl.idl";[object,uuid(378B8756-3FD2-4CCC-9A50-C1352C7D5FBC),dual,nonextensible,helpstring("INumbers2 接口"),pointer_default(unique)]interface INumbers2 : IDispatch{[propget, id(1), helpstring("属性 Count")] HRESULT Count([out, retval] LONG* pVal);[propget, id(DISPID_VALUE), helpstring("属性 Item")] HRESULT Item([in] LONG n, [out, retval] LONG* pVal);[propget, id(DISPID_NEWENUM), helpstring("属性 _NewEnum")] HRESULT _NewEnum([out, retval] IUnknown** pVal);};[uuid(AEA414AB-1F5B-47C0-80E6-BB2738BECA9F),version(1.0),helpstring("MyNumbers2 1.0 类型库")]library MyNumbers2Lib{importlib("stdole2.tlb");[uuid(3CAE6047-35AB-4855-8B03-6957665B49EC),helpstring("Numbers2 Class")]coclass Numbers2{[default] interface INumbers2;};};

 

2.Numbers2.h

 class _CopyVariantFromLong{public:static HRESULT copy(VARIANT* p1, const long* p2) { p1->vt = VT_I4;p1->lVal = *p2;return S_OK;}static void init(VARIANT *p) {VariantInit(p);}static void destroy(VARIANT *p) {VariantClear(p);}};typedef CComEnumOnSTL<IEnumVARIANT,&IID_IEnumVARIANT,VARIANT,_CopyVariantFromLong,std::vector<long> > MyNumbersOnSTL;// CNumbers2class ATL_NO_VTABLE CNumbers2 :public CComObjectRootEx<CComSingleThreadModel>,public CComCoClass<CNumbers2, &CLSID_Numbers2>,public IDispatchImpl<INumbers2, &IID_INumbers2, &LIBID_MyNumbers2Lib, /*wMajor =*/ 1, /*wMinor =*/ 0>{public:CNumbers2(){m_MyNum.push_back(100);m_MyNum.push_back(200);m_MyNum.push_back(300);m_MyNum.push_back(400);m_MyNum.push_back(500);}DECLARE_REGISTRY_RESOURCEID(IDR_NUMBERS2)BEGIN_COM_MAP(CNumbers2)COM_INTERFACE_ENTRY(INumbers2)COM_INTERFACE_ENTRY(IDispatch)END_COM_MAP()DECLARE_PROTECT_FINAL_CONSTRUCT()HRESULT FinalConstruct(){return S_OK;}void FinalRelease(){}public:public:STDMETHOD(get_Count)(LONG* pVal);public:STDMETHOD(get_Item)(LONG n, LONG* pVal);public:STDMETHOD(get__NewEnum)(IUnknown** pVal);private:std::vector<long> m_MyNum;};OBJECT_ENTRY_AUTO(__uuidof(Numbers2), CNumbers2)

 

 3.Numbers2.cpp

STDMETHODIMP CNumbers2::get_Count(LONG* pVal){// TODO: 在此添加实现代码*pVal = (long)m_MyNum.size();return S_OK;}STDMETHODIMP CNumbers2::get_Item(LONG n, LONG* pVal){// TODO: 在此添加实现代码long lSize = (long)m_MyNum.size();if (n >= lSize){return E_OUTOFMEMORY;}*pVal = m_MyNum[n];return S_OK;}STDMETHODIMP CNumbers2::get__NewEnum(IUnknown** pVal){// TODO: 在此添加实现代码CComObject<MyNumbersOnSTL>* pNum = NULL;HRESULT hr = CComObject<MyNumbersOnSTL>::CreateInstance(&pNum);if (FAILED(hr)){return E_FAIL;}pNum->AddRef();hr = pNum->Init(this->GetUnknown(),m_MyNum);if (FAILED(hr)){pNum->Release();return E_FAIL;}hr = pNum->QueryInterface(IID_IUnknown,(void**)pVal);if (FAILED(hr)){pNum->Release();return E_FAIL;}pNum->Release();return S_OK;}

 

4.测试Client

#include <iostream>#include <atlbase.h>#include <atlcomcli.h>#include <atlcom.h>#include "../MyNumbers2/MyNumbers2.h"#include "../MyNumbers2/MyNumbers2_i.c"using namespace std;int main(){CoInitialize(NULL);try{CComPtr<INumbers2> pNum = NULL;HRESULT hr;hr = CoCreateInstance(CLSID_Numbers2,NULL,CLSCTX_INPROC,IID_INumbers2,(void**)&pNum);if (FAILED(hr)){cout << "CoCrateInstance error." << endl;CoUninitialize();return -1;}long lSize = 0;pNum->get_Count(&lSize);cout << "Size: " << lSize << endl;CComPtr<IEnumVARIANT> pEnum = NULL;CComPtr<IUnknown> pUnknown = NULL;pNum->get__NewEnum(&pUnknown);pUnknown->QueryInterface(IID_IEnumVARIANT,(void**)&pEnum);for (int i=1;i<=lSize;i++){VARIANT varTemp[1];ULONG lFetched = 0;hr = pEnum->Next(1,&varTemp[0],&lFetched);if (SUCCEEDED(hr)){CComVariant comvar;comvar.Attach(&varTemp[0]);wcout << "Number " << i << " is " << comvar.lVal << endl;}}}catch (...){cout << "an expection occur......" << endl;}CoUninitialize();system("pause");return 0;}