feat: Reduce UI Support (#500)

Signed-off-by: veds-g <[email protected]>
numaproj · Feb 14, 2023 · 2844cfb · 2844cfb
1 parent 57f43e5
commit 2844cfb
Show file tree

Hide file tree

Showing 15 changed files with 327 additions and 205 deletions.
diff --git a/pkg/apis/proto/daemon/daemon.pb.go b/pkg/apis/proto/daemon/daemon.pb.go
diff --git a/pkg/apis/proto/daemon/daemon.proto b/pkg/apis/proto/daemon/daemon.proto
@@ -67,11 +67,10 @@ message GetBufferResponse {
 message GetVertexMetricsRequest {
   required string pipeline = 2;
   required string vertex = 3;
-
 }
 
 message GetVertexMetricsResponse {
-  required VertexMetrics vertex = 1;
+  repeated VertexMetrics vertexMetrics = 1;
 }
 
 /* Watermark */

diff --git a/pkg/daemon/client/daemon_client.go b/pkg/daemon/client/daemon_client.go
@@ -87,14 +87,14 @@ func (dc *DaemonClient) GetPipelineBuffer(ctx context.Context, pipeline, buffer
 	}
 }
 
-func (dc *DaemonClient) GetVertexMetrics(ctx context.Context, pipeline, vertex string) (*daemon.VertexMetrics, error) {
+func (dc *DaemonClient) GetVertexMetrics(ctx context.Context, pipeline, vertex string) ([]*daemon.VertexMetrics, error) {
 	if rspn, err := dc.client.GetVertexMetrics(ctx, &daemon.GetVertexMetricsRequest{
 		Pipeline: &pipeline,
 		Vertex:   &vertex,
 	}); err != nil {
 		return nil, err
 	} else {
-		return rspn.Vertex, nil
+		return rspn.VertexMetrics, nil
 	}
 }