public override DbExpression Visit(DbEqualExpression exp)
{
DbExpression left = exp.Left;
DbExpression right = exp.Right;
left = DbExpressionExtensions.ParseDbExpression(left);
right = DbExpressionExtensions.ParseDbExpression(right);
//明确 left right 其中一边一定为 null
if (DbExpressionExtensions.AffirmExpressionRetValueIsNull(right))
{
left.Accept(this);
this._sqlBuilder.Append(" IS NULL");
return(exp);
}
if (DbExpressionExtensions.AffirmExpressionRetValueIsNull(left))
{
right.Accept(this);
this._sqlBuilder.Append(" IS NULL");
return(exp);
}
left.Accept(this);
this._sqlBuilder.Append(" = ");
right.Accept(this);
return(exp);
}
public override DbExpression Visit(DbEqualExpression exp)
{
DbExpression left = exp.Left;
DbExpression right = exp.Right;
left = DbExpressionHelper.OptimizeDbExpression(left);
right = DbExpressionHelper.OptimizeDbExpression(right);
//明确 left right 其中一边一定为 null 或 空字符串。
if (DbExpressionHelper.AffirmExpressionRetValueIsNullOrEmpty(right))
{
left.Accept(this);
this._sqlBuilder.Append(" IS NULL");
return(exp);
}
if (DbExpressionHelper.AffirmExpressionRetValueIsNullOrEmpty(left))
{
right.Accept(this);
this._sqlBuilder.Append(" IS NULL");
return(exp);
}
AmendDbInfo(left, right);
left.Accept(this);
this._sqlBuilder.Append(" = ");
right.Accept(this);
return(exp);
}
public virtual int Delete <TEntity>(TEntity entity, string table)
{
Utils.CheckNull(entity);
TypeDescriptor typeDescriptor = TypeDescriptor.GetDescriptor(entity.GetType());
EnsureEntityHasPrimaryKey(typeDescriptor);
MappingMemberDescriptor keyMemberDescriptor = typeDescriptor.PrimaryKey;
MemberInfo keyMember = typeDescriptor.PrimaryKey.MemberInfo;
var keyVal = keyMemberDescriptor.GetValue(entity);
if (keyVal == null)
{
throw new ChloeException(string.Format("The primary key '{0}' could not be null.", keyMember.Name));
}
DbTable dbTable = table == null ? typeDescriptor.Table : new DbTable(table, typeDescriptor.Table.Schema);
DbExpression left = new DbColumnAccessExpression(dbTable, keyMemberDescriptor.Column);
DbExpression right = new DbParameterExpression(keyVal);
DbExpression conditionExp = new DbEqualExpression(left, right);
DbDeleteExpression e = new DbDeleteExpression(dbTable, conditionExp);
return(this.ExecuteSqlCommand(e));
}
public virtual int Delete <T>(T entity)
{
Utils.CheckNull(entity);
TypeDescriptor typeDescriptor = TypeDescriptor.GetDescriptor(entity.GetType());
EnsureMappingTypeHasPrimaryKey(typeDescriptor);
MappingMemberDescriptor keyMemberDescriptor = typeDescriptor.PrimaryKey;
var keyMember = typeDescriptor.PrimaryKey.MemberInfo;
var keyVal = keyMemberDescriptor.GetValue(entity);
if (keyVal == null)
{
throw new Exception(string.Format("实体主键 {0} 值为 null", keyMember.Name));
}
DbExpression left = new DbColumnAccessExpression(typeDescriptor.Table, keyMemberDescriptor.Column);
DbExpression right = new DbParameterExpression(keyVal);
DbExpression conditionExp = new DbEqualExpression(left, right);
DbDeleteExpression e = new DbDeleteExpression(typeDescriptor.Table, conditionExp);
return(this.ExecuteSqlCommand(e));
}
/// <summary>
/// 修改
/// </summary>
/// <param name="entity"></param>
/// <returns></returns>
public TEntity Edit(TEntity entity)
{
Checks.NotNull(entity, "entity");
TypeDescriptor typeDescriptor = TypeDescriptor.GetDescriptor(entity.GetType());
object keyVal = null;
MappingMemberDescriptor keyMemberDescriptor = null;
Dictionary <MappingMemberDescriptor, DbExpression> updateColumns = new Dictionary <MappingMemberDescriptor, DbExpression>();
foreach (var kv in typeDescriptor.MappingMemberDescriptors)
{
MemberInfo member = kv.Key;
MappingMemberDescriptor memberDescriptor = kv.Value;
object val = memberDescriptor.GetValue(entity);
if (memberDescriptor.SZColumnAttribute.IsKey)
{
if (val == null)
{
throw new SZORMException("主键值不允许为空.");
}
keyVal = val;
keyMemberDescriptor = memberDescriptor;
continue;
}
if (memberDescriptor.SZColumnAttribute.IsEditTime)
{
val = DateTime.Now;
memberDescriptor.SetValue(entity, val);
}
DbExpression valExp = DbExpression.Parameter(val, memberDescriptor.MemberInfoType);
updateColumns.Add(memberDescriptor, valExp);
}
DbExpression left = new DbColumnAccessExpression(typeDescriptor.Table, keyMemberDescriptor.Column);
DbExpression right = DbExpression.Parameter(keyVal, keyMemberDescriptor.MemberInfoType);
DbExpression conditionExp = new DbEqualExpression(left, right);
DbUpdateExpression e = new DbUpdateExpression(typeDescriptor.Table, conditionExp);
foreach (var item in updateColumns)
{
e.UpdateColumns.Add(item.Key.Column, item.Value);
}
IDbExpressionTranslator translator = this.DbContext.DatabaseProvider.CreateDbExpressionTranslator();
List <DbParam> parameters;
string sql = translator.Translate(e, out parameters);
this.DbContext.ExecuteNoQuery(sql, parameters.ToArray());
return(entity);
}
public override DbExpression Visit(DbEqualExpression exp)
{
DbExpression left = exp.Left;
DbExpression right = exp.Right;
left = DbExpressionExtension.StripInvalidConvert(left);
right = DbExpressionExtension.StripInvalidConvert(right);
MethodInfo method_Sql_Equals = PublicConstants.MethodInfo_Sql_Equals.MakeGenericMethod(left.Type);
/* Sql.Equals(left, right) */
DbMethodCallExpression left_equals_right = DbExpression.MethodCall(null, method_Sql_Equals, new List <DbExpression>(2)
{
left, right
});
if (right.NodeType == DbExpressionType.Parameter || right.NodeType == DbExpressionType.Constant || left.NodeType == DbExpressionType.Parameter || left.NodeType == DbExpressionType.Constant || right.NodeType == DbExpressionType.SubQuery || left.NodeType == DbExpressionType.SubQuery || !left.Type.CanNull() || !right.Type.CanNull())
{
/*
* a.Name == name --> a.Name == name
* a.Id == (select top 1 T.Id from T) --> a.Id == (select top 1 T.Id from T)
* 对于上述查询，我们不考虑 null
*/
left_equals_right.Accept(this);
return(exp);
}
/*
* a.Name == a.XName --> a.Name == a.XName or (a.Name is null and a.XName is null)
*/
/* Sql.Equals(left, null) */
var left_is_null = DbExpression.MethodCall(null, method_Sql_Equals, new List <DbExpression>(2)
{
left, DbExpression.Constant(null, left.Type)
});
/* Sql.Equals(right, null) */
var right_is_null = DbExpression.MethodCall(null, method_Sql_Equals, new List <DbExpression>(2)
{
right, DbExpression.Constant(null, right.Type)
});
/* Sql.Equals(left, null) && Sql.Equals(right, null) */
var left_is_null_and_right_is_null = DbExpression.And(left_is_null, right_is_null);
/* Sql.Equals(left, right) || (Sql.Equals(left, null) && Sql.Equals(right, null)) */
var left_equals_right_or_left_is_null_and_right_is_null = DbExpression.Or(left_equals_right, left_is_null_and_right_is_null);
left_equals_right_or_left_is_null_and_right_is_null.Accept(this);
return(exp);
}
public override DbExpression Visit(DbEqualExpression exp)
{
DbExpression left = exp.Left;
DbExpression right = exp.Right;
left = DbExpressionHelper.OptimizeDbExpression(left);
right = DbExpressionHelper.OptimizeDbExpression(right);
MethodInfo method_Sql_Equals = UtilConstants.MethodInfo_Sql_Equals.MakeGenericMethod(left.Type);
/* Sql.Equals(left, right) */
DbMethodCallExpression left_equals_right = DbExpression.MethodCall(null, method_Sql_Equals, new List <DbExpression>(2)
{
left, right
});
if (right.NodeType == DbExpressionType.Parameter || right.NodeType == DbExpressionType.Constant || left.NodeType == DbExpressionType.Parameter || left.NodeType == DbExpressionType.Constant)
{
/*
* a.Name == name --> a.Name == name
*/
left_equals_right.Accept(this);
return(exp);
}
/*
* a.Name == a.XName --> a.Name == a.XName or (a.Name is null and a.XName is null)
*/
/* Sql.Equals(left, null) */
var left_is_null = DbExpression.MethodCall(null, method_Sql_Equals, new List <DbExpression>(2)
{
left, DbExpression.Constant(null, left.Type)
});
/* Sql.Equals(right, null) */
var right_is_null = DbExpression.MethodCall(null, method_Sql_Equals, new List <DbExpression>(2)
{
right, DbExpression.Constant(null, right.Type)
});
/* Sql.Equals(left, null) && Sql.Equals(right, null) */
var left_is_null_and_right_is_null = DbExpression.And(left_is_null, right_is_null);
/* Sql.Equals(left, right) || (Sql.Equals(left, null) && Sql.Equals(right, null)) */
var left_equals_right_or_left_is_null_and_right_is_null = DbExpression.Or(left_equals_right, left_is_null_and_right_is_null);
left_equals_right_or_left_is_null_and_right_is_null.Accept(this);
return(exp);
}
public static DbExpression MakeCondition(PairList <PrimitivePropertyDescriptor, object> propertyValuePairs, DbTable dbTable)
{
DbExpression conditionExp = null;
foreach (var pair in propertyValuePairs)
{
PrimitivePropertyDescriptor propertyDescriptor = pair.Item1;
object val = pair.Item2;
DbExpression left = new DbColumnAccessExpression(dbTable, propertyDescriptor.Column);
DbExpression right = DbExpression.Parameter(val, propertyDescriptor.PropertyType, propertyDescriptor.Column.DbType);
DbExpression equalExp = new DbEqualExpression(left, right);
conditionExp = conditionExp.And(equalExp);
}
return(conditionExp);
}
public void Process(DbMethodCallExpression exp, SqlGenerator generator)
{
DbExpression e = exp.Arguments.First();
DbEqualExpression equalNullExpression = DbExpression.Equal(e, DbExpression.Constant(null, UtilConstants.TypeOfString));
DbEqualExpression equalEmptyExpression = DbExpression.Equal(e, DbExpression.Constant(string.Empty));
DbOrExpression orExpression = DbExpression.Or(equalNullExpression, equalEmptyExpression);
DbCaseWhenExpression.WhenThenExpressionPair whenThenPair = new DbCaseWhenExpression.WhenThenExpressionPair(orExpression, DbConstantExpression.One);
List<DbCaseWhenExpression.WhenThenExpressionPair> whenThenExps = new List<DbCaseWhenExpression.WhenThenExpressionPair>(1);
whenThenExps.Add(whenThenPair);
DbCaseWhenExpression caseWhenExpression = DbExpression.CaseWhen(whenThenExps, DbConstantExpression.Zero, UtilConstants.TypeOfBoolean);
var eqExp = DbExpression.Equal(caseWhenExpression, DbConstantExpression.One);
eqExp.Accept(generator);
}
public override DbExpression Visit(DbEqualExpression exp)
{
/*
* join 的条件不考虑 null 问题
*/
DbExpression left = exp.Left;
DbExpression right = exp.Right;
MethodInfo method_Sql_Equals = UtilConstants.MethodInfo_Sql_Equals.MakeGenericMethod(left.Type);
/* Sql.Equals(left, right) */
DbMethodCallExpression left_equals_right = DbExpression.MethodCall(null, method_Sql_Equals, new List <DbExpression>(2)
{
left.Accept(this), right.Accept(this)
});
return(left_equals_right);
}
protected override DbExpression VisitBinary_Equal(BinaryExpression exp)
{
var left = exp.Left;
var right = exp.Right;
// 对于应用于 bool 类型的相等情况
if (left.Type == UtilConstants.TypeOfBoolean)
{
return(this.VisitBinary_Equal_Boolean(exp));
}
// 对于应用于 Nullable<bool> 类型的相等情况
if (left.Type == UtilConstants.TypeOfBoolean_Nullable)
{
return(this.VisitBinary_Equal_NullableBoolean(exp));
}
DbEqualExpression dbExp = DbExpression.Equal(this.Visit(left), this.Visit(right));
return(dbExp);
}
static DbExpression MakeCondition(Dictionary <PropertyDescriptor, object> keyValueMap, DbTable dbTable)
{
DbExpression conditionExp = null;
foreach (var kv in keyValueMap)
{
PropertyDescriptor keyPropertyDescriptor = kv.Key;
object keyVal = kv.Value;
if (keyVal == null)
{
throw new ArgumentException(string.Format("The primary key '{0}' could not be null.", keyPropertyDescriptor.Property.Name));
}
DbExpression left = new DbColumnAccessExpression(dbTable, keyPropertyDescriptor.Column);
DbExpression right = DbExpression.Parameter(keyVal, keyPropertyDescriptor.PropertyType, keyPropertyDescriptor.Column.DbType);
DbExpression equalExp = new DbEqualExpression(left, right);
conditionExp = conditionExp == null ? equalExp : DbExpression.And(conditionExp, equalExp);
}
return(conditionExp);
}
public override int Update <TEntity>(TEntity entity, string table)
{
Utils.CheckNull(entity);
TypeDescriptor typeDescriptor = TypeDescriptor.GetDescriptor(entity.GetType());
EnsureMappingTypeHasPrimaryKey(typeDescriptor);
MappingMemberDescriptor keyMemberDescriptor = typeDescriptor.PrimaryKey;
MemberInfo keyMember = keyMemberDescriptor.MemberInfo;
object keyVal = null;
IEntityState entityState = this.TryGetTrackedEntityState(entity);
Dictionary <MappingMemberDescriptor, DbExpression> updateColumns = new Dictionary <MappingMemberDescriptor, DbExpression>();
foreach (var kv in typeDescriptor.MappingMemberDescriptors)
{
MemberInfo member = kv.Key;
MappingMemberDescriptor memberDescriptor = kv.Value;
if (member == keyMember)
{
keyVal = memberDescriptor.GetValue(entity);
keyMemberDescriptor = memberDescriptor;
continue;
}
SequenceAttribute attr = (SequenceAttribute)memberDescriptor.GetCustomAttribute(typeof(SequenceAttribute));
if (attr != null)
{
continue;
}
object val = memberDescriptor.GetValue(entity);
if (entityState != null && !entityState.HasChanged(memberDescriptor, val))
{
continue;
}
DbExpression valExp = DbExpression.Parameter(val, memberDescriptor.MemberInfoType);
updateColumns.Add(memberDescriptor, valExp);
}
if (keyVal == null)
{
throw new ChloeException(string.Format("The primary key '{0}' could not be null.", keyMember.Name));
}
if (updateColumns.Count == 0)
{
return(0);
}
DbTable dbTable = table == null ? typeDescriptor.Table : new DbTable(table, typeDescriptor.Table.Schema);
DbExpression left = new DbColumnAccessExpression(dbTable, keyMemberDescriptor.Column);
DbExpression right = DbExpression.Parameter(keyVal, keyMemberDescriptor.MemberInfoType);
DbExpression conditionExp = new DbEqualExpression(left, right);
DbUpdateExpression e = new DbUpdateExpression(dbTable, conditionExp);
foreach (var item in updateColumns)
{
e.UpdateColumns.Add(item.Key.Column, item.Value);
}
int ret = this.ExecuteSqlCommand(e);
if (entityState != null)
{
entityState.Refresh();
}
return(ret);
}
public override DbExpression Visit(DbEqualExpression exp)
{
return(this.ConvertDbBooleanExpression(exp));
}
public abstract T Visit(DbEqualExpression exp);
public override int Update <T>(T entity)
{
Utils.CheckNull(entity);
TypeDescriptor typeDescriptor = TypeDescriptor.GetDescriptor(entity.GetType());
EnsureMappingTypeHasPrimaryKey(typeDescriptor);
object keyVal = null;
MappingMemberDescriptor keyMemberDescriptor = typeDescriptor.PrimaryKey;
MemberInfo keyMember = keyMemberDescriptor.MemberInfo;
IEntityState entityState = this.TryGetTrackedEntityState(entity);
Dictionary <MappingMemberDescriptor, DbExpression> updateColumns = new Dictionary <MappingMemberDescriptor, DbExpression>();
foreach (var kv in typeDescriptor.MappingMemberDescriptors)
{
var member = kv.Key;
var memberDescriptor = kv.Value;
if (member == keyMember)
{
keyVal = memberDescriptor.GetValue(entity);
keyMemberDescriptor = memberDescriptor;
continue;
}
AutoIncrementAttribute attr = (AutoIncrementAttribute)memberDescriptor.GetCustomAttribute(typeof(AutoIncrementAttribute));
if (attr != null)
{
continue;
}
var val = memberDescriptor.GetValue(entity);
if (entityState != null && !entityState.IsChanged(member, val))
{
continue;
}
DbExpression valExp = DbExpression.Parameter(val, memberDescriptor.MemberInfoType);
updateColumns.Add(memberDescriptor, valExp);
}
if (keyVal == null)
{
throw new Exception(string.Format("实体主键 {0} 值为 null", keyMember.Name));
}
if (updateColumns.Count == 0)
{
return(0);
}
DbExpression left = new DbColumnAccessExpression(typeDescriptor.Table, keyMemberDescriptor.Column);
DbExpression right = DbExpression.Parameter(keyVal, keyMemberDescriptor.MemberInfoType);
DbExpression conditionExp = new DbEqualExpression(left, right);
DbUpdateExpression e = new DbUpdateExpression(typeDescriptor.Table, conditionExp);
foreach (var item in updateColumns)
{
e.UpdateColumns.Add(item.Key.Column, item.Value);
}
int ret = this.ExecuteSqlCommand(e);
if (entityState != null)
{
entityState.Refresh();
}
return(ret);
}
