如何实现Ecto中preload的反向操作？移除已预加载关联

当然有办法实现！Ecto本身并没有提供像Repo.preload/2那样直接的反向unload函数，但我们可以手动模拟这个行为——因为Ecto里未加载的关联本质上是一个Ecto.Association.NotLoaded结构体实例，我们只需要把已加载的关联字段替换成这个实例就行。

核心原理 #

当Ecto关联未被加载时，对应的字段值是#Ecto.Association.NotLoaded<association :comments is not loaded>，这其实是Ecto.Association.NotLoaded的实例，包含两个关键属性：

• __field__: 关联字段名（比如:comments）
• __owner__: 所属的Ecto Schema模块（比如Post）

我们只需要手动构造这个实例，替换掉结构体中已加载的关联字段即可。

具体实现 #

1. 单个结构体的Unload操作

如果你只需要处理单个Post结构体，可以直接修改字段值：

# 假设post是已经加载了comments的Post实例
post = %Post{post | comments: %Ecto.Association.NotLoaded{__field__: :comments, __owner__: Post}}

2. 封装通用的Unload函数

为了方便处理单个或多个结构体（比如从Repo查询回来的列表），可以封装一个通用函数：

defmodule RepoHelpers do
  @doc """
  将结构体或结构体列表中的指定关联重置为未加载状态
  """
  def unload(struct_or_structs, association) when is_list(struct_or_structs) do
    Enum.map(struct_or_structs, &unload(&1, association))
  end

  def unload(%module{} = struct, association) do
    not_loaded = %Ecto.Association.NotLoaded{__field__: association, __owner__: module}
    Map.put(struct, association, not_loaded)
  end
end

3. 使用示例

就像你描述的场景一样，查询所有Post后，调用这个函数即可将comments关联变回未加载状态：

posts = Repo.all(Post) |> RepoHelpers.unload(:comments)

注意事项 #

• 这个方法适用于所有类型的Ecto关联（一对一、一对多、多对多等），只要传入正确的关联字段名即可。
• 操作是纯内存修改，不会影响数据库中的数据。

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