动态确定返回给使用者的列
PROVIDER_COLUMN_ENTRY 宏通常处理 IColumnsInfo::GetColumnsInfo 调用。 但是，由于使用者可能选择使用书签，因此提供者必须能够根据使用者是否请求书签来更改返回的列。
要处理 IColumnsInfo::GetColumnsInfo 调用，请从 CustomRS.h 中的 CCustomWindowsFile 用户记录中用于删除定义函数 GetColumnInfo 的 PROVIDER_COLUMN_MAP，并将其替换为你自己的 GetColumnInfo 函数的定义：
////////////////////////////////////////////////////////////////////////
// CustomRS.H
class CCustomWindowsFile
{
public:
DWORD dwBookmark;
static const int iSize = 256;
TCHAR szCommand[iSize];
TCHAR szText[iSize];
TCHAR szCommand2[iSize];
TCHAR szText2[iSize];
static ATLCOLUMNINFO* GetColumnInfo(void* pThis, ULONG* pcCols);
bool operator==(const CCustomWindowsFile& am)
{
return (lstrcmpi(szCommand, am.szCommand) == 0);
}
};
接下来，在 CustomRS.cpp 中实现 GetColumnInfo 函数，如下代码所示。
GetColumnInfo 首先检查是否设置了 OLE DB 属性 DBPROP_BOOKMARKS。 要获取该属性，GetColumnInfo 将使用指向行集对象的指针 (pRowset)。 pThis 指针表示创建行集的类，即存储属性映射的类。 GetColumnInfo 类型可将 pThis 指针类型转换为 RCustomRowset 指针。
要检查 DBPROP_BOOKMARKS 属性，GetColumnInfo 将使用 IRowsetInfo 接口，你可以通过对 pRowset 接口调用 QueryInterface 来获取该接口。 作为替代方案，你可以改用 ATL CComQIPtr 方法。
////////////////////////////////////////////////////////////////////
// CustomRS.cpp
ATLCOLUMNINFO* CCustomWindowsFile::GetColumnInfo(void* pThis, ULONG* pcCols)
{
static ATLCOLUMNINFO _rgColumns[5];
ULONG ulCols = 0;
// Check the property flag for bookmarks; if it is set, set the zero
// ordinal entry in the column map with the bookmark information.
CCustomRowset* pRowset = (CCustomRowset*) pThis;
CComQIPtr<IRowsetInfo, &IID_IRowsetInfo> spRowsetProps = pRowset;
CDBPropIDSet set(DBPROPSET_ROWSET);
set.AddPropertyID(DBPROP_BOOKMARKS);
DBPROPSET* pPropSet = NULL;
ULONG ulPropSet = 0;
HRESULT hr;
if (spRowsetProps)
hr = spRowsetProps->GetProperties(1, &set, &ulPropSet, &pPropSet);
if (pPropSet)
{
CComVariant var = pPropSet->rgProperties[0].vValue;
CoTaskMemFree(pPropSet->rgProperties);
CoTaskMemFree(pPropSet);
if (SUCCEEDED(hr) && (var.boolVal == VARIANT_TRUE))
{
ADD_COLUMN_ENTRY_EX(ulCols, OLESTR("Bookmark"), 0, sizeof(DWORD),
DBTYPE_BYTES, 0, 0, GUID_NULL, CCustomWindowsFile, dwBookmark,
DBCOLUMNFLAGS_ISBOOKMARK)
ulCols++;
}
}
// Next, set the other columns up.
ADD_COLUMN_ENTRY(ulCols, OLESTR("Command"), 1, 256, DBTYPE_STR, 0xFF, 0xFF,
GUID_NULL, CCustomWindowsFile, szCommand)
ulCols++;
ADD_COLUMN_ENTRY(ulCols, OLESTR("Text"), 2, 256, DBTYPE_STR, 0xFF, 0xFF,
GUID_NULL, CCustomWindowsFile, szText)
ulCols++;
ADD_COLUMN_ENTRY(ulCols, OLESTR("Command2"), 3, 256, DBTYPE_STR, 0xFF, 0xFF,
GUID_NULL, CCustomWindowsFile, szCommand2)
ulCols++;
ADD_COLUMN_ENTRY(ulCols, OLESTR("Text2"), 4, 256, DBTYPE_STR, 0xFF, 0xFF,
GUID_NULL, CCustomWindowsFile, szText2)
ulCols++;
if (pcCols != NULL)
*pcCols = ulCols;
return _rgColumns;
}
此示例使用静态数组来保存列信息。 如果使用者不想要使用书签列，则数组中的一个条目会变为未使用状态。 要处理这些信息，你需要创建两个数组宏：ADD_COLUMN_ENTRY 和 ADD_COLUMN_ENTRY_EX。 ADD_COLUMN_ENTRY_EX 采用额外参数 flags，如果你指定书签列，则需要该参数。
////////////////////////////////////////////////////////////////////////
// CustomRS.h
#define ADD_COLUMN_ENTRY(ulCols, name, ordinal, colSize, type, precision, scale, guid, dataClass, member) \
_rgColumns[ulCols].pwszName = (LPOLESTR)name; \
_rgColumns[ulCols].pTypeInfo = (ITypeInfo*)NULL; \
_rgColumns[ulCols].iOrdinal = (ULONG)ordinal; \
_rgColumns[ulCols].dwFlags = 0; \
_rgColumns[ulCols].ulColumnSize = (ULONG)colSize; \
_rgColumns[ulCols].wType = (DBTYPE)type; \
_rgColumns[ulCols].bPrecision = (BYTE)precision; \
_rgColumns[ulCols].bScale = (BYTE)scale; \
_rgColumns[ulCols].cbOffset = offsetof(dataClass, member);
#define ADD_COLUMN_ENTRY_EX(ulCols, name, ordinal, colSize, type, precision, scale, guid, dataClass, member, flags) \
_rgColumns[ulCols].pwszName = (LPOLESTR)name; \
_rgColumns[ulCols].pTypeInfo = (ITypeInfo*)NULL; \
_rgColumns[ulCols].iOrdinal = (ULONG)ordinal; \
_rgColumns[ulCols].dwFlags = flags; \
_rgColumns[ulCols].ulColumnSize = (ULONG)colSize; \
_rgColumns[ulCols].wType = (DBTYPE)type; \
_rgColumns[ulCols].bPrecision = (BYTE)precision; \
_rgColumns[ulCols].bScale = (BYTE)scale; \
_rgColumns[ulCols].cbOffset = offsetof(dataClass, member); \
memset(&(_rgColumns[ulCols].columnid), 0, sizeof(DBID)); \
_rgColumns[ulCols].columnid.uName.pwszName = (LPOLESTR)name;
在 GetColumnInfo 函数中，书签宏的使用如下：
ADD_COLUMN_ENTRY_EX(ulCols, OLESTR("Bookmark"), 0, sizeof(DWORD),
DBTYPE_BYTES, 0, 0, GUID_NULL, CAgentMan, dwBookmark,
DBCOLUMNFLAGS_ISBOOKMARK)
你现在可以编译和运行增强的提供程序。 要测试提供程序，请修改测试使用者，如实现简单使用者中所述。 使用提供程序运行测试使用者，并验证测试使用者是否从提供程序检索正确的字符串。