sql >> Base de Datos >  >> RDS >> Sqlserver

Cómo agregar una restricción de clave principal a las columnas de identidad a todas las tablas en la base de datos de SQL Server - Tutorial de SQL Server / TSQL, parte 63

Escenario:

Estás trabajando como desarrollador de SQL Server. Obtuvo este requisito en el que debe crear una restricción de clave principal en todas las columnas de identidad en todas las tablas de la base de datos de SQL Server si no existe.
El nombre de la clave principal debe ser Pk_SchemaName_TableName_ColumnName.

Solución:

En primer lugar, debemos obtener la lista de tablas que tienen una columna de identidad y la restricción de clave principal no se crea en ellas. Usaremos tablas del sistema y vistas del sistema para obtener todas las tablas con columnas de identidad que no tienen restricciones de clave principal.
Luego vamos a usar el Cursor con SQL dinámico para generar una instrucción Alter para la tabla para agregar la restricción de clave principal y ejecutar el script.
USE YourDatabaseName
GO

--Declare Variables
DECLARE @DatabaseName AS VARCHAR(128)
DECLARE @SchemaName AS VARCHAR(128)
DECLARE @TableName AS VARCHAR(128)
DECLARE @ColumnName AS VARCHAR(128)

DECLARE CUR CURSOR
FOR
--Get list of tables with Identity Column
--on which have no Primary Key
SELECT DB_Name() AS DatabaseName
    ,Schema_Name(Schema_id) AS TableSchema
    ,OBJECT_NAME(c.OBJECT_ID) AS TableName
    ,c.NAME AS ColumnName
FROM sys.columns c
INNER JOIN sys.tables t ON c.object_id = t.object_id
WHERE is_identity = 1
    AND type_desc = 'USER_TABLE'

EXCEPT

SELECT TC.Table_Catalog AS DatabaseName
    ,TC.Table_Schema AS TableSchema
    ,TC.Table_Name AS TableName
    ,CCU.Column_Name AS ColumnName
FROM information_Schema.Table_Constraints TC
INNER JOIN Information_Schema.constraint_column_usage CCU 
ON TC.Constraint_Name = CCU.Constraint_Name
    AND TC.Table_Name = CCU.Table_Name
WHERE Constraint_Type = 'PRIMARY KEY'

OPEN Cur

FETCH NEXT
FROM Cur
INTO @DatabaseName
    ,@SchemaName
    ,@TableName
    ,@ColumnName

WHILE @@FETCH_STATUS = 0
BEGIN
    --Build dynamic sql for each database 
    DECLARE @SQL VARCHAR(MAX) = NULL

    SET @SQL = 'Alter Table [' + @SchemaName + '].[' + @TableName + ']' + CHAR(10)
    SET @SQL += 'Add Constraint [Pk_' + @SchemaName 
+ '_' + @TableName 
             + '_' + @ColumnName + '] '
    SET @SQL += 'Primary Key ([' + @ColumnName + '])'

    EXEC (@SQL)

    PRINT @SQL

    FETCH NEXT
    FROM Cur
    INTO @DatabaseName
        ,@SchemaName
        ,@TableName
        ,@ColumnName
END

CLOSE Cur

DEALLOCATE Cur
 
 
 
 
 Si desea simplemente generar declaraciones de modificación para agregar Restricción de clave principal, puede agregar un comentario a la parte Exec(@SQL). El script anterior imprimirá declaraciones de modificación y podrá ejecutarlas según le convenga. Si no le gusta agregar la parte del nombre del esquema a la restricción de clave principal, puede eliminar @SchemaName.
Ejecuté la consulta anterior y creó una restricción de clave principal en las columnas de identidad donde faltaba. Alterar scripts generados por script para agregar restricción de clave principal a las columnas de identidad

Verifiqué las tablas y las restricciones de clave principal se crearon correctamente. Cómo crear una restricción de clave principal en columnas de identidad en la base de datos de SQL Server