using EFCore.Sharding;
using Microsoft.Extensions.DependencyInjection;
using System;
using System.Data;
using System.Linq;
using System.Threading.Tasks;
namespace Core.Util
{
/// <summary>
/// 使用事务包裹
/// </summary>
public class TransactionalAttribute : BaseAOPAttribute
{
private readonly IsolationLevel _isolationLevel;
public TransactionalAttribute(IsolationLevel isolationLevel = IsolationLevel.ReadCommitted)
{
_isolationLevel = isolationLevel;
}
private TransactionContainer _container;
public override async Task Befor(IAOPContext context)
{
_container = context.ServiceProvider.GetService<TransactionContainer>();
if (!_container.TransactionOpened)
{
_container.TransactionOpened = true;
await _container.BeginTransactionAsync(_isolationLevel);
}
}
public override async Task After(IAOPContext context)
{
_container = context.ServiceProvider.GetService<TransactionContainer>();
try
{
if (_container.TransactionOpened)
{
_container.CommitTransaction();
}
}
catch (Exception ex)
{
_container.RollbackTransaction();
throw new Exception("系统异常", ex);
}
if (_container.TransactionOpened)
{
_container.TransactionOpened = false;
}
await Task.CompletedTask;
}
}
public class TransactionContainer : IScopedDependency, IDistributedTransaction
{
public TransactionContainer(IServiceProvider serviceProvider)
{
_distributedTransaction = DistributedTransactionFactory.GetDistributedTransaction();
var allRepositoryInterfaces = GlobalData.AllFxTypes.Where(x =>
typeof(IRepository).IsAssignableFrom(x)
&& x.IsInterface
&& x != typeof(IRepository)
).ToList();
allRepositoryInterfaces.Add(typeof(IRepository));
var repositories = allRepositoryInterfaces
.Select(x => serviceProvider.GetService(x) as IRepository)
.ToArray();
_distributedTransaction.AddRepository(repositories);
}
private readonly IDistributedTransaction _distributedTransaction;
public bool TransactionOpened { get; set; }
public void Dispose()
{
_distributedTransaction.Dispose();
}
public Task<(bool Success, Exception ex)> RunTransactionAsync(Func<Task> action, IsolationLevel isolationLevel = IsolationLevel.ReadCommitted)
{
return _distributedTransaction.RunTransactionAsync(action, isolationLevel);
}
public void AddRepository(params IRepository[] repositories)
{
_distributedTransaction.AddRepository(repositories);
}
public (bool Success, Exception ex) RunTransaction(Action action, IsolationLevel isolationLevel = IsolationLevel.ReadCommitted)
{
return _distributedTransaction.RunTransaction(action, isolationLevel);
}
public void BeginTransaction(IsolationLevel isolationLevel)
{
_distributedTransaction.BeginTransaction(isolationLevel);
}
public Task BeginTransactionAsync(IsolationLevel isolationLevel)
{
return _distributedTransaction.BeginTransactionAsync(isolationLevel);
}
public void CommitTransaction()
{
_distributedTransaction.CommitTransaction();
}
public void RollbackTransaction()
{
_distributedTransaction.RollbackTransaction();
}
public void DisposeTransaction()
{
_distributedTransaction.DisposeTransaction();
}
}
}
(精华)2020年6月27日 C#类库 TransactionalAttribute(Castle-AOP的封装)
猜你喜欢
转载自blog.csdn.net/aa2528877987/article/details/106984762
今日推荐
周排行