当前位置:操作系统 > Unix/Linux >>

从数据库中获取Insert语句

 经常需要从开发库中提交生成数据的sql到测试库中。之前采用的方式:
    1、自己写sql,将sql保存起来,等测试的时候提交这些sql。

    2、在ERP系统中添加数据使用SQL SERVER Profiler跟踪SQL,保存起来,等测试的时候提交这些sql。

    两种方式存在的问题:

    1、如果修改一些数据,必须将更新语句也保存起来。

    2、保存的sql如果没有很好的管理机制,容易丢失。

     所以,在网上找了一下,写了一个根据数据库表名称和过滤条件生成insert的语句的存储过程,当提交测试的时候根据这个存储过程产生insert语句就行了。主要代码来自:Generating INSERT statements in SQL Server 。我根据业务需求做了相应的调整,代码如下:

使用这个存储过程:

SET QUOTED_IDENTIFIER OFF
GO
SET ANSI_NULLS ON
GO
DROP   PROC InsertGenerator
go

CREATE PROC InsertGenerator
    (
      @tableName VARCHAR(100) ,
      @filterCondition VARCHAR(100)
    )
AS --定义一个游标获取数据表列的相关信息
    DECLARE cursCol CURSOR FAST_FORWARD
    FOR
        SELECT  column_name ,
                data_type
        FROM    information_schema.columns
        WHERE   table_name = @tableName
    OPEN cursCol
    DECLARE @string NVARCHAR(3000) --存放insert语句的前一半
    DECLARE @stringData NVARCHAR(3000) --存放 (VALUES)
    DECLARE @dataType NVARCHAR(1000) --列的数据类型
    SET @string = 'INSERT ' + @tableName + '('
    SET @stringData = ''

    DECLARE @colName NVARCHAR(50)

    FETCH NEXT FROM cursCol INTO @colName, @dataType

    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 + '+'''''
                        + ''''',''NULL'')+'',''+'
                END
            ELSE
                IF @dataType IN ( 'text', 'ntext' ) --text 类型
                    BEGIN
                        SET @stringData = @stringData
                            + '''''''''+isnull(cast(' + @colName
                            + ' as varchar(2000)),'''')+'''''',''+'
                    END
                ELSE
                    IF @dataType = 'money' -- money 类型
                        BEGIN
                            SET @stringData = @stringData
                                + '''convert(money,''''''+isnull(cast('
                                + @colName
                                + ' as varchar(200)),''0.0000'')+''''''),''+'
                        END
                    ELSE
                        IF @dataType = 'datetime'-- datetime 类型
                            BEGIN
                                SET @stringData = @stringData
                                    + '''convert(datetime,' + '''+isnull('''''
                                    + '''''+convert(varchar(200),' + @colName
&nbs

CopyRight © 2012 站长网 编程知识问答 www.zzzyk.com All Rights Reserved
部份技术文章来自网络,