Ecto项目指南：使用Multi实现可组合的事务处理-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00807/article/details/148467907

Ecto项目指南：使用Multi实现可组合的事务处理

ecto A toolkit for data mapping and language integrated query. 项目地址: https://gitcode.com/gh_mirrors/ec/ecto

引言

在数据库操作中，事务(Transaction)是确保数据一致性的关键机制。Ecto作为Elixir生态中强大的数据库包装器，提供了完善的事务支持。本文将深入探讨Ecto中如何使用Ecto.Multi模块来构建可组合、声明式的事务处理流程。

传统事务处理的局限性

基础事务示例

考虑一个银行转账场景，需要原子性地完成两个账户的余额更新：

Repo.transaction(fn ->
  # 给Mary账户增加10元
  mary_update = from(Account, where: [id: ^mary.id], update: [inc: [balance: +10]])
  {1, _} = Repo.update_all(mary_update, [])
  
  # 从John账户扣除10元
  john_update = from(Account, where: [id: ^john.id], update: [inc: [balance: -10]])
  {1, _} = Repo.update_all(john_update, [])
end)

带错误处理的事务

当需要检查每个操作的状态时，代码会变得复杂：

Repo.transaction(fn ->
  case Repo.update_all(mary_update, []) do
    {1, _} ->
      case Repo.update_all(john_update, []) do
        {1, _} -> {:ok, {mary, john}}
        {_, _} -> Repo.rollback({:failed_transfer, john})
      end
    {_, _} ->
      Repo.rollback({:failed_transfer, mary})
  end
end)

嵌套事务的问题

虽然嵌套事务可以提高代码可读性，但仍然存在大量样板代码：

Repo.transaction(fn ->
  case transfer_money(mary, john, 10) do
    {:ok, {mary, john}} ->
      transfer = %Transfer{from: mary.id, to: john.id, amount: 10}
      Repo.insert!(transfer)
    {:error, error} ->
      Repo.rollback(error)
  end
end)

Ecto.Multi解决方案

Ecto.Multi提供了一种声明式的方式来定义事务操作，将操作定义与执行上下文解耦。

基本用法

重写银行转账示例：

Ecto.Multi.new()
|> Ecto.Multi.update_all(:mary, mary_update, [])
|> Ecto.Multi.run(:check_mary, fn
  _repo, %{mary: {1, _}} -> {:ok, nil}
  _repo, %{mary: {_, _}} -> {:error, {:failed_transfer, mary}}
end)
|> Ecto.Multi.update_all(:john, john_update, [])
|> Ecto.Multi.run(:check_john, fn
  _repo, %{john: {1, _}} -> {:ok, nil}
  _repo, %{john: {_, _}} -> {:error, {:failed_transfer, john}}
end)

组合多个操作

可以轻松组合多个Multi操作：

transfer_money(mary, john, 10)
|> Ecto.Multi.insert(:transfer, %Transfer{
  from: mary.id, 
  to: john.id, 
  amount: 10
})

执行事务

最终执行事务并处理结果：

transfer_money(mary, john, 10)
|> Ecto.Multi.insert(:transfer, transfer)
|> Repo.transaction()
|> case do
  {:ok, %{transfer: transfer}} ->
    # 成功处理
  {:error, name, value, changes_so_far} ->
    # 失败处理
end

高级用法：依赖值处理

实际案例：带标签的文章发布

考虑一个文章发布场景，需要同时处理文章内容和标签：

def insert_or_update_post_with_tags(post, params) do
  Ecto.Multi.new()
  |> Ecto.Multi.run(:tags, fn _repo, _changes ->
    insert_and_get_all_tags(params)
  end)
  |> Ecto.Multi.run(:post, fn _repo, %{tags: tags} ->
    insert_or_update_post(post, tags, params)
  end)
  |> Repo.transaction()
end