Сделать два зависимых INSERT в разные таблицы одним запросом postgresql?

Question

[Виктор]

Здравствуйте.
Подскажите, необходимо сделать два insert в разные таблицы, в одном запросе. При том, что второй insert зависит от результатов первого. И при этом, данные точно должны быть вставлены в обе таблицы, и если по какой-то причине данные не вставились в одну из таблиц, то и во второй их быть не должно.
Если что, запрос отправляю из Node в postgresql.
Понимаю, что нужна транзакция, но пока не могу понять, как использовать результат вставки.
Пример запросов.

INSERT INTO public.users(first_name, second_name, password) VALUES ($1, $2, $3) RETURNING id

INSERT INTO public.user_phone_number(user_id, phone_number) VALUES ($1, $2) RETURNING phone_number

В переменной $1 второго запроса, нужно использовать id, возвращаемый первым запросом.

Comments

[Fedor Vlasenko]

Напишите триггер

[Ипатьев]

непонятно, что здесь может быть непонятно.
выполнить первый запрос
получить результат
выполнить второй

[Akina]

PostgreSQL допускает применение INSERT в CTE с возвратом вставленных (в т.ч. генерируемых и присваиваемых в триггере) значений (RETURNING clause), которые можно использовать в основном запросе либо следующем CTE. Такой запрос выполняется как атом - соответственно если где-то ошибка, откатывается всё. что было изменено.

[Виктор]

Роман Юрьевич Ипатьев, Я так изначально и делаю, но это не правильно, это две ф-йй, выполняющие два запроса, а должна быть одна, и один запрос на добавление данных в две таблицы.

[Ипатьев]

с чего это взяли?
никому ваша функция ничего не должна
с чего вы решили что два запроса хуже, чем один?

[Виктор]

Роман Юрьевич Ипатьев, Тем, что я не контролирую целостность данных.

[Ипатьев]

Как это не контролируете?
Вы же сами написали

Понимаю, что нужна транзакция,

которая вам прекрасно всё и контролирует

[Виктор]

Роман Юрьевич Ипатьев, Хорошо, я инициализировал таблицу, в которую поместил результат первого инсерта и использовал его во втором, но я никак не могу вернуть результат этих запросов.

DO $$
DECLARE tableId bigint;
BEGIN
INSERT INTO public.users(first_name, second_name, password) VALUES ('dfg', 'ddf', 'dfgfg') RETURNING id INTO tableId;
INSERT INTO public.user_phone_number(user_id, phone_number) VALUES (tableId, '343443');
COMMIT;
END $$;

Returning не работает, и SELECT тоже не могу использовать внутри этой транзакции. Тут только отдельный запрос? Для генерации токена, необходим данные после вставки, и теперь я вижу только выход второго запроса в виде поиска по введённым данным.

Answer 1

[Виктор]

Решить вопрос можно двумя способами.
Первый

WITH user_data AS (INSERT INTO public.users(first_name, second_name, password) 
				VALUES ($1, $2, $3) RETURNING id, first_name, second_name), 
				user_phone AS (INSERT INTO public.user_phone_number(user_id, phone_number) 
				VALUES ((SELECT id FROM user_data), $4) RETURNING user_id, phone_number) 
				SELECT ud.id, ud.first_name, ud.second_name, up.phone_number 
				FROM user_data AS ud INNER JOIN user_phone AS up 
				ON up.user_id = ud.id

Второй

DO $$
DECLARE tableId bigint;
BEGIN
INSERT INTO public.users(first_name, second_name, password) VALUES ($1, $2, $3) RETURNING id INTO tableId;
INSERT INTO public.user_phone_number(user_id, phone_number) VALUES (tableId, $4);
COMMIT;
END $$;

Answer 2

[Andrey Tsvetkov]

https://stackoverflow.com/questions/6560447/can-i-...