如何使用Scripter仅创建特定的删除语句

时间:2015-08-11 19:50:36

标签: sql-server powershell smo

我正在使用Scripter类为我提供现有数据库中的数据脚本。我想编写一个可插入生产数据库的数据集。我们这样做是为了测试我们Software的安装是否正确。

不幸的是,数据集必须在以后删除而不留下任何条目,这样它就不会干扰我们客户的数据。所以我需要的是INSERT和DELTE语句。这些都是在负担过重的情况下手动维护的。

很好,所以我去了两次执行Scripter(一次用于INSERT,一次用于DELETE)

问题是,当将ScriptDrops设置为true时,输出的格式为

DELETE FROM [dbo].[TableName]

我想要的是形式:

DELETE FROM [dbo].[TableName] WHERE ID = 'GUID'

从技术上讲,这是可能的,因为所有表都有主键。 Scripter类还必须以某种形式知道这些事情,因为它还通过外键获得DELETE语句(依赖项)的顺序。

对此有任何帮助将不胜感激。

以下是我用于导出数据的2个PowerShell脚本:

ScriptRepositoryData.ps1

$scriptPath = $MyInvocation.MyCommand.Path
$scriptDirectory = Split-Path $scriptPath -Parent

. $scriptDirectory\DatabaseScripting.ps1

$filepath='c:\data.sql'
$database='ECMS_Repository'

$tablesToExclude = @(
  "SomeUnwantedTable"
  )

$tablesListFromDatabase = GetTableList $database

$tablesArray = @()

$tablesListFromDatabase |% {
  if (-not $tablesToExclude.Contains($_.Name.ToString()))
  {
    $tablesArray += $_.Name
  }
}

ScriptInsert $database $tablesArray $filepath

DatabaseScripting.ps1

[System.Reflection.Assembly]::LoadWithPartialName("Microsoft.SqlServer.SMO") | out-null
[System.Reflection.Assembly]::LoadWithPartialName("Microsoft.SqlServer.SMOExtended") | out-null

Function GetTableList ($database)
{
  Invoke-SqlCmd -Database $database -query "SELECT * FROM sys.tables"
}

Function ScriptInsert ($database, $tables, $destination)
{
  try {

    $serverMO = new-object ("Microsoft.SqlServer.Management.Smo.Server") "localhost"
    if ($serverMO.Version -eq  $null) {Throw "Can't find the instance localhost"}

    $urnsToScript = New-Object Microsoft.SqlServer.Management.Smo.UrnCollection

    $databaseMO = $serverMO.Databases.Item("ECMS_Repository")
    if ($databaseMO.Name -ne $database) {Throw "Can't find the database $database"}

    $tables |% {

      $tableListMO = $databaseMO.Tables.Item($_, "dbo")

      $tableListMO |% {
        $urnsToScript.Add($_.Urn)
      } 
    }

    $scripter = new-object ('Microsoft.SqlServer.Management.Smo.Scripter') $serverMO
    $scripter.Options.ScriptSchema = $False;
    $scripter.Options.ScriptData = $true;
    $scripter.Options.ScriptDrops = $true;
    $scripter.Options.ScriptAlter = $true;

    $scripter.Options.NoCommandTerminator = $true;
    $scripter.Options.Filename = $destination;
    $scripter.Options.ToFileOnly = $true
    $scripter.Options.Encoding = [System.Text.Encoding]::UTF8

    $scripter.EnumScript($urnsToScript)

    Write-Host -ForegroundColor Green "Done"
  }
  catch {
    Write-Host
    Write-Host -ForegroundColor Red "Error occured"
    Write-Host
    Write-Host $_.Exception.ToString()
    Write-Host
  }
}

1 个答案:

答案 0 :(得分:0)

不幸的是我没有找到使用Sql管理对象的方法。

无论如何,我现在使用Scripter的输出并选择每个表的ID。然后我使用ID来更改看起来像

的每一行
DELETE FROM [dbo].[tableName]

到此:

DELETE FROM [dbo].[tableName] WHERE ID IN ('guid1', 'guid2')

我是这样做的:

$content = Get-Content $destination
Clear-Content $destination

$content |% {
  $line = $_
  $table = $line.Replace("DELETE FROM [dbo].[","").Replace("]","")
  $query = "SELECT ID, ClassID FROM" + $_

  $idsAsQueryResult = Invoke-SqlCmd -Database $database -query $query
  $ids = $idsAsQueryResult | Select-Object -Expand ID

  if ($ids -ne $null) {
    $joinedIDs = [string]::Join("','",$ids)
    $newLine = $line + " WHERE ID IN ('" + $joinedIDs + "')"

    Add-Content $destination $newLine
  }
}

其中$ destination是使用Scripter类生成的脚本,$ database是包含数据库名称的字符串。

我必须选择第二列(由于我们的OR mapper重新存储而在所有表上存在ClassID),​​因为Select-Object中的一些奇怪的错误我不完全理解。

这当然只能起作用,因为所有表都有主键,所有主键都是命名ID,而不是组合主键或其他东西。 当然,通过SQL管理对象提取主键信息,您可以为其他更复杂的数据库模式实现相同的目的。