Create admin API to update namespace

temporalio · Nov 16, 2021 · 7ddd561 · 7ddd561
1 parent b9b0615
commit 7ddd561
Show file tree

Hide file tree

Showing 11 changed files with 1,391 additions and 445 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
@@ -287,6 +287,20 @@ func (c *clientImpl) RegisterNamespace(
 	return client.RegisterNamespace(ctx, request, opts...)
 }
 
+func (c *clientImpl) UpdateNamespace(
+	ctx context.Context,
+	request *adminservice.UpdateNamespaceRequest,
+	opts ...grpc.CallOption,
+) (*adminservice.UpdateNamespaceResponse, error) {
+	client, err := c.getRandomClient()
+	if err != nil {
+		return nil, err
+	}
+	ctx, cancel := c.createContext(ctx)
+	defer cancel()
+	return client.UpdateNamespace(ctx, request, opts...)
+}
+
 func (c *clientImpl) GetNamespaceReplicationMessages(
 	ctx context.Context,
 	request *adminservice.GetNamespaceReplicationMessagesRequest,

diff --git a/client/admin/metricClient.go b/client/admin/metricClient.go
@@ -335,6 +335,24 @@ func (c *metricClient) RegisterNamespace(
 	return resp, err
 }
 
+func (c *metricClient) UpdateNamespace(
+	ctx context.Context,
+	request *adminservice.UpdateNamespaceRequest,
+	opts ...grpc.CallOption,
+) (*adminservice.UpdateNamespaceResponse, error) {
+
+	c.metricsClient.IncCounter(metrics.AdminClientUpdateNamespaceScope, metrics.ClientRequests)
+
+	sw := c.metricsClient.StartTimer(metrics.AdminClientUpdateNamespaceScope, metrics.ClientLatency)
+	resp, err := c.client.UpdateNamespace(ctx, request, opts...)
+	sw.Stop()
+
+	if err != nil {
+		c.metricsClient.IncCounter(metrics.AdminClientUpdateNamespaceScope, metrics.ClientFailures)
+	}
+	return resp, err
+}
+
 func (c *metricClient) GetNamespaceReplicationMessages(
 	ctx context.Context,
 	request *adminservice.GetNamespaceReplicationMessagesRequest,

diff --git a/client/admin/retryableClient.go b/client/admin/retryableClient.go
@@ -303,6 +303,22 @@ func (c *retryableClient) RegisterNamespace(
 	return resp, err
 }
 
+func (c *retryableClient) UpdateNamespace(
+	ctx context.Context,
+	request *adminservice.UpdateNamespaceRequest,
+	opts ...grpc.CallOption,
+) (*adminservice.UpdateNamespaceResponse, error) {
+
+	var resp *adminservice.UpdateNamespaceResponse
+	op := func() error {
+		var err error
+		resp, err = c.client.UpdateNamespace(ctx, request, opts...)
+		return err
+	}
+	err := backoff.Retry(op, c.policy, c.isRetryable)
+	return resp, err
+}
+
 func (c *retryableClient) GetNamespaceReplicationMessages(
 	ctx context.Context,
 	request *adminservice.GetNamespaceReplicationMessagesRequest,