Ecto迁移中同连接使用Repo执行数据迁移的问题咨询

首先得指出你示例代码里的一个关键问题：你把数据查询的逻辑写在了模块的顶层，这部分代码会在编译时执行，而不是迁移运行的时候——也就是说，alter table还没执行，你的查询就先跑了，这才是导致找不到status字段的真正原因，而不是连接的问题。

针对你的场景（数据量小、可接受停机、希望保证一致性），正确的做法是把DDL修改和数据迁移逻辑都放在迁移的up函数里，利用PostgreSQL支持DDL事务的特性，让整个操作在同一个事务、同一个连接中完成。

下面是修正后的完整迁移代码：

import Ecto.Query
defmodule MyApp.Repo.Migrations.AddStatus do
  use Ecto.Migration

  def up do
    # 第一步：在事务内添加新字段
    alter table(:foo) do
      add(:status, :text)
    end

    # 第二步：同一事务中查询数据并更新
    foos = from(f in MyApp.Foo) |> MyApp.Repo.all()
    Enum.each(foos, fn foo ->
      # 这里写你的复杂逻辑计算status值
      new_status = cond do
        foo.is_enabled -> "active"
        foo.last_login_at == nil -> "inactive"
        true -> "pending"
      end

      # 更新单条记录
      MyApp.Repo.update!(Ecto.Changeset.change(foo, status: new_status))
    end)
  end

  def down do
    # 回滚操作：移除status字段
    alter table(:foo) do
      remove(:status)
    end
  end
end

为什么这样能解决问题？ #

• Ecto在PostgreSQL环境下，默认会把整个up函数的逻辑包装在一个事务里执行，所以alter table和后续的查询、更新操作都会使用同一个数据库连接，字段添加后立刻就能被后续操作感知到。
• 把所有逻辑放在up函数内，确保代码在迁移运行时执行，而不是编译时，避免了字段还未创建就查询的问题。

额外优化建议 #

如果你的数据量虽然小，但逐条更新还是有点慢，可以用update_all做批量更新，效率会更高：

# 替换掉Enum.each的部分，用批量更新
from(f in MyApp.Foo,
  update: [
    set: [
      status: fragment("
        CASE
          WHEN ? THEN 'active'
          WHEN ? IS NULL THEN 'inactive'
          ELSE 'pending'
        END
      ", f.is_enabled, f.last_login_at)
    ]
  ]
) |> MyApp.Repo.update_all([])

这种方式直接在数据库层面完成计算和更新，不需要把所有数据加载到应用内存里，性能更好。

内容的提问来源于stack exchange，提问作者seddy