From 4978787d49490275249b66f1fa506dea2589dbd6 Mon Sep 17 00:00:00 2001
From: releng <vertica-releng@opentext>
Date: Tue, 12 Nov 2024 20:04:54 -0500
Subject: [PATCH] Sync from server repo (5d462da8224)

---
 vclusterops/https_update_node_state_op.go |  7 +++-
 vclusterops/stop_node.go                  | 40 ++++++++++++++++++++++-
 2 files changed, 45 insertions(+), 2 deletions(-)

diff --git a/vclusterops/https_update_node_state_op.go b/vclusterops/https_update_node_state_op.go
index a8a8703..e11dd20 100644
--- a/vclusterops/https_update_node_state_op.go
+++ b/vclusterops/https_update_node_state_op.go
@@ -95,7 +95,12 @@ func (op *httpsUpdateNodeStateOp) processResult(execContext *opEngineExecContext
 			if !ok {
 				return fmt.Errorf("cannot find host %s in vdb", host)
 			}
-			vnode.State = util.NodeUnknownState
+			// Compute nodes will persistently fail the precondition, and shouldn't have status overwritten.
+			// Note that if the vdb was constructed by querying node(s) from a different sandbox than the
+			// compute node, it will already have UNKNOWN state in the vdb and that will not change here.
+			if vnode.State != util.NodeComputeState {
+				vnode.State = util.NodeUnknownState
+			}
 
 			continue
 		}
diff --git a/vclusterops/stop_node.go b/vclusterops/stop_node.go
index 61a4f60..5d70932 100644
--- a/vclusterops/stop_node.go
+++ b/vclusterops/stop_node.go
@@ -107,6 +107,12 @@ func (vcc VClusterCommands) VStopNode(options *VStopNodeOptions) error {
 		return err
 	}
 
+	// stop_node should fail if the nodes to stop belong to different sandboxes
+	targetSandbox, err := checkTargetNodesSandboxes(&vdb, options)
+	if err != nil {
+		return err
+	}
+
 	options.completeVDBSetting(&vdb)
 
 	// stop_node is aborted if requirements are not met.
@@ -122,12 +128,44 @@ func (vcc VClusterCommands) VStopNode(options *VStopNodeOptions) error {
 	}
 
 	clusterOpEngine := makeClusterOpEngine(instructions, options)
-	if runError := clusterOpEngine.run(vcc.Log); runError != nil {
+	if runError := clusterOpEngine.runInSandbox(vcc.Log, &vdb, targetSandbox); runError != nil {
 		return fmt.Errorf("fail to complete stop node operation, %w", runError)
 	}
 	return nil
 }
 
+// checkStopNodeRequirements checks whether the nodes to stop belong to the same sandbox (or main cluster)
+func checkTargetNodesSandboxes(vdb *VCoordinationDatabase,
+	options *VStopNodeOptions) (targetSandbox string, err error) {
+	sandboxHostMap := make(map[string][]string)
+	for _, host := range options.StopHosts {
+		vnode, exists := vdb.HostNodeMap[host]
+		if !exists {
+			return "", fmt.Errorf("cannot find host %s in vdb", host)
+		}
+
+		var sandboxDisplayName string
+		if vnode.Sandbox == util.MainClusterSandbox {
+			sandboxDisplayName = "main-cluster"
+		} else {
+			sandboxDisplayName = vnode.Sandbox
+		}
+
+		if _, exists := sandboxHostMap[sandboxDisplayName]; !exists {
+			sandboxHostMap[sandboxDisplayName] = []string{host}
+		} else {
+			sandboxHostMap[sandboxDisplayName] = append(sandboxHostMap[sandboxDisplayName], host)
+		}
+
+		targetSandbox = vnode.Sandbox
+	}
+	if len(sandboxHostMap) > 1 {
+		return "", fmt.Errorf("the hosts to stop cannot belong to different sandboxes %+v", sandboxHostMap)
+	}
+
+	return targetSandbox, nil
+}
+
 // checkStopNodeRequirements returns an error if at least one of the nodes
 // to stop does not exist in db.
 func checkStopNodeRequirements(vdb *VCoordinationDatabase, hostsToStop []string) error {