From c0aa87d6da9bb0a5b04e70e4d2bb3cd187c566b6 Mon Sep 17 00:00:00 2001
From: Aleksandr Nogikh <nogikh@google.com>
Date: Fri, 18 Oct 2024 14:10:33 +0200
Subject: pkg/manager: support multiple pools in Web UI

---
 pkg/rpcserver/rpcserver.go | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

(limited to 'pkg/rpcserver/rpcserver.go')

diff --git a/pkg/rpcserver/rpcserver.go b/pkg/rpcserver/rpcserver.go
index 76f1ddc68..10455cee7 100644
--- a/pkg/rpcserver/rpcserver.go
+++ b/pkg/rpcserver/rpcserver.go
@@ -8,6 +8,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"net/url"
 	"slices"
 	"sort"
 	"strings"
@@ -103,12 +104,27 @@ type Stats struct {
 }
 
 func NewStats() Stats {
+	return NewNamedStats("")
+}
+
+func NewNamedStats(name string) Stats {
+	suffix := name
+	if suffix != "" {
+		suffix = " [" + suffix + "]"
+	}
+	vmsLink := "/vms"
+	if name != "" {
+		vmsLink += "?pool=" + url.QueryEscape(name)
+	}
 	return Stats{
-		StatExecs: stat.New("exec total", "Total test program executions",
-			stat.Console, stat.Rate{}, stat.Prometheus("syz_exec_total")),
-		StatNumFuzzing: stat.New("fuzzing VMs",
-			"Number of VMs that are currently fuzzing", stat.Graph("fuzzing VMs")),
-		StatVMRestarts: stat.New("vm restarts", "Total number of VM starts",
+		StatExecs: stat.New("exec total"+suffix, "Total test program executions",
+			stat.Console, stat.Rate{}, stat.Prometheus("syz_exec_total"+name),
+		),
+		StatNumFuzzing: stat.New("fuzzing VMs"+suffix,
+			"Number of VMs that are currently fuzzing", stat.Graph("fuzzing VMs"),
+			stat.Link(vmsLink),
+		),
+		StatVMRestarts: stat.New("vm restarts"+suffix, "Total number of VM starts",
 			stat.Rate{}, stat.NoGraph),
 	}
 }
-- 
cgit mrf-deployment