Ustaw IDENTITY_INSERT OFF dla wszystkich tabel

Question

Mam skrypt, który tworzy całą bazę danych i wstawia wszystkie rekordy do kilkudziesięciu tabel. Działa wspaniale, chyba że wystąpi jakiś problem podczas przetwarzania, a tabela zostanie pozostawiona z IDENTITY_INSERT ON, gdy skrypt ulegnie uszkodzeniu podczas wstawiania i zanim będzie można go ponownie wyłączyć.

Kiedy to się dzieje, skrypt automatycznie zawiedzie przy próbie uruchomienia go ponownie, z błędem "IDENTITY_INSERT jest już włączony dla tabeli xx", gdy wchodzimy do wstawiania dla pierwszej tabeli.

Jako zabezpieczenie przed uszkodzeniem chciałbym się upewnić, że IDENTITY_INSERT jest ustawione na WYŁ. Dla wszystkich tabel, przed uruchomieniem reszty przetwarzania w skrypcie instalacyjnym.

Jako alternatywę, moglibyśmy ewentualnie zamknąć połączenie MS SQL i otworzyć je ponownie, co, jak rozumiem, wyczyściłoby wszystkie wartości IDENTITY_INSERT dla sesji połączenia.

Jaki jest najlepszy sposób, aby to zrobić i zapobiec błędom "już trwają"?

Answer 1

EXEC sp_MSforeachtable @command1="SET IDENTITY_INSERT ? OFF" 

Answer 2

Dynamiczny SQL:

select 'set identity_insert ['+s.name+'].['+o.name+'] off' 
from sys.objects o 
inner join sys.schemas s on s.schema_id=o.schema_id 
where o.[type]='U' 
and exists(select 1 from sys.columns where object_id=o.object_id and is_identity=1) 

Następnie skopiuj & wklej wygenerowany kod SQL w innym oknie zapytań i uruchomić

Answer 3

EXEC sp_MSforeachtable @command1="PRINT '?'; SET IDENTITY_INSERT ? OFF", 
@whereand = ' AND EXISTS (SELECT 1 FROM sys.columns WHERE object_id = o.id AND is_identity = 1)' 

Opierając się na odpowiedź Lynn, w przypadku jesteś zbyt leniwy, aby wykonaj to w więcej niż jednym kroku - powinno to działać na wszystkich tabelach, w których znajduje się kolumna tożsamości.

Zastrzeżenie jest testowany tylko w 2012 i sp_MSforeachtable jest oczywiście całkowicie nieobsługiwanym ...

Answer 4

Opierając się na odpowiedź użytkownika @ KevD - To działa dobrze dla wyłączenia, ale tutaj jest bardziej dla umożliwiając również.

Aby wyłączyć wszystkie wkładki tożsamości, gdzie muszą być wyłączone, zastosowanie -

EXEC sp_MSforeachtable @command1="PRINT '?'; SET IDENTITY_INSERT ? OFF", 
@whereand = ' AND EXISTS (SELECT 1 FROM sys.columns WHERE object_id = o.id 
AND is_identity = 1) and o.type = ''U''' 

Aby włączyć wszystkie wkładki tożsamości, gdzie muszą być włączone, zastosowanie -

EXEC sp_MSforeachtable @command1="PRINT '?'; SET IDENTITY_INSERT ? ON", 
@whereand = ' AND EXISTS (SELECT 1 FROM sys.columns WHERE object_id = o.id 
AND is_identity = 1) and o.type = ''U''' 

Testowane na serwerze SQL 2014 i 2016

Answer 5

Miałem podobny problem, ale wolałbym nie używać nieudokumentowanych procedur przechowywanych w produkcji. Aby zautomatyzować to, wbudowałem odpowiedź @Johna Deweya i umieściłem ją w kursorze. Powtarza to ponad 699 tabel w 407 ms.

DECLARE @sql NVARCHAR(500) -- SQL command to execute 

DECLARE sql_cursor CURSOR LOCAL FAST_FORWARD FOR 
SELECT 'SET identity_insert ['+s.name+'].['+o.name+'] OFF' 
FROM sys.objects o 
INNER JOIN sys.schemas s on s.schema_id=o.schema_id 
WHERE o.[type]='U' 
AND EXISTS(SELECT 1 FROM sys.columns WHERE object_id=o.object_id AND is_identity=1) 

OPEN sql_cursor 
FETCH NEXT FROM sql_cursor INTO @sql 

WHILE @@FETCH_STATUS = 0 
    BEGIN 
     EXECUTE sp_executesql @sql --> Comment this out to test 
     -- PRINT @sql --> Uncomment to test or if logging is desired 
     FETCH NEXT FROM sql_cursor INTO @sql 
    END 

CLOSE sql_cursor 
DEALLOCATE sql_cursor 

Jeśli grasz przeciwko kursorom, możesz łatwo przekształcić je w pętlę while.