Add admin api to list all cluster information (#2377)

* Add admin api to list all cluster information
temporalio · Jan 13, 2022 · eb95443 · eb95443
1 parent 72e5b67
commit eb95443
Show file tree

Hide file tree

Showing 13 changed files with 1,069 additions and 263 deletions.
diff --git a/api/adminservice/v1/request_response.pb.go b/api/adminservice/v1/request_response.pb.go
diff --git a/api/adminservice/v1/service.pb.go b/api/adminservice/v1/service.pb.go
diff --git a/api/adminservicemock/v1/service.pb.mock.go b/api/adminservicemock/v1/service.pb.mock.go
diff --git a/client/admin/client.go b/client/admin/client.go
@@ -259,6 +259,20 @@ func (c *clientImpl) DescribeCluster(
 	return client.DescribeCluster(ctx, request, opts...)
 }
 
+func (c *clientImpl) ListClusters(
+	ctx context.Context,
+	request *adminservice.ListClustersRequest,
+	opts ...grpc.CallOption,
+) (*adminservice.ListClustersResponse, error) {
+	client, err := c.getRandomClient()
+	if err != nil {
+		return nil, err
+	}
+	ctx, cancel := c.createContext(ctx)
+	defer cancel()
+	return client.ListClusters(ctx, request, opts...)
+}
+
 func (c *clientImpl) ListClusterMembers(
 	ctx context.Context,
 	request *adminservice.ListClusterMembersRequest,

diff --git a/client/admin/metricClient.go b/client/admin/metricClient.go
@@ -300,6 +300,24 @@ func (c *metricClient) DescribeCluster(
 	return resp, err
 }
 
+func (c *metricClient) ListClusters(
+	ctx context.Context,
+	request *adminservice.ListClustersRequest,
+	opts ...grpc.CallOption,
+) (*adminservice.ListClustersResponse, error) {
+
+	c.metricsClient.IncCounter(metrics.AdminClientListClustersScope, metrics.ClientRequests)
+
+	sw := c.metricsClient.StartTimer(metrics.AdminClientListClustersScope, metrics.ClientLatency)
+	resp, err := c.client.ListClusters(ctx, request, opts...)
+	sw.Stop()
+
+	if err != nil {
+		c.metricsClient.IncCounter(metrics.AdminClientListClustersScope, metrics.ClientFailures)
+	}
+	return resp, err
+}
+
 func (c *metricClient) ListClusterMembers(
 	ctx context.Context,
 	request *adminservice.ListClusterMembersRequest,

diff --git a/client/admin/retryableClient.go b/client/admin/retryableClient.go
@@ -273,6 +273,23 @@ func (c *retryableClient) DescribeCluster(
 	err := backoff.Retry(op, c.policy, c.isRetryable)
 	return resp, err
 }
+
+func (c *retryableClient) ListClusters(
+	ctx context.Context,
+	request *adminservice.ListClustersRequest,
+	opts ...grpc.CallOption,
+) (*adminservice.ListClustersResponse, error) {
+
+	var resp *adminservice.ListClustersResponse
+	op := func() error {
+		var err error
+		resp, err = c.client.ListClusters(ctx, request, opts...)
+		return err
+	}
+	err := backoff.Retry(op, c.policy, c.isRetryable)
+	return resp, err
+}
+
 func (c *retryableClient) ListClusterMembers(
 	ctx context.Context,
 	request *adminservice.ListClusterMembersRequest,