Add role states in walle_status_description in cms

765fc5287a82f0295eff3f39559f13f0b187f4e4

internal/cms/helpers.go

@@ -2,9 +2,11 @@ package cms
 
 import (
 	"context"
+	"strings"
 
 	"go.ytsaurus.tech/library/go/ptr"
 	"go.ytsaurus.tech/yt/admin/cms/internal/models"
+	"go.ytsaurus.tech/yt/go/ypath"
 	"go.ytsaurus.tech/yt/go/yt"
 	"go.ytsaurus.tech/yt/go/ytsys"
 )
@@ -137,3 +139,11 @@ func markNodeDecommissioned(
 	}
 	return dc.MarkNodeDecommissioned(ctx, node.Addr, decommisionMsg)
 }
+
+func getPodIDFromYPath(path ypath.Path) string {
+	pathSplit := strings.Split(path.String(), "/")
+	addr := pathSplit[len(pathSplit)-1]
+	shortAddr := strings.Split(addr, ".")[0]
+
+	return shortAddr
+}

internal/cms/task_processor.go

@@ -2,6 +2,7 @@ package cms
 
 import (
 	"context"
+	"fmt"
 	"sort"
 	"strings"
 	"sync"
@@ -1084,6 +1085,11 @@ func (p *TaskProcessor) processTask(ctx context.Context, t *models.Task) {
 		p.processHost(taskCtx, t.HostStates[h])
 	}
 
+	if desc := p.generateWalleDescription(t); t.WalleStatusDescription != desc {
+		t.WalleStatusDescription = desc
+		p.tryUpdateTaskInStorage(ctx, t)
+	}
+
 	// Update task state in storage.
 	if t.AllHostsFinished() {
 		t.SetFinished()
@@ -1352,3 +1358,15 @@ func (p *TaskProcessor) hasTaskUpgrade(t *models.Task) bool {
 func IsNOCTask(t *models.Task) bool {
 	return t.IsGroupTask && t.Action == walle.ActionTemporaryUnreachable
 }
+
+func (p *TaskProcessor) generateWalleDescription(t *models.Task) string {
+	var desc []string
+
+	for _, h := range t.HostStates {
+		for path, r := range h.Roles {
+			desc = append(desc, fmt.Sprintf("%s: %s", getPodIDFromYPath(path), r.GetState()))
+		}
+	}
+
+	return strings.Join(desc, " | ")
+}