PRODENG-2826 MCR Uninstall now swarm drains and prunes volumes

Signed-off-by: Dimitar <[email protected]>

pkg/product/mke/phase/uninstall_mcr.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/Mirantis/mcc/pkg/phase"
 	"github.com/Mirantis/mcc/pkg/product/mke/api"
+	"github.com/Mirantis/mcc/pkg/swarm"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -30,6 +31,19 @@ func (p *UninstallMCR) Run() error {
 func (p *UninstallMCR) uninstallMCR(h *api.Host, c *api.ClusterConfig) error {
 	log.Infof("%s: uninstalling container runtime", h)
 
+	leader := c.Spec.SwarmLeader()
+
+	if err := p.drainNode(leader, h); err != nil {
+		return fmt.Errorf("%s: drain node: %w", h, err)
+	}
+
+	uVolumeCmd := h.Configurer.DockerCommandf("volume prune -f")
+	log.Infof("%s: unmounted dangling volumes", h)
+
+	if err := h.Exec(uVolumeCmd); err != nil {
+		return fmt.Errorf("%s: failed to unmount dangling volumes: %w", h, err)
+	}
+
 	if err := h.Configurer.UninstallMCR(h, h.Metadata.MCRInstallScript, c.Spec.MCR); err != nil {
 		return fmt.Errorf("%s: uninstall container runtime: %w", h, err)
 	}
@@ -38,3 +52,22 @@ func (p *UninstallMCR) uninstallMCR(h *api.Host, c *api.ClusterConfig) error {
 
 	return nil
 }
+
+// DrainNode drains a node from the workload via docker drain command.
+func (p *UninstallMCR) drainNode(lead *api.Host, h *api.Host) error {
+	nodeID, err := swarm.NodeID(h)
+	if err != nil {
+		return fmt.Errorf("failed to get node ID for %s: %w", h, err)
+	}
+
+	drainCmd := lead.Configurer.DockerCommandf("node update --availability drain %s", nodeID)
+	if err := lead.Exec(drainCmd); err != nil {
+		return fmt.Errorf("%s: failed to run MKE uninstaller: %w", lead, err)
+	}
+	if err := lead.Exec(drainCmd); err != nil {
+		return fmt.Errorf("failed to drain node %s: %w", nodeID, err)
+	}
+
+	log.Infof("%s: node %s drained", lead, nodeID)
+	return nil
+}