160 lines
4.0 KiB
C++
160 lines
4.0 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (c) 1997-1999, Microsoft Corp. All rights reserved.
|
|
//
|
|
// FILE
|
|
//
|
|
// SimTable.h
|
|
//
|
|
// SYNOPSIS
|
|
//
|
|
// This file describes the class CSimpleTable
|
|
//
|
|
// MODIFICATION HISTORY
|
|
//
|
|
// 10/31/1997 Original version.
|
|
// 02/09/1998 Reorganized some things to make is easier to extend.
|
|
// Thierry Perraut
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef _SIMTABLE_H_
|
|
#define _SIMTABLE_H_
|
|
|
|
#include <atlbase.h>
|
|
#include <oledb.h>
|
|
#include <bitvec.h>
|
|
struct DBBinding;
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CLASS
|
|
//
|
|
// CSimpleTable
|
|
//
|
|
// DESCRIPTION
|
|
//
|
|
// This class provides a simple read-only wrapper for iterating through a
|
|
// rowset and retrieving information. The interface is based on the ATL
|
|
// CTable<> class. I kept all the function signatures the same, so the two
|
|
// should be almost interchangeable. The main difference is that CTable<>
|
|
// opens a table and retrieves a rowset, while CSimpleTable is handed a
|
|
// rowset that was retrieved elsewhere.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
class CSimpleTable
|
|
{
|
|
public:
|
|
|
|
CSimpleTable();
|
|
~CSimpleTable();
|
|
|
|
HRESULT Attach(IRowset* pRowset);
|
|
IRowset* Detach();
|
|
|
|
HRESULT MoveFirst();
|
|
HRESULT MoveNext();
|
|
|
|
HRESULT Insert();
|
|
HRESULT Delete();
|
|
HRESULT SetData();
|
|
|
|
void DiscardChanges()
|
|
{
|
|
dirty.reset();
|
|
}
|
|
|
|
DBORDINAL GetColumnCount() const
|
|
{
|
|
return numColumns;
|
|
}
|
|
|
|
DBCOLUMNFLAGS GetColumnFlags(DBORDINAL nOrdinal) const
|
|
{
|
|
return columnInfo[OrdinalToColumn(nOrdinal)].dwFlags;
|
|
}
|
|
|
|
LPCWSTR GetColumnName(DBORDINAL nOrdinal) const
|
|
{
|
|
return columnInfo[OrdinalToColumn(nOrdinal)].pwszName;
|
|
}
|
|
|
|
DBTYPE GetColumnType(DBORDINAL nOrdinal) const
|
|
{
|
|
return columnInfo[OrdinalToColumn(nOrdinal)].wType;
|
|
}
|
|
|
|
DBLENGTH GetLength(DBORDINAL nOrdinal) const;
|
|
|
|
bool GetOrdinal(LPCWSTR szColumnName, DBORDINAL* pOrdinal) const;
|
|
|
|
DBSTATUS GetStatus(DBORDINAL nOrdinal) const;
|
|
|
|
const void* GetValue(DBORDINAL nOrdinal) const
|
|
{
|
|
return _GetDataPtr(nOrdinal);
|
|
}
|
|
|
|
template <class T>
|
|
void SetValue(DBORDINAL nOrdinal, const T& t)
|
|
{
|
|
*(T*)_GetDataPtr(nOrdinal) = t;
|
|
}
|
|
|
|
void SetValue(DBORDINAL nOrdinal, PCSTR szValue)
|
|
{
|
|
strcpy((PSTR)_GetDataPtr(nOrdinal), szValue);
|
|
}
|
|
|
|
void SetValue(DBORDINAL nOrdinal, PSTR szValue)
|
|
{
|
|
strcpy((PSTR)_GetDataPtr(nOrdinal), szValue);
|
|
}
|
|
|
|
bool HasBookmark() const
|
|
{
|
|
return (numColumns > 0) && (columnInfo->iOrdinal == 0);
|
|
}
|
|
|
|
protected:
|
|
|
|
enum { FETCH_QUANTUM = 256 }; // The number of rows fetched at a time.
|
|
|
|
HRESULT CreateAccessorForWrite(HACCESSOR* phAccessor);
|
|
|
|
void* _GetDataPtr(DBORDINAL nOrdinal);
|
|
|
|
const void* _GetDataPtr(DBORDINAL nOrdinal) const
|
|
{
|
|
return buffer +
|
|
(ULONG_PTR)columnInfo[OrdinalToColumn(nOrdinal)].pTypeInfo;
|
|
}
|
|
|
|
HRESULT ReleaseRows();
|
|
|
|
DBORDINAL OrdinalToColumn(DBORDINAL nOrdinal) const
|
|
{
|
|
return nOrdinal -= columnInfo->iOrdinal;
|
|
}
|
|
|
|
// Various representations of the rowset being manipulated.
|
|
CComPtr<IRowset> rowset;
|
|
CComPtr<IAccessor> accessor;
|
|
CComPtr<IRowsetChange> rowsetChange;
|
|
|
|
DBORDINAL numColumns; // Number of columns in the table.
|
|
DBCOLUMNINFO* columnInfo; // Column info.
|
|
OLECHAR* stringsBuffer; // Buffer used by columnInfo.
|
|
DBBinding* columnBinding; // Column bindings.
|
|
HACCESSOR readAccess; // Handle for read accessor.
|
|
PBYTE buffer; // Accessor buffer.
|
|
DBLENGTH bufferLength; // Length of accessor buffer.
|
|
HROW row[FETCH_QUANTUM]; // Array of row handles.
|
|
DBCOUNTITEM numRows; // Number of rows in the row array.
|
|
DBCOUNTITEM currentRow; // Current row being accessed.
|
|
BitVector dirty; // Columns that have been modified.
|
|
bool endOfRowset; // True if we've reached the end of the rowset.
|
|
};
|
|
|
|
#endif // _SIMTABLE_H_
|