You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
4.2 KiB
113 lines
4.2 KiB
//========= Copyright Valve Corporation, All rights reserved. ============// |
|
// |
|
// Purpose: |
|
// |
|
// $NoKeywords: $ |
|
//============================================================================= |
|
|
|
#ifndef SQLUTIL_H |
|
#define SQLUTIL_H |
|
|
|
#ifdef _WIN32 |
|
#pragma once |
|
#endif |
|
|
|
|
|
namespace GCSDK |
|
{ |
|
typedef CFmtStrN< 4096 > TSQLCmdStr; |
|
|
|
// Returns a long (1024 char) string of "?,?,?,?,?"... for use in IN clauses or INSERT statements |
|
const char *GetInsertArgString(); |
|
// Returns the number of characters that should be used for the specified number of parameters |
|
uint32 GetInsertArgStringChars( uint32 nNumParams ); |
|
// Returns the maximum number of parameters that are supported by the GetInsertArgString string |
|
uint32 GetInsertArgStringMaxParams(); |
|
|
|
|
|
void ConvertFieldToText( EGCSQLType eFieldType, uint8 *pubRecord, int cubRecord, char *rgchField, int cchField, bool bQuoteString = true ); |
|
void ConvertFieldArrayToInText( const CColumnInfo &columnInfo, byte *pubData, int cubData, char *rgchResult, bool bForPreparedStatement ); |
|
char *SQLTypeFromField( const CColumnInfo &colInfo, char *pchBuf, int cchBuf ); |
|
void EscapeStringValue( char *rgchField, int cchField ); |
|
void AppendConstraints( const CRecordInfo *pRecordInfo, const CColumnInfo *pColumnInfo, bool bForAdd, CFmtStrMax & sCmd ); |
|
void AppendTableConstraints( CRecordInfo *pRecordInfo, CFmtStrMax & sCmd ); |
|
void AppendConstraint( const char *pchTableName, const char *pchColumnName, int nColFlagConstraint, bool bForAdd, bool bClustered, |
|
CFmtStrMax & sCmd, int nFillFactor ); |
|
void BuildTablePKConstraintText( TSQLCmdStr *psStatement, CRecordInfo *pRecordInfo ); |
|
//void BuildSelectStatementText( CUtlVector<CQuery> *pVecQuery, bool bForPreparedStatement, char *pchStatement, int cchStatement ); |
|
|
|
void BuildInsertStatementText( TSQLCmdStr *psStatement, const CRecordInfo *pRecordInfo ); |
|
void BuildInsertAndReadStatementText( TSQLCmdStr *psStatement, CUtlVector<int> *pvecOutputFields, const CRecordInfo *pRecordInfo ) ; |
|
void BuildMergeStatementTextOnPKWhenMatchedUpdateWhenNotMatchedInsert( TSQLCmdStr *psStatement, const CRecordInfo *pRecordInfo ); |
|
void BuildMergeStatementTextOnPKWhenNotMatchedInsert( TSQLCmdStr *psStatement, const CRecordInfo *pRecordInfo ); |
|
void BuildSelectStatementText( TSQLCmdStr *psStatement, const CColumnSet & selectSet, const char *pchTopClause = NULL ); |
|
void BuildUpdateStatementText( TSQLCmdStr *psStatement, const CColumnSet & columnSet ); |
|
void BuildDeleteStatementText( TSQLCmdStr *psStatement, const CRecordInfo* pRecordInfo ); |
|
void AppendWhereClauseText( TSQLCmdStr *psClause, const CColumnSet & columnSet ); |
|
void BuildOutputClauseText( TSQLCmdStr *psClause, const CColumnSet & columnSet ); |
|
|
|
template< typename T > |
|
bool CopyResultToSchVector( IGCSQLResultSet *pResultSet, const CColumnSet & readSet, CUtlVector< T > *pvecRecords ) |
|
{ |
|
if ( pResultSet->GetRowCount() == 0 ) |
|
return true; |
|
|
|
FOR_EACH_COLUMN_IN_SET( readSet, nColumnIndex ) |
|
{ |
|
EGCSQLType eRecordType = readSet.GetColumnInfo( nColumnIndex ).GetType(); |
|
EGCSQLType eResultType = pResultSet->GetColumnType( nColumnIndex ); |
|
|
|
Assert( eResultType == eRecordType ); |
|
if( eRecordType != eResultType ) |
|
return false; |
|
} |
|
|
|
|
|
for( CSQLRecord sqlRecord( 0, pResultSet ); sqlRecord.IsValid(); sqlRecord.NextRow() ) |
|
{ |
|
int nRecord = pvecRecords->AddToTail(); |
|
|
|
FOR_EACH_COLUMN_IN_SET( readSet, nColumnIndex ) |
|
{ |
|
uint8 *pubData; |
|
uint32 cubData; |
|
|
|
DbgVerify( sqlRecord.BGetColumnData( nColumnIndex, &pubData, (int*)&cubData ) ); |
|
DbgVerify( pvecRecords->Element( nRecord ).BSetField( readSet.GetColumn( nColumnIndex), pubData, cubData ) ); |
|
} |
|
} |
|
return true; |
|
} |
|
|
|
//EResult UpdateOrInsertUnique( CSQLAccess &sqlAccess, int iTable, int iField, CRecordBase *pRecordBase, int iIndexID ); |
|
//bool BIsDuplicateInsertAttempt( const CSQLErrorInfo *pErr ); |
|
|
|
|
|
#define EXIT_WITH_SQL_FAILURE( ret ) { nRet = ret; goto Exit; } |
|
|
|
#define EXIT_ON_BOOL_FAILURE( bRet, msg ) \ |
|
{ \ |
|
if ( false == bRet ) \ |
|
{ \ |
|
SetSQLError( msg ); \ |
|
nRet = SQL_ERROR; \ |
|
goto Exit; \ |
|
} \ |
|
} |
|
|
|
#define SAFE_CLOSE_STMT( x ) \ |
|
if ( NULL != (x) ) \ |
|
{ \ |
|
SQLFreeHandle( SQL_HANDLE_STMT, (x) ); \ |
|
(x) = NULL; \ |
|
} |
|
|
|
#define SAFE_FREE_HANDLE( x, y ) \ |
|
if ( NULL != (x) ) \ |
|
{ \ |
|
SQLFreeHandle( y, (x) ); \ |
|
(x) = NULL; \ |
|
} |
|
|
|
} // namespace GCSDK |
|
#endif // SQLUTIL_H
|
|
|