Ecto - 如何按照确切的顺序获取ID记录

Question

我有一个记录ID列表 - [9, 1, 4, 3]。

我想从postgresql中检索记录，并希望它们按照此ID列表进行排序。但是当我进行查询时，记录以任意顺序返回：

Ecto.Query.from(r in Record, where: r.id in [9, 1, 4, 3]) 
  |> Repo.all()
  |> Enum.map(&Map.get(&1, :id)) # => [4, 9, 1, 3]

如何检索具有相同订单的记录？

Answer 1

我不认为在数据库中有任何简单的方法可以做到这一点，但这里有一种方法可以在Elixir中执行此操作：

ids = [123, 4, 1, 3, 2, 456]
posts = from(p in Post, where: p.id in ^ids, select: {p.id, p}) |> Repo.all |> Map.new
posts = for id <- ids, posts[id], do: posts[id]
posts |> Enum.map(&(&1.id)) |> IO.inspect

输出：

[4, 1, 3, 2]

首先，我们建立一个id => post地图。然后，对于ids中的每个id，我们得到相应的Post，如果找到的话。在我的应用程序中，没有ID为123或456的帖子，所以在for中忽略了它们。

Answer 2

你可以使用PostgreSQL的array_position function和Ecto的fragment function。在您的情况下，它看起来像：

Ecto.Query.from(r in Record, where: r.id in [9, 1, 4, 3])
  |> Ecto.Query.order_by([r], fragment("array_position(?, ?)", [9, 1, 4, 3], r.id) 
  |> Repo.all()

我会避免在数据库引擎之外处理数据。在这个非常简单的例子中应该没关系。但是，它可能会对更大的数据集或更复杂的数据结构的性能产生影响，因为首先，您必须将结果加载到内存中，然后对它们执行操作才能更改顺序。