开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 查看源码
DBREAD.H
资源名称：MSDN_VC98.zip　[点击查看]
上传用户：bangxh
上传日期：2007-01-31
资源大小：42235k
文件大小：3k
源码类别：

Windows编程

开发平台：

Visual C++

DBREAD.H：源码内容
  1. #include <atldbcli.h>
  3. #if 1
  4. class CProduct
  5. {
  6. public:
  7. long    nProductID;
  8. char*   pName;
  9. long    nUnitsInStock;
  10. long    nReorderLevel;
  12. BEGIN_ACCESSOR_MAP(CProduct, 1)
  13. BEGIN_ACCESSOR(0, false)
  14. COLUMN_ENTRY(1, nProductID)
  15. COLUMN_ENTRY_TYPE(2, DBTYPE_STR | DBTYPE_BYREF, pName)
  16. COLUMN_ENTRY(7, nUnitsInStock)
  17. COLUMN_ENTRY(9, nReorderLevel)
  18. END_ACCESSOR();
  19. END_ACCESSOR_MAP()
  21. DEFINE_COMMAND(CProduct, _T("SELECT * FROM Products"))
  22. };
  23. #else
  24. class CProduct
  25. {
  26. public:
  27. long id;
  28. char Name[80];
  29. char PhoneNumber[50];
  30. char Location[50];
  31. char Title[50];
  32. char Company[50];
  33. char Email[50];
  35. BEGIN_COLUMN_MAP(CProduct)
  36. COLUMN_ENTRY(1, id)
  37. COLUMN_ENTRY(2, Name)
  38. COLUMN_ENTRY(3, PhoneNumber)
  39. COLUMN_ENTRY(4, Location)
  40. COLUMN_ENTRY(5, Title)
  41. COLUMN_ENTRY(6, Company)
  42. COLUMN_ENTRY(7, Email)
  43. END_COLUMN_MAP()
  44. };
  45. #endif
  47. void ProcessRecord(CProduct& product);
  49. class CMyRowset : public CRowset
  50. {
  51. public:
  52. HRESULT MoveAndProcess()
  53. {
  54. CProduct product;
  55. HRESULT hr;
  56. hr = MoveNext();
  57. if (hr == S_OK)
  58. {
  59. hr = GetDataHere(&product);
  60. ATLASSERT(SUCCEEDED(hr));
  61. ProcessRecord(product);
  62. }
  63. return hr;
  64. }
  65. };
  67. ULONG WINAPI ReadTable(CTable<CAccessor<CProduct>, CMyRowset>* pProduct)
  68. {
  69. CMyRowset rowset = *((CMyRowset*)pProduct);
  70. HRESULT   hr;
  71. ULONG     nRows = 0;
  73. while ((hr=rowset.MoveAndProcess()) == S_OK)
  74. {
  75. nRows++;
  76. }
  77. ATLTRACE("Thread %d processed %d records. Last HRESULT = %Xn", GetCurrentThreadId(), nRows, hr);
  78. return nRows;
  79. }
  81. void ProcessRecord(CProduct& product)
  82. {
  83. DWORD dwThreadId = GetCurrentThreadId();
  84. ATLTRACE("%d, %4d, %sn", dwThreadId, product.nProductID, product.pName);
  85. }
  87. HRESULT ReadRecords(int nThreads, DWORD* pdwTime, DWORD* pRowsProcessed)
  88. {
  89. CDataSource     db;
  90. CSession        session;
  91. CTable<CAccessor<CProduct>, CMyRowset>  product;
  92. HRESULT         hr;
  93. DWORD           dwTime;
  95. ATLTRACE("DoStuff: Startedn");
  96. hr = db.Open(_T("MSDASQL"), _T("OLE_DB_NWind_Jet"));
  97. ATLASSERT(SUCCEEDED(hr));
  98. if (FAILED(hr))
  99. return hr;
  101. hr = session.Open(db);
  102. ATLASSERT(SUCCEEDED(hr));
  103. if (FAILED(hr))
  104. return hr;
  106. CDBPropSet propset(DBPROPSET_ROWSET);
  107. propset.AddProperty(DBPROP_CANHOLDROWS, true);
  109. hr = product.Open(session, _T("Products"), &propset);
  110. ATLASSERT(SUCCEEDED(hr));
  111. if (FAILED(hr))
  112. return hr;
  114. DWORD  idThread;
  115. int i;
  116. HANDLE* phThread = new HANDLE[nThreads];
  117. if (phThread == NULL)
  118. return E_FAIL;
  120. dwTime = GetTickCount();
  121. for (i=0; i<nThreads; i++)
  122. {
  123. *(phThread+i) = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)ReadTable, &product, 0, &idThread);
  124. if (*(phThread+i) == NULL)
  125. ATLTRACE(_T("Couldn't create thread %dn"), i);
  126. }
  128. // Ensure all threads are finished before we end
  129. WaitForMultipleObjects(nThreads, phThread, TRUE, INFINITE);
  130. // Return the total time
  131. *pdwTime = GetTickCount() - dwTime;
  133. DWORD nTotalRows = 0, nRows;
  134. for (i=0; i<nThreads; i++)
  135. {
  136. GetExitCodeThread(*(phThread + i), &nRows);
  137. nTotalRows += nRows;
  138. CloseHandle(*(phThread + i));
  139. }
  140. *pRowsProcessed = nTotalRows;
  141. ATLTRACE("DoStuff: Endedn");
  142. return S_OK;
  143. }