实现GBase中的ExecuteDeleted方法

This commit is contained in:
tobybain
2025-11-22 12:00:45 +08:00
parent fc718d2b9c
commit 14c165754b
2 changed files with 125 additions and 4 deletions

View File

@@ -2,6 +2,7 @@ using FreeSql.DataAnnotations;
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Linq; using System.Linq;
using System.Threading.Tasks;
using Xunit; using Xunit;
namespace FreeSql.Tests.GBase namespace FreeSql.Tests.GBase
@@ -80,9 +81,28 @@ namespace FreeSql.Tests.GBase
[Fact] [Fact]
public void ExecuteDeleted() public void ExecuteDeleted()
{ {
Assert.Throws<NotImplementedException>(() => delete.Where(a => a.Id > 0).ExecuteDeleted()); g.gbase.Delete<Topic>().Where(a => a.Id > 0).ExecuteAffrows();
var list = new[] { new Topic { Title = "t1" }, new Topic { Title = "t2" } };
g.gbase.Insert<Topic>().ExecuteAffrows();
var datas = delete.Where(a => a.Id > 0).ExecuteDeleted();
foreach (var data in datas)
{
Assert.Contains(list, it => it.Title == data.Title);
}
} }
[Fact]
public async Task ExecuteDeletedAsync()
{
await g.gbase.Delete<Topic>().Where(a => a.Id > 0).ExecuteAffrowsAsync();
var list = new[] { new Topic { Title = "t1" }, new Topic { Title = "t2" } };
var cnt = await g.gbase.Insert(list).ExecuteAffrowsAsync();
var datas = await delete.Where(a => a.Id > 0).ExecuteDeletedAsync();
foreach (var data in datas)
{
Assert.Contains(list, it => it.Title == data.Title);
}
}
[Fact] [Fact]
public void AsTable() public void AsTable()
{ {

View File

@@ -2,6 +2,7 @@
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Data; using System.Data;
using System.Data.Common;
using System.Text; using System.Text;
using System.Threading; using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
@@ -16,11 +17,111 @@ namespace FreeSql.GBase.Curd
{ {
} }
public override List<T1> ExecuteDeleted() => throw new NotImplementedException($"FreeSql.Provider.GBase {CoreErrorStrings.S_Not_Implemented_Feature}"); public override List<T1> ExecuteDeleted()
{
var ret = new List<T1>();
DbParameter[] dbParms = null;
StringBuilder sbret = null;
ToSqlFetch(sb =>
{
if (dbParms == null)
{
dbParms = _params.ToArray();
sbret = new StringBuilder();
var colidx = 0;
foreach (var col in _table.Columns.Values)
{
if (colidx > 0) sbret.Append(", ");
sbret.Append(_commonUtils.RereadColumn(col, _commonUtils.QuoteSqlName(col.Attribute.Name))).Append(" as ").Append(_commonUtils.QuoteSqlName(col.CsName));
++colidx;
}
}
var delSql = sb.ToString();
var validx = delSql.IndexOf(" WHERE ");
if (validx == -1) throw new ArgumentException(CoreErrorStrings.S_NotFound_Name("WHERE"));
var wherePart = delSql.Substring(validx);
var selectSql = new StringBuilder()
.Append("SELECT ").Append(sbret)
.Append(" FROM ").Append(_commonUtils.QuoteSqlName(TableRuleInvoke()))
.Append(wherePart);
var before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Delete, string.Concat(selectSql.ToString(), "; ", delSql, ";"), dbParms);
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
Exception exception = null;
try
{
ret.AddRange(_orm.Ado.Query<T1>(_table.TypeLazy ?? _table.Type, _connection, _transaction, CommandType.Text, selectSql.ToString(), _commandTimeout, dbParms));
_orm.Ado.ExecuteNonQuery(_connection, _transaction, CommandType.Text, delSql, _commandTimeout, dbParms);
}
catch (Exception ex)
{
exception = ex;
throw;
}
finally
{
var after = new Aop.CurdAfterEventArgs(before, exception, ret);
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
}
});
return ret;
}
#if net40 #if net40
#else #else
public override Task<List<T1>> ExecuteDeletedAsync(CancellationToken cancellationToken = default) => throw new NotImplementedException($"FreeSql.Provider.GBase {CoreErrorStrings.S_Not_Implemented_Feature}"); async public override Task<List<T1>> ExecuteDeletedAsync(CancellationToken cancellationToken = default)
{
var ret = new List<T1>();
DbParameter[] dbParms = null;
StringBuilder sbret = null;
await ToSqlFetchAsync(async sb =>
{
if (dbParms == null)
{
dbParms = _params.ToArray();
sbret = new StringBuilder();
var colidx = 0;
foreach (var col in _table.Columns.Values)
{
if (colidx > 0) sbret.Append(", ");
sbret.Append(_commonUtils.RereadColumn(col, _commonUtils.QuoteSqlName(col.Attribute.Name))).Append(" as ").Append(_commonUtils.QuoteSqlName(col.CsName));
++colidx;
}
}
var delSql = sb.ToString();
var validx = delSql.IndexOf(" WHERE ");
if (validx == -1) throw new ArgumentException(CoreErrorStrings.S_NotFound_Name("WHERE"));
var wherePart = delSql.Substring(validx);
var selectSql = new StringBuilder()
.Append("SELECT ").Append(sbret)
.Append(" FROM ").Append(_commonUtils.QuoteSqlName(TableRuleInvoke()))
.Append(wherePart);
var before = new Aop.CurdBeforeEventArgs(_table.Type, _table, Aop.CurdType.Delete, string.Concat(selectSql.ToString(), "; ", delSql, ";"), dbParms);
_orm.Aop.CurdBeforeHandler?.Invoke(this, before);
Exception exception = null;
try
{
ret.AddRange(await _orm.Ado.QueryAsync<T1>(_table.TypeLazy ?? _table.Type, _connection, _transaction, CommandType.Text, selectSql.ToString(), _commandTimeout, dbParms, cancellationToken));
await _orm.Ado.ExecuteNonQueryAsync(_connection, _transaction, CommandType.Text, delSql, _commandTimeout, dbParms, cancellationToken);
}
catch (Exception ex)
{
exception = ex;
throw;
}
finally
{
var after = new Aop.CurdAfterEventArgs(before, exception, ret);
_orm.Aop.CurdAfterHandler?.Invoke(this, after);
}
});
return ret;
}
#endif #endif
} }
} }