Backport of Removed nodename and status from consul snapshot save -ap…

…pend-filename command and using leader version in version into release/1.15.x (#18685) * backport of commit e8f4bfe * backport of commit 5072958 * Removed nodename and status from consul snapshot save -append-filename command and using leader version in version (#18680) * init * fix tests * fix tests lint * fix api call inside dc * updated doc * address comments --------- Co-authored-by: absolutelightning <[email protected]> Co-authored-by: Ashesh Vidyut <[email protected]>
hashicorp · Sep 6, 2023 · fa9e837 · fa9e837
1 parent 6bdd979
commit fa9e837
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 55 deletions.
diff --git a/command/snapshot/save/snapshot_save.go b/command/snapshot/save/snapshot_save.go
@@ -33,7 +33,7 @@ type cmd struct {
 func (c *cmd) getAppendFileNameFlag() *flag.FlagSet {
 	fs := flag.NewFlagSet("", flag.ContinueOnError)
 	fs.Var(&c.appendFileNameFlag, "append-filename", "Append filename flag supports the following "+
-		"comma-separated arguments. 1. version, 2. dc. 3. node 4. status. It appends these values to the filename provided in the command")
+		"comma-separated arguments. 1. version, 2. dc. It appends these values to the filename provided in the command")
 	return fs
 }
 
@@ -71,50 +71,35 @@ func (c *cmd) Run(args []string) int {
 	appendFileNameFlags := strings.Split(c.appendFileNameFlag.String(), ",")
 
 	if len(appendFileNameFlags) != 0 && len(c.appendFileNameFlag.String()) > 0 {
-		agentSelfResponse, err := client.Agent().Self()
-		if err != nil {
-			c.UI.Error(fmt.Sprintf("Error connecting to Consul agent and fetching datacenter/version: %s", err))
-			return 1
-		}
-
 		fileExt := filepath.Ext(file)
 		fileNameWithoutExt := strings.TrimSuffix(file, fileExt)
 
 		if slices.Contains(appendFileNameFlags, "version") {
-			if config, ok := agentSelfResponse["Config"]; ok {
-				if version, ok := config["Version"]; ok {
-					fileNameWithoutExt = fileNameWithoutExt + "-" + version.(string)
+			operatorHealthResponse, err := client.Operator().AutopilotServerHealth(nil)
+			if err != nil {
+				c.UI.Error(fmt.Sprintf("Error fetching version of Consul agent Leader: %s", err))
+				return 1
+			}
+			var version string
+			for _, server := range operatorHealthResponse.Servers {
+				if server.Leader {
+					version = server.Version
+					break
 				}
 			}
+			fileNameWithoutExt = fileNameWithoutExt + "-" + version
 		}
 
 		if slices.Contains(appendFileNameFlags, "dc") {
-			if config, ok := agentSelfResponse["Config"]; ok {
-				if datacenter, ok := config["Datacenter"]; ok {
-					fileNameWithoutExt = fileNameWithoutExt + "-" + datacenter.(string)
-				}
+			agentSelfResponse, err := client.Agent().Self()
+			if err != nil {
+				c.UI.Error(fmt.Sprintf("Error connecting to Consul agent and fetching datacenter/version: %s", err))
+				return 1
 			}
-		}
 
-		if slices.Contains(appendFileNameFlags, "node") {
 			if config, ok := agentSelfResponse["Config"]; ok {
-				if nodeName, ok := config["NodeName"]; ok {
-					fileNameWithoutExt = fileNameWithoutExt + "-" + nodeName.(string)
-				}
-			}
-		}
-
-		if slices.Contains(appendFileNameFlags, "status") {
-			if status, ok := agentSelfResponse["Stats"]; ok {
-				if config, ok := status["consul"]; ok {
-					configMap := config.(map[string]interface{})
-					if leader, ok := configMap["leader"]; ok {
-						if leader == "true" {
-							fileNameWithoutExt = fileNameWithoutExt + "-" + "leader"
-						} else {
-							fileNameWithoutExt = fileNameWithoutExt + "-" + "follower"
-						}
-					}
+				if datacenter, ok := config["Datacenter"]; ok {
+					fileNameWithoutExt = fileNameWithoutExt + "-" + datacenter.(string)
 				}
 			}
 		}

diff --git a/command/snapshot/save/snapshot_save_test.go b/command/snapshot/save/snapshot_save_test.go
@@ -85,34 +85,34 @@ func TestSnapshotSaveCommandWithAppendFileNameFlag(t *testing.T) {
 	dir := testutil.TempDir(t, "snapshot")
 	file := filepath.Join(dir, "backup.tgz")
 	args := []string{
-		"-append-filename=version,dc,node,status",
+		"-append-filename=version,dc",
 		"-http-addr=" + a.HTTPAddr(),
 		file,
 	}
 
-	stats := a.Stats()
-
-	status := "follower"
-
-	if stats["consul"]["leader"] == "true" {
-		status = "leader"
-	}
-
 	// We need to use the self endpoint here for ENT, which returns the product suffix (+ent)
 	self, err := client.Agent().Self()
 	require.NoError(t, err)
 
 	cfg, ok := self["Config"]
 	require.True(t, ok)
 
-	versionAny, ok := cfg["Version"]
+	dc, ok := cfg["Datacenter"]
 	require.True(t, ok)
 
-	version, ok := versionAny.(string)
-	require.True(t, ok)
+	datacenter := dc.(string)
+
+	operatorHealth, error := client.Operator().AutopilotServerHealth(nil)
+	require.NoError(t, error)
+
+	var version string
+	for _, server := range operatorHealth.Servers {
+		if server.Leader {
+			version = server.Version
+		}
+	}
 
-	newFilePath := filepath.Join(dir, "backup"+"-"+version+"-"+a.Config.Datacenter+
-		"-"+a.Config.NodeName+"-"+status+".tgz")
+	newFilePath := filepath.Join(dir, "backup"+"-"+version+"-"+datacenter+".tgz")
 
 	code := c.Run(args)
 	if code != 0 {

diff --git a/website/content/commands/snapshot/save.mdx b/website/content/commands/snapshot/save.mdx
@@ -77,20 +77,20 @@ $ consul snapshot save -stale backup.snap
 # ...
 ```
 
-To create snapshot file with consul version, datacenter, node name and leader/follower info,
-run
+This is useful for situations where a cluster is in a degraded state and no
+leader is available. To target a specific server for a snapshot, you can run
+the `consul snapshot save` command on that specific server.
+
+To create snapshot file with consul version and datacenter run
 
 ```shell-session
-$ consul snapshot save -append-filename node,status,version,dc backup.snap
+$ consul snapshot save -append-filename version,dc backup.snap
 #...
 ```
 
-File name created will be like backup-%CONSUL_VERSION%-%DC_NAME%-%NODE_NAME%-%STATUS.snap
+File name created will be like backup-%CONSUL_VERSION%-%DC_NAME%.snap
 example - backup-1.17.0-dc1-local-machine-leader.tgz
-
-This is useful for situations where a cluster is in a degraded state and no
-leader is available. To target a specific server for a snapshot, you can run
-the `consul snapshot save` command on that specific server.
+Note Version is always the leader's consul version
 
 Please see the [HTTP API](/consul/api-docs/snapshot) documentation for
 more details about snapshot internals.