private static void writeGetRowMatchingIdMethod(
DBConnection cn, TextWriter writer, Database database, string table, TableColumns tableColumns, bool isSmallTable, bool tableUsesRowVersionedCaching,
bool isRevisionHistoryTable)
{
writer.WriteLine("public static Row GetRowMatchingId( " + tableColumns.KeyColumns.Single().DataTypeName + " id, bool returnNullIfNoMatch = false ) {");
if (isSmallTable)
{
writer.WriteLine("var cache = Cache.Current;");
var commandConditionsExpression = isRevisionHistoryTable ? "getLatestRevisionsCondition().ToSingleElementArray()" : "new InlineDbCommandCondition[ 0 ]";
writer.WriteLine("cache.Queries.GetResultSet( " + commandConditionsExpression + ", commandConditions => {");
writeResultSetCreatorBody(cn, writer, database, table, tableColumns, tableUsesRowVersionedCaching, isRevisionHistoryTable, "true");
writer.WriteLine("} );");
var rowsByPkExpression = "cache.{0}RowsByPk".FormatWith(isRevisionHistoryTable ? "LatestRevision" : "");
writer.WriteLine("if( !returnNullIfNoMatch )");
writer.WriteLine("return {0}[ System.Tuple.Create( id ) ];".FormatWith(rowsByPkExpression));
writer.WriteLine("Row row;");
writer.WriteLine("return {0}.TryGetValue( System.Tuple.Create( id ), out row ) ? row : null;".FormatWith(rowsByPkExpression));
}
else
{
writer.WriteLine(
"var rows = GetRows( new " + DataAccessStatics.GetEqualityConditionClassName(cn, database, table, tableColumns.KeyColumns.Single()) + "( id ) );");
writer.WriteLine("return returnNullIfNoMatch ? rows.SingleOrDefault() : rows.Single();");
}
writer.WriteLine("}");
}
private static void writeCreateForSingleRowUpdateMethod(
DBConnection cn, string tableName, bool isRevisionHistoryTable, bool isRevisionHistoryClass, string methodNameSuffix)
{
// header
CodeGenerationStatics.AddSummaryDocComment(
writer,
"Creates a modification object in single-row update mode with the specified current data. All column values in this object will have HasChanged = false, despite being initialized. This object can then be used to do a piecemeal update of the " +
tableName + " table.");
writer.Write(
"public static " + GetClassName(cn, tableName, isRevisionHistoryTable, isRevisionHistoryClass) + " CreateForSingleRowUpdate" + methodNameSuffix +
"( ");
writeColumnParameterDeclarations(columns.AllColumnsExceptRowVersion);
writer.WriteLine(" ) {");
// body
writer.WriteLine(
"var mod = new " + GetClassName(cn, tableName, isRevisionHistoryTable, isRevisionHistoryClass) + " { modType = ModificationType.Update };");
// Use the values of key columns as conditions.
writer.WriteLine("mod.conditions = new List<" + DataAccessStatics.GetTableConditionInterfaceName(cn, database, tableName) + ">();");
foreach (var column in columns.KeyColumns)
{
writer.WriteLine(
"mod.conditions.Add( new " + DataAccessStatics.GetEqualityConditionClassName(cn, database, tableName, column) + "( " +
EwlStatics.GetCSharpIdentifier(column.CamelCasedName) + " ) );");
}
writeColumnValueAssignmentsFromParameters(columns.AllColumnsExceptRowVersion, "mod");
writer.WriteLine("mod.markColumnValuesUnchanged();");
writer.WriteLine("return mod;");
writer.WriteLine("}");
}
private static void writeCreateForUpdateMethod(
DBConnection cn, string tableName, bool isRevisionHistoryTable, bool isRevisionHistoryClass, string methodNameSuffix)
{
// header
CodeGenerationStatics.AddSummaryDocComment(
writer,
"Creates a modification object in update mode with the specified conditions, which can be used to do a piecemeal update of the " + tableName + " table." );
writer.WriteLine(
"public static " + GetClassName( cn, tableName, isRevisionHistoryTable, isRevisionHistoryClass ) + " CreateForUpdate" + methodNameSuffix + "( " +
getConditionParameterDeclarations( cn, tableName ) + " ) {" );
// body
writer.WriteLine(
"var mod = new " + GetClassName( cn, tableName, isRevisionHistoryTable, isRevisionHistoryClass ) +
" { modType = ModificationType.Update, conditions = getConditionList( requiredCondition, additionalConditions ) };" );
// Set column values that correspond to modification conditions to the values of those conditions. One reason this is important is so the primary
// key can be retrieved in a consistent way regardless of whether the modification object is an insert or an update.
writer.WriteLine( "foreach( var condition in mod.conditions ) {" );
var prefix = "if";
foreach( var column in columns.AllColumnsExceptRowVersion ) {
writer.WriteLine( prefix + "( condition is " + DataAccessStatics.GetEqualityConditionClassName( cn, database, tableName, column ) + " )" );
writer.WriteLine(
"mod." + getColumnFieldName( column ) + ".Value = ( condition as " + DataAccessStatics.GetEqualityConditionClassName( cn, database, tableName, column ) +
" ).Value;" );
prefix = "else if";
}
writer.WriteLine( "}" );
writer.WriteLine( writer.NewLine + "mod.markColumnValuesUnchanged();" );
writer.WriteLine( "return mod;" );
writer.WriteLine( "}" );
}
private static void writeGetRowsMethod(
DBConnection cn, TextWriter writer, IDatabase database, string table, TableColumns tableColumns, bool isSmallTable, bool tableUsesRowVersionedCaching,
bool isRevisionHistoryTable, bool excludePreviousRevisions, int?commandTimeoutSeconds)
{
// header
var methodName = "GetRows" + (isSmallTable ? "MatchingConditions" : "") +
(isRevisionHistoryTable && !excludePreviousRevisions ? "IncludingPreviousRevisions" : "");
CodeGenerationStatics.AddSummaryDocComment(
writer,
"Retrieves the rows from the table that match the specified conditions, ordered in a stable way." +
(isSmallTable ? " Since the table is specified as small, you should only use this method if you cannot filter the rows in code." : ""));
writer.WriteLine(
"public static IEnumerable<Row> " + methodName + "( params " + DataAccessStatics.GetTableConditionInterfaceName(cn, database, table) + "[] conditions ) {");
// body
// If it's a primary key query, use RowsByPk if possible.
foreach (var i in tableColumns.KeyColumns)
{
var equalityConditionClassName = DataAccessStatics.GetEqualityConditionClassName(cn, database, table, i);
writer.WriteLine("var {0}Condition = conditions.OfType<{1}>().FirstOrDefault();".FormatWith(i.CamelCasedName, equalityConditionClassName));
}
writer.WriteLine("var cache = Cache.Current;");
var pkConditionVariableNames = tableColumns.KeyColumns.Select(i => i.CamelCasedName + "Condition");
writer.WriteLine(
"var isPkQuery = " + StringTools.ConcatenateWithDelimiter(" && ", pkConditionVariableNames.Select(i => i + " != null").ToArray()) +
" && conditions.Count() == " + tableColumns.KeyColumns.Count() + ";");
writer.WriteLine("if( isPkQuery ) {");
writer.WriteLine("Row row;");
writer.WriteLine(
"if( cache." + (excludePreviousRevisions ? "LatestRevision" : "") + $"RowsByPk.TryGetValue( {TypeNames.Tuple}.Create( " +
StringTools.ConcatenateWithDelimiter(", ", pkConditionVariableNames.Select(i => i + ".Value").ToArray()) + " ), out row ) )");
writer.WriteLine("return new [] {row};");
writer.WriteLine("}");
var commandConditionsExpression = "conditions.Select( i => i.CommandCondition )";
if (excludePreviousRevisions)
{
commandConditionsExpression += ".Concat( new [] {getLatestRevisionsCondition()} )";
}
writer.WriteLine("return cache.Queries.GetResultSet( " + commandConditionsExpression + ", commandConditions => {");
writeResultSetCreatorBody(
cn,
writer,
database,
table,
tableColumns,
tableUsesRowVersionedCaching,
excludePreviousRevisions,
"!isPkQuery",
commandTimeoutSeconds);
writer.WriteLine("} );");
writer.WriteLine("}");
}
private static void writeGetRowMatchingPkMethod(
DBConnection cn, TextWriter writer, IDatabase database, string table, TableColumns tableColumns, bool isSmallTable, bool tableUsesRowVersionedCaching,
bool isRevisionHistoryTable, int?commandTimeoutSeconds)
{
var pkIsId = tableColumns.KeyColumns.Count() == 1 && tableColumns.KeyColumns.Single().Name.ToLower().EndsWith("id");
var methodName = pkIsId ? "GetRowMatchingId" : "GetRowMatchingPk";
var pkParameters = pkIsId
? tableColumns.KeyColumns.Single().DataTypeName + " id"
: tableColumns.KeyColumns.Select(i => $"{i.DataTypeName} {i.CamelCasedName}").GetCommaDelimitedList();
const string returnNullIfNoMatch = nameof(returnNullIfNoMatch);
const string id = nameof(id);
writer.CodeBlock(
$"public static Row {methodName}( {pkParameters}, bool {returnNullIfNoMatch} = false ) {{",
() => {
if (isSmallTable)
{
writer.WriteLine("var cache = Cache.Current;");
var commandConditionsExpression = isRevisionHistoryTable ? "new [] {getLatestRevisionsCondition()}" : "new InlineDbCommandCondition[ 0 ]";
writer.WriteLine("cache.Queries.GetResultSet( " + commandConditionsExpression + ", commandConditions => {");
writeResultSetCreatorBody(
cn,
writer,
database,
table,
tableColumns,
tableUsesRowVersionedCaching,
isRevisionHistoryTable,
"true",
commandTimeoutSeconds);
writer.WriteLine("} );");
var rowsByPkExpression = $"cache.{( isRevisionHistoryTable ? "LatestRevision" : "" )}RowsByPk";
var pkTupleCreationArguments = pkIsId ? id : StringTools.ConcatenateWithDelimiter(", ", tableColumns.KeyColumns.Select(i => i.CamelCasedName));
writer.WriteLine($"if( !{returnNullIfNoMatch} )");
writer.WriteLine($"return {rowsByPkExpression}[ {TypeNames.Tuple}.Create( {pkTupleCreationArguments} ) ];");
writer.WriteLine("Row row;");
writer.WriteLine($"return {rowsByPkExpression}.TryGetValue( {TypeNames.Tuple}.Create( {pkTupleCreationArguments} ), out row ) ? row : null;");
}
else
{
var condition = pkIsId
? $"new {DataAccessStatics.GetEqualityConditionClassName( cn, database, table, tableColumns.KeyColumns.Single() )}( {id} )"
: (from keyColumn in tableColumns.KeyColumns
let className = DataAccessStatics.GetEqualityConditionClassName(cn, database, table, keyColumn)
select $"new {className}( {keyColumn.CamelCasedName} )").GetCommaDelimitedList();
writer.WriteLine($"return GetRows( {condition} ).PrimaryKeySingle({returnNullIfNoMatch});");
}
});
}
private static void writeGetRowMatchingPkMethod(
DBConnection cn, TextWriter writer, Database database, string table, TableColumns tableColumns, bool isSmallTable, bool tableUsesRowVersionedCaching,
bool isRevisionHistoryTable)
{
var pkIsId = tableColumns.KeyColumns.Count() == 1 && tableColumns.KeyColumns.Single().Name.ToLower().EndsWith("id");
var methodName = pkIsId ? "GetRowMatchingId" : "GetRowMatchingPk";
var pkParameters = pkIsId
? "{0} id".FormatWith(tableColumns.KeyColumns.Single().DataTypeName)
: StringTools.ConcatenateWithDelimiter(
", ",
tableColumns.KeyColumns.Select(i => "{0} {1}".FormatWith(i.DataTypeName, i.CamelCasedName)).ToArray());
writer.WriteLine("public static Row " + methodName + "( " + pkParameters + ", bool returnNullIfNoMatch = false ) {");
if (isSmallTable)
{
writer.WriteLine("var cache = Cache.Current;");
var commandConditionsExpression = isRevisionHistoryTable ? "getLatestRevisionsCondition().ToCollection()" : "new InlineDbCommandCondition[ 0 ]";
writer.WriteLine("cache.Queries.GetResultSet( " + commandConditionsExpression + ", commandConditions => {");
writeResultSetCreatorBody(cn, writer, database, table, tableColumns, tableUsesRowVersionedCaching, isRevisionHistoryTable, "true");
writer.WriteLine("} );");
var rowsByPkExpression = "cache.{0}RowsByPk".FormatWith(isRevisionHistoryTable ? "LatestRevision" : "");
var pkTupleCreationArguments =
pkIsId ? "id" : StringTools.ConcatenateWithDelimiter(", ", tableColumns.KeyColumns.Select(i => i.CamelCasedName).ToArray());
writer.WriteLine("if( !returnNullIfNoMatch )");
writer.WriteLine("return {0}[ System.Tuple.Create( {1} ) ];".FormatWith(rowsByPkExpression, pkTupleCreationArguments));
writer.WriteLine("Row row;");
writer.WriteLine(
"return {0}.TryGetValue( System.Tuple.Create( {1} ), out row ) ? row : null;".FormatWith(rowsByPkExpression, pkTupleCreationArguments));
}
else
{
writer.WriteLine(
"var rows = GetRows( {0} );".FormatWith(
pkIsId
? "new {0}( id )".FormatWith(DataAccessStatics.GetEqualityConditionClassName(cn, database, table, tableColumns.KeyColumns.Single()))
: StringTools.ConcatenateWithDelimiter(
", ",
tableColumns.KeyColumns.Select(
i => "new {0}( {1} )".FormatWith(DataAccessStatics.GetEqualityConditionClassName(cn, database, table, i), i.CamelCasedName))
.ToArray())));
writer.WriteLine("return returnNullIfNoMatch ? rows.SingleOrDefault() : rows.Single();");
}
writer.WriteLine("}");
}
private static void writeExecuteInsertOrUpdateMethod(
DBConnection cn, string tableName, bool isRevisionHistoryClass, IEnumerable <Column> keyColumns, Column identityColumn)
{
writer.WriteLine("private void executeInsertOrUpdate( bool isLongRunning ) {");
writer.WriteLine("try {");
if (isRevisionHistoryClass)
{
writer.WriteLine(DataAccessStatics.GetConnectionExpression(database) + ".ExecuteInTransaction( delegate {");
}
// insert
writer.WriteLine("if( modType == ModificationType.Insert ) {");
// If this is a revision history table, write code to insert a new revision when a row is inserted into this table.
if (isRevisionHistoryClass)
{
writer.WriteLine("var revisionHistorySetup = RevisionHistoryStatics.SystemProvider;");
writer.WriteLine(getColumnFieldName(columns.PrimaryKeyAndRevisionIdColumn) + ".Value = revisionHistorySetup.GetNextMainSequenceValue();");
writer.WriteLine(
"revisionHistorySetup.InsertRevision( System.Convert.ToInt32( " + getColumnFieldName(columns.PrimaryKeyAndRevisionIdColumn) +
".Value ), System.Convert.ToInt32( " + getColumnFieldName(columns.PrimaryKeyAndRevisionIdColumn) + ".Value ), " +
DataAccessStatics.GetConnectionExpression(database) + ".GetUserTransactionId() );");
}
writer.WriteLine("var insert = new InlineInsert( \"" + tableName + "\" );");
writer.WriteLine("insert.AddColumnModifications( getColumnModificationValues() );");
if (identityColumn != null)
{
// One reason the ChangeType call is necessary: SQL Server identities always come back as decimal, and you can't cast a boxed decimal to an int.
writer.WriteLine(
"{0}.Value = {1};".FormatWith(
getColumnFieldName(identityColumn),
identityColumn.GetIncomingValueConversionExpression(
"EwlStatics.ChangeType( insert.Execute( {0}, isLongRunning: isLongRunning ), typeof( {1} ) )".FormatWith(
DataAccessStatics.GetConnectionExpression(database),
identityColumn.UnconvertedDataTypeName))));
}
else
{
writer.WriteLine("insert.Execute( {0}, isLongRunning: isLongRunning );".FormatWith(DataAccessStatics.GetConnectionExpression(database)));
}
// Future calls to Execute should perform updates, not inserts. Use the values of key columns as conditions.
writer.WriteLine("modType = ModificationType.Update;");
writer.WriteLine("conditions = new List<" + DataAccessStatics.GetTableConditionInterfaceName(cn, database, tableName) + ">();");
foreach (var column in keyColumns)
{
writer.WriteLine(
"conditions.Add( new " + DataAccessStatics.GetEqualityConditionClassName(cn, database, tableName, column) + "( " +
EwlStatics.GetCSharpIdentifier(column.PascalCasedNameExceptForOracle) + " ) );");
}
writer.WriteLine("}"); // if insert
// update
writer.WriteLine("else {");
writer.WriteLine("var modificationValues = getColumnModificationValues();");
writer.WriteLine("if( modificationValues.Any() ) {");
if (isRevisionHistoryClass)
{
writer.WriteLine("copyLatestRevisions( conditions, isLongRunning );");
}
writer.WriteLine("var update = new InlineUpdate( \"" + tableName + "\" );");
writer.WriteLine("update.AddColumnModifications( modificationValues );");
writer.WriteLine("conditions.ForEach( condition => update.AddCondition( condition.CommandCondition ) );");
if (isRevisionHistoryClass)
{
writer.WriteLine("update.AddCondition( getLatestRevisionsCondition() );");
}
writer.WriteLine("update.Execute( {0}, isLongRunning: isLongRunning );".FormatWith(DataAccessStatics.GetConnectionExpression(database)));
writer.WriteLine("}");
writer.WriteLine("}"); // else
if (isRevisionHistoryClass)
{
writer.WriteLine("} );"); // cn.ExecuteInTransaction
}
writer.WriteLine("}"); // try
writer.WriteLine("catch( System.Exception e ) {");
writer.WriteLine("rethrowAsDataModificationExceptionIfNecessary( e );");
writer.WriteLine("throw;");
writer.WriteLine("}"); // catch
writer.WriteLine("}"); // method
}
private static void writeResultSetCreatorBody(
DBConnection cn, TextWriter writer, IDatabase database, string table, TableColumns tableColumns, bool tableUsesRowVersionedCaching,
bool excludesPreviousRevisions, string cacheQueryInDbExpression, int?commandTimeoutSeconds)
{
if (tableUsesRowVersionedCaching)
{
writer.WriteLine("var results = new List<Row>();");
writer.WriteLine(DataAccessStatics.DataAccessStateCurrentDatabaseConnectionExpression + ".ExecuteInTransaction( () => {");
// Query for the cache keys of the results.
writer.WriteLine(
"var keyCommand = {0};".FormatWith(
getInlineSelectExpression(
table,
tableColumns,
"{0}, \"{1}\"".FormatWith(
StringTools.ConcatenateWithDelimiter(", ", tableColumns.KeyColumns.Select(i => "\"{0}\"".FormatWith(i.Name)).ToArray()),
cn.DatabaseInfo is OracleInfo ? "ORA_ROWSCN" : tableColumns.RowVersionColumn.Name),
cacheQueryInDbExpression,
commandTimeoutSeconds)));
writer.WriteLine(getCommandConditionAddingStatement("keyCommand"));
writer.WriteLine($"var keys = new List<{TypeNames.Tuple}<{getPkAndVersionTupleTypeArguments( cn, tableColumns )}>>();");
var concatenateWithDelimiter = StringTools.ConcatenateWithDelimiter(
", ",
tableColumns.KeyColumns.Select((c, i) => c.GetDataReaderValueExpression("r", ordinalOverride: i)));
var o = cn.DatabaseInfo is OracleInfo
? "({0})r.GetValue( {1} )".FormatWith(oracleRowVersionDataType, tableColumns.KeyColumns.Count())
: tableColumns.RowVersionColumn.GetDataReaderValueExpression("r", ordinalOverride: tableColumns.KeyColumns.Count());
writer.WriteLine(
"keyCommand.Execute( " + DataAccessStatics.DataAccessStateCurrentDatabaseConnectionExpression + ", r => { while( r.Read() ) keys.Add( " +
$"{TypeNames.Tuple}.Create( {concatenateWithDelimiter}, {o} )" + " ); } );");
writer.WriteLine("var rowsByPkAndVersion = getRowsByPkAndVersion();");
writer.WriteLine("var cachedKeyCount = keys.Where( i => rowsByPkAndVersion.ContainsKey( i ) ).Count();");
// If all but a few results are cached, execute a single-row query for each missing result.
writer.CodeBlock(
"if( cachedKeyCount >= keys.Count() - 1 || cachedKeyCount >= keys.Count() * .99 ) {",
() => {
writer.WriteLine("foreach( var key in keys ) {");
writer.WriteLine("results.Add( new Row( rowsByPkAndVersion.GetOrAdd( key, () => {");
writer.WriteLine("var singleRowCommand = {0};".FormatWith(getInlineSelectExpression(table, tableColumns, "\"*\"", "false", commandTimeoutSeconds)));
foreach (var i in tableColumns.KeyColumns.Select((c, i) => new { column = c, index = i }))
{
writer.WriteLine(
"singleRowCommand.AddCondition( ( ({0})new {1}( key.Item{2} ) ).CommandCondition );".FormatWith(
DataAccessStatics.GetTableConditionInterfaceName(cn, database, table),
DataAccessStatics.GetEqualityConditionClassName(cn, database, table, i.column),
i.index + 1));
}
writer.WriteLine("var singleRowResults = new List<BasicRow>();");
writer.WriteLine(
"singleRowCommand.Execute( " + DataAccessStatics.DataAccessStateCurrentDatabaseConnectionExpression +
", r => { while( r.Read() ) singleRowResults.Add( new BasicRow( r ) ); } );");
writer.WriteLine("return singleRowResults.Single();");
writer.WriteLine("} ) ) );");
writer.WriteLine("}");
});
// Otherwise, execute the full query.
writer.CodeBlock(
"else {",
() => {
writer.WriteLine(
"var command = {0};".FormatWith(
getInlineSelectExpression(
table,
tableColumns,
cn.DatabaseInfo is OracleInfo ? "\"{0}.*\", \"ORA_ROWSCN\"".FormatWith(table) : "\"*\"",
cacheQueryInDbExpression,
commandTimeoutSeconds)));
writer.WriteLine(getCommandConditionAddingStatement("command"));
writer.WriteLine("command.Execute( " + DataAccessStatics.DataAccessStateCurrentDatabaseConnectionExpression + ", r => {");
writer.WriteLine(
"while( r.Read() ) results.Add( new Row( rowsByPkAndVersion.GetOrAdd( System.Tuple.Create( {0}, {1} ), () => new BasicRow( r ) ) ) );".FormatWith(
StringTools.ConcatenateWithDelimiter(", ", tableColumns.KeyColumns.Select(i => i.GetDataReaderValueExpression("r")).ToArray()),
cn.DatabaseInfo is OracleInfo
? "({0})r.GetValue( {1} )".FormatWith(oracleRowVersionDataType, tableColumns.AllColumns.Count())
: tableColumns.RowVersionColumn.GetDataReaderValueExpression("r")));
writer.WriteLine("} );");
});
writer.WriteLine("} );");
}
else
{
writer.WriteLine(
"var command = {0};".FormatWith(getInlineSelectExpression(table, tableColumns, @"""*""", cacheQueryInDbExpression, commandTimeoutSeconds)));
writer.WriteLine(getCommandConditionAddingStatement("command"));
writer.WriteLine("var results = new List<Row>();");
writer.WriteLine(
"command.Execute( " + DataAccessStatics.DataAccessStateCurrentDatabaseConnectionExpression +
", r => { while( r.Read() ) results.Add( new Row( new BasicRow( r ) ) ); } );");
}
// Add all results to RowsByPk.
writer.WriteLine("foreach( var i in results ) {");
var pkTupleCreationArgs = tableColumns.KeyColumns.Select(i => "i." + Utility.GetCSharpIdentifier(i.PascalCasedNameExceptForOracle));
var pkTuple = "System.Tuple.Create( " + StringTools.ConcatenateWithDelimiter(", ", pkTupleCreationArgs.ToArray()) + " )";
writer.WriteLine($"cache.RowsByPk[ {pkTuple} ] = i;");
if (excludesPreviousRevisions)
{
writer.WriteLine($"cache.LatestRevisionRowsByPk[ {pkTuple} ] = i;");
}
writer.WriteLine("}");
writer.WriteLine("return results;");
}
