C++ COM组件字符串数组的传递

haha0518

在ATL下开发简单的COM组件很简单，唯一的难点在于复杂参数的传递。比如说字符串数组，需要学习SAFFARRAY，bstr以及VARIANT的用法。

下面贴出代码，展示一个字符串数组如何传递。

在COM组件的方法里面：

1.   STDMETHODIMP CwordCountService::analyzeTheWordFile(BSTR filename, VARIANT* wordsSafearray, VARIANT* frequencySafearray)
   
2. {
   
3.         // TODO: 在此添加实现代码
   
4.         MYWORD **wordArray;
   
5.         int wordCount;
   
6.         char *cFilename = _com_util::ConvertBSTRToString(filename);
   
7.         //调用analyzeTxtWordFile函数对指定文档进行统计分析
   
8.         analyzeTxtWordFile(cFilename, &wordArray, &wordCount);
   
9.         delete cFilename;
   
10.         cFilename = nullptr;
    
11. 
    
12.         //把统计到的结果，存放到安全数组里面
    
13.         VARIANT var1, var2;
    
14.         VariantInit(&var1);
    
15.         VariantInit(&var2);
    
16.         var1.vt = VT_BSTR;
    
17.         var2.vt = VT_INT;
    
18.         SAFEARRAY *pArray1, *pArray2;
    
19.         pArray1 = SafeArrayCreateVector(VT_VARIANT, 0, wordCount);
    
20.         pArray2 = SafeArrayCreateVector(VT_VARIANT, 0, wordCount);
    
21.         for (long i = 0; i < wordCount; i++)
    
22.         {
    
23.                 var1.bstrVal = _com_util::ConvertStringToBSTR(wordArray[i]->MYWORD);
    
24.                 var2.intVal = wordArray[i]->count;
    
25.                 SafeArrayPutElement(pArray1, &i, &var1);
    
26.                 SafeArrayPutElement(pArray2, &i, &var2);
    
27.                 free(wordArray[i]);
    
28.         }
    
29.         free(wordArray);
    
30. 
    
31.         VariantClear(&var1);
    
32.         VariantClear(&var2);
    
33.         wordsSafearray->vt = VT_ARRAY | VT_VARIANT;
    
34.         frequencySafearray->vt = VT_ARRAY | VT_VARIANT;
    
35.         wordsSafearray->parray = pArray1;
    
36.         frequencySafearray->parray = pArray2;
    
37.         return S_OK;
    
38. }

复制代码

调用COM组件的C++代码：

1. #include <atlbase.h>  
   
2. #import "D:\wordCountComponent.dll" no_namespace
   
3. 
   
4. int main()
   
5. {
   
6.         //根据接口的clsid来初始化com组件
   
7.         CoInitialize(NULL);
   
8.         CLSID clsid;
   
9.         HRESULT hr = CLSIDFromString(OLESTR("{40F2E186-5636-4F0D-87AB-DF351B41D133}"), &clsid);
   
10. 
    
11.         //用智能指针去获取接口指针
    
12.         CComPtr<IwordCountService> wordCountServicePtr;
    
13.         hr = wordCountServicePtr.CoCreateInstance(clsid);
    
14.         if (NOERROR != hr)
    
15.         {
    
16.                 printf("初始化com失败！\n");
    
17.                 return -1;
    
18.         }
    
19. 
    
20.         VARIANT var1, var2;
    
21.         VariantInit(&var1);
    
22.         VariantInit(&var2);
    
23.         SAFEARRAY *pArray1, *pArray2;
    
24. 
    
25.         BSTR fileName = _com_util::ConvertStringToBSTR("d:/Romeo and Juliet.txt");
    
26.         //调用analyzeTheWordFile函数分析文件，分析得到的结果存放在var1, var2的安全数组中
    
27.         wordCountServicePtr->analyzeTheWordFile(fileName, &var1, &var2);
    
28.         ::SysFreeString(fileName);
    
29.         BSTR fileName2 = _com_util::ConvertStringToBSTR("d:/log.txt");
    
30.         //分析完之后，可以利用得到的结果，调用createTxtLogFile生成统计文件
    
31.         wordCountServicePtr->createTxtLogFile(fileName2, var1, var2);
    
32.         ::SysFreeString(fileName2);
    
33. 
    
34.         //下列代码展示了如何把数据从安全数组里面读取出来，并且保证内存不泄露
    
35.         pArray1 = var1.parray;
    
36.         pArray2 = var2.parray;
    
37. 
    
38.         //获取安全数组的上下边界以及长度
    
39.         long plLowBound, plUpperBound;
    
40.         SafeArrayGetLBound(pArray1, 1, &plLowBound);
    
41.         SafeArrayGetUBound(pArray1, 1, &plUpperBound);
    
42.         int wordCount = plUpperBound - plLowBound + 1;
    
43.         char **wordArray = new char*[wordCount];
    
44.         int *frequencyArray = new int[wordCount];
    
45.         VARIANT temp1, temp2;
    
46.         VariantInit(&temp1);
    
47.         VariantInit(&temp2);
    
48.         for (long i = plLowBound; i <= plUpperBound; i++)
    
49.         {
    
50.                 ::SafeArrayGetElement(pArray1, &i, &temp1);
    
51.                 ::SafeArrayGetElement(pArray2, &i, &temp2);
    
52.                 wordArray[i] = _com_util::ConvertBSTRToString(temp1.bstrVal);
    
53.                 frequencyArray[i] = temp2.intVal;
    
54.         }
    
55. 
    
56.         //把从安全数组读取出来的数据打印到屏幕上
    
57.         for (long i = plLowBound; i <= plUpperBound; i++)
    
58.         {
    
59.                 printf("%s %d\n", wordArray[i], frequencyArray[i]);
    
60.         }
    
61. 
    
62.         //调用searchWord搜索指定单词出现频率
    
63.         BSTR theWord = _com_util::ConvertStringToBSTR("the");
    
64.         long frequency = 0;
    
65.         wordCountServicePtr->searchWord(theWord, var1, var2, &frequency);
    
66.         ::SysFreeString(theWord);
    
67.         printf("所搜索的单词在文本出现频数为%ld次\n", frequency);
    
68. 
    
69.         for (long i = plLowBound; i <= plUpperBound; i++)
    
70.         {
    
71.                 ::SafeArrayGetElement(pArray1, &i, &temp1);
    
72.                 //释放掉安全数组里面的bstr内存
    
73.                 ::SysFreeString(temp1.bstrVal);
    
74.                 //释放掉字符串内存
    
75.                 delete wordArray[i];
    
76.         }
    
77.         delete[] wordArray;
    
78.         delete[] frequencyArray;
    
79. 
    
80.         //安全数组使用完毕之后记得销毁
    
81.         SafeArrayDestroy(pArray1);
    
82.         SafeArrayDestroy(pArray2);
    
83.         //释放智能指针
    
84.         wordCountServicePtr.Release();
    
85.         //卸载com组件
    
86.         CoUninitialize();
    
87.         return 0;
    
88. }

复制代码