Ecto preload为nil时创建空has_one关联及现有用户补StripeProfile方案

一、是否应该把批量创建逻辑写在迁移文件中？ #

不建议，原因如下：

• Ecto 迁移的设计定位是处理数据库结构变更，耦合业务数据写入逻辑会让迁移职责不清晰，后续维护成本很高
• 迁移执行时的运行环境和正常业务运行环境可能存在差异，如果后续StripeProfile的业务校验规则更新，迁移中的硬编码写入逻辑很容易触发数据异常
• 若存量用户量级较大，迁移执行时间过长会锁表，影响线上服务可用性

二、存量用户批量创建StripeProfile的实现方案 #

方案1：一次性数据补丁脚本（适合用户量<10w的场景） #

你可以写一个专属的Elixir脚本，或者在应用中写一个临时的mix任务来执行：

# lib/mix/tasks/populate_stripe_profiles.ex
defmodule Mix.Tasks.Populate.StripeProfiles do
  use Mix.Task
  import Ecto.Query
  alias MyApp.Repo
  alias MyApp.Users.User
  alias MyApp.Users.StripeProfile

  @shortdoc "为没有StripeProfile的存量用户创建空关联记录"
  def run(_) do
    Mix.Task.run("app.start")

    Repo.transaction(fn ->
      # 查所有没有关联StripeProfile的用户
      users_without_profile = 
        from u in User,
          left_join: sp in assoc(u, :stripe_profile),
          where: is_nil(sp.id),
          select: u.id

      # 批量构造插入数据
      profile_attrs = 
        users_without_profile
        |> Repo.all()
        |> Enum.map(fn user_id -> 
          %{user_id: user_id, inserted_at: DateTime.utc_now(), updated_at: DateTime.utc_now()} 
        end)

      # 批量插入，冲突直接忽略（避免并发写入重复）
      Repo.insert_all(StripeProfile, profile_attrs, on_conflict: :nothing)
    end)

    Mix.shell().info("所有存量用户的StripeProfile初始化完成")
  end
end

执行时直接跑命令 mix populate.stripe_profiles 即可。

方案2：分批次处理（适合用户量较大的场景） #

如果用户量超过10w，建议按id区间分批次插入，避免一次性占用过多数据库资源：

# 核心分批逻辑示例
batch_size = 500
max_user_id = Repo.one(from u in User, select: max(u.id))

for start_id <- 1..max_user_id//batch_size do
  end_id = start_id + batch_size - 1

  users = 
    from u in User,
      left_join: sp in assoc(u, :stripe_profile),
      where: is_nil(sp.id),
      where: u.id >= ^start_id and u.id <= ^end_id,
      select: u.id

  # 同方案1的插入逻辑
  profiles = Enum.map(users, &%{user_id: &1, inserted_at: DateTime.utc_now(), updated_at: DateTime.utc_now()})
  Repo.insert_all(StripeProfile, profiles, on_conflict: :nothing)

  # 每批插入后加少量延迟，避免压垮数据库
  Process.sleep(100)
end

三、实现preload关联为空时自动创建空记录的能力 #

不需要修改Ecto的preload逻辑，你可以选择以下两种方案实现：

方法1：封装查询方法（更推荐，侵入性低） #

# 在User的上下文模块里加这个方法
def get_user_with_stripe_profile(user_id) do
  user = Repo.get(User, user_id) |> Repo.preload(:stripe_profile)

  if is_nil(user.stripe_profile) do
    {:ok, profile} = create_stripe_profile(%{user_id: user.id})
    %{user | stripe_profile: profile}
  else
    user
  end
end

方法2：用after_load回调（自动触发，适合所有查询场景） #

# 在User的schema里加回调
schema "users" do
  # ... 原有字段定义
  has_one :stripe_profile, StripeProfile
  after_load :ensure_stripe_profile_exists
end

defp ensure_stripe_profile_exists(user) do
  if Ecto.assoc_loaded?(user.stripe_profile) and is_nil(user.stripe_profile) do
    # after_load是只读回调，不能直接写库，异步触发创建即可
    Task.start(fn -> 
      MyApp.Users.create_stripe_profile(%{user_id: user.id})
    end)
    # 临时构造空关联返回
    %{user | stripe_profile: %StripeProfile{user_id: user.id}}
  else
    user
  end
end

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