v3.2.666-preview20220730 - 修复 Clickhouse Delete bug;

This commit is contained in:
2881099
2022-07-28 09:35:12 +08:00
parent 0949b28096
commit ae2f43c4ec
5 changed files with 502 additions and 4 deletions

View File

@ -2,6 +2,7 @@
using System;
using System.Collections.Generic;
using System.Data;
using System.Data.Common;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
@ -20,12 +21,76 @@ namespace FreeSql.ClickHouse.Curd
public override string ToSql()
{
return base.ToSql()?.Replace("DELETE FROM ", "ALTER TABLE ").Replace(" WHERE ", " DELETE WHERE ");
return ReplaceDeleteSql(base.ToSql());
}
string ReplaceDeleteSql(string sql) => sql?.Replace("DELETE FROM ", "ALTER TABLE ").Replace(" WHERE ", " DELETE WHERE ");
public override int ExecuteAffrows()
{
var affrows = 0;
DbParameter[] dbParms = null;
ToSqlFetch(sb =>
{
if (dbParms == null) dbParms = _params.ToArray();
var sql = ReplaceDeleteSql(sb.ToString());
var before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Delete, sql, dbParms);
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
Exception exception = null;
try
{
affrows += _orm.Ado.ExecuteNonQuery(_connection, _transaction, CommandType.Text, sql, _commandTimeout, dbParms);
}
catch (Exception ex)
{
exception = ex;
throw;
}
finally
{
var after = new Aop.CurdAfterEventArgs(before, exception, affrows);
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
}
});
if (dbParms != null) this.ClearData();
return affrows;
}
#if net40
#else
public override Task<List<T1>> ExecuteDeletedAsync(CancellationToken cancellationToken = default) => throw new NotImplementedException($"FreeSql.Provider.ClickHouse {CoreStrings.S_Not_Implemented_Feature}");
async public override Task<int> ExecuteAffrowsAsync(CancellationToken cancellationToken = default)
{
var affrows = 0;
DbParameter[] dbParms = null;
await ToSqlFetchAsync(async sb =>
{
if (dbParms == null) dbParms = _params.ToArray();
var sql = ReplaceDeleteSql(sb.ToString());
var before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Delete, sql, dbParms);
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
Exception exception = null;
try
{
affrows += await _orm.Ado.ExecuteNonQueryAsync(_connection, _transaction, CommandType.Text, sql, _commandTimeout, dbParms, cancellationToken);
}
catch (Exception ex)
{
exception = ex;
throw;
}
finally
{
var after = new Aop.CurdAfterEventArgs(before, exception, affrows);
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
}
});
if (dbParms != null) this.ClearData();
return affrows;
}
#endif
}
}