如何使 FreeSql 和 CAP 进行集成

[yang-xiaodong]

本文用于描述如何使用 FreeSql 和 CAP 进行事务集成。

背景描述

在CAP中，事务对象需要交给CAP进行提交从而在事务实现提交后对缓存消息到 Broker 的 Flush 动作，而目前的Orm大部分都有自己的事务管理对象进行事务的提交。CAP官方直接原生支持使用 ADO.NET 和 EntityFrameworkCore 进行事务集成，而对于第三方ORM本文提供了一种扩展用以集成的示例。

示例

以下为在 FreeSql 中和 CAP 进行集成的示例：

1. FreeSql Repository+UnitOfWork 事务模式

public class FreeSqlRepositoryPatternTransaction : CapTransactionBase
{
    public FreeSqlRepositoryPatternTransaction(IDispatcher dispatcher, IUnitOfWork uow) : base(dispatcher)
    {
        Uow = uow;
    }

    public IUnitOfWork Uow { get; }

    public override object? DbTransaction => Uow.GetOrBeginTransaction();

    public override void Commit()
    {
        Uow.Commit();
        Flush();
    }

    public override Task CommitAsync(CancellationToken cancellationToken = default)
    {
        throw new NotImplementedException();
    }

    public override void Rollback()
    {
        Uow.Rollback();
    }

    public override Task RollbackAsync(CancellationToken cancellationToken = default)
    {
        throw new NotImplementedException();
    }

    public override void Dispose()
    {
        Uow.Dispose();
    }
}

public static class Extensions
{
      // 注意：你可以酌情修改此扩展以支持你的使用习惯，参考下方讨论内容
      public static ICapTransaction BeginTransaction(this IFreeSql freeSql,
          ICapPublisher publisher, out IRepositoryUnitOfWork uow, bool autoCommit = false)
      {
          var dispatcher = publisher.ServiceProvider.GetRequiredService<IDispatcher>();
          uow = freeSql.CreateUnitOfWork();
          var transaction = new FreeSqlRepositoryPatternTransaction(dispatcher, uow)
          {
              AutoCommit = autoCommit
          };
          return publisher.Transaction= transaction;
      }
}

使用方式：

[Route("~/with/test")]
public IActionResult WithTransaction()
{
      using (var transaction = _freeSql.BeginTransaction(_capBus, out var uow, false))
      {
          _capBus.Publish("sample.rabbitmq.mysql", DateTime.Now);

          var person = _freeSql.GetRepository<Person2>();
          person.UnitOfWork = uow;
          person.Insert(new Person2() { Name = "HelloWorld2" });

          transaction.Commit();
      }

    return Ok();
}

2. FreeSql DbContext 事务模式

此扩展方式不支持使用 dbContext.UnitOfWork.Orm Api 来获得 IFreeSql 对象，请直接通过 dbContext 来获取。

public class CapUnitOfWork : IUnitOfWork
{
    private readonly CapTransactionBase _transactionBase;

    public CapUnitOfWork(CapTransactionBase transactionBase)
    {
        _transactionBase = transactionBase;
    }

    public void Dispose()
    {
        _transactionBase.Dispose();
    }

    public DbTransaction GetOrBeginTransaction(bool isCreate = true)
    {
        return _transactionBase.DbTransaction as DbTransaction;
    }

    public void Commit()
    {
        _transactionBase.Commit();

        if (EntityChangeReport.OnChange != null && EntityChangeReport.Report.Any())
        {
            EntityChangeReport.OnChange.Invoke(EntityChangeReport.Report); 
        }
    }

    public void Rollback()
    {
        _transactionBase.Rollback();
    }

    public IFreeSql Orm => throw new NotSupportedException("不支持使用此方式获取 IFreeSql 对象");
    public IsolationLevel? IsolationLevel { get; set; }
    public DbContext.EntityChangeReport EntityChangeReport { get; } = new();
    public Dictionary<string, object> States { get; } = new();
}


public static class Extensions
{
    public static DbContext WithCap(this DbContext dbContext, ICapPublisher publisher)
    {
        var dispatcher = publisher.ServiceProvider.GetRequiredService<IDispatcher>();
        var mysqlTransaction = new MySqlCapTransaction(dispatcher)
        {
            AutoCommit = false,
            DbTransaction = dbContext.UnitOfWork.GetOrBeginTransaction(true)
        };
        dbContext.UnitOfWork = new CapUnitOfWork(mysqlTransaction);
        publisher.Transaction= mysqlTransaction;
        return dbContext;
    }
}

使用示例

[Route("~/with/test")]
public IActionResult WithTransaction()
{
    using var dbContext = _freeSql.CreateDbContext().WithCap(_capBus);

    _capBus.Publish("sample.rabbitmq.mysql", DateTime.Now);

    dbContext.Add(new Person2() { Name = "AAAA" });

    dbContext.SaveChanges();

    return Ok();
}

[luoyunchong]

每次CreateDbContext后，TraceAfter 事件 是否会重复绑定

[2881099]

可以可以，throw new NotSupportExceition

[yang-xiaodong]

好的，感谢澄清，示例已更新

[zfchai]

请教 dotnetcore/CAP#1526 这个该怎么搞呢？

[yang-xiaodong]

你的问题是什么？

[zfchai]

使用 FreeSql DbContext 事务模式时，下面的 Extensions 类中 MySqlCapTransaction 替换使用 pg 的事务处理该怎么弄？

[yang-xiaodong]

使用 FreeSql DbContext 事务模式时，下面的 Extensions 类中 MySqlCapTransaction 替换使用 pg 的事务处理该怎么弄？

替换 MySqlCapTransaction --> PostgreSqlCapTransaction 即可。

[zfchai]

好的，请问 PostgreSqlCapTransaction 这个需要依赖什么 ns 或者 nuget 包呢？

我还有个问题请教，描述如下：

• 使用 FreeSqlRepositoryPatternTransaction 模式时，

因为 FreeSql 的 IUnitOfWork 暂时不支持 CommitAsync 和 RollbackAsync 异步操作，请问扩展方法可以这样使用麽？

using (var transaction = freeSql.BeginTransaction(capBus, out var uow, false))
{
    // 使用 capBus 提供的相关异步操作方法
    await capBus.PublishDelayAsync(TimeSpan.FromSeconds(10), name, DateTime.Now);

    var person = freeSql.GetRepository<Person>();
    person.UnitOfWork = uow;
    // 使用 FreeSql 提供的相关异步操作方法
    await person.InsertAsync(new Person() { Name = "HelloWorld2" });

    // 事务同步提交
    transaction.Commit();
}

你上面例子给出都是同步的操作方法，请问这样使用恰当麽，谢谢！

[zfchai]

替换 MySqlCapTransaction --> PostgreSqlCapTransaction 即可。
需要导入对应的 CAP 包，DotNetCore.CAP.PostgreSql

[yang-xiaodong]

应该可以混用，你试试