如何在作为变量传递的表名上设置identity_insert

时间:2015-04-23 19:32:01

标签: sql sql-server sql-server-2008 sql-server-2012

我在@strDBName中有数据库名称。我构建SET IDENTITY_INSERT并执行它。没有错误,但后续插入失败。下面的代码说明了问题。

  Declare @Query Varchar(MAX)
  SET @Query = 'SET IDENTITY_INSERT '+ @strDBName +'..TableName ON'
  EXEC(@Query)


  INSERT INTO [TableName] ... (MAX) Value from another table and other applicable record.


  SET @Query = 'SET IDENTITY_INSERT '+ @strDBName +'..TableName OFF'
  EXEC(@Query)

1 个答案:

答案 0 :(得分:2)

只是为了备份Brad在评论中给出的答案,这里是一个在单个动态查询中完成整个插入序列的MVCE。根据克里斯'注释,确保数据库名称是白名单,因为查询容易受到SqlInjection的影响(遗憾的是,数据库名称无法通过sp_executesql在动态sql中参数化)

假设:

CREATE TABLE TableName
(
    ID INT IDENTITY(1,1)
);

可以执行单个批处理:

DECLARE @strDBName VARCHAR(100) = 'MyDatabase';
Declare @Query Varchar(MAX)
SET @Query = 'SET IDENTITY_INSERT '+ @strDBName +'..TableName ON; '
SET @Query = @Query + 'INSERT INTO '+ @strDBName 
 +'..[TableName](ID) SELECT COALESCE(MAX(ID), 0)+1 FROM '+ @strDBName +'..TableName; '
SET @Query = @Query + 'SET IDENTITY_INSERT '+ @strDBName +'..TableName OFF;'
EXEC(@Query)