Update logic

Azure · Aug 14, 2024 · cdb8399 · cdb8399
1 parent 55bb394
commit cdb8399
Showing 1 changed file with 15 additions and 9 deletions.
diff --git a/FunctionApp/Modules/SessionHostReplacer/functions/Get-SHRHostPoolDecision.ps1 b/FunctionApp/Modules/SessionHostReplacer/functions/Get-SHRHostPoolDecision.ps1
@@ -70,20 +70,26 @@ function Get-SHRHostPoolDecision {
     Write-PSFMessage -Level Host -Message "We have a buffer of {0} session hosts more than the target." -StringValues $TargetSessionHostBuffer
 
     $weCanDeployUpTo = $TargetSessionHostCount + $TargetSessionHostBuffer - $SessionHosts.count - $RunningDeployments.SessionHostNames.Count
-    if ($weCanDeployUpTo -ge 0) { Write-PSFMessage -Level Host -Message "We can deploy up to {0} session hosts" -StringValues $weCanDeployUpTo }
-    else { Write-PSFMessage -Level Host -Message "Buffer is full. We can not deploy more session hosts" }
-
-    $weNeedToDeploy = $TargetSessionHostCount - $sessionHostsCurrentTotal.Count
-    if ($weNeedToDeploy -gt 0) {
-        Write-PSFMessage -Level Host -Message "We need to deploy {0} new session hosts" -StringValues $weNeedToDeploy
-        $weCanDeploy = if ($weNeedToDeploy -gt $weCanDeployUpTo) { $weCanDeployUpTo } else { $weNeedToDeploy } # If we need to deploy 10 machines, and we can deploy 5, we should only deploy 5.
-        Write-PSFMessage -Level Host -Message "Buffer allows deploying {0} session hosts" -StringValues $weCanDeploy
+    if ($weCanDeployUpTo -ge 0) {
+        Write-PSFMessage -Level Host -Message "We can deploy up to {0} session hosts" -StringValues $weCanDeployUpTo
+
+        $weNeedToDeploy = $TargetSessionHostCount - $sessionHostsCurrentTotal.Count
+        if ($weNeedToDeploy -gt 0) {
+            Write-PSFMessage -Level Host -Message "We need to deploy {0} new session hosts" -StringValues $weNeedToDeploy
+            $weCanDeploy = if ($weNeedToDeploy -gt $weCanDeployUpTo) { $weCanDeployUpTo } else { $weNeedToDeploy } # If we need to deploy 10 machines, and we can deploy 5, we should only deploy 5.
+            Write-PSFMessage -Level Host -Message "Buffer allows deploying {0} session hosts" -StringValues $weCanDeploy
+        }
+        else {
+            $weCanDeploy = 0
+            Write-PSFMessage -Level Host -Message "We have enough session hosts in good shape."
+        }
     }
     else {
+        Write-PSFMessage -Level Host -Message "Buffer is full. We can not deploy more session hosts"
         $weCanDeploy = 0
-        Write-PSFMessage -Level Host -Message "We have enough session hosts in good shape."
     }
 
+
     $weCanDelete = $SessionHosts.Count - $TargetSessionHostCount
     if ($weCanDelete -gt 0) {
         Write-PSFMessage -Level Host -Message "We need to delete {0} session hosts" -StringValues $weCanDelete