将表里的数据批量生成INSERT语句的存储过程 继续增强版

9/1/2015来源:SQL技巧人气:1738

将表里的数据批量生成INSERT语句的存储过程 继续增强版

文章继续 桦仔兄的文章将表里的数据批量生成INSERT语句的存储过程 增强版继续增强...

本来打算将该内容回复于桦仔兄的文章的下面的,但是不知为何博客园就是不让提交!....

所以在这里贴出来吧,算作继续增加文章中解决的:根据查询条件自动生成插入脚本的需求,其实这种需求还是蛮常见的。

本文着重解决了文中的脚本的schema问题,给调整了下,现在脚本能自动识别出不同的schema下同名的表的语句

修改后脚本如下:

-- Author:      <桦仔>-- Blog:        <http://www.cnblogs.com/lyhabc/>-- Create date: <2014/10/18>-- Description: <根据查询条件导出表数据的insert脚本>-- =============================================ALTER  PROCEDURE InsertGenerator    (      @tableName NVARCHAR(MAX),      @whereClause NVARCHAR(MAX)    )AS --Then it includes a cursor to fetch column specific information (column name and the data type thereof) --from information_schema.columns pseudo entity and loop through for building the INSERT and VALUES clauses --of an INSERT DML statement.    DECLARE @string NVARCHAR(MAX) --for storing the first half of INSERT statement    DECLARE @stringData NVARCHAR(MAX) --for storing the data (VALUES) related statement    DECLARE @dataType NVARCHAR(MAX) --data types returned for respective columns    DECLARE @schemaName NVARCHAR(MAX) --schema name returned from sys.schemas    DECLARE @schemaNameCount int--shema count    DECLARE @QueryString  NVARCHAR(MAX) -- provide for the whole query,     set @QueryString=' '     --如果有多个schema,选择其中一个schema    SELECT @schemaNameCount=COUNT(*)    FROM    sys.tables t            INNER JOIN sys.schemas s ON t.schema_id = s.schema_id    WHERE   t.name = @tableName    WHILE(@schemaNameCount>0)    BEGIN    --如果有多个schema,依次指定    select @schemaName = name     from     (        SELECT ROW_NUMBER() over(order by  s.schema_id) RowID,s.name        FROM    sys.tables t                INNER JOIN sys.schemas s ON t.schema_id = s.schema_id        WHERE   t.name =  @tableName    ) as v    where RowID=@schemaNameCount    --Declare a cursor to retrieve column specific information     --for the specified table    DECLARE cursCol CURSOR FAST_FORWARD    FOR        SELECT  column_name ,                data_type        FROM    information_schema.columns        WHERE   table_name = @tableName                AND table_schema = @schemaName     OPEN cursCol    SET @string = 'INSERT INTO [' + @schemaName + '].[' + @tableName + ']('    SET @stringData = ''    DECLARE @colName NVARCHAR(500)    FETCH NEXT FROM cursCol INTO @colName, @dataType    PRINT @schemaName    PRINT @colName    IF @@fetch_status <> 0        BEGIN            PRINT 'Table ' + @tableName + ' not found, processing skipped.'            CLOSE curscol            DEALLOCATE curscol            RETURN        END    WHILE @@FETCH_STATUS = 0        BEGIN            IF @dataType IN ( 'varchar', 'char', 'nchar', 'nvarchar' )                BEGIN                       SET @stringData = @stringData + '''''''''+                            isnull(' + @colName + ','''')+'''''',''+'                END            ELSE                IF @dataType IN ( 'text', 'ntext' ) --if the datatype                                  --is text or something else                     BEGIN                        SET @stringData = @stringData + '''''''''+          isnull(cast(' + @colName + ' as nvarchar(max)),'''')+'''''',''+'                    END                ELSE                    IF @dataType = 'money' --because money doesn't get converted                        --from varchar implicitly                        BEGIN                            SET @stringData = @stringData                                + '''convert(money,''''''+        isnull(cast(' + @colName                                + ' as nvarchar(max)),''0.0000'')+''''''),''+'                        END                    ELSE                        IF @dataType = 'datetime'                            BEGIN                                SET @stringData = @stringData                                    + '''convert(datetime,''''''+        isnull(cast(' + @colName + ' as nvarchar(max)),''0'')+''''''),''+'                            END                        ELSE                            IF @dataType = 'image'                                BEGIN                                    SET @stringData = @stringData + '''''''''+       isnull(cast(convert(varbinary,' + @colName + ')        as varchar(6)),''0'')+'''''',''+'                                END                            ELSE --presuming the data type is int,bit,numeric,decimal                             BEGIN                                    SET @stringData = @stringData + '''''''''+          isnull(cast(' + @colName + ' as nvarchar(max)),''0'')+'''''',''+'                                END            SET @string = @string + '[' + @colName + ']' + ','            FETCH NEXT FROM cursCol INTO @colName, @dataType        END--After both of the clauses are built, the VALUES clause contains a trailing comma which needs to be replaced with a single quote. The prefixed clause will only face removal of the trailing comma.    DECLARE @Query NVARCHAR(MAX) -- provide for the whole query,                               -- you may increase the size    PRINT @whereClause    IF ( @whereClause IS NOT NULL         AND @whereClause <> ''       )        BEGIN              SET @query = 'SELECT ''' + SUBSTRING(@string, 0, LEN(@string))                + ') VALUES(''+ ' + SUBSTRING(@stringData, 0,                                              LEN(@stringData) - 2)                + '''+'')''    FROM ' +@schemaName+'.'+ @tableName + ' WHERE ' + @whereClause            PRINT @query           -- EXEC sp_executesql @query --load and run the built query--Eventually, close and de-allocate the cursor created for columns information.        END    ELSE  BEGIN             SET @query = 'SELECT ''' + SUBSTRING(@string, 0, LEN(@string))                + ') VALUES(''+ ' + SUBSTRING(@stringData, 0,                                              LEN(@stringData) - 2)                + '''+'')''     FROM ' + @schemaName+'.'+ @tableName        END    CLOSE cursCol    DEALLOCATE cursCol    SET @schemaNameCount=@schemaNameCount-1    IF(@schemaNameCount=0)    BEGIN       SET @QueryString=@QueryString+@query    END    ELSE    BEGIN        SET @QueryString=@QueryString+@query+' UNION ALL '    END    PRINT convert(varchar(max),@schemaNameCount)+'---'+@QueryString    END    EXEC sp_executesql @QueryString --load and run the built query--Eventually, close and de-allocate the cursor created for columns information.

1、测试脚本如下:

INSERT INTO test1.[customer]([city],[region]) VALUES('2','3')InsertGenerator 'customer', null

效果如下:

2、增加筛选条件

InsertGenerator 'customer', 'city=1'

其它内容可以参照桦仔兄的原文章地址。