functions/Copy-SqlDataCollector.ps1

Function Copy-SqlDataCollector
{
<#
.SYNOPSIS
Migrates user SQL Data Collector collection sets. SQL Data Collector configuration is on the agenda, but it's hard.
 
.DESCRIPTION
By default, all data collector objects are migrated. If the object already exists on the destination, it will be skipped unless -Force is used.
     
The -CollectionSets parameter is autopopulated for command-line completion and can be used to copy only specific objects.
 
THIS CODE IS PROVIDED "AS IS", WITH NO WARRANTIES.
 
.PARAMETER Source
Source SQL Server. You must have sysadmin access and server version must be SQL Server version 2008 or higher.
 
.PARAMETER Destination
Destination Sql Server. You must have sysadmin access and server version must be SQL Server version 2008 or higher.
 
.PARAMETER SourceSqlCredential
Allows you to login to servers using SQL Logins as opposed to Windows Auth/Integrated/Trusted. To use:
 
$scred = Get-Credential, then pass $scred object to the -SourceSqlCredential parameter.
 
Windows Authentication will be used if DestinationSqlCredential is not specified. SQL Server does not accept Windows credentials being passed as credentials.
To connect as a different Windows user, run PowerShell as that user.
 
.PARAMETER DestinationSqlCredential
Allows you to login to servers using SQL Logins as opposed to Windows Auth/Integrated/Trusted. To use:
 
$dcred = Get-Credential, then pass this $dcred to the -DestinationSqlCredential parameter.
 
Windows Authentication will be used if DestinationSqlCredential is not specified. SQL Server does not accept Windows credentials being passed as credentials.
To connect as a different Windows user, run PowerShell as that user.
 
.PARAMETER Force
If collection sets exists on destination server, it will be dropped and recreated.
 
.PARAMETER WhatIf
Shows what would happen if the command were to run. No actions are actually performed.
 
.PARAMETER Confirm
Prompts you for confirmation before executing any changing operations within the command.
 
.PARAMETER NoServerReconfig
Upcoming parameter to enable server reconfiguration
 
.NOTES
Tags: Migration
Author: Chrissy LeMaire (@cl), netnerds.net
Requires: sysadmin access on SQL Servers
 
dbatools PowerShell module (https://dbatools.io, clemaire@gmail.com)
Copyright (C) 2016 Chrissy LeMaire
 
This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
 
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License along with this program. If not, see <http://www.gnu.org/licenses/>.
 
.LINK
https://dbatools.io/Copy-SqlDataCollector
 
.EXAMPLE
Copy-SqlDataCollector -Source sqlserver2014a -Destination sqlcluster
 
Copies all Data Collector Objects and Configurations from sqlserver2014a to sqlcluster, using Windows credentials.
 
.EXAMPLE
Copy-SqlDataCollector -Source sqlserver2014a -Destination sqlcluster -SourceSqlCredential $cred
 
Copies all Data Collector Objects and Configurations from sqlserver2014a to sqlcluster, using SQL credentials for sqlserver2014a and Windows credentials for sqlcluster.
 
.EXAMPLE
Copy-SqlDataCollector -Source sqlserver2014a -Destination sqlcluster -WhatIf
 
Shows what would happen if the command were executed.
     
.EXAMPLE
Copy-SqlDataCollector -Source sqlserver2014a -Destination sqlcluster -CollectionSets 'Server Activity', 'Table Usage Analysis'
 
Copies two Collection Sets, Server Activity and Table Usage Analysis, from sqlserver2014a to sqlcluster.
#>

    [CmdletBinding(DefaultParameterSetName = "Default", SupportsShouldProcess = $true)]
    param (
        [parameter(Mandatory = $true)]
        [object]$Source,
        [parameter(Mandatory = $true)]
        [object]$Destination,
        [System.Management.Automation.PSCredential]$SourceSqlCredential,
        [System.Management.Automation.PSCredential]$DestinationSqlCredential,
        [switch]$NoServerReconfig,
        [switch]$Force
    )
    
    DynamicParam { if ($source) { return (Get-ParamSqlDataCollectionSets -SqlServer $Source -SqlCredential $SourceSqlCredential) } }
    
    BEGIN
    {
        if ([System.Reflection.Assembly]::LoadWithPartialName("Microsoft.SqlServer.Management.Collector") -eq $null)
        {
            throw "SMO version is too old. To migrate collection sets, you must have SQL Server Management Studio 2008 R2 or higher installed."
        }
        
        $sourceserver = Connect-SqlServer -SqlServer $Source -SqlCredential $SourceSqlCredential
        $destserver = Connect-SqlServer -SqlServer $Destination -SqlCredential $DestinationSqlCredential
        
        $source = $sourceserver.DomainInstanceName
        $destination = $destserver.DomainInstanceName
        
        # New name?
        $collectionSets = $psboundparameters.CollectionSets
        
        if ($sourceserver.versionMajor -lt 10 -or $destserver.versionMajor -lt 10)
        {
            throw "Collection Sets are only supported in SQL Server 2008 and above. Quitting."
        }
        
    }
    process
    {
        if ($NoServerReconfig -eq $false) 
        {
            Write-Warning "Server reconfiguration not yet supported. Only Collection Set migration will be migrated at this time."
            $NoServerReconfig = $true
        }
        
        $sourceSqlConn = $sourceserver.ConnectionContext.SqlConnectionObject
        $sourceSqlStoreConnection = New-Object Microsoft.SqlServer.Management.Sdk.Sfc.SqlStoreConnection $sourceSqlConn
        $sourceStore = New-Object Microsoft.SqlServer.Management.Collector.CollectorConfigStore $sourceSqlStoreConnection
        
        $destSqlConn = $destserver.ConnectionContext.SqlConnectionObject
        $destSqlStoreConnection = New-Object Microsoft.SqlServer.Management.Sdk.Sfc.SqlStoreConnection $destSqlConn
        $destStore = New-Object Microsoft.SqlServer.Management.Collector.CollectorConfigStore $destSqlStoreConnection
        
        $configdb = $sourceStore.ScriptAlter().GetScript() | Out-String
        $configdb = $configdb -replace "'$source'", "'$destination'"
        
        if (!$NoServerReconfig)
        {
            if ($Pscmdlet.ShouldProcess($destination, "Attempting to modify Data Collector configuration"))
            {
                try
                {
                    $sql = "Unknown at this time"
                    $destserver.ConnectionContext.ExecuteNonQuery($sql)
                    $destStore.Alter()
                }
                catch 
                { 
                    Write-Exception $_ 
                }
            }
        }
        
        if ($deststore.Enabled -eq $false) {
            Write-Warning "The Data Collector must be setup initially for Collection Sets to be migrated. "
            Write-Warning "Setup the Data Collector and try again."
            return
        }
        
        $storeCollectionSets = $sourceStore.CollectionSets | Where-Object { $_.isSystem -eq $false }
        if ($collectionSets.length -gt 0) { $storeCollectionSets = $storeCollectionSets | Where-Object { $collectionSets -contains $_.Name } }
        
        Write-Output "Migrating collection sets"
        foreach ($set in $storeCollectionSets)
        {
            $collectionName = $set.name
            if ($deststore.CollectionSets[$collectionName] -ne $null)
            {
                if ($force -eq $false)
                {
                    Write-Warning "Collection Set '$collectionName' was skipped because it already exists on $destination"
                    Write-Warning "Use -Force to drop and recreate"
                    continue
                }
                else
                {
                    if ($Pscmdlet.ShouldProcess($destination, "Attempting to drop $collectionName"))
                    {
                        Write-Verbose "Collection Set '$collectionName' exists on $destination"
                        Write-Verbose "Force specified. Dropping $collectionName."
                        
                        try
                        {
                            $deststore.CollectionSets[$collectionName].Drop()
                        }
                        catch
                        {
                            Write-Exception "Unable to drop: $_ Moving on."
                            continue
                        }
                    }
                }
            }
            
            if ($Pscmdlet.ShouldProcess($destination, "Migrating collection set $collectionName"))
            {
                try
                {
                    $sql = $set.ScriptCreate().GetScript() | Out-String
                    $sql = $sql -replace [Regex]::Escape("'$source'"), [Regex]::Escape("'$destination'")
                    Write-Verbose $sql
                    Write-Output "Migrating collection set $collectionName"
                    $null = $destserver.ConnectionContext.ExecuteNonQuery($sql)
                    
                    if ($set.IsRunning)
                    {
                        Write-Output "Starting collection set $collectionName"
                        $deststore.CollectionSets.Refresh()
                        $deststore.CollectionSets[$collectionName].Start()
                    }
                }
                catch
                {
                    Write-Exception $_
                }
            }
        }
    }
    
    end
    {
        $sourceserver.ConnectionContext.Disconnect()
        $destserver.ConnectionContext.Disconnect()
        If ($Pscmdlet.ShouldProcess("console", "Showing finished message")) { Write-Output "Data Collector migration finished" }
    }
}