Framework/Core/SVT/Services/Search.ps1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#using namespace Microsoft.Azure.Commands.Search.Models
Set-StrictMode -Version Latest 
class Search: SVTBase
{       
    hidden [PSObject] $ResourceObject;

    Search([string] $subscriptionId, [string] $resourceGroupName, [string] $resourceName): 
        Base($subscriptionId, $resourceGroupName, $resourceName) 
    { 
        $this.GetResourceObject();
    }

    Search([string] $subscriptionId, [SVTResource] $svtResource): 
        Base($subscriptionId, $svtResource) 
    { 
        $this.GetResourceObject();
    }

    hidden [PSObject] GetResourceObject()
    {
        if (-not $this.ResourceObject) {
            $this.ResourceObject = Get-AzResource -Name $this.ResourceContext.ResourceName  `
                                        -ResourceType  "Microsoft.Search/searchServices" `
                                        -ResourceGroupName $this.ResourceContext.ResourceGroupName `
                                        -ErrorAction Stop

            if(-not $this.ResourceObject)
            {
                throw ([SuppressedException]::new(("Resource '{0}' not found under Resource Group '{1}'" -f ($this.ResourceContext.ResourceName), ($this.ResourceContext.ResourceGroupName)), [SuppressedExceptionType]::InvalidOperation))
            }
        }

        return $this.ResourceObject;
    }

    hidden [ControlResult] CheckSearchReplicaCount([ControlResult] $controlResult)
   {
        $replicaCount = $this.ResourceObject.Properties.replicaCount
        $isCompliant =  $replicaCount -ge 3
        if($isCompliant) 
        {
          $controlResult.AddMessage([VerificationResult]::Passed,
                                    [MessageData]::new("Replica count for resource " + $this.ResourceContext.ResourceName + " is " + $replicaCount)); 
        }
        else
        {
          $controlResult.AddMessage([VerificationResult]::Failed,
                                    [MessageData]::new("Replica count for resource " + $this.ResourceContext.ResourceName + " is " + $replicaCount));
        }
        
        return $controlResult;
    }
}