检查行是否已经存在，如果存在则告诉引用的表 id

Question

假设我有一本桌上杂志：

CREATE TABLE magazine
(
  magazine_id integer NOT NULL DEFAULT nextval(('public.magazine_magazine_id_seq'::text)::regclass),
  longname character varying(1000),
  shortname character varying(200),
  issn character varying(9),
  CONSTRAINT pk_magazine PRIMARY KEY (magazine_id)
);

还有一个表问题：

CREATE TABLE issue
(
  issue_id integer NOT NULL DEFAULT nextval(('public.issue_issue_id_seq'::text)::regclass),
  number integer,
  year integer,
  volume integer,
  fk_magazine_id integer,
  CONSTRAINT pk_issue PRIMARY KEY (issue_id),
  CONSTRAINT fk_magazine_id FOREIGN KEY (fk_magazine_id)
      REFERENCES magazine (magazine_id) MATCH SIMPLE
      ON UPDATE NO ACTION ON DELETE NO ACTION
);

当前插入：

INSERT INTO magazine (longname,shotname,issn)
VALUES ('a long name','ee','1111-2222');

INSERT INTO issue (fk_magazine_id,number,year,volume)
VALUES (currval('magazine_magazine_id_seq'),'8','1982','6');

现在只有在“杂志”不存在的情况下才可以插入一行。但是，如果存在，则表 'issue' 需要获取已存在的行的 'magazine_id' 才能建立引用。

我该怎么做？

提前谢谢！

Answer 1

您如何知道杂志是否已经在magazine 表中？ issn 列是否定义杂志？如果是，那么它应该是一个主键，或者至少是unique。

最简单的方法是在您的客户端应用程序中检查杂志是否存在，如下所示（在伪代码中）：

function insert_issue(longname, shotname, issn, number,year,volume) {
    /* extensive comments for newbies */
    start_transaction();
    q_get_magazine_id = prepare_query(
      'select magazine_id from magazine where issn=?'
    );
    magazine_id = execute_query(q_get_magazine_id, issn);
    /* if magazine_id is null now then there’s no magazine with this issn */
    /* and we have to add it */
    if ( magazine_id == NULL ) {
      q_insert_magazine = prepare_query(
        'insert into magazine (longname, shotname, issn)
          values (?,?,?) returning magazine_id'
      );
      magazine_id = execute_query(q_insert_magazine, longname, shortname, issn);
      /* we have tried to add a new magazine; */
      /* if we failed (magazine_id==NULL) then somebody else just added it */
      if ( magazine_id == NULL ) { 
        /* other, parerelly connected client just inserted this magazine, */
        /* this is unlikely but possible */
        rollback();
        start_transaction();
        magazine_id = execute_query(q_get_magazine_id, issn);
      }
    }
    /* now magazine_id is an id of magazine, */
    /* added if it was not in a database before, new otherwise */
    q_insert_issue = prepare_query(
      'insert into issue (fk_magazine_id,number,year,volume)
         values (?,?,?,?)'
    );
    execute_query(q_insert_issue, magazine_id, number, year, volume);
    /* we have inserted a new issue referencing old, */
    /* or if it was needed new, magazine */
    if ( ! commit() ) {
      rollback();
      raise "Unable to insert an issue";
    }
}

如果您只需要在一个查询中执行此操作，那么您可以将此伪代码实现为数据库中的 pl/pgsql 函数，只需 select insert_issue(?, ?, ?, ?, ?, ?)。

Answer 2

如果您使用的是 PostgreSQL 9.1 或更高版本，您可以执行以下操作：

WITH ref_key (id) AS (
    WITH ins (id) AS (
       INSERT INTO magazine (longname,shotname,issn)
       VALUES ('a long name','ee','1111-2222')
       RETURNING id
    )
    SELECT id 
      FROM magazine 
      LEFT JOIN ins USING id
     WHERE issn = '1111-2222'
)
INSERT INTO INTO issue (fk_magazine_id,number,year,volume)
SELECT id,'8','1982','6'
  FROM ref_key;

可写 CTE 的 FTW！

Answer 3

我不确定您是否可以使用 SQL 执行此操作。我知道 Oracle 可以用于触发器，但我认为 SQL 不能。如果我错了，请有人纠正我。