private SqlString GenerateLockString()
{
ISessionFactoryImplementor factory = lockable.Factory;
SqlSimpleSelectBuilder select = new SqlSimpleSelectBuilder(factory.Dialect, factory)
.SetLockMode(lockMode)
.SetTableName(lockable.RootTableName)
.AddColumn(lockable.RootTableIdentifierColumnNames[0])
.SetIdentityColumn(lockable.RootTableIdentifierColumnNames, lockable.IdentifierType);
if (lockable.IsVersioned)
{
select.SetVersionColumn(new string[] { lockable.VersionColumnName }, lockable.VersionType);
}
if (factory.Settings.IsCommentsEnabled)
{
select.SetComment(lockMode + " lock " + lockable.EntityName);
}
return select.ToSqlString();
}
public void SimpleSelectStringSqlTest()
{
Configuration cfg = new Configuration();
ISessionFactory factory = cfg.BuildSessionFactory();
ISessionFactoryImplementor factoryImpl = (ISessionFactoryImplementor) factory;
SqlSimpleSelectBuilder select = new SqlSimpleSelectBuilder(factoryImpl.Dialect, factoryImpl);
select.SetTableName("test_simple_select_builder");
select.AddColumn("column_no_alias");
select.AddColumn("aliased_column", "aliased_column_alias");
select.AddColumns(new string[] {"column1_no_alias", "column2_no_alias"});
select.AddColumns(new string[] {"column1_with_alias", "column2_with_alias"}, new string[] {"c1_alias", "c2_alias"});
select.SetIdentityColumn(new string[] {"identity_column"}, NHibernateUtil.Int64);
select.SetVersionColumn(new string[] {"version_column"}, (IVersionType) NHibernateUtil.Int32);
select.AddWhereFragment(new string[] {"where_frag_column"}, NHibernateUtil.Int32, " = ");
SqlString sqlString = select.ToSqlString();
Parameter[] actualParams = new Parameter[3];
string expectedSql = new StringBuilder().Append("SELECT ")
.Append("column_no_alias, ")
.Append("aliased_column AS aliased_column_alias, ")
.Append("column1_no_alias, ")
.Append("column2_no_alias, ")
.Append("column1_with_alias AS c1_alias, ")
.Append("column2_with_alias AS c2_alias ")
.Append("FROM test_simple_select_builder ")
.Append("WHERE identity_column = ? AND version_column = ?")
.Append(" AND where_frag_column = ?")
.ToString();
Assert.AreEqual(expectedSql, sqlString.ToString(), "SQL String");
Assert.AreEqual(3, sqlString.GetParameterCount(), "3 parameters");
}
/// <summary>
/// Generate the SQL that pessimistic locks a row by id (and version)
/// </summary>
/// <param name="sqlString">An existing SqlString to copy for then new SqlString.</param>
/// <param name="forUpdateFragment"></param>
/// <returns>A new SqlString</returns>
/// <remarks>
/// The parameter <c>sqlString</c> does not get modified. It is Cloned to make a new SqlString.
/// If the parameter<c>sqlString</c> is null a new one will be created.
/// </remarks>
protected override SqlString GenerateLockString( SqlString sqlString, string forUpdateFragment )
{
SqlStringBuilder sqlBuilder = null;
if( sqlString == null )
{
SqlSimpleSelectBuilder builder = new SqlSimpleSelectBuilder( factory );
// set the table name and add the columns to select
builder.SetTableName( qualifiedTableName )
.AddColumn( base.IdentifierColumnNames[ 0 ] );
// add the parameters to use in the WHERE clause
builder.SetIdentityColumn( base.IdentifierColumnNames, IdentifierType );
if( IsVersioned )
{
builder.SetVersionColumn( new string[ ] {VersionColumnName}, VersionType );
}
sqlBuilder = new SqlStringBuilder( builder.ToSqlString() );
}
else
{
sqlBuilder = new SqlStringBuilder( sqlString );
}
// add any special text that is contained in the forUpdateFragment
if( forUpdateFragment != null )
{
sqlBuilder.Add( forUpdateFragment );
}
return sqlBuilder.ToSqlString();
}
public void SimpleSelectStringSqlTest()
{
Configuration cfg = new Configuration();
ISessionFactory factory = cfg.BuildSessionFactory( );
ISessionFactoryImplementor factoryImpl = (ISessionFactoryImplementor)factory;
SqlSimpleSelectBuilder select = new SqlSimpleSelectBuilder(factoryImpl);
select.SetTableName("test_simple_select_builder");
select.AddColumn("column_no_alias");
select.AddColumn("aliased_column", "aliased_column_alias");
select.AddColumns(new string[]{"column1_no_alias", "column2_no_alias"});
select.AddColumns(new string[]{"column1_with_alias", "column2_with_alias"}, new string[] {"c1_alias", "c2_alias"});
select.SetIdentityColumn(new string[]{"identity_column"}, NHibernateUtil.Int64);
select.SetVersionColumn(new string[]{"version_column"}, (IVersionType)NHibernateUtil.Int32);
select.AddWhereFragment(new string[]{"where_frag_column"}, NHibernateUtil.Int32, " = ");
SqlString sqlString = select.ToSqlString();
Parameter[] actualParams = new Parameter[3];
int numOfParameters = 0;
string expectedSql = new StringBuilder().Append("SELECT ")
.Append("column_no_alias, ")
.Append("aliased_column AS aliased_column_alias, ")
.Append("column1_no_alias, ")
.Append("column2_no_alias, ")
.Append("column1_with_alias AS c1_alias, ")
.Append("column2_with_alias AS c2_alias ")
.Append("FROM test_simple_select_builder ")
.Append("WHERE identity_column = :identity_column AND version_column = :version_column")
.Append(" AND where_frag_column = :where_frag_column")
.ToString();
Assert.AreEqual(expectedSql , sqlString.ToString(), "SQL String");
foreach(object part in sqlString.SqlParts)
{
if(part is Parameter)
{
actualParams[numOfParameters] = (Parameter)part;
numOfParameters++;
}
}
Assert.AreEqual(3, numOfParameters, "3 parameters");
Parameter firstParam = new Parameter( "identity_column", new SqlTypes.Int64SqlType() );
Parameter secondParam = new Parameter( "version_column", new SqlTypes.Int32SqlType() );
Parameter thirdParam = new Parameter( "where_frag_column", new SqlTypes.Int32SqlType() );
Assert.AreEqual(firstParam.SqlType.DbType, actualParams[0].SqlType.DbType, "First Parameter Type");
Assert.AreEqual(firstParam.Name, actualParams[0].Name, "First Parameter Name");
Assert.AreEqual(secondParam.SqlType.DbType, actualParams[1].SqlType.DbType, "Second Parameter Type");
Assert.AreEqual(secondParam.Name, actualParams[1].Name, "Second Parameter Name");
Assert.AreEqual(thirdParam.SqlType.DbType, actualParams[2].SqlType.DbType, "Third Parameter Type");
Assert.AreEqual(thirdParam.Name, actualParams[2].Name, "Third Parameter Name");
}