Как я могу вставить данные в две таблицы одновременно в SQL Server?

Question

Скажем, моя структура таблицы выглядит так:

CREATE TABLE [dbo].[table1] (
    [id] [int] IDENTITY(1,1) NOT NULL, 
    [data] [varchar](255) NOT NULL, 
    CONSTRAINT [PK_table1] PRIMARY KEY CLUSTERED ([id] ASC) 
) 

CREATE TABLE [dbo].[table2] (
    [id] [int] IDENTITY(1,1) NOT NULL, 
    [table1_id] [int] NOT NULL, 
    [data] [varchar](255) NOT NULL, 
    CONSTRAINT [PK_table2] PRIMARY KEY CLUSTERED ([id] ASC) 
) 

[id] поле первой таблицы соответствует [table1_id] поле второго. То, что я хотел бы сделать, это вставить данные в обе таблицы в одну транзакцию. Теперь я уже знаю, как это сделать, выполнив INSERT-SELECT-INSERT, например:

BEGIN TRANSACTION; 
DECLARE @id [int]; 
INSERT INTO [table1] ([data]) VALUES ('row 1'); 
SELECT @id = SCOPE_IDENTITY(); 
INSERT INTO [table2] ([table1_id], [data]) VALUES (@id, 'more of row 1'); 
COMMIT TRANSACTION; 

Это все хорошо и прекрасно подходит для небольших подобных случаев, когда вы только вставки, может быть несколько строк. Но мне нужно вставить пару сотен тысяч строк или, возможно, даже миллион строк, все сразу. Данные поступают из другой таблицы, так что, если я только вставить его в одну таблицу, это было бы легко, я просто должен сделать это:

INSERT INTO [table] ([data]) 
SELECT [data] FROM [external_table]; 

Но как бы я это сделать и расколоть данных в [table1] и [table2], и все еще обновлять [table2] с соответствующим [table1_id], как я это делаю? Возможно ли это?

Answer 1

Попробуйте это:

insert into [table] ([data]) 
output inserted.id, inserted.data into table2 
select [data] from [external_table] 

UPDATE: Re:

Denis - this seems very close to what I want to do, but perhaps you could fix the following SQL statement for me? Basically the [data] in [table1] and the [data] in [table2] represent two different/distinct columns from [external_table]. The statement you posted above only works when you want the [data] columns to be the same.

INSERT INTO [table1] ([data]) 
OUTPUT [inserted].[id], [external_table].[col2] 
INTO [table2] SELECT [col1] 
FROM [external_table] 

Это невозможно для вывода внешних колонок в качестве insert заявления, так что я думаю, что вы могли бы сделать что-то подобное

merge into [table1] as t 
using [external_table] as s 
on 1=0 --modify this predicate as necessary 
when not matched then insert (data) 
values (s.[col1]) 
output inserted.id, s.[col2] into [table2] 
; 

Answer 2

Вы можете написать хранимую процедуру, которая выполняет итерацию по транзакции, которую вы предложили. Итератором будет курсор для таблицы, содержащей исходные данные.

Answer 3

BEGIN TRANSACTION; 

DECLARE @tblMapping table(sourceid int, destid int) 

INSERT INTO [table1] ([data]) 
OUTPUT source.id, new.id 
Select [data] from [external_table] source; 

INSERT INTO [table2] ([table1_id], [data]) 
Select map.destid, source.[more data] 
from [external_table] source 
    inner join @tblMapping map on source.id=map.sourceid; 

COMMIT TRANSACTION; 

Answer 4

Обратите внимание, что SQL Server поддерживает инструкцию INSERT ALL. Oracle имеет уже, это выглядит следующим образом (SQL Cookbook):

insert all 
    when loc in ('NEW YORK', 'BOSTON') THEN 
    into dept_east(deptno, dname, loc) values(deptno, dname, loc) 
    when loc in ('CHICAGO') THEN 
    into dept_mid(deptno, dname, loc) values(deptno, dname, loc) 
    else 
    into dept_west(deptno, dname, loc) values(deptno, dname, loc) 
select deptno, dname, loc 
    from dept 

Answer 5

Другим вариантом является запуск двух вставок отдельно, оставляя колонки нуль FK, затем запустить обновление для poulate правильно.

Если в двух таблицах нет ничего естественного, которые соответствуют одной записи другим (скорее всего), создайте временный столбец GUID и заполните это в своих данных и вставьте в оба поля. Затем вы можете обновить с помощью FK и исключить GUID.

т.д .:

CREATE TABLE [dbo].[table1] ( 
    [id] [int] IDENTITY(1,1) NOT NULL, 
    [data] [varchar](255) NOT NULL, 
    CONSTRAINT [PK_table1] PRIMARY KEY CLUSTERED ([id] ASC), 
    JoinGuid UniqueIdentifier NULL 
) 

CREATE TABLE [dbo].[table2] ( 
    [id] [int] IDENTITY(1,1) NOT NULL, 
    [table1_id] [int] NULL, 
    [data] [varchar](255) NOT NULL, 
    CONSTRAINT [PK_table2] PRIMARY KEY CLUSTERED ([id] ASC), 
    JoinGuid UniqueIdentifier NULL 
) 


INSERT INTO Table1.... 

INSERT INTO Table2.... 

UPDATE b 
SET table1_id = a.id 
FROM Table1 a 
JOIN Table2 b on a.JoinGuid = b.JoinGuid 
WHERE b.table1_id IS NULL 

UPDATE Table1 SET JoinGuid = NULL 
UPDATE Table2 SET JoinGuid = NULL 

Answer 6

Я также борется с этой проблемой, и считают, что лучший способ заключается в использовании CURSOR.

Я пробовал решение Denis с OUTPUT, но, как он упоминает, невозможно выводить внешние столбцы в инструкции insert, а MERGE не может работать, когда вы вставляете несколько строк по выбору.

Итак, я использовал КУРСОР, для каждой строки во внешней таблице я сделал INSERT, а затем использовал @@ IDENTITY для другого INSERT.

DECLARE @OuterID int 

DECLARE MY_CURSOR CURSOR 
    LOCAL STATIC READ_ONLY FORWARD_ONLY 
FOR 
SELECT ID FROM [external_Table] 

OPEN MY_CURSOR 
FETCH NEXT FROM MY_CURSOR INTO @OuterID 

WHILE @@FETCH_STATUS = 0 
BEGIN 
INSERT INTO [Table] (data) 
    SELECT data 
    FROM  [external_Table] where ID = @OuterID 

    INSERT INTO [second_table] (FK,OuterID) 
    VALUES(@OuterID,@@identity) 

    FETCH NEXT FROM MY_CURSOR INTO @OuterID 
END 
CLOSE MY_CURSOR 
DEALLOCATE MY_CURSOR