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

时间:2017-04-08 17:42:52

标签: elixir phoenix-framework ecto

我有一份记录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]

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

2 个答案:

答案 0 :(得分:6)

您可以使用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()

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

答案 1 :(得分:3)

我认为在数据库中没有任何简单的方法可以做到这一点,但是这里有一种方法可以在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,如果找到它,我们会得到相应的帖子。在我的应用中,没有ID为123或456的帖子,因此在for中忽略了它们。