52 lines
1.1 KiB
C#
52 lines
1.1 KiB
C#
using New_College.IRepository.UnitOfWork;
|
||
using SqlSugar;
|
||
using System;
|
||
|
||
namespace New_College.Repository.UnitOfWork
|
||
{
|
||
public class UnitOfWork : IUnitOfWork
|
||
{
|
||
private readonly ISqlSugarClient _sqlSugarClient;
|
||
|
||
public UnitOfWork(ISqlSugarClient sqlSugarClient)
|
||
{
|
||
_sqlSugarClient = sqlSugarClient;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取DB，保证唯一性
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public SqlSugarClient GetDbClient()
|
||
{
|
||
// 必须要as，后边会用到切换数据库操作
|
||
return _sqlSugarClient as SqlSugarClient;
|
||
}
|
||
|
||
public void BeginTran()
|
||
{
|
||
GetDbClient().BeginTran();
|
||
}
|
||
|
||
public void CommitTran()
|
||
{
|
||
try
|
||
{
|
||
GetDbClient().CommitTran(); //
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
GetDbClient().RollbackTran();
|
||
throw ex;
|
||
}
|
||
}
|
||
|
||
public void RollbackTran()
|
||
{
|
||
GetDbClient().RollbackTran();
|
||
}
|
||
|
||
}
|
||
|
||
}
|