From 22f6a6d14f3e1cba3f8b88041e3b2fdcce5d9705 Mon Sep 17 00:00:00 2001 From: Radovan Zvoncek Date: Tue, 3 Oct 2023 14:26:36 +0300 Subject: [PATCH] Make MedusaRestoreJob properly match nodes when restoring in-place --- pkg/medusa/hostmap.go | 4 +++- pkg/medusa/hostmap_test.go | 34 +++++++++++++++++----------------- 2 files changed, 20 insertions(+), 18 deletions(-) diff --git a/pkg/medusa/hostmap.go b/pkg/medusa/hostmap.go index a8f378bb8..57278fa70 100644 --- a/pkg/medusa/hostmap.go +++ b/pkg/medusa/hostmap.go @@ -139,11 +139,13 @@ func sortLocations(m map[NodeLocation][]string) []rack { Rack: k, DC: DC, } + nodes := m[loc] + sort.Strings(nodes) out = append(out, rack{ Location: loc, Size: len(m[loc]), - Nodes: m[loc], + Nodes: nodes, }, ) } diff --git a/pkg/medusa/hostmap_test.go b/pkg/medusa/hostmap_test.go index 67cffe818..f27320ff3 100644 --- a/pkg/medusa/hostmap_test.go +++ b/pkg/medusa/hostmap_test.go @@ -18,34 +18,34 @@ func (client mockgRPCClient) GetBackups(ctx context.Context) ([]*BackupSummary, BackupName: "remote-backupname", Nodes: []*BackupNode{ { - Host: "192.168.1.2", + Host: "test-cluster-test-dc1-default-sts-0", Datacenter: "test-dc1", Rack: "default", }, { - Host: "192.168.1.3", + Host: "test-cluster-test-dc1-default-sts-2", Datacenter: "test-dc1", Rack: "default", }, { - Host: "192.168.1.4", + Host: "test-cluster-test-dc1-default-sts-1", Datacenter: "test-dc1", Rack: "default", }, { - Host: "192.168.1.5", + Host: "test-cluster-test-dc2-test-rack1-sts-0", Datacenter: "test-dc2", Rack: "test-rack1", }, { - Host: "192.168.1.6", + Host: "test-cluster-test-dc2-test-rack3-sts-0", Datacenter: "test-dc2", - Rack: "test-rack2", + Rack: "test-rack3", }, { - Host: "192.168.1.7", + Host: "test-cluster-test-dc2-test-rack2-sts-0", Datacenter: "test-dc2", - Rack: "test-rack3", + Rack: "test-rack2", }, }, }, @@ -79,9 +79,9 @@ func TestGetSourceRacksIPs(t *testing.T) { sourceRacks, err := getSourceRacksIPs(*medusaBackup, mockgRPCClient, ctx, "test-dc2") assert.NoError(t, err, err) expectedSourceRacks := map[NodeLocation][]string{ - {Rack: "test-rack1", DC: "test-dc2"}: {"192.168.1.5"}, - {Rack: "test-rack2", DC: "test-dc2"}: {"192.168.1.6"}, - {Rack: "test-rack3", DC: "test-dc2"}: {"192.168.1.7"}, + {Rack: "test-rack1", DC: "test-dc2"}: {"test-cluster-test-dc2-test-rack1-sts-0"}, + {Rack: "test-rack2", DC: "test-dc2"}: {"test-cluster-test-dc2-test-rack2-sts-0"}, + {Rack: "test-rack3", DC: "test-dc2"}: {"test-cluster-test-dc2-test-rack3-sts-0"}, } assert.Equal(t, expectedSourceRacks, sourceRacks) } @@ -176,15 +176,15 @@ func TestGetHostMap(t *testing.T) { assert.NoError(t, err, err) expected := HostMappingSlice{ { - Source: "192.168.1.2", + Source: "test-cluster-test-dc1-default-sts-0", Target: "test-cluster-test-dc1-default-sts-0", }, { - Source: "192.168.1.3", + Source: "test-cluster-test-dc1-default-sts-1", Target: "test-cluster-test-dc1-default-sts-1", }, { - Source: "192.168.1.4", + Source: "test-cluster-test-dc1-default-sts-2", Target: "test-cluster-test-dc1-default-sts-2", }, } @@ -201,15 +201,15 @@ func TestGetHostMap(t *testing.T) { cassDc.ObjectMeta.Name = "test-dc2" expected = HostMappingSlice{ { - Source: "192.168.1.5", + Source: "test-cluster-test-dc2-test-rack1-sts-0", Target: "test-cluster-test-dc2-rack1-sts-0", }, { - Source: "192.168.1.6", + Source: "test-cluster-test-dc2-test-rack2-sts-0", Target: "test-cluster-test-dc2-rack2-sts-0", }, { - Source: "192.168.1.7", + Source: "test-cluster-test-dc2-test-rack3-sts-0", Target: "test-cluster-test-dc2-rack3-sts-0", }, }