Вставить вставить идентификатор в другую таблицу

Question

Вот сценарий:

create table a (
id serial primary key, 
val text 
); 

create table b (
id serial primary key, 
a_id integer references a(id) 
); 

create rule a_inserted as on insert to a do also insert into b (a_id) values (new.id); 

Я пытаюсь создать запись в b ссылающийся на a при вставке в a таблице. Но я получаю, что new.id имеет значение NULL, поскольку он автоматически генерируется из последовательности. Я также попробовал спусковой крючок AFTER вставить FOR EACH ROW, но результат был тот же. Любой способ справиться с этим?

Answer 1

Избегайте правил, так как они вернутся, чтобы укусить вас.

Используйте триггер после таблицы a, который запускается для каждой строки. Это должно выглядеть примерно так (непроверенный):

create function a_ins() returns trigger as $$ 
begin 
    insert into b (a_id) values (new.id); 
    return null; 
end; 
$$ language plpgsql; 

create trigger a_ins after insert on a 
for each row execute procedure a_ins(); 

Answer 2

Чтобы сохранить его простым, вы также можете просто использовать data-modifying CTE (а не триггер или правило):

WITH ins_a AS (
    INSERT INTO a(val) 
    VALUES ('foo') 
    RETURNING a_id 
    ) 
INSERT INTO b(a_id) 
SELECT a_id 
FROM ins_a 
RETURNING b.*; -- last line optional if you need the values in return 

Родственный ответ с более подробной информации:

• PostgreSQL multi INSERT...RETURNING with multiple columns

Или у НУ может работать с currval() and lastval():

• How to get the serial id just after inserting a row?
• Reference value of serial column in another column during same INSERT

Answer 3

Не использовать триггеры или другие базы данных кунг-фу. Эта ситуация происходит каждый момент где-то в мире - существует простое решение:

После вставки используйте функцию LASTVAL(), которая возвращает значение последней последовательности, которая была автоматически увеличена.

Ваш код будет выглядеть следующим образом:

insert into a (val) values ('foo'); 
insert into b (a_id, val) values (lastval(), 'bar'); 

Легко читать, поддерживать и понимать.