12

在 Postgres 9.4 和 Rails的列中使用 JSON 数组jsonb，我可以设置一个范围，该范围返回包含传递给范围方法的数组中的任何元素的所有行 - 如下所示：

scope :tagged, ->(tags) {
  where(["data->'tags' ?| ARRAY[:tags]", { tags: tags }])
}

我还想根据数组中匹配元素的数量对结果进行排序。

我很感激我可能需要跳出 ActiveRecord 的范围才能做到这一点，所以一个普通的 Postgres SQL 答案也很有帮助，但如果它可以包含在 ActiveRecord 中，那么它就可以成为一个可链接的范围。

根据要求，这是一个示例表。（实际的模式要复杂得多，但这就是我所关心的。）

 id |               data                
----+-----------------------------------
  1 | {"tags": ["foo", "bar", "baz"]}
  2 | {"tags": ["bish", "bash", "baz"]}
  3 |
  4 | {"tags": ["foo", "foo", "foo"]}

用例是根据标签查找相关内容。更多匹配的标签更相关，因此结果应按匹配数排序。在 Ruby 中，我有一个像这样的简单方法：

Page.tagged(['foo', 'bish', 'bash', 'baz']).all

应该按以下顺序返回页面：2, 1, 4.

4

2 回答 2

7

您的数组仅包含原始值，嵌套文档会更复杂。

询问

jsonb_array_elements_text()在LATERAL连接和计数匹配中取消嵌套找到的行的 JSON 数组：

SELECT *
FROM  (
   SELECT *
   FROM   tbl
   WHERE  data->'tags' ?| ARRAY['foo', 'bar']
   ) t
, LATERAL (
   SELECT count(*) AS ct
   FROM   jsonb_array_elements_text(t.data->'tags') a(elem)
   WHERE  elem = ANY (ARRAY['foo', 'bar'])  -- same array parameter
   ) ct
ORDER  BY ct.ct DESC;  -- more expressions to break ties?

与INSTERSECT. 这是我们可以使用此基本 SQL 功能的少数情况之一：

SELECT *
FROM  (
   SELECT *
   FROM   tbl
   WHERE  data->'tags' ?| '{foo, bar}'::text[]  -- alt. syntax w. array
   ) t
, LATERAL (
   SELECT count(*) AS ct
   FROM  (
      SELECT * FROM jsonb_array_elements_text(t.data->'tags')
      INTERSECT ALL
      SELECT * FROM unnest('{foo, bar}'::text[])  -- same array literal
      ) i
   ) ct
ORDER  BY ct.ct DESC;

请注意一个细微的区别：这会在匹配时消耗每个元素，因此它不会data->'tags'像第一个变体那样计算不匹配的重复项。有关详细信息，请参阅下面的演示。

还演示了传递数组参数的另一种方法：作为数组文字：'{foo, bar}'。对于某些客户来说，这可能更容易处理：

或者您可以创建一个带有VARIADIC参数的服务器端搜索函数并传递可变数量的普通text值：

有关的：

指数

确保有一个功能性 GIN 索引来支持jsonb存在运算符?|：

CREATE INDEX tbl_dat_gin ON tbl USING gin (data->'tags');

重复的细微差别

根据评论中的要求进行澄清。比如说，我们有一个带有两个重复标签（总共 4 个）的 JSON 数组：

jsonb '{"tags": ["foo", "bar", "foo", "bar"]}'

并使用包含两个标签的 SQL 数组参数进行搜索，其中一个重复（共 3 个）：

'{foo, bar, foo}'::text[]

考虑这个演示的结果：

SELECT *
FROM  (SELECT jsonb '{"tags":["foo", "bar", "foo", "bar"]}') t(data)

, LATERAL (
   SELECT count(*) AS ct
   FROM   jsonb_array_elements_text(t.data->'tags') e
   WHERE  e = ANY ('{foo, bar, foo}'::text[])
   ) ct

, LATERAL (
   SELECT count(*) AS ct_intsct_all
   FROM  (
      SELECT * FROM jsonb_array_elements_text(t.data->'tags')
      INTERSECT ALL
      SELECT * FROM unnest('{foo, bar, foo}'::text[])
      ) i
   ) ct_intsct_all

