DSCResources/MSFT_SPSearchCrawlerImpactRule/MSFT_SPSearchCrawlerImpactRule.psm1
|
function Get-TargetResource { [CmdletBinding()] [OutputType([System.Collections.Hashtable])] param ( [parameter(Mandatory = $true)] [System.String] $ServiceAppName, [parameter(Mandatory = $true)] [System.String] $Name, [System.UInt32] $RequestLimit = 0, [System.UInt32] $WaitTime = 0, [ValidateSet("Present","Absent")] [System.String] $Ensure = "Present", [System.Management.Automation.PSCredential] $InstallAccount ) Write-Verbose -Message "Getting Crawler Impact Rule Setting for '$Name'" if(($RequestLimit -gt 0) -and ($WaitTime -gt 0)) { throw "Only one Crawler Impact Rule HitRate argument (RequestLimit, WaitTime) can be specified" } $result = Invoke-SPDSCCommand -Credential $InstallAccount ` -Arguments $PSBoundParameters ` -ScriptBlock { $params = $args[0] $nullReturn = @{ ServiceAppName = $params.ServiceAppName Name = $params.Name RequestLimit = $null WaitTime = $null Ensure = "Absent" InstallAccount = $params.InstallAccount } $serviceApp = Get-SPEnterpriseSearchServiceApplication -Identity $params.ServiceAppName if($null -eq $serviceApp) { $nullReturn.ServiceAppName = $null return $nullReturn } else { $crawlerImpactRule = Get-SPEnterpriseSearchSiteHitRule -Identity $params.Name -SearchService $serviceApp if($null -eq $crawlerImpactRule) { return $nullReturn } else { if($crawlerImpactRule.Behavior -eq "0") { return @{ ServiceAppName = $params.ServiceAppName Name = $params.Name RequestLimit = $crawlerImpactRule.HitRate WaitTime = 0 Ensure = "Present" InstallAccount = $params.InstallAccount } } else { return @{ ServiceAppName = $params.ServiceAppName Name = $params.Name RequestLimit = 0 WaitTime = $crawlerImpactRule.HitRate Ensure = "Present" InstallAccount = $params.InstallAccount } } } } } return $result } function Set-TargetResource { [CmdletBinding()] param ( [parameter(Mandatory = $true)] [System.String] $ServiceAppName, [parameter(Mandatory = $true)] [System.String] $Name, [System.UInt32] $RequestLimit = 0, [System.UInt32] $WaitTime = 0, [ValidateSet("Present","Absent")] [System.String] $Ensure = "Present", [System.Management.Automation.PSCredential] $InstallAccount ) Write-Verbose -Message "Setting Crawler Impact Rule Setting for '$Name'" if(($RequestLimit -gt 0) -and ($WaitTime -gt 0)) { throw "Only one Crawler Impact Rule HitRate argument (RequestLimit, WaitTime) can be specified" } $result = Get-TargetResource @PSBoundParameters if ($result.Ensure -eq "Absent" -and $Ensure -eq "Present") { Write-Verbose -Message "Creating Crawler Impact Rule $Name" Invoke-SPDSCCommand -Credential $InstallAccount ` -Arguments $PSBoundParameters ` -ScriptBlock { $params = $args[0] $behavior = "0" $hitRate = 0 if($null -eq $params.RequestLimit) { $behavior = "1" $hitRate = $params.WaitTime } else { $behavior = "0" $hitRate = $params.RequestLimit } $serviceApp = Get-SPEnterpriseSearchServiceApplication -Identity $params.ServiceAppName if($null -eq $serviceApp) { throw "The Search Service Application does not exist." } New-SPEnterpriseSearchSiteHitRule -Name $params.Name ` -Behavior $behavior ` -HitRate $hitRate ` -SearchService $serviceApp } } if ($result.Ensure -eq "Present" -and $Ensure -eq "Present") { Write-Verbose -Message "Updating Crawler Impact Rule $Name" Invoke-SPDSCCommand -Credential $InstallAccount ` -Arguments $PSBoundParameters ` -ScriptBlock { $params = $args[0] $behavior = "0" $hitRate = 0 if($null -eq $params.RequestLimit) { $behavior = "1" $hitRate = $params.WaitTime } else { $behavior = "0" $hitRate = $params.RequestLimit } $serviceApp = Get-SPEnterpriseSearchServiceApplication -Identity $params.ServiceAppName if($null -eq $serviceApp) { throw "The Search Service Application does not exist." } Remove-SPEnterpriseSearchSiteHitRule -Identity $params.Name ` -SearchService $serviceApp ` -ErrorAction SilentlyContinue New-SPEnterpriseSearchSiteHitRule -Name $params.Name ` -Behavior $behavior ` -HitRate $hitRate ` -SearchService $serviceApp } } if($Ensure -eq "Absent") { Write-Verbose -Message "Removing Crawler Impact Rule $Name" Invoke-SPDSCCommand -Credential $InstallAccount ` -Arguments $PSBoundParameters ` -ScriptBlock { $params = $args[0] $serviceApp = Get-SPEnterpriseSearchServiceApplication -Identity $params.ServiceAppName if($null -eq $serviceApp) { throw "The Search Service Application does not exist." } Remove-SPEnterpriseSearchSiteHitRule -Identity $params.Name ` -SearchService $serviceApp ` -ErrorAction SilentlyContinue } } } function Test-TargetResource { [CmdletBinding()] [OutputType([System.Boolean])] param ( [parameter(Mandatory = $true)] [System.String] $ServiceAppName, [parameter(Mandatory = $true)] [System.String] $Name, [System.UInt32] $RequestLimit = 0, [System.UInt32] $WaitTime = 0, [ValidateSet("Present","Absent")] [System.String] $Ensure = "Present", [System.Management.Automation.PSCredential] $InstallAccount ) Write-Verbose -Message "Testing Crawler Impact Rule Setting for '$Name'" if(($RequestLimit -gt 0) -and ($WaitTime -gt 0)) { throw "Only one Crawler Impact Rule HitRate argument (RequestLimit, WaitTime) can be specified" } $behavior = "" if($RequestLimit -ne 0) { $behavior = "RequestLimit" } else { $behavior = "WaitTime" } $CurrentValues = Get-TargetResource @PSBoundParameters if($Ensure -eq "Present") { return Test-SPDscParameterState -CurrentValues $CurrentValues ` -DesiredValues $PSBoundParameters ` -ValuesToCheck @("ServiceAppName", "Name", $behavior) } else { return Test-SPDscParameterState -CurrentValues $CurrentValues ` -DesiredValues $PSBoundParameters ` -ValuesToCheck @("ServiceAppName", "Name", "Ensure") } } Export-ModuleMember -Function *-TargetResource |