using Microsoft.Extensions.Logging;
|
using SqlSugar;
|
using System;
|
using System.Collections.Generic;
|
using System.Linq;
|
using System.Text;
|
using System.Threading.Tasks;
|
|
namespace WIDESEA_Core.BaseRepository
|
{
|
public class UnitOfWork : IDisposable
|
{
|
// 定义一个ILogger类型的Logger属性
|
public ILogger Logger { get; set; }
|
// 定义一个ISqlSugarClient类型的Db属性
|
public ISqlSugarClient Db { get; internal set; }
|
|
// 定义一个ITenant类型的Tenant属性
|
public ITenant Tenant { get; internal set; }
|
|
// 定义一个bool类型的IsTran属性，表示是否开启事务
|
public bool IsTran { get; internal set; }
|
|
// 定义一个bool类型的IsCommit属性，表示是否提交事务
|
public bool IsCommit { get; internal set; }
|
|
// 定义一个bool类型的IsClose属性，表示是否关闭连接
|
public bool IsClose { get; internal set; }
|
|
// 实现IDisposable接口的Dispose方法
|
public void Dispose()
|
{
|
// 如果开启事务且未提交事务，则回滚事务
|
if (IsTran && !IsCommit)
|
{
|
Logger.LogDebug("UnitOfWork RollbackTran");
|
this.Tenant.RollbackTran();
|
}
|
|
// 如果事务不为空或者已经关闭连接，则直接返回
|
if (this.Db.Ado.Transaction != null || this.IsClose)
|
return;
|
// 关闭连接
|
this.Db.Close();
|
}
|
|
// 提交事务
|
public bool Commit()
|
{
|
// 如果开启事务且未提交事务，则提交事务
|
if (this.IsTran && !this.IsCommit)
|
{
|
Logger.LogDebug("UnitOfWork CommitTran");
|
this.Tenant.CommitTran();
|
this.IsCommit = true;
|
}
|
|
// 如果事务为空且未关闭连接，则关闭连接
|
if (this.Db.Ado.Transaction == null && !this.IsClose)
|
{
|
this.Db.Close();
|
this.IsClose = true;
|
}
|
|
// 返回是否提交事务
|
return this.IsCommit;
|
}
|
}
|
}
|
