Public/Start-RestoreSqlDatabase.ps1
|
function Restore-SqlDatabase { <# .SYNOPSIS Restores database on MSSQL Server. .DESCRIPTION Uses Start-RestoreSqlDatabase.sql sql script to restore database. .EXAMPLE Start-RestoreSqlDatabase -DatabaseName "DbName" -ConnectionString "data source=localhost;integrated security=True" -Path "C:\database.bak" #> [CmdletBinding()] [OutputType([void])] param( [Parameter(Mandatory=$true)] [string] $ConnectionString, # Database name - if not specified, Initial Catalog from ConnectionString will be used. [Parameter(Mandatory=$false)] [string] $DatabaseName, # Backup file path. [Parameter(Mandatory=$true)] [string] $Path, # Remote share credential to use if $Path is an UNC path. Note the file will be copied to localhost if this set, and this will work only if # you're connecting to local database. [Parameter(Mandatory=$false)] [PSCredential] $RemoteShareCredential, # Timeout for executing sql restore command. [Parameter(Mandatory=$false)] [int] $QueryTimeoutInSeconds = 3600 ) try { if (!$DatabaseName) { $csb = New-Object -TypeName System.Data.SqlClient.SqlConnectionStringBuilder -ArgumentList $ConnectionString $DatabaseName = $csb.InitialCatalog } if (!$DatabaseName) { throw "No database name - please specify -DatabaseName or add Initial Catalog to ConnectionString." } if ($RemoteShareCredential) { $shareDir = Split-Path -Path $Path -Parent $username = $RemoteShareCredential.UserName $password = $RemoteShareCredential.GetNetworkCredential().Password Write-Log -Info "Connecting to $Path using username $username." & net use $shareDir $password /user:$username $tempDir = Join-Path -Path $env:TEMP -ChildPath 'PPoshSqlToolsTemp' if (Test-Path -LiteralPath $tempDir) { Write-Log -Info "Deleting temp directory '$tempDir'." [void](Remove-Item -LiteralPath $tempDir -Force -Recurse) } [void](New-Item -Path $tempDir -ItemType Directory -Force) Write-Log -Info "Copying '$Path' to '$tempDir'" Copy-Item -Path $Path -Destination $tempDir -Force #TODO: unhardcode this user Set-SimpleAcl -Path $tempDir -User 'NT Service\MSSQLSERVER' -Permission 'Read' -Type 'Allow' $Path = Join-Path -Path $tempDir -ChildPath (Split-Path -Path $Path -Leaf) } $sqlScript = Join-Path -Path $PSScriptRoot -ChildPath "Start-RestoreSqlDatabase.sql" $parameters = @{ "DatabaseName" = $DatabaseName } $parameters += @{ "Path" = $Path } [void](Invoke-Sql -ConnectionString $ConnectionString -InputFile $sqlScript -SqlCmdVariables $parameters -QueryTimeoutInSeconds $QueryTimeoutInSeconds -DatabaseName '') } finally { if ($RemoteShareCredential) { & net use $shareDir /DELETE Remove-TempDirectory } } } |