如何在psql的UPDATE命令中使用UNION语句或or语句？

Question

我有一个应用程序，通过api向用户提供“代码”。一个代码属于一个代码池，当用户到达一个端点时，他/她将从这个“池”中得到一个代码。目前，只有一个“池”的代码可以被提供。这个想法最好在下面的sql中表达。

<<-SQL
    UPDATE codes SET vended_at = NOW()
      WHERE id = (
        SELECT "codes"."id"
        FROM "codes"
        INNER JOIN "code_batches" ON "code_batches"."id" = "codes"."code_batch_id"
        WHERE "codes"."vended_at" IS NULL
          AND "code_batches"."active" = true
        ORDER BY "code_batches"."end_at" ASC
        FOR UPDATE OF "codes" SKIP LOCKED
        LIMIT 1 
      )
    RETURNING *;
 SQL

因此，基本上，当端点被点击时，我将返回一个代码，即active，其vended_at字段为NULL。

现在，我需要做的是构建这个sql，这样用户就可以从这个池或第二个池中获得代码。例如，如果用户无法从这个池获得代码(我们将称它为由上面的sql表示的A)，我需要从另一个池(我们将称之为B)提供一个代码。

我查阅了postgresql的文档，我想我要做的是做任何一件事。以某种方式使用UNION将池A和B组合成一个超级池来销售代码，或者如果我无法通过池A出售代码，则使用postgresql的OR子句从池B中进行选择。

问题是我似乎不能使用这些语法中的任何一个。我试过这样的方法，用不同的变体来调整它。

<<-SQL
    UPDATE codes SET vended_at = NOW()
      WHERE id = (
        SELECT "codes"."id"
        FROM "codes"
        INNER JOIN "code_batches" ON "code_batches"."id" = "codes"."code_batch_id"
        WHERE "codes"."vended_at" IS NULL
          AND "code_batches"."active" = true
        ORDER BY "code_batches"."end_at" ASC
        FOR UPDATE OF "codes" SKIP LOCKED
        LIMIT 1 
      ) UNION (
          ######## SELECT SOME OTHER STUFF #########
      )
    RETURNING *;
 SQL

或

<<-SQL
    UPDATE codes SET vended_at = NOW()
      WHERE id = (
        SELECT "codes"."id"
        FROM "codes"
        INNER JOIN "code_batches" ON "code_batches"."id" = "codes"."code_batch_id"
        WHERE "codes"."vended_at" IS NULL
          AND "code_batches"."active" = true
        ORDER BY "code_batches"."end_at" ASC
        FOR UPDATE OF "codes" SKIP LOCKED
        LIMIT 1 
      ) OR (
          ######## SELECT SOME OTHER STUFF USING OR #########
      )
    RETURNING *;
 SQL

到目前为止，语法已经关闭，我开始怀疑是否可以使用这种方法来完成我想要做的事情。我无法确定我的方法是否错误，或者我是否使用了UNION、OR和SUB-SELECTS错误。有谁能给我什么建议来达到我的目标吗？谢谢。

#######编辑########

为了说明和使这个概念更容易，我基本上想要这样做。

<<-SQL
    UPDATE codes SET vended_at = NOW()
      WHERE id = (
        CRITERIA 1
      ) 
      OR/UNION 
      (
        CRITERIA 2
      )
    RETURNING *;
 SQL

Answer 1

使用一个表存储两个池。

向codes表中添加一个pool_number列，以指示代码位于哪个池中，然后只需添加

ORDER BY pool_number

到您现有的查询。