, LATERAL (
   SELECT count(DISTINCT e) AS ct_dist
   FROM   jsonb_array_elements_text(t.data->'tags') e
   WHERE  e = ANY ('{foo, bar, foo}'::text[])
   ) ct_dist

, LATERAL (
   SELECT count(*) AS ct_intsct
   FROM  (
      SELECT * FROM jsonb_array_elements_text(t.data->'tags')
      INTERSECT
      SELECT * FROM unnest('{foo, bar, foo}'::text[])
      ) i
   ) ct_intsct;

结果：

data                                     | ct | ct_intsct_all | ct_dist | ct_intsct
-----------------------------------------+----+---------------+---------+----------
'{"tags": ["foo", "bar", "foo", "bar"]}' | 4  | 3             | 2       | 2

将 JSON 数组中的元素与数组参数中的元素进行比较：

  • 4 个标签匹配任何搜索元素：ct.
  • 集合中的3 个标签相交（可以逐个匹配）ct_intsct_all：.
  • 可以识别2 个不同的ct_dist匹配标签：或ct_intsct.

如果您没有受骗者或不想排除他们，请使用前两种技术之一。另外两个要慢一些（除了不同的结果），因为他们必须检查是否有欺骗。

于 2015-07-17T02:32:35.140 回答
2

我在 Ruby 中发布了我的解决方案的详细信息，以防它对解决相同问题的任何人有用。

最后，我认为范围不合适，因为该方法将返回对象数组（不是可链接的ActiveRecord::Relation），因此我编写了一个类方法并提供了一种通过块将链接范围传递给它的方法：

def self.with_any_tags(tags, &block)
  composed_scope = (
    block_given? ? yield : all
  ).where(["data->'tags' ?| ARRAY[:tags]", { tags: tags }])

  t   = Arel::Table.new('t',  ActiveRecord::Base)
  ct  = Arel::Table.new('ct', ActiveRecord::Base)

  arr_sql = Arel.sql "ARRAY[#{ tags.map { |t| Arel::Nodes::Quoted.new(t).to_sql }.join(', ') }]"
  any_tags_func = Arel::Nodes::NamedFunction.new('ANY', [arr_sql])

  lateral = ct
    .project(Arel.sql('e').count(true).as('ct'))
    .from(Arel.sql "jsonb_array_elements_text(t.data->'tags') e")
    .where(Arel::Nodes::Equality.new Arel.sql('e'), any_tags_func)

  query = t
    .project(t[Arel.star])
    .from(composed_scope.as('t'))
    .join(Arel.sql ", LATERAL (#{ lateral.to_sql }) ct")
    .order(ct[:ct].desc)

  find_by_sql query.to_sql
end

这可以像这样使用：

Page.with_any_tags(['foo', 'bar'])

# SELECT "t".*
# FROM (
#   SELECT "pages".* FROM "pages"
#   WHERE data->'tags' ?| ARRAY['foo','bar']
#   ) t,
# LATERAL (
#   SELECT COUNT(DISTINCT e) AS ct
#   FROM jsonb_array_elements_text(t.data->'tags') e
#   WHERE e = ANY(ARRAY['foo', 'bar'])
#   ) ct
# ORDER BY "ct"."ct" DESC

Page.with_any_tags(['foo', 'bar']) do
  Page.published
end

# SELECT "t".*
# FROM (
#   SELECT "pages".* FROM "pages"
#   WHERE pages.published_at <= '2015-07-19 15:11:59.997134'
#   AND pages.deleted_at IS NULL
#   AND data->'tags' ?| ARRAY['foo','bar']
#   ) t,
# LATERAL (
#   SELECT COUNT(DISTINCT e) AS ct
#   FROM jsonb_array_elements_text(t.data->'tags') e
#   WHERE e = ANY(ARRAY['foo', 'bar'])
#   ) ct
# ORDER BY "ct"."ct" DESC
于 2015-07-19T15:17:07.837 回答