2

我正在尝试编写一个存储过程来将一组表中的数据子集复制到不同数据库中的一组相同表中。“源”数据库需要是存储过程的参数。

我已经为此苦苦挣扎了两天，我认为我有一个很好的解决方案：

  1. 验证模式是否相同。
  2. 使用动态 SQL 为源表创建临时“rmt”同义词。
  3. 使用 INSERT INTO A SELECT * FROM rmtA WHERE <条件>复制数据
  4. 删除同义词。

这对于大多数表来说效果很好，但是对于包含标识列的表，我不仅被迫设置 IDENTITY_INSERT ON & OFF，而且更糟糕的是，我不能使用 SELECT *; 我必须明确指定所有列。如果我稍后添加或删除列，这将是一场噩梦。

我必须得到一些东西，所以我现在要使用这个解决方案，但我想认为在某个地方有更好的解决方案。

帮助？

4

1 回答 1

1

听起来您在存储过程中使用了动态 SQL，因此您已准备好在 SELECT 子句中动态创建列列表。

您可以从 sys.columns 中进行选择以获取列列表，并了解该表是否具有标识列。这是一个查询，显示了创建列列表所需的信息。

SELECT c.name, is_identity
FORM sys.columns c
WHERE object_id = object_id('MyTable')

简而言之，如果is_identity至少一列为 1，则需要包含SET IDENTITY_INSERT. 而且，您将从 SELECT 子句 where is_identity= 1 中排除任何列。

而且，这种方法将适应您添加到表中的新列。

这是一个例子

DECLARE @TableName varchar(128) = 'MyTableName'
DECLARE @ColumnName varchar(128)
DECLARE @IsIdentity bit
DECLARE @TableHasIdentity bit = 0
DECLARE @sql varchar(2000) = 'SELECT '

-- create cursor to step through list of columns
DECLARE MyCurs CURSOR FOR
SELECT c.name, is_identity
FROM sys.columns c
WHERE object_id = object_id(@TableName)
ORDER BY column_id

-- open cursor and get first row
OPEN MyCurs
FETCH NEXT FROM MyCurs INTO @ColumnName, @IsIdentity

-- process each column in the table
WHILE @@FETCH_STATUS = 0
  BEGIN
    IF @IsIdentity = 0
-- add column to the SELECT clause
        SET @sql = @sql + @ColumnName + ', '
    ELSE
-- indicate that table has identity column  
        SET @TableHasIdentity = 1

-- get next column
    FETCH NEXT FROM MyCurs INTO @ColumnName, @IsIdentity
  END

-- cursor cleanup
CLOSE MyCurs
DEALLOCATE MyCurs

-- add FROM clause
SET @sql = LEFT(@sql, LEN(@sql)-1) + CHAR(13) + CHAR(10) + 'FROM ' + @TableName

-- add SET IDENTITY if necessary
IF @TableHasIdentity = 1
    SET @sql = 'SET IDENTITY_INSERT ' + @TableName + ' ON' + CHAR(13) + CHAR (10)
                + @sql + CHAR(13) + CHAR (10)
                + 'SET IDENTITY_INSERT ' + @TableName + ' OFF'

PRINT @sql
于 2010-08-04T04:10:27.080 回答