source-engine/gcsdk/sqlaccess/sqlutil.cpp

919 lines
29 KiB
C++
Raw Normal View History

2020-04-22 16:56:21 +00:00
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//=============================================================================
#include "stdafx.h"
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
namespace GCSDK
{
const char *GetInsertArgString()
{
static char s_str[1024];
static bool s_bInit = false;
if ( !s_bInit )
{
for ( int i = 0; i < 1023; i++ )
{
s_str[i] = i % 2 == 0 ? '?' : ',';
}
s_str[1023] = NULL;
s_bInit = true;
}
return s_str;
}
uint32 GetInsertArgStringChars( uint32 nNumParams )
{
AssertMsg( nNumParams <= GetInsertArgStringMaxParams(), "Error: Requested more characters than are provided by the GetInsertArgString" );
if( nNumParams == 0 )
return 0;
return nNumParams * 2 - 1;
}
uint32 GetInsertArgStringMaxParams()
{
return 512;
}
//-----------------------------------------------------------------------------
// Purpose: Converts array of field data to text for SQL IN clause
// Input: columnInfo - schema of column being converted
// pubData - pointer to array of data to convert
// cubData - size of array of data
// rgchResult - pointer to output buffer
// cubResultLen - size of output buffer
// bForPreparedStatement - Should we prepare the text for a prepared statement or directly place the values?
//-----------------------------------------------------------------------------
void ConvertFieldArrayToInText( const CColumnInfo &columnInfo, byte *pubData, int cubData, char *rgchResult, int cubResultLen, bool bForPreparedStatement )
{
int32 cubLength = columnInfo.GetFixedSize();
Assert( cubData % cubLength == 0 );
int32 nArrayCount = cubData / cubLength;
int32 len = 0;
rgchResult[len++] = '(';
for( int i = 0; i < nArrayCount; ++i )
{
if ( bForPreparedStatement )
{
if ( i < nArrayCount - 1 )
{
rgchResult[len++] = '?';
rgchResult[len++] = ',';
}
else
{
rgchResult[len++] = '?';
rgchResult[len++] = ')';
}
}
else
{
switch ( columnInfo.GetType() )
{
case k_EGCSQLType_int8:
if ( i < nArrayCount - 1 )
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%d,", *( (byte *) pubData ) );
else
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%d)", *( (byte *) pubData ) );
break;
case k_EGCSQLType_int16:
if ( i < nArrayCount - 1 )
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%d,", *( (short *) pubData ) );
else
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%d)", *( (short *) pubData ) );
break;
case k_EGCSQLType_int32:
if ( i < nArrayCount - 1 )
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%d,", *( (int *) pubData ) );
else
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%d)", *( (int *) pubData ) );
break;
case k_EGCSQLType_int64:
if ( i < nArrayCount - 1 )
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%lld,", *( (int64 *) pubData ) );
else
len += Q_snprintf( rgchResult + len, cubResultLen - len, "%lld)", *( (int64 *) pubData ) );
break;
default:
AssertMsg( false, "Unsupported data type for non prepares statement with IN clause\n" );
rgchResult[0] = 0;
return;
}
}
if( len >= cubResultLen - 1 )
{
AssertMsg( false, "Generation of IN clause foverflowed\n" );
rgchResult[0] = 0;
return;
}
pubData += cubLength;
}
rgchResult[len] = 0;
return;
}
//-----------------------------------------------------------------------------
// Purpose: Converts field data to text equivalent for SQL statement
// Input: eFieldType - The type of the field to convert to text
// pubRecord - pointer to record data to convert
// cubRecord - size of record data
// rgchField - pointer to output buffer
// cchField - size of output buffer
//-----------------------------------------------------------------------------
void ConvertFieldToText( EGCSQLType eFieldType, uint8 *pubRecord, int cubRecord, char *rgchField, int cchField, bool bQuoteString )
{
char rgchTmp[k_cMedBuff];
switch ( eFieldType )
{
case k_EGCSQLType_int8:
Q_snprintf( rgchField, cchField, "%d", *( (byte *) pubRecord ) );
break;
case k_EGCSQLType_int16:
Q_snprintf( rgchField, cchField, "%d", *( (short *) pubRecord ) );
break;
case k_EGCSQLType_int32:
Q_snprintf( rgchField, cchField, "%d", *( (int *) pubRecord ) );
break;
case k_EGCSQLType_int64:
Q_snprintf( rgchField, cchField, "%lld", *( (int64 *) pubRecord ) );
break;
case k_EGCSQLType_float:
Q_snprintf( rgchField, cchField, "%f", *((float*) pubRecord) );
break;
case k_EGCSQLType_double:
Q_snprintf( rgchField, cchField, "%f", *((double*) pubRecord) );
break;
case k_EGCSQLType_String:
if ( pubRecord && *pubRecord )
{
Assert( cubRecord + 1 < Q_ARRAYSIZE( rgchTmp ) );
Q_memcpy( rgchTmp, (char *) pubRecord, cubRecord );
rgchTmp[cubRecord] = 0;
if ( bQuoteString )
{
EscapeStringValue( rgchTmp, Q_ARRAYSIZE( rgchTmp ) );
Q_snprintf( rgchField, cchField, "'%s'", rgchTmp );
}
else
{
Q_strncpy( rgchField, rgchTmp, cchField );
}
}
else
{
if ( bQuoteString )
{
Q_strncpy( rgchField, "''", cchField );
}
else
{
Q_strncpy( rgchField, "", cchField );
}
}
break;
case k_EGCSQLType_Blob:
case k_EGCSQLType_Image:
Q_strncpy( rgchField, "0x", cchField );
Q_binarytohex( pubRecord, cubRecord, rgchField + 2, cchField - 2 );
break;
default:
Assert( false );
break;
}
}
//-----------------------------------------------------------------------------
// Purpose: Returns the text SQL type for a given field
// Input: field - field to determine type for
// pchBuf - pointer to output buffer
// cchBuf - size of output buffer
// Output: returns pchBuf for convenience of one-line usage
//-----------------------------------------------------------------------------
char *SQLTypeFromField( const CColumnInfo &colInfo, char *pchBuf, int cchBuf )
{
EGCSQLType eType = colInfo.GetType();
*pchBuf = 0;
switch ( eType )
{
case k_EGCSQLType_int8:
Q_strncpy( pchBuf, "TINYINT", cchBuf );
break;
case k_EGCSQLType_int16:
Q_strncpy( pchBuf, "SMALLINT", cchBuf );
break;
case k_EGCSQLType_int32:
Q_strncpy( pchBuf, "INT", cchBuf );
break;
case k_EGCSQLType_int64:
Q_strncpy( pchBuf, "BIGINT", cchBuf );
break;
case k_EGCSQLType_float:
Q_strncpy( pchBuf, "REAL", cchBuf );
break;
case k_EGCSQLType_double:
Q_strncpy( pchBuf, "FLOAT", cchBuf );
break;
case k_EGCSQLType_String:
Q_snprintf( pchBuf, cchBuf, "VARCHAR(%d)", colInfo.GetMaxSize() );
break;
case k_EGCSQLType_Blob:
Q_snprintf( pchBuf, cchBuf, "VARBINARY(%d)", colInfo.GetMaxSize() );
break;
case k_EGCSQLType_Image:
Q_strncpy( pchBuf, "IMAGE", cchBuf );
break;
default:
Assert( false );
break;
}
return pchBuf;
}
//-----------------------------------------------------------------------------
// Purpose: Escapes any single quotes to a string value to double single quotes
// Input: rgchField - text to escape
// cchField - size of text buffer
// Notes: The text will be escaped and expanded in place in the buffer.
// In the worst case, the text may expand by 2x. (If the field is all
// single quotes.) So, you must pass in a buffer which is at least
// twice as long as the text length so we can guarantee to be able to
// escape the string.
//-----------------------------------------------------------------------------
void EscapeStringValue( char *rgchField, int cchField )
{
// TODO - what else do we need to escape? %() ...
char *pubCur = rgchField;
int nLen = 0;
int cSingleQuotes = 0;
// This function gets called on every text field we write but most text fields
// don't need to be escaped, so try to be as fast as possible in the normal case.
// first, walk through the string and count the string length and number of single quotes
while ( *pubCur )
{
if ( '\'' == *pubCur )
cSingleQuotes++;
nLen ++;
pubCur++;
}
// if no single quotes, nothing to do
if ( !cSingleQuotes )
return;
// caller must pass in a buffer that's long enough for expansion
Assert( nLen + cSingleQuotes + 1 <= cchField );
if ( !( nLen + cSingleQuotes + 1 <= cchField ) )
return;
// We know exactly how many characters the string will expand by (the # of single quotes). Walk backward
// and copy the characters into the right places. This touches each character only once.
pubCur = rgchField + nLen + cSingleQuotes;
*pubCur = 0;
pubCur--;
while ( pubCur > rgchField && cSingleQuotes > 0 )
{
// read pointer is offset from write pointer by # of remaining single quotes
char *pubRead = pubCur - cSingleQuotes;
Assert( pubRead >= rgchField );
// copy each character
*pubCur = *pubRead;
if ( '\'' == *pubRead )
{
// if the character is a single quote, back up one more and insert another single quote to escape it
pubCur --;
*pubCur = '\'';
// decrement # of single quotes remaining
cSingleQuotes --;
Assert( cSingleQuotes >= 0 );
}
pubCur--;
}
}
//-----------------------------------------------------------------------------
// Purpose: Adds constraint information to a SQL command to add or remove constraint
// Input: pchTableName - name of table
// pchColumnName - name of column
// nColFlagConstraint - flag with which constraint to
// bForAdd - whether constraint is being added or removed
// pchCmd - buffer to append SQL command to
// cchCmd - size of buffer
//-----------------------------------------------------------------------------
void AppendConstraint( const char *pchTableName, const char *pchColumnName, int nColFlagConstraint, bool bForAdd,
bool bClustered, CFmtStrMax & sCmd, int nFillFactor )
{
Assert( pchTableName && pchTableName[0] );
Assert( pchColumnName && pchColumnName[0] );
switch ( nColFlagConstraint )
{
case k_nColFlagPrimaryKey:
sCmd.AppendFormat( " CONSTRAINT %s_%s_PrimaryKey", pchTableName, pchColumnName);
if ( bForAdd )
{
sCmd += " PRIMARY KEY ";
if ( bClustered )
{
sCmd.AppendFormat( " CLUSTERED WITH (FILLFACTOR = %d) ", nFillFactor );
}
else
{
sCmd += "NONCLUSTERED";
}
}
break;
case k_nColFlagUnique:
/* do nothing - the uniqueness will be handled by creation of an index */
break;
case k_nColFlagAutoIncrement:
sCmd += " IDENTITY";
break;
default:
AssertMsg( false, "CSQLThread::AppendContraint: invalid constraint type" );
break;
}
}
//-----------------------------------------------------------------------------
// Purpose: Adds constraint information to a SQL command to add or remove constraint
// Input: pRecordInfo - record info describing table
// pColumnInfo - record info describing column
// bForAdd - whether constraint is being added or removed
// pchCmd - buffer to append SQL command to
// cchCmd - size of buffer
//-----------------------------------------------------------------------------
void AppendConstraints( const CRecordInfo *pRecordInfo, const CColumnInfo *pColumnInfo, bool bForAdd, CFmtStrMax & sCmd )
{
Assert( pRecordInfo != NULL );
Assert( pColumnInfo != NULL );
if ( pColumnInfo->BIsPrimaryKey() )
{
// any column in a PK can't be NULL.
if ( bForAdd )
{
sCmd += " NOT NULL";
}
// only add primary key constraint here if it is a single-column PK
if ( pRecordInfo->GetPrimaryKeyType() == k_EPrimaryKeyTypeSingle )
{
// get the fields on the primary key
const CUtlVector< FieldSet_t > &refFields = pRecordInfo->GetIndexFields( );
int nFillFactor = refFields.Element( pRecordInfo->GetPKIndex() ).GetFillFactor();
AppendConstraint( pRecordInfo->GetName(), pColumnInfo->GetName(), k_nColFlagPrimaryKey, bForAdd, pColumnInfo->BIsClustered(), sCmd, nFillFactor );
}
}
else if ( pColumnInfo->BIsUnique() )
{
AppendConstraint( pRecordInfo->GetName(), pColumnInfo->GetName(), k_nColFlagUnique, bForAdd, pColumnInfo->BIsClustered(), sCmd, 0 );
}
if ( pColumnInfo->BIsAutoIncrement() )
{
AppendConstraint( pRecordInfo->GetName(), pColumnInfo->GetName(), k_nColFlagAutoIncrement, bForAdd, pColumnInfo->BIsClustered(), sCmd, 0 );
}
}
//-----------------------------------------------------------------------------
// Purpose: Generates the "CONSTRAINT ..." text for the table primary key
//-----------------------------------------------------------------------------
void BuildTablePKConstraintText( TSQLCmdStr *psStatement, CRecordInfo *pRecordInfo )
{
const FieldSet_t& vecFields = pRecordInfo->GetPKFields( );
psStatement->sprintf( "CONSTRAINT %s_PrimaryKey PRIMARY KEY %s ( ",
pRecordInfo->GetName(),
vecFields.IsClustered() ? "CLUSTERED" : "NONCLUSTERED" );
for ( int nField = 0; nField < vecFields.GetCount(); nField++ )
{
// what field is the next column in our index?
int nThisField = vecFields.GetField( nField );
const CColumnInfo& columnInfo = pRecordInfo->GetColumnInfo(nThisField);
if (nField != 0)
{
*psStatement += ", ";
}
*psStatement += columnInfo.GetName();
}
// close our list
*psStatement += ") ";
if ( vecFields.GetFillFactor() != 0 )
{
// non-default fill factor, so specify it
psStatement->AppendFormat( " WITH FILLFACTOR = %d ",
vecFields.GetFillFactor() );
}
}
//-----------------------------------------------------------------------------
// Purpose: Adds constraint information to a SQL command to add or remove table-level constraints
// Input: pRecordInfo - record info describing table
// pchCmd - buffer to append SQL command to
// cchCmd - size of buffer
//-----------------------------------------------------------------------------
void AppendTableConstraints( CRecordInfo *pRecordInfo, CFmtStrMax & sCmd )
{
// the only supported table constraint is for PKs or FKs
if ( pRecordInfo->GetPrimaryKeyType() == k_EPrimaryKeyTypeMulti )
{
TSQLCmdStr tmp;
BuildTablePKConstraintText( &tmp, pRecordInfo );
sCmd += ", ";
sCmd += tmp;
}
// Look for FKs required on this table
// the only supported table constraint is for PKs or FKs
int cFKs = pRecordInfo->GetFKCount();
for( int i=0; i < cFKs; ++i )
{
FKData_t &fkData = pRecordInfo->GetFKData( i );
CFmtStr sColumns, sParentColumns;
FOR_EACH_VEC( fkData.m_VecColumnRelations, nCol )
{
FKColumnRelation_t &colRelation = fkData.m_VecColumnRelations[nCol];
if ( nCol > 0)
{
sColumns += ",";
sParentColumns += ",";
}
sColumns += colRelation.m_rgchCol;
sParentColumns += colRelation.m_rgchParentCol;
}
TSQLCmdStr sTmp;
sTmp.sprintf( ", CONSTRAINT %s FOREIGN KEY (%s) REFERENCES %s(%s) ON DELETE %s ON UPDATE %s",
fkData.m_rgchName, sColumns.Access(), fkData.m_rgchParentTableName, sParentColumns.Access(),
PchNameFromEForeignKeyAction( fkData.m_eOnDeleteAction ), PchNameFromEForeignKeyAction( fkData.m_eOnUpdateAction ) );
// add to the command
sCmd += sTmp;
}
}
//-----------------------------------------------------------------------------
// Purpose: Builds a SQL INSERT statement
// Input: psStatement - The string to put the statement into
// pRecordInfo - record info describing table inserting into
//-----------------------------------------------------------------------------
void BuildInsertStatementText( TSQLCmdStr *psStatement, const CRecordInfo *pRecordInfo )
{
psStatement->sprintf("INSERT INTO %s.%s (", GSchemaFull().GetDefaultSchemaNameForCatalog( pRecordInfo->GetESchemaCatalog() ), pRecordInfo->GetName() );
// build a string of the field names
int cColumns = pRecordInfo->GetNumColumns();
int nInsertable = 0;
bool bAddedBefore = false;
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( !columnInfo.BIsInsertable() )
continue;
nInsertable++;
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( columnInfo.GetName() );
}
psStatement->AppendFormat( ") VALUES (%.*s)", GetInsertArgStringChars( nInsertable ), GetInsertArgString() );
}
//-----------------------------------------------------------------------------
// Purpose: Builds a SQL INSERT statement
// IMPORTANT NOTE - This Insert statement will use the Microsoft SQL Server
// specific clause 'OUTPUT Inserted.ColumnName'
// The result of that will be that the SQL statement will return to us
// the columns that could not be specified by the Insert.
// At the time of writing, that is primarily AutoIncrement columns,
// however in theory we should be able to recover any computed column
// from SQL server, with the caveats specified at :
// http://msdn.microsoft.com/en-us/library/ms177564.aspx
//
// Input: psStatement - The output statement string
// pRecordInfo - record info describing table inserting into
//-----------------------------------------------------------------------------
void BuildInsertAndReadStatementText( TSQLCmdStr *psStatement, CUtlVector<int> *pvecOutputFields, const CRecordInfo *pRecordInfo )
{
psStatement->sprintf("INSERT INTO %s.%s (", GSchemaFull().GetDefaultSchemaNameForCatalog( pRecordInfo->GetESchemaCatalog() ), pRecordInfo->GetName() );
// build a string of the field names
int nInsertable = 0;
int cColumns = pRecordInfo->GetNumColumns();
bool bAddedBefore = false;
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( !columnInfo.BIsInsertable() )
continue;
nInsertable++;
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( columnInfo.GetName() );
}
bAddedBefore = false ;
int nOutputColumn = 0;
for( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn ) ;
//
// If we can't Insert it - we want SQL Server to tell us what value was stored
// in the column !!
//
if( !columnInfo.BIsInsertable() )
{
if( bAddedBefore )
psStatement->Append( ", INSERTED." );
else
psStatement->Append( ") OUTPUT INSERTED." );
bAddedBefore = true ;
psStatement->Append( columnInfo.GetName() );
pvecOutputFields->AddToTail( iColumn );
nOutputColumn++;
}
}
// add field values to SQL statement
psStatement->AppendFormat( " VALUES (%.*s)", GetInsertArgStringChars( nInsertable ), GetInsertArgString() );
}
//-----------------------------------------------------------------------------
// Purpose: Builds a SQL MERGE statement update or insert using in-flight values table
// Input: psStatement - The string to put the statement into
// pRecordInfo - record info describing table inserting into
//-----------------------------------------------------------------------------
void BuildMergeStatementTextOnPKWhenMatchedUpdateWhenNotMatchedInsert( TSQLCmdStr *psStatement, const CRecordInfo *pRecordInfo )
{
psStatement->sprintf( "MERGE INTO %s.%s WITH( HOLDLOCK, ROWLOCK ) T USING ( VALUES (%.*s) ) AS S(",
GSchemaFull().GetDefaultSchemaNameForCatalog( pRecordInfo->GetESchemaCatalog() ), pRecordInfo->GetName(),
GetInsertArgStringChars( pRecordInfo->GetNumColumns() ), GetInsertArgString() );
{
int cColumns = pRecordInfo->GetNumColumns();
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( iColumn )
psStatement->Append( ',' );
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( ") ON " );
// build a string of the PK columns
const FieldSet_t &fsPK = pRecordInfo->GetIndexFields()[pRecordInfo->GetPKIndex()];
{
int cColumns = fsPK.GetCount();
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( fsPK.GetField( iColumn ) );
if ( iColumn )
psStatement->Append( " AND " );
psStatement->Append( "T." );
psStatement->Append( columnInfo.GetName() );
psStatement->Append( "=S." );
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( " WHEN MATCHED THEN UPDATE SET " );
// build the update string
{
int cColumns = pRecordInfo->GetNumColumns();
bool bAddedBefore = false;
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
bool bThisColumnIsPartOfPK = false;
for ( int ipkCheck = 0; ipkCheck < fsPK.GetCount(); ++ipkCheck )
{
if ( iColumn == fsPK.GetField( ipkCheck ) )
{
bThisColumnIsPartOfPK = true;
break;
}
}
if ( bThisColumnIsPartOfPK )
continue;
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( columnInfo.GetName() );
psStatement->Append( "=S." );
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( " WHEN NOT MATCHED BY TARGET THEN INSERT (" );
// build a string of the field names
{
int cColumns = pRecordInfo->GetNumColumns();
bool bAddedBefore = false;
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( !columnInfo.BIsInsertable() )
continue;
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( ") VALUES (" );
{
int cColumns = pRecordInfo->GetNumColumns();
bool bAddedBefore = false;
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( !columnInfo.BIsInsertable() )
continue;
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( "S." );
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( ");" );
}
//-----------------------------------------------------------------------------
// Purpose: Builds a SQL MERGE statement using CTE_MergeParams as supplied table holding rows
// Input: psStatement - The string to put the statement into
// pRecordInfo - record info describing table inserting into
//-----------------------------------------------------------------------------
void BuildMergeStatementTextOnPKWhenNotMatchedInsert( TSQLCmdStr *psStatement, const CRecordInfo *pRecordInfo )
{
psStatement->sprintf( "MERGE INTO %s.%s WITH( HOLDLOCK, ROWLOCK ) T USING ( VALUES (%.*s) ) AS S(",
GSchemaFull().GetDefaultSchemaNameForCatalog( pRecordInfo->GetESchemaCatalog() ), pRecordInfo->GetName(),
GetInsertArgStringChars( pRecordInfo->GetNumColumns() ), GetInsertArgString() );
{
int cColumns = pRecordInfo->GetNumColumns();
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( iColumn )
psStatement->Append( ',' );
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( ") ON " );
// build a string of the PK columns
const FieldSet_t &fsPK = pRecordInfo->GetIndexFields()[pRecordInfo->GetPKIndex()];
{
int cColumns = fsPK.GetCount();
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( fsPK.GetField( iColumn ) );
if ( iColumn )
psStatement->Append( " AND " );
psStatement->Append( "T." );
psStatement->Append( columnInfo.GetName() );
psStatement->Append( "=S." );
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( " WHEN NOT MATCHED BY TARGET THEN INSERT (" );
// build a string of the field names
{
int cColumns = pRecordInfo->GetNumColumns();
bool bAddedBefore = false;
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( !columnInfo.BIsInsertable() )
continue;
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( ") VALUES (" );
{
int cColumns = pRecordInfo->GetNumColumns();
bool bAddedBefore = false;
for ( int iColumn = 0; iColumn < cColumns; iColumn++ )
{
const CColumnInfo &columnInfo = pRecordInfo->GetColumnInfo( iColumn );
if ( !columnInfo.BIsInsertable() )
continue;
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( "S." );
psStatement->Append( columnInfo.GetName() );
}
}
psStatement->Append( ");" );
}
void BuildSelectStatementText( TSQLCmdStr *psStatement, const CColumnSet & selectSet, const char *pchTopClause )
{
*psStatement = "SELECT ";
if( pchTopClause )
{
psStatement->Append( pchTopClause );
psStatement->Append( ' ' );
}
// build a string of the field names
bool bAddedBefore = false;
FOR_EACH_COLUMN_IN_SET( selectSet, nColumnIndex )
{
const CColumnInfo &columnInfo = selectSet.GetColumnInfo( nColumnIndex );
if ( bAddedBefore )
psStatement->Append( ',' );
bAddedBefore = true;
psStatement->Append( columnInfo.GetName() );
}
psStatement->Append( " FROM ");
psStatement->Append( GSchemaFull().GetDefaultSchemaNameForCatalog( selectSet.GetRecordInfo()->GetESchemaCatalog() ) );
psStatement->Append( '.' );
psStatement->Append( selectSet.GetRecordInfo()->GetName() );
}
//-----------------------------------------------------------------------------
// Purpose: Builds a SQL UPDATE statement
// Input: pRecordInfo - record info describing table inserting into
// bForPreparedStatement - if true, inserts values as '?' for later
// binding. If false, values are inserted in text.
// pchStatement - pointer to buffer to build statement in
// cchStatement - size of buffer
// pSQLRecord - pointer to record with data to update
// iColumnMatch - column to use for WHERE condition
// pvMatch - data value to use for WHERE condition
// cubMatch - size of pvMatch data
// rgiColumnUpdate - array of column #'s to update
// ciColumnUpdate - count of column #'s to update
//-----------------------------------------------------------------------------
void BuildUpdateStatementText( TSQLCmdStr *psStatement, const CColumnSet & updateColumns )
{
// build the UPDATE statement
psStatement->sprintf( "UPDATE %s.%s SET ", GSchemaFull().GetDefaultSchemaNameForCatalog( updateColumns.GetRecordInfo()->GetESchemaCatalog() ), updateColumns.GetRecordInfo()->GetName() );
// add each field we're updating to the UPDATE statement
FOR_EACH_COLUMN_IN_SET( updateColumns, nColumnIndex )
{
const CColumnInfo &columnInfo = updateColumns.GetColumnInfo( nColumnIndex );
if( nColumnIndex > 0 )
psStatement->Append( ',' );
psStatement->Append( columnInfo.GetName() );
psStatement->Append( "=?" );
}
}
//-----------------------------------------------------------------------------
// Purpose: Builds a SQL UPDATE statement
//-----------------------------------------------------------------------------
void BuildDeleteStatementText( TSQLCmdStr *psStatement, const CRecordInfo *pRecordInfo )
{
psStatement->sprintf( "DELETE FROM %s.%s", GSchemaFull().GetDefaultSchemaNameForCatalog( pRecordInfo->GetESchemaCatalog() ), pRecordInfo->GetName() );
}
//-----------------------------------------------------------------------------
// Purpose: Builds a where clause for the provided fields
//-----------------------------------------------------------------------------
void AppendWhereClauseText( TSQLCmdStr *psClause, const CColumnSet & columnSet )
{
// add each field we're updating to the UPDATE statement
FOR_EACH_COLUMN_IN_SET( columnSet, nColumnIndex )
{
const CColumnInfo &columnInfo = columnSet.GetColumnInfo( nColumnIndex );
if( nColumnIndex > 0 )
psClause->Append( " AND ");
psClause->Append( columnInfo.GetName() );
psClause->Append( "=?" );
}
}
//-----------------------------------------------------------------------------
// Purpose: Builds an OUTPUT [fields] INTO [table] for the provided fields/data
//-----------------------------------------------------------------------------
void BuildOutputClauseText( TSQLCmdStr *psClause, const CColumnSet & columnSet )
{
*psClause = " OUTPUT ";
FOR_EACH_COLUMN_IN_SET( columnSet, nColumnIndex )
{
const CColumnInfo &columnInfo = columnSet.GetColumnInfo( nColumnIndex );
if( nColumnIndex > 0 )
psClause->Append( ", ");
psClause->Append( " ? AS " );
psClause->Append( columnInfo.GetName() );
}
psClause->Append( " INTO " );
psClause->Append( columnSet.GetRecordInfo()->GetName() );
}
////-----------------------------------------------------------------------------
//// Purpose: our own special "upsert" into a column with a uniqueness constraint
////-----------------------------------------------------------------------------
//EResult UpdateOrInsertUnique( CSQLAccess &sqlAccess, int iTable, int iField, CRecordBase *pRecordBase, int iIndexID )
//{
// // attempt an update - if it fails due to duplicate primary key, they can't use this
// // url (it's taken) - if it succeeds but affects 0 rows, they didn't have a vanity url
// // and we need to do an insert (which could again fail due to primary key constraints)
// int cRecordsUpdated = 0;
// bool bRet = sqlAccess.BYieldingUpdateFieldFromRecordWithIndex( iTable, &cRecordsUpdated, iField, pRecordBase, iIndexID );
// if ( !bRet )
// {
// // ODBC is the suck - give me Spring JDBC templates, please.
// if ( sqlAccess.GetLastError()->IsDuplicateInsertAttempt() )
// {
// return k_EResultDuplicateName;
// }
// return k_EResultFail;
// }
// else if ( 0 == cRecordsUpdated )
// {
// // the user didn't have an entry, so insert one.
// bRet = sqlAccess.BYieldingInsertRecord( iTable, pRecordBase );
// if ( !bRet )
// {
// // ODBC is the suck - give me Spring JDBC templates, please.
// if ( sqlAccess.GetLastError()->IsDuplicateInsertAttempt() )
// {
// return k_EResultDuplicateName;
// }
// return k_EResultFail;
// }
// }
// return k_EResultOK;
//}
//
} // namespace GCSDK