From c83e518775760717706eb03605be8cfa66082401 Mon Sep 17 00:00:00 2001
From: lub <git@lubiland.de>
Date: Fri, 2 Apr 2021 17:21:05 +0200
Subject: [PATCH] add count=0 nodes to the distribution

---
 failover.ps1 | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/failover.ps1 b/failover.ps1
index bd01752..638a929 100644
--- a/failover.ps1
+++ b/failover.ps1
@@ -4,22 +4,29 @@ $nodeList = docker node ls --format '{{json .}}'| ConvertFrom-Json | ForEach-Obj
     docker node inspect $_.ID | ConvertFrom-Json
 }
 $onlineNodeList = $nodeList | Where-Object {$_.Spec.Availability -eq 'active'}
-$offlineNodeList = $nodeList | Where-Object {$_.Spec.Availability -ne 'active'}
 
-if (!$offlineNodeList.Count) {
+if (!($nodeList | Where-Object {$_.Spec.Availability -ne 'active'})) {
     Write-Output 'all hosts online, nothing to failover'
     exit
 }
 
 $failoverAddressList = (Invoke-RestMethod -Uri 'https://robot-ws.your-server.de/failover' -Authentication Basic -Credential $hetznerCredentials).failover
-$offlineAddressList = $failoverAddressList | Where-Object active_server_ip -notin $onlineNodeList.Status.Addr
+$onlineNodeAddressList = $onlineNodeList.Description.Hostname | ForEach-Object {
+    dig +short $_
+}
+$offlineAddressList = $failoverAddressList | Where-Object active_server_ip -notin $onlineNodeAddressList
 
 $maximumAddressCount = [Math]::Ceiling($failoverAddressList.Count / $onlineNodeList.Count)
 
 $i = 0
-$addressGroupList = $failoverAddressList+($onlineNodeList.Status.Addr | ForEach-Object {@{active_server_ip=$_}}) `
-| Where-Object active_server_ip -in $onlineNodeList.Status.Addr `
-| Group-Object active_server_ip `
+$addressGroupList = $onlineNodeAddressList
+| ForEach-Object {
+    $nodeAddress = $_
+    @{
+        Count = [int]($addressGroupList | Where-Object {$_.Name -eq $nodeAddress}).Count
+        NodeAddress = $nodeAddress
+    }
+}
 | Sort-Object Count
 $distributionTarget = :distribution foreach ($addressGroup in $addressGroupList) {
     $currentAddressCount = $addressGroup.Count - 1
@@ -31,9 +38,9 @@ $distributionTarget = :distribution foreach ($addressGroup in $addressGroupList)
         }
 
         @{
-            active_server_ip = $addressGroup.Name
+            active_server_ip = $addressGroup.NodeAddress
             ip = $offlineAddressList[$i++].ip
-            node = $nodeList | Where-Object {$_.Status.Addr -eq $addressGroup.Name}
+            node = (dig +short -x $($addressGroup.NodeAddress)) -replace '\.$'
         }
         $currentAddressCount++
     }
@@ -49,7 +56,7 @@ $distributionTarget | ForEach-Object {
     $cleanupNodeList = docker node ls --filter $filter --format '{{json .}}' | ConvertFrom-Json
 
     $label = $labelName+'=true'
-    docker node update --label-add $label $($_.node.ID)
+    docker node update --label-add $label $($_.node)
 
     $cleanupNodeList | ForEach-Object {
         docker node update --label-rm $labelName $($_.ID)
@@ -57,4 +64,4 @@ $distributionTarget | ForEach-Object {
 }
 $distributionTarget | ForEach-Object -ThrottleLimit $failoverAddressList.Count -Parallel {
     Invoke-RestMethod -Method Post -Uri ('https://robot-ws.your-server.de/failover/{0}' -f $_.ip) -Authentication Basic -Credential $using:hetznerCredentials -Body @{active_server_ip=$_.active_server_ip}
-}
\ No newline at end of file
+}