web-dev-qa-db-fra.com

Insérer dans select dans plusieurs tables liées à l'aide de INSERT_IDENTITY

D'accord pour préparer le terrain. J'ai trois tables, (Table1, Table2 et DataTable) et je souhaite insérer dans Table1 et Table2 en utilisant DataTable comme source. Donc, pour chaque ligne dans DataTable je veux une ligne dans Table1 et Table2, et Table2 doit avoir le id (PK) inséré de Table1...

Si je devais faire ça ...

INSERT INTO Table1 SELECT A, B, C FROM MyTable
INSERT INTO Table2 SELECT IDENTITY_INSERT(), D, E, F FROM MyTable

J'obtiendrais le ID du dernier enregistrement inséré dans Table1.

Une boucle CURSOR ou WHILE est-elle la seule façon de procéder?

sql-serverinsertinheritance
10
m4rc

Une solution qui pourrait fonctionner pour vous est d'utiliser la clause OUTPUT, qui crache toutes les lignes insérées, vous pouvez donc les réinsérer dans une autre table . Cependant, cela impose des limitations sur les contraintes de clé étrangère sur Table2, si la mémoire est utilisée.

Quoi qu'il en soit, la solution ressemblerait à ceci:

MERGE INTO Table1 AS t1
USING MyTable ON 1=0 -- always generates "not matched by target"

WHEN NOT MATCHED BY TARGET THEN
    -- INSERT into Table1:
    INSERT (A, B, C) VALUES (t1.A, t1.B, t1.C)

--- .. and INSERT into Table2:
OUTPUT inserted.ID, MyTable.D, MyTable.E, MyTable.F
INTO Table2 (ID, D, E, F);

MERGE, contrairement aux autres instructions DML, peut référencer d'autres tables que inserted et deleted, ce qui vous est utile ici.

Plus: http://sqlsunday.com/2013/08/04/cool-merge-features/

10
Daniel Hutmacher

Si vous envisagez de le faire régulièrement (c'est-à-dire que cela fait partie de la logique d'application et non d'un exercice de transformation de données ponctuel), vous pouvez utiliser une vue sur Table1 et Table2 avec un déclencheur INSTEAD OF INSERT Pour gérer diviser les données (et organiser les clés/relations) - alors vous feriez simplement:

INSERT newView SELECT NEWID(), A, B, C, D, E, F FROM MyTable

et le déclencheur pourrait être aussi simple que:

CREATE trg_newview_insert TRIGGER newView INSTEAD OF UPDATE AS 
    INSERT table1 SELECT ID, A, B, C FROM inserted
    INSERT table2 SELECT ID, D, E, F FROM inserted
GO

en supposant que la vue est quelque chose comme:

CREATE VIEW newView AS 
SELECT table1.ID, A, B, C, D, E, F 
FROM table1 
    JOIN table2 ON table1.ID = table2.ID;

ou s'il peut y avoir des lignes dans chaque table sans lignes correspondantes dans l'autre:

CREATE VIEW newView AS 
SELECT ISNULL(table1.ID, table2.ID), A, B, C, D, E, F 
FROM table1 
    FULL OUTER JOIN table2 ON table1.ID = table2.ID;

(bien sûr, quelles lignes sont sorties lorsque vous SELECT à partir de la vue n'a pas d'importance si vous n'avez pas l'intention de SELECT à partir de celle-ci et il n'existe que pour fournir un modèle à INSERT pour que le déclencheur fasse sa magie)

Cela suppose que vous avez l'intention d'utiliser un type UUID pour votre clé primaire dans ce cas - si vous utilisez une clé entière incrémentée automatiquement sur table1, il y a un peu plus de travail à faire. Quelque chose comme ce qui suit pourrait fonctionner:

CREATE trg_newview_insert TRIGGER newView INSTEAD OF UPDATE AS 
    INSERT table1 (A, B, C) 
    SELECT A, B, C 
    FROM inserted;
    INSERT table2 (ID, D, E, F) 
    SELECT ID, D, E, F 
    FROM table1 AS t 
        JOIN inserted AS i ON t.A = i.A AND t.B = i.B AND t.C = i.C;
GO

et en fait, cette paire d'instructions INSERT peut fonctionner directement comme une seule fois comme cela (que vous utilisiez un type INT IDENTITY ou UNIQUEIDENTIFIER DEFAULT NEWID() pour le clé):

INSERT table1 (A, B, C) 
SELECT A, B, C 
FROM MyTable;
INSERT table2 (ID, D, E, F) 
SELECT ID, D, E, F 
FROM table1 AS t 
    JOIN MyTable AS i ON t.A = i.A AND t.B = i.B AND t.C = i.C;

annulant complètement le besoin de la vue et du déclencheur, bien que s'il s'agit d'une opération que vous effectuerez souvent dans votre code, le déclencheur vue + devrait encore être envisagé de faire abstraction du besoin de plusieurs instructions à chaque fois.

CAVEAT: tout le SQL ci-dessus a été tapé de la pensée et non testé, il aura besoin de travail avant qu'il n'y ait une garantie qu'il fonctionnera comme vous en avez besoin.

4
David Spillett

On dirait que tu veux:

INSERT dbo.Table1(A,B,C) SELECT A,B,C 
  FROM dbo.DataTable WHERE <identify one row>;

INSERT dbo.Table2(ID,D,E,F) SELECT SCOPE_IDENTITY(),D,E,F
  FROM dbo.DataTable WHERE <identify that same row>;

Ou peut-être utilisez-vous une seule table, si vous avez toujours une ligne dans chaque table ... avez-vous une bonne raison de les diviser en plusieurs tables?

3
Aaron Bertrand

En lisant votre question et les commentaires sur les autres réponses, il semble que vous tentiez de résoudre un problème avec DataTable en le divisant en deux nouveaux tableaux.

Je suppose que DataTable n'a pas déjà un seul champ unique tel qu'une IDENTITY(1,1)? Sinon, vous devriez peut-être en ajouter un que vous pourriez utiliser pour insérer des données dans Table1 Et Table2.

À titre d'exemple; J'ai créé un exemple de schéma, inséré des données de test dans DataTable, modifié DataTable pour avoir une colonne IDENTITY(1,1), puis utilisé cela pour insérer des données dans les deux Table1 et Table2:

USE tempdb;
GO

CREATE TABLE dbo.DataTable
(
    A INT
    , B INT
    , C INT
    , D INT
    , E INT
    , F INT
);

INSERT INTO dbo.DataTable (A, B, C, D, E, F)
VALUES (1, 2, 3, 11, 12, 13)
    , (4, 5, 6, 14, 15, 16)
    , (7, 8, 9, 17, 18, 19);

CREATE TABLE dbo.Table1
(
    Table1PK INT NOT NULL CONSTRAINT PK_Table1 PRIMARY KEY CLUSTERED IDENTITY(1,1)
    , A INT
    , B INT
    , C INT
);

CREATE TABLE dbo.Table2
(
    Table2PK INT NOT NULL CONSTRAINT PK_Table2 PRIMARY KEY CLUSTERED IDENTITY(1,1)
    , Table1PK INT NOT NULL CONSTRAINT FK_Table2_Table1_PK FOREIGN KEY REFERENCES dbo.Table1(Table1PK)
    , D INT
    , E INT
    , F INT
);

ALTER TABLE dbo.DataTable ADD TempCol INT NOT NULL IDENTITY(1,1);

SET IDENTITY_INSERT dbo.Table1 ON;

INSERT INTO Table1 (Table1PK, A, B, C)
SELECT TempCol, A, B, C 
FROM DataTable;

SET IDENTITY_INSERT dbo.Table1 OFF;

INSERT INTO Table2 
SELECT Table1PK, D, E, F 
FROM dbo.DataTable DT
    INNER JOIN dbo.Table1 T ON DT.TempCol = T.Table1PK;

SELECT *
FROM dbo.Table1;

SELECT *
FROM dbo.Table2;
1
Max Vernon