SQL Server中中數據行批量插入腳本的存儲實現

本文轉載自查看原文 2015-12-23 17:12 2722 存儲過程/ 批量插入腳本/ MSSQL

看到博友SQL Server MVP 樺仔的一篇博文 “將表里的數據批量生成INSERT語句的存儲過程的實現”。我仔細看來博文中的兩個存儲代碼，自我感覺兩個都不太滿意，都是生成的單行模式的插入，數據行稍微大些性能會受影響的。所在公司本來就存在第二個版本的類似實現，但是是基於多行模式的，還是需要手工添加UNAION ALL來滿足多行模式的插入。看到這篇博文和基於公司數據行批量腳本的存儲的缺點，這次改寫和增強該存儲的功能。

本存儲運行於SQL Server 2005或以上版本，T-SQL代碼如下：

  1 IF OBJECT_ID(N'dbo.usp_GetInsertSQL', 'P') IS NOT NULL
  2 BEGIN
  3     DROP PROCEDURE dbo.usp_GetInsertSQL;
  4 END
  5 GO
  6  
  7 --==================================
  8 -- 功能: 獲取數據表記錄插入的SQL腳本
  9 -- 說明: 具體實現闡述 
 10 -- 作者: XXX
 11 -- 創建: yyyy-MM-dd
 12 -- 修改: yyyy-MM-dd XXX 修改內容描述
 13 --==================================
 14 CREATE PROCEDURE dbo.usp_GetInsertSQL 
 15 (
 16     @chvnTable NVARCHAR(128),            -- 數據表名稱(建議只使用表名稱，不要帶有分隔符[])
 17     @chvnWhere NVARCHAR(1000) = N'',    -- where查詢條件(不帶WHERE關鍵字）
 18     @bitIsSingleRow BIT = 1                -- 是否單行模式，默認為單行模式(單行模式為單行INSERT INTO VALUES格式；非單行模式（多行模式）為多行INSERT INTO SELECT格式)
 19 )
 20     --$Encode$--
 21 AS
 22 BEGIN
 23     SET NOCOUNT ON;
 24  
 25     SET @bitIsSingleRow = ISNULL(@bitIsSingleRow, 1);
 26  
 27     DECLARE
 28         @intTableID AS INT,
 29         @chvnSchemaTableName NVARCHAR(261);/*格式：[schema].[table]--1+128+1+1+1+128+1（各部分對應字符數）*/
 30  
 31     SELECT
 32         @intTableID = 0,
 33         @chvnSchemaTableName = N'';
 34  
 35     SELECT
 36         @intTableID = object_id
 37         ,@chvnSchemaTableName = QUOTENAME(SCHEMA_NAME(schema_id)) + N'.' + QUOTENAME(@chvnTable) /*組合架構名稱和表名稱的連接*/
 38     FROM sys.objects
 39     WHERE name = @chvnTable
 40         AND type = 'U';
 41  
 42     DECLARE
 43         @chvnColumnNames NVARCHAR(2000),                -- 字段列名集，多個以逗號','分隔，格式如：[column_name],[column_name],...
 44         @chvnColumnValues AS NVARCHAR(MAX);                -- 字段列值集，多個以逗號','分隔    
 45  
 46     DECLARE 
 47         @chvnTSQL AS NVARCHAR(MAX),                        -- TSQL腳本變量
 48         @chvnInsertIntoBoday AS NVARCHAR(2100);            -- InsertInto主體變量
 49     SELECT
 50         @chvnTSQL = N'',
 51         @chvnInsertIntoBoday = N'';
 52  
 53     SELECT
 54         @chvnColumnNames = ISNULL(@chvnColumnNames + N',', N'') +  QUOTENAME(T.column_name)
 55         ,@chvnColumnValues = ISNULL(@chvnColumnValues + N' + '','' + ', N'') + CAST(T.column_value AS NVARCHAR(800))
 56     FROM (SELECT 
 57                 name AS column_name /*字段列名*/
 58                 /*字段列值*/
 59                 ,column_value = CASE
 60                     WHEN system_type_id IN (48, 52, 56, 59, 60, 62, 104, 106, 108, 122, 127) /*數字數據類型：整數數據類型(bit、tinyint、smallint、int、bigint)，帶精度和小數的數據類型(decimal、numeric)和貨幣數據類型（monery和smallmoney*/                           
 61                             THEN 'CASE WHEN '+ name + ' IS NULL THEN ''NULL'' ELSE CAST(' + name + ' AS VARCHAR) END'
 62                     WHEN system_type_id IN (58, 61, 40, 41, 42) /*日期和時間數據類型：datetime、smalldatetime（兼容sql server 2008新增 date、datetime2和time）*/
 63                             THEN 'CASE WHEN '+ name + ' IS NULL THEN ''NULL'' ELSE '''''''' + REPLACE(CONVERT(VARCHAR(22), ' + name + ', 23), '' 00:00:00.000'', '''') + '''''''' END'
 64                     WHEN system_type_id IN (167) /*字符串數據類型：varchar*/
 65                             THEN 'CASE WHEN '+ name + ' IS NULL THEN ''NULL'' ELSE '''''''' + REPLACE(' + name + ', '''''''', '''''''''''') + '''''''' END'
 66                     WHEN system_type_id IN (231) /*Unicode字符串數據類型：nvarchar*/
 67                             THEN 'CASE WHEN '+ name + ' IS NULL THEN ''NULL'' ELSE ''N'''''' + REPLACE(' + name + ', '''''''','''''''''''') + '''''''' END'
 68                     WHEN system_type_id IN (175) /*字符串數據類型：char*/
 69                             THEN 'CASE WHEN '+ name + ' IS NULL THEN ''NULL'' ELSE '''''''' + CAST(REPLACE(' + name + ', '''''''' ,'''''''''''') AS CHAR(' + CAST(max_length AS VARCHAR)  + ')) + '''''''' END'
 70                     WHEN system_type_id IN (239) /*nicode字符串數據類型：nchar*/
 71                             THEN 'CASE WHEN '+ name + ' IS NULL THEN ''NULL'' ELSE ''N'''''' + CAST(REPLACE(' + name + ', '''''''' ,'''''''''''') AS CHAR(' + CAST(max_length AS VARCHAR)  + ')) + '''''''' END'
 72                     ELSE '''NULL'''    END
 73             FROM sys.columns  
 74             WHERE object_id = @intTableID
 75     ) AS T;
 76  
 77     SET @chvnInsertIntoBoday = N'''INSERT INTO '+  @chvnSchemaTableName + N' (' + @chvnColumnNames + N')''';    
 78     -- 方式一、代碼格式使用了GOTO和Label
 79     --BEGIN
 80     --    IF @bitIsSingleRow = 0 /*多行模式*/
 81     --    BEGIN    
 82     --        SET @chvnTSQL = N'SELECT ''SELECT '' + ' + @chvnColumnValues + ' AS RowData, ROW_NUMBER() OVER(ORDER BY (SELECT NULL)) AS RowNum FROM ' + @chvnSchemaTableName
 83  
 84  
 85     --        -- 此處不能使用GOTO WhereCondition;,因為之后的代碼不會被執行
 86     --        IF @chvnWhere > ''
 87     --        BEGIN
 88     --            SET @chvnTSQL = @chvnTSQL + ' WHERE ' + @chvnWhere;
 89     --        END
 90  
 91     --        -- 處理多行模式，需要使用ROW_NUMBER窗口函數        
 92     --        SET @chvnTSQL = N'SELECT CASE WHEN T.rownum = 1 THEN REPLICATE(N'' '', LEN(N''UNION ALL '') + 1) +  T.RowData ELSE N''UNION ALL '' + T.RowData END' +
 93     --                        N' FROM (' + @chvnTSQL + N') AS T';
 94  
 95     --        SET @chvnTSQL = N'SELECT '+ @chvnInsertIntoBoday + N';' +
 96     --                        @chvnTSQL;
 97  
 98     --        GOTO MultiRow;
 99     --    END
100     --    ELSE IF @bitIsSingleRow = 1 /*當行模式*/
101     --    BEGIN
102     --        SET @chvnTSQL = N'SELECT ' + @chvnInsertIntoBoday + 
103     --                        N' + ''VALUES('' + ' + @chvnColumnValues + ' + '');'' FROM ' + @chvnSchemaTableName;
104  
105     --        GOTO WhereCondition;
106     --    END
107  
108     --    -- where查詢條件
109     --    WhereCondition:
110     --    IF @chvnWhere > ''
111     --    BEGIN
112     --        SET @chvnTSQL = @chvnTSQL + ' WHERE ' + @chvnWhere;
113     --    END
114  
115  
116     --    MultiRow:/*多行模式GOTO的Label空標記*/
117  
118     --END
119  
120     -- 方式二、存在部分代碼的冗余
121     BEGIN
122         IF @bitIsSingleRow = 0 /*多行模式*/
123         BEGIN    
124             SET @chvnTSQL = N'SELECT ''SELECT '' + ' + @chvnColumnValues + ' AS RowData, ROW_NUMBER() OVER(ORDER BY (SELECT NULL)) AS RowNum FROM ' + @chvnSchemaTableName    
125  
126             IF @chvnWhere > ''
127             BEGIN
128                 SET @chvnTSQL = @chvnTSQL + ' WHERE ' + @chvnWhere;
129             END
130  
131             -- 多行模式特殊代碼，需要使用ROW_NUMBER窗口函數    
132             SET @chvnTSQL = N'SELECT CASE WHEN T.rownum = 1 THEN REPLICATE(N'' '', LEN(N''UNION ALL '') + 1) +  T.RowData ELSE N''UNION ALL '' + T.RowData END' +
133                             N' FROM (' + @chvnTSQL + N') AS T';
134  
135             SET @chvnTSQL = N'SELECT '+ @chvnInsertIntoBoday + N';' +
136                             @chvnTSQL;
137         END
138         ELSE IF @bitIsSingleRow = 1 /*單行模式*/
139         BEGIN
140             SET @chvnTSQL = N'SELECT ' + @chvnInsertIntoBoday + 
141                             N' + ''VALUES('' + ' + @chvnColumnValues + ' + '');'' FROM ' + @chvnSchemaTableName;
142  
143             IF @chvnWhere > ''
144             BEGIN
145                 SET @chvnTSQL = @chvnTSQL + ' WHERE ' + @chvnWhere;
146             END
147         END
148     END
149  
150     PRINT @chvnTSQL;        
151     EXEC(@chvnTSQL);
152 END
153 GO

為了測試以上存儲的效果，下面准備一個有數據的數據表，T-SQL代碼如下：

 1 IF OBJECT_ID(N'dbo.UserLoginInfo', N'U') IS NOT NULL
 2 BEGIN
 3     DROP TABLE dbo.UserLoginInfo;
 4 END
 5 GO
 6  
 7 -- create testing table UserLoginInfo
 8 CREATE TABLE dbo.UserLoginInfo (
 9     ID INT IDENTITY(1, 1) PRIMARY KEY,
10     Name VARCHAR(50) NOT NULL,
11     LoginTime DATETIME NOT NULL
12 );
13 GO
14  
15 -- insert testing data
16 INSERT  dbo.UserLoginInfo (Name, LoginTime) VALUES 
17 ('zhang', '2015-11-10 12:01:50')
18 ,('li', '2015-11-11 11:01:50')
19 ,('wang', '2015-11-9 11:01:50')
20 ,('zhang', '2015-11-11 12:01:50')
21 ,('li', '2015-11-11 12:01:50')
22 ,('wang', '2015-11-11 11:01:50')
23 ,('zhang', '2015-11-12 12:01:50')
24 ,('li', '2015-11-13 13:01:50')
25 ,('wang', '2015-11-12 11:01:50')
26 ,('zhang', '2015-11-13 12:01:50')
27 ,('li', '2015-11-14 11:01:50')
28 ,('wang', '2015-11-14 11:01:50')
29 ,('zhang', '2015-11-10 12:01:50')
30 ,('li', '2013-10-05 11:01:50')
31 ,('li', '2013-10-06 11:01:50')
32 ,('li', '2014-10-05 11:01:50')
33 ,('li', '2014-10-06 11:01:50')
34 ,('li', '2015-10-05 11:01:50')
35 ,('li', '2015-10-06 11:01:50')
36 ,('li', '2015-11-10 11:01:50')
37 ,('li', '2015-11-11 11:01:50')
38 ,('wang', '2015-11-09 11:01:50')
39 ,('zhang', '2015-11-11 12:01:50')
40 ,('li', '2015-11-11 12:01:50')
41 ,('wang', '2015-11-11 11:01:50')
42 ,('zhang', '2015-11-12 12:01:50')
43 ,('li', '2015-11-13 13:01:50')
44 ,('wang', '2015-11-12 11:01:50')
45 ,('zhang', '2015-11-13 12:01:50')
46 ,('li', '2015-11-14 11:01:50')
47 ,('wang', '2015-11-14 11:01:50');
48 GO

先測試單行模式的效果，相應的T-SQL代碼如下：

1 EXEC dbo.usp_GetInsertSQL
2     @chvnTable = N'UserLoginInfo',            -- nvarchar(128)
3     @chvnWhere = N'',                        -- nvarchar(1000)
4     @bitIsSingleRow = 1;                    -- bit
5 GO

執行后的查詢結果如下：

再測試多行模式的效果，相應的T-SQL代碼如下：

1 EXEC dbo.usp_GetInsertSQL
2     @chvnTable = N'UserLoginInfo',            -- nvarchar(128)
3     @chvnWhere = N'',                        -- nvarchar(1000)
4     @bitIsSingleRow = 0;                    -- bit
5 GO

執行后的查詢效果如下：

注意：多行模式，還需要將以上的兩個結果前后合並在一個文件就可以啦。

博友如有其他更好的解決方案，也請不吝賜教，萬分感謝。

免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。

猜您在找 SQL Server批量向表中插入多行數據語句 SQL server 批量插入和更新數據 SQL Server中數據的存儲 python腳本批量生成50000條插入數據的sql語句 SQL Server 批量插入數據的兩種方法 sql Server如何執行批量插入和批量刪除 SQL server存儲過程：數據的插入和更新 sql server存儲過程實現批量刪除 mysql 批量插入數據存儲過程 MySQL高級（十）--- 批量插入數據腳本