如何在 postgresql 的单个查询中选择一行并计算行数？

Question

我在 postgresql 中有一个如下表：

 id | chat_id |  content  | time |                 read_times                 
----+---------+-----------+------+-------------------------------------------------------------------------
  1 | chat_10 | content_1 |  t1  | [{"username": "user1", "time": 123}, {"username": "user2", "time": 111}]
  2 | chat_10 | content_2 |  t2  | [{"username": "user2", "time": 1235}]
  3 | chat_10 | content_3 |  t3  | []
  4 | chat_11 | content_4 |  t4  | [{"username": "user1", "time": 125}, {"username": "user3", "time": 121}]
  5 | chat_11 | content_5 |  t5  | [{"username": "user1", "time": 126}, {"username": "user3", "time": 127}]

注意：t1 < t2 < t3 < t4 < t5

每个用户阅读一条消息后，我们将其注册到read_times 列（user2 在时间1235 阅读了一条 id 为2 的消息），现在我想获取带有未读计数聊天的用户聊天列表。对于user1，结果如下：

 chat_id |  content  |  unread_count
 --------+-----------+--------------
 chat_10 | content_3 |       2
 chat_11 | content_5 |       0

注意：unread_count 是用户未在caht_id 中阅读的消息数。

一个查询可以吗？

Answer 1

首先，您必须使用json_array_elements() 函数和FIRST_VALUE() 窗口函数提取每个chat_id 和content 的用户名，获取每个chat_id 的最后一个content。
然后聚合并结合SUM()窗口函数和MAX()聚合函数得到列unread_count：

WITH cte AS (
  SELECT t.chat_id, t.content,
         FIRST_VALUE(t.content) OVER (PARTITION BY t.chat_id ORDER BY t.time DESC) last_content,
         (r->>'username') username
  FROM tablename t LEFT JOIN json_array_elements(read_times::json) r ON true
)
SELECT DISTINCT c.chat_id, MAX(c.last_content) "content",
       SUM((MAX((COALESCE(username, '') = 'user1')::int) = 0)::int) OVER (PARTITION BY c.chat_id) unread_count
FROM cte c
GROUP BY c.chat_id, c.content
ORDER BY c.chat_id

请参阅demo。