Question

我正在node.js中编写一个函数来查询PostgreSQL表如果该行存在，我想从行返回id列如果它不存在，我想插入它并返回id（insert into ... returning id）。

我一直在尝试使用case和if else语句的变体，似乎无法使其发挥作用。

Answer 1

单个SQL语句中的解决方案。虽然需要PostgreSQL 8.4 或更高版本请考虑以下演示：

测试设置：

CREATE TEMP TABLE tbl (
  id  serial PRIMARY KEY
 ,txt text   UNIQUE   -- obviously there is unique column (or set of columns)
);

INSERT INTO tbl(txt) VALUES ('one'), ('two');

INSERT / SELECT命令：

WITH v AS (SELECT 'three'::text AS txt)
    ,s AS (SELECT id FROM tbl JOIN v USING (txt))
    ,i AS (
       INSERT INTO tbl (txt)
       SELECT txt
       FROM   v
       WHERE  NOT EXISTS (SELECT * FROM s)
       RETURNING id
       )
SELECT id, 'i'::text AS src FROM i
UNION  ALL
SELECT id, 's' FROM s;

第一个 CTE v 并非绝对必要，但您只需输入值一次。
如果“行”存在，则第二个 CTE从id中选择 tbl。
第三个 CTE我将“行”插入<{strong> tbl如果（并且仅当）它不存在，则返回id。

最终SELECT会返回id。我添加了一列src，表示“来源” - “行”是否已预先存在且id来自SELECT，还是“行”是新的，id也是如此

此版本应尽可能快，因为它不需要来自tbl的额外SELECT，而是使用CTE。

为了在多用户环境中使其免受可能的竞争条件的影响：
此外，对于在Postgres 9.5 或更高版本中使用新UPSERT的更新技术：

Is SELECT or INSERT in a function prone to race conditions?

Answer 2

我建议在数据库端进行检查，然后将id返回给nodejs。

示例：

CREATE OR REPLACE FUNCTION foo(p_param1 tableFoo.attr1%TYPE, p_param2 tableFoo.attr1%TYPE) RETURNS tableFoo.id%TYPE AS $$
  DECLARE
  v_id tableFoo.pk%TYPE;
  BEGIN
    SELECT id
    INTO v_id
    FROM tableFoo
    WHERE attr1 = p_param1
    AND attr2 = p_param2;

    IF v_id IS NULL THEN
      INSERT INTO tableFoo(id, attr1, attr2) VALUES (DEFAULT, p_param1, p_param2)
      RETURNING id INTO v_id;
    END IF;

    RETURN v_id:

  END;
$$ LANGUAGE plpgsql;

而不是在Node.js端（我在这个例子中使用了node-postgres）：

var pg = require('pg');
pg.connect('someConnectionString', function(connErr, client){

  //do some errorchecking here

  client.query('SELECT id FROM foo($1, $2);', ['foo', 'bar'], function(queryErr, result){

    //errorchecking

    var id = result.rows[0].id;      

  };

});

Answer 3

如果您使用的是PostgreSQL 9.1

，请执行此类操作

with test_insert as (
   insert into foo (id, col1, col2)
   select 42, 'Foo', 'Bar'
   where not exists (select * from foo where id = 42)
   returning foo.id, foo.col1, foo.col2
)
select id, col1, col2
from test_insert
union 
select id, col1, col2
from foo
where id = 42;

它有点长，你需要重复id来测试几次，但我想不出涉及单个SQL语句的不同解决方案。

如果存在id=42的行，则可写CTE将不会插入任何内容，因此第二个并集部分将返回现有行。

测试时我实际上认为新行将返回两次（因此union而不是union all）但事实证明第二个select语句的结果实际上是在整个之前进行评估语句运行，它不会看到新插入的行。因此，如果插入了新行，它将从“返回”部分中获取。

Answer 4

create table t (
    id serial primary key,
    a integer
)
;

insert into t (a)
select 2
from (
    select count(*) as s
    from t
    where a = 2
    ) s
where s.s = 0
;
select id
from t
where a = 2
;

如果存在行则返回id，否则返回INSERT

4 个答案: