119 lines
4.9 KiB
C++
119 lines
4.9 KiB
C++
#include <windows.h>
|
||
#define OLEDBVER 0x0250 // enable ICommandTree interface
|
||
#include <ole2.h>
|
||
#include <oledb.h>
|
||
#include <cmdtree.h>
|
||
#include <ntquery.h>
|
||
|
||
// the maximum number of columns that the user can request in their
|
||
// comma delimited list
|
||
#define MAX_COLUMNS 6
|
||
#define MAX_FRIENDLYNAME 128
|
||
|
||
typedef HRESULT
|
||
(STDAPICALLTYPE *PCIMAKEICOMMAND)(ICommand ** ppQuery,
|
||
ULONG cScope,
|
||
DWORD const * aDepths,
|
||
WCHAR const * const * awcsScope,
|
||
WCHAR const * const * awcsCat,
|
||
WCHAR const * const * awcsMachine );
|
||
|
||
typedef HRESULT
|
||
(STDAPICALLTYPE *PCITEXTTOFULLTREE)(WCHAR const * pwszRestriction,
|
||
WCHAR const * pwszColumns,
|
||
WCHAR const * pwszSortColumns, // may be NULL
|
||
WCHAR const * pwszGroupings, // may be NULL
|
||
DBCOMMANDTREE * * ppTree,
|
||
ULONG cProperties,
|
||
/*optional*/ CIPROPERTYDEF * pReserved,
|
||
LCID LocaleID );
|
||
|
||
class CIndexServerQuery {
|
||
public:
|
||
// This is used to globally initalize the CIndexServerQuery classes
|
||
// by having it load the necessary bits that it needs from Tripoli.
|
||
// It should be called on service startup by any service which
|
||
// expects to use CIndexServerQuery
|
||
static HRESULT GlobalInitialize();
|
||
// This is the global shutdown code... it should be called on service
|
||
// shutdown
|
||
static HRESULT GlobalShutdown();
|
||
// constructor
|
||
CIndexServerQuery();
|
||
//
|
||
// start the query going.
|
||
//
|
||
// arguments:
|
||
// [in] bDeepQuery - TRUE if deep query, FALSE if shallow
|
||
// [in] pwszQueryString - the tripoli query string
|
||
// [in] pwszMachine - the machine to query against (NULL for localhost)
|
||
// [in] pwszCatalog - the tripoli catalog to query against (name or
|
||
// path is okay).
|
||
// [in] pwszScope - the tripoli scope to query against. NULL for the
|
||
// default scope (\).
|
||
// [in] pwszColumns - the columns to return. supported columns are
|
||
// filename,newsarticleid,newsgroup,newsmsgid.
|
||
// note: this string gets altered internally, so
|
||
// it might change from what you pass in.
|
||
// [in] pwszSortOrder - sort priority for the columns. NULL to return
|
||
// unsorted
|
||
//
|
||
HRESULT MakeQuery( BOOL bDeepQuery, WCHAR const *pwszQueryString,
|
||
WCHAR const *pwszMachine, WCHAR const *pwszCatalog,
|
||
WCHAR const *pwszScope, WCHAR *pwszColumns, WCHAR const *pwszSortOrder,
|
||
LCID LocaleID = GetSystemDefaultLCID(), DWORD cMaxRows = 0);
|
||
//
|
||
// get the results from the query
|
||
//
|
||
// arguments:
|
||
// [in] pcResults - pointer to the a size of the ppvResults array
|
||
// [out] pcResults - the number of items put into ppvResults
|
||
// [in/out] ppvResults - an array of pointers to PROPVARIANTS. this is
|
||
// filled in by column for up to *pcResults
|
||
// rows.
|
||
// [out] pfMore - set to TRUE if there are more results, FALSE if
|
||
// this is the last set of results.
|
||
//
|
||
// usage:
|
||
// DWORD cResults;
|
||
// PROPVARIANT *rgpvResults[COLUMNS * ROWS];
|
||
// BOOL fMore;
|
||
// cResults = ROWS;
|
||
// HRESULT hr;
|
||
// hr = GetQueryResults(&cResults, rgpvResults, &fMore);
|
||
// if (FAILED(hr)) /* handle error */
|
||
// else {
|
||
// for (i = 0; i < ROWS; i++) {
|
||
// PROPVARIANT **ppvColumn = rgpvResults + (j * ROWS);
|
||
// /* ppvColumn[0] has column 0 in row j */
|
||
// /* ppvColumn[1] has column 1 in row j */
|
||
// /* etc... */
|
||
// }
|
||
// }
|
||
//
|
||
HRESULT GetQueryResults(DWORD *pcResults, PROPVARIANT **ppvResults,
|
||
BOOL *pfMore);
|
||
~CIndexServerQuery();
|
||
|
||
private:
|
||
// class globals
|
||
static HMODULE m_hmQuery;
|
||
static PCIMAKEICOMMAND m_pfnCIMakeICommand;
|
||
static PCITEXTTOFULLTREE m_pfnCITextToFullTree;
|
||
|
||
// class variables
|
||
HACCESSOR m_hAccessor;
|
||
IRowset *m_pRowset;
|
||
DWORD m_cCols;
|
||
BOOL m_fNoMoreRows;
|
||
HROW *m_phRows;
|
||
DBCOUNTITEM m_cRowHandlesAllocated;
|
||
DBCOUNTITEM m_cRowHandlesInUse;
|
||
struct tagCIPROPERTYDEF *m_pPropDef;
|
||
DWORD m_cPropDef;
|
||
|
||
HRESULT CreateAccessor(WCHAR *szColumns);
|
||
HRESULT BuildFriendlyNames(const WCHAR *pwszQueryString);
|
||
void ReleaseAccessor();
|
||
};
|
||
|