Change zbus client getter to be public

threefoldtech · Oct 16, 2023 · 687d2c8 · 687d2c8
1 parent f70015a
commit 687d2c8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pkg/perf/cpubench_task.go b/pkg/perf/cpubench_task.go
@@ -59,7 +59,7 @@ func (c *CPUBenchmarkTask) Run(ctx context.Context) (interface{}, error) {
 	if err != nil {
 		return nil, fmt.Errorf("failed to parse cpubench output: %w", err)
 	}
-	client := getZbusClient(ctx)
+	client := GetZbusClient(ctx)
 	statistics := stubs.NewStatisticsStub(client)
 
 	workloads, err := statistics.Workloads(ctx)

diff --git a/pkg/perf/monitor.go b/pkg/perf/monitor.go
@@ -101,6 +101,7 @@ func withZbusClient(ctx context.Context, client zbus.Client) context.Context {
 	return context.WithValue(ctx, zbusClient{}, client)
 }
 
-func getZbusClient(ctx context.Context) zbus.Client {
+// GetZbusClient gets zbus client from the given context
+func GetZbusClient(ctx context.Context) zbus.Client {
 	return ctx.Value(zbusClient{}).(zbus.Client)
 }