postgresql中如何处理json数据并放入数组

Question

我有一个包含两列的 postgresql 表。名称 VARCHAR(255) 并注释 JSON。示例数据集可能如下所示：

| name      | notes                            |
|-----------|----------------------------------|
| 'anna'    | {'link_to': ['bob']}             |
| 'bob'     | {'link_to': ['anna', 'claudia']} |
| 'claudia' | {'link_to': []}                  |

现在我想做两件事。

1. 将来自'link_to' 的json 列表放入另一个名为referres_to 的列中（那么它必须是数组VARCHAR 类型）。以我为例：

   name	notes	referrals_to
   'anna'	{'link_to': ['bob']}	['bob']
   'bob'	{'link_to': ['anna', 'claudia']}	['anna', 'claudia']
   'claudia'	{'link_to': []}	[]

2. 创建另一个名为referres_from 的列，我想在其中存储引用了某个名称的所有名称。在我的例子中：

   name	notes	referrals_from
   'anna'	{'link_to': ['bob']}	['bob']
   'bob'	{'link_to': ['anna', 'claudia']}	['anna']
   'claudia'	{'link_to': []}	['bob']

如何使用 postgresql 查询来做到这一点？我可以使用 python 轻松完成，但我猜这会比直接使用 postgresql 慢。

Answer 1

referrals_to 列可以使用-> 运算符提取数组。对于referrals_from 列，我将使用标量子选择将所有引用名称收集到单个列（JSON 数组）中

select name, 
       notes, 
       notes -> 'link_to' as referrals_to,
       (select jsonb_agg(name)
        from the_table t2
        where t2.name <> t1.name
        and t2.notes -> 'link_to' ? t1.name) as referrals_from
from the_table t1
;

? 运算符测试 JSON 数组是否包含特定字符串。在这种情况下，它会测试 link_to 数组是否包含来自外部查询的名称。

Online Example

假设 name 在您的表中是唯一的，您可以使用该查询来更新表中的新列：

update the_table 
   set referrals_to = notes -> 'link_to', 
       referrals_from = t.referrals_from
from (
  select t1.name, 
         (select jsonb_agg(name)
          from the_table t2
          where t2.name <> t1.name
          and t2.notes -> 'link_to' ? t1.name) as referrals_from
  from the_table t1
) t 
where t.name = the_table.name;

Answer 2

或多或少您在步骤中描述的内容：

WItH cte AS (
  SELECT
    name,
    BTRIM(value:: text, '"') AS referrals
  FROM
    (
      SELECT
        name,
        foo.notes:: json - > > 'link_to' AS link_to
      FROM
        foo
    ) a
    left JOIN LATERAL json_array_elements (a.link_to:: json) ON TRUE
)
SELECT
  a1.name,
  array_to_json(array_agg(a1.referrals))
FROM
  cte a1
  JOIN cte a2 ON a1.name = a2.referrals
GROUP BY
  a1.name