wip

Signed-off-by: Michal Maslanka <michal@redpanda.com>

src/go/rpk/pkg/adminapi/api_broker.go

@@ -23,12 +23,12 @@ const (
 
 type MaintenanceStatus struct {
 	Draining     bool `json:"draining"`
-	Finished     bool `json:"finished"`
-	Errors       bool `json:"errors"`
-	Partitions   int  `json:"partitions"`
-	Eligible     int  `json:"eligible"`
-	Transferring int  `json:"transferring"`
-	Failed       int  `json:"failed"`
+	Finished     *bool `json:"finished"`
+	Errors       *bool `json:"errors"`
+	Partitions   *int  `json:"partitions"`
+	Eligible     *int  `json:"eligible"`
+	Transferring *int  `json:"transferring"`
+	Failed       *int  `json:"failed"`
 }
 
 // MembershipStatus enumerates possible membership states for brokers.

src/go/rpk/pkg/cli/cluster/maintenance/enable.go

@@ -112,7 +112,7 @@ node exists that is already in maintenance mode then an error will be returned.
 				}
 				addBrokerMaintenanceReport(table, b)
 				table.Flush()
-				if b.Maintenance.Draining && b.Maintenance.Finished {
+				if b.Maintenance.Draining && b.Maintenance.Finished != nil && *b.Maintenance.Finished {
 					fmt.Printf("\nAll partitions on node %d have drained.\n", nodeID)
 					return
 				}

src/go/rpk/pkg/cli/cluster/maintenance/status.go

@@ -10,6 +10,8 @@
 package maintenance
 
 import (
+	"fmt"
+
 	"github.com/redpanda-data/redpanda/src/go/rpk/pkg/adminapi"
 	"github.com/redpanda-data/redpanda/src/go/rpk/pkg/config"
 	"github.com/redpanda-data/redpanda/src/go/rpk/pkg/out"
@@ -25,16 +27,34 @@ func newMaintenanceReportTable() *out.TabWriter {
 	return out.NewTable(headers...)
 }
 
+func nullableBool2str(v *bool) string {
+
+	if v == nil {
+		return "-"
+	} else {
+		return fmt.Sprint(*v)
+	}
+}
+
+func nullableInt2str(v *int) string {
+
+	if v == nil {
+		return "-"
+	} else {
+		return fmt.Sprint(*v)
+	}
+}
 func addBrokerMaintenanceReport(table *out.TabWriter, b adminapi.Broker) {
+
 	table.Print(
 		b.NodeID,
 		b.Maintenance.Draining,
-		b.Maintenance.Finished,
-		b.Maintenance.Errors,
-		b.Maintenance.Partitions,
-		b.Maintenance.Eligible,
-		b.Maintenance.Transferring,
-		b.Maintenance.Failed)
+		nullableBool2str(b.Maintenance.Finished),
+		nullableBool2str(b.Maintenance.Errors),
+		nullableInt2str(b.Maintenance.Partitions),
+		nullableInt2str(b.Maintenance.Eligible),
+		nullableInt2str(b.Maintenance.Transferring),
+		nullableInt2str(b.Maintenance.Failed))
 }
 
 func newStatusCommand(fs afero.Fs, p *config.Params) *cobra.Command {

src/v/redpanda/admin_server.cc

@@ -842,13 +842,6 @@ fill_maintenance_status(const cluster::broker_state& b_state) {
 
     ret.draining = b_state.get_maintenance_state()
                    == model::maintenance_state::active;
-    // ensure that the output json has all fields
-    ret.finished = false;
-    ret.errors = false;
-    ret.partitions = 0;
-    ret.transferring = 0;
-    ret.eligible = 0;
-    ret.failed = 0;
 
     return ret;
 }

tests/rptest/clients/rpk.py

@@ -989,14 +989,20 @@ def parse(line):
             line = [x.strip() for x in line]
             if line[0] == "NODE-ID":
                 return None
+            def bool_or_none(value: str):
+                return None if value == "-" else  value == "true"
+            def int_or_none(value: str):
+                return None if value == "-" else  int(value)
+
+
             return RpkMaintenanceStatus(node_id=int(line[0]),
                                         enabled=line[1] == "true",
-                                        finished=line[2] == "true",
-                                        errors=line[3] == "true",
-                                        partitions=int(line[4]),
-                                        eligible=int(line[5]),
-                                        transferring=int(line[6]),
-                                        failed=int(line[7]))
+                                        finished=bool_or_none(line[2]),
+                                        errors=bool_or_none(line[3]),
+                                        partitions=int_or_none(line[4]),
+                                        eligible=int_or_none(line[5]),
+                                        transferring=int_or_none(line[6]),
+                                        failed=int_or_none(line[7]))
 
         cmd = [
             self._rpk_binary(),

tests/rptest/tests/maintenance_test.py

@@ -56,8 +56,11 @@ def _in_maintenance_mode(self, node):
 
     def _in_maintenance_mode_fully(self, node):
         status = self.admin.maintenance_status(node)
-        return status["finished"] and not status["errors"] and \
-                status["partitions"] > 0
+        if all([key in status for key in ['finished', 'errors', 'partitions']]):
+            return status["finished"] and not status["errors"] and \
+                    status["partitions"] > 0
+        else:
+            return False
 
     def _verify_broker_metadata(self, maintenance_enabled, node):
         """
@@ -80,17 +83,18 @@ def _verify_broker_metadata(self, maintenance_enabled, node):
             return False
         # check status wanted
         if maintenance_enabled:
-            return status['draining'] and status['finished']
+            return status['draining'] and status[
+                'finished'] if 'finished' in status else True
         else:
             return not status['draining']
 
-    def _verify_maintenance_status(self, node, draining):
+    def _verify_maintenance_status(self, node, enabled):
         """
         Check that cluster reports maintenance status as expected through
         both rpk status tooling as well as raw admin interface.
         """
         # get status for this node via rpk
-        node_id = self.redpanda.idx(node)
+        node_id = self.redpanda.node_id(node)
         statuses = self.rpk.cluster_maintenance_status()
         self.logger.debug(f"finding node_id {node_id} in rpk "
                           "maintenance status: {statuses}")
@@ -108,7 +112,7 @@ def _verify_maintenance_status(self, node, draining):
                           "{node.name}: {admin_status}")
 
         # ensure that both agree on expected outcome
-        return admin_status["draining"] == rpk_status.enabled == draining
+        return admin_status["draining"] == rpk_status.enabled == enabled
 
     def _enable_maintenance(self, node):
         """
@@ -325,6 +329,8 @@ def _check_maintenance_status_on_each_broker(status):
             True
         ), "All the nodes should keep reporting the state of node in maintenance mode"
 
+        # self._verify_maintenance_status(target, True)
+
         if self._use_rpk:
             self.rpk.cluster_maintenance_disable(target)
         else: