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

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

Windows编程

开发平台：

Visual C++

SESSION.H：源码内容
  1. // Session.h : Declaration of the CDirSession
  2. //
  3. // This is a part of the ActiveX Template Library.
  4. // Copyright (C) 1998 Microsoft Corporation
  5. // All rights reserved.
  6. //
  7. // This source code is only intended as a supplement to the
  8. // ActiveX Template Library Reference and related
  9. // electronic documentation provided with the library.
  10. // See these sources for detailed information regarding the
  11. // ActiveX Template Library product.
  13. #ifndef __SESSION_H_
  14. #define __SESSION_H_
  16. #include "resource.h"       // main symbols
  17. #include "rowset.h"
  20. class CDirTRSchemaRowset;
  21. class CDirColSchemaRowset;
  22. class CDirPTSchemaRowset;
  23. /////////////////////////////////////////////////////////////////////////////
  24. // CDirSession
  25. class ATL_NO_VTABLE CDirSession :
  26. public CComObjectRootEx<CComMultiThreadModel>,
  27. public IGetDataSourceImpl<CDirSession>,
  28. public ISessionPropertiesImpl<CDirSession>,
  29. public IObjectWithSiteSessionImpl<CDirSession>,
  30. public IDBCreateCommandImpl<CDirSession, CDirCommand>,
  31. public IOpenRowsetImpl<CDirSession>,
  32. public IDBSchemaRowsetImpl<CDirSession>
  33. {
  34. public:
  36. CDirSession()
  37. {
  38. }
  39. HRESULT FinalConstruct()
  40. {
  41. return FInit();
  42. }
  43. STDMETHOD(OpenRowset)(IUnknown *pUnkOuter, DBID *pTableID, DBID *pIndexID, REFIID riid,
  44.    ULONG cPropertySets, DBPROPSET rgPropertySets[], IUnknown **ppRowset)
  45. {
  46. CDirRowset* pRowset;
  47. return CreateRowset(pUnkOuter, pTableID, pIndexID, riid, cPropertySets,
  48. rgPropertySets, ppRowset, pRowset);
  49. }
  51. BEGIN_PROPSET_MAP(CDirSession)
  52. CHAIN_PROPERTY_SET(CDirCommand)
  53. END_PROPSET_MAP()
  55. BEGIN_COM_MAP(CDirSession)
  56. COM_INTERFACE_ENTRY(IGetDataSource)
  57. COM_INTERFACE_ENTRY(IDBSchemaRowset)
  58. COM_INTERFACE_ENTRY(ISessionProperties)
  59. COM_INTERFACE_ENTRY(IObjectWithSite)
  60. COM_INTERFACE_ENTRY(IOpenRowset)
  61. COM_INTERFACE_ENTRY(IDBCreateCommand)
  62. END_COM_MAP()
  64. BEGIN_SCHEMA_MAP(CDirSession)
  65. SCHEMA_ENTRY(DBSCHEMA_TABLES, CDirTRSchemaRowset)
  66. SCHEMA_ENTRY(DBSCHEMA_COLUMNS, CDirColSchemaRowset)
  67. SCHEMA_ENTRY(DBSCHEMA_PROVIDER_TYPES, CDirPTSchemaRowset)
  68. END_SCHEMA_MAP()
  69. };
  71. class CDirTRSchemaRowset :
  72. public CRowsetImpl< CDirTRSchemaRowset, CTABLESRow, CDirSession>
  73. {
  74. public:
  75. HRESULT Execute(LONG* pcRowsAffected, ULONG, const VARIANT*)
  76. {
  77. USES_CONVERSION;
  78. CWindowsFile wf;
  79. CTABLESRow trData;
  80. lstrcpyW(trData.m_szType, OLESTR("TABLE"));
  81. lstrcpyW(trData.m_szDesc, OLESTR("The Directory Table"));
  83. HANDLE hFile = INVALID_HANDLE_VALUE;
  84. TCHAR szDir[MAX_PATH + 1];
  85. DWORD cbCurDir = GetCurrentDirectory(MAX_PATH, szDir);
  86. lstrcat(szDir, _T("\*.*"));
  87. hFile = FindFirstFile(szDir, &wf);
  88. if (hFile == INVALID_HANDLE_VALUE)
  89. return E_FAIL; // User doesn't have a c: drive
  90. FindClose(hFile);
  91. lstrcpynW(trData.m_szTable, T2OLE(szDir), SIZEOF_MEMBER(CTABLESRow, m_szTable));
  92. if (!m_rgRowData.Add(trData))
  93. return E_OUTOFMEMORY;
  94. *pcRowsAffected = 1;
  95. return S_OK;
  96. }
  98. };
  100. class CDirColSchemaRowset :
  101. public CRowsetImpl< CDirColSchemaRowset, CCOLUMNSRow, CDirSession>
  102. {
  103. public:
  104. HRESULT Execute(LONG* pcRowsAffected, ULONG, const VARIANT*)
  105. {
  106. USES_CONVERSION;
  107. CWindowsFile wf;
  108. HANDLE hFile = INVALID_HANDLE_VALUE;
  109. TCHAR szDir[MAX_PATH + 1];
  110. DWORD cbCurDir = GetCurrentDirectory(MAX_PATH, szDir);
  111. lstrcat(szDir, _T("\*.*"));
  112. hFile = FindFirstFile(szDir, &wf);
  113. if (hFile == INVALID_HANDLE_VALUE)
  114. return E_FAIL; // User doesn't have a c: drive
  115. FindClose(hFile);// szDir has got the tablename
  117. DBID dbid;
  118. memset(&dbid, 0, sizeof(DBID));
  119. dbid.uName.pwszName = T2OLE(szDir);
  120. dbid.eKind = DBKIND_NAME;
  121. return InitFromRowset < _RowsetArrayType > (m_rgRowData, &dbid, NULL, m_spUnkSite, pcRowsAffected);
  122. }
  123. };
  125. class CDirPTSchemaRowset :
  126. public CRowsetImpl< CDirColSchemaRowset, CPROVIDER_TYPERow, CDirSession>
  127. {
  128. public:
  129. HRESULT Execute(LONG* pcRowsAffected, ULONG, const VARIANT*)
  130. {
  131. return S_OK;
  132. }
  133. };
  135. #endif //__SESSION_H_