Powershell--批量拆分SQL语句为事务并批处理

时间：2016-09-07 17:33:53 阅读：253 评论：0 收藏：0 [点我收藏+]

标签：

作为DBA，时不时会遇到将数据导入到数据库的情况，假设业务或研发提供一个包含上百万行INSERT语句的脚本文件，而且这些INSERT 语句没有使用GO来进行批处理拆分，那么直接使用SQLCMD来执行会发现该进程消耗大量物理内存并迟迟没有数据写入，即使脚本中每一行都添加了GO，你依然会发现这插入效率太差，让你无法忍受，怎么搞呢，下面小代码帮你折腾下:

$old_file_path= ‘C:\insert_scripts.sql‘
$new_file_path=‘C:\insert_scripts_new.sql‘
$tran_rows=50

$line_content="
SET NOCOUNT ON
GO
BEGIN TRAN
BEGIN TRY
"
$line_content |Out-File $new_file_path
$line_num=1

$sr1=[IO.StreamReader]$old_file_path
$line_content=""
while(-not $sr1.EndOfStream)
{
    
    $tmp_content= $sr1.ReadLine()
    $line_content=$line_content+"`r"+$tmp_content
    $line_num_mode =$line_num % $tran_rows
    
    if($line_num_mode -eq 0)
    {
        $line_content=$line_content+"
COMMIT
PRINT ‘执行第"+($line_num-$tran_rows)+"行到第"+($line_num)+"行成功‘
END TRY
BEGIN CATCH
ROLLBACK
PRINT ‘执行第"+($line_num-$tran_rows)+"行到第"+($line_num)+"行失败‘
END CATCH
GO
BEGIN TRAN
BEGIN TRY
"
        $line_content | Out-File -Append $new_file_path
        Write-Host "处理到行" $line_num
        $line_content=""
    }

    $line_num=$line_num+1
    
}

$line_content=$line_content+"
COMMIT
PRINT ‘执行第"+($line_num-$tran_rows)+"行到第"+($line_num)+"行完成‘
END TRY
BEGIN CATCH
ROLLBACK
PRINT ‘执行第"+($line_num-$tran_rows)+"行到第"+($line_num)+"行失败‘
END CATCH
GO
"
$line_content | Out-File -Append $new_file_path

$sr1.Close()