From b30ae8befedb6dfb57182c59ed7cefe17f8bcfad Mon Sep 17 00:00:00 2001 From: andreacappelletti97 Date: Tue, 4 Apr 2023 14:22:55 -0500 Subject: [PATCH] Client Get call to /database-number-of-workers --- dax/controller/client/client.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/dax/controller/client/client.go b/dax/controller/client/client.go index fc40acc73..f3506d486 100644 --- a/dax/controller/client/client.go +++ b/dax/controller/client/client.go @@ -701,3 +701,22 @@ func (c *Client) SnapshotTable(ctx context.Context, qtid dax.QualifiedTableID) e return nil } + +func (c *Client) GetDatabaseNumberOfWorkers(ctx context.Context, qdbid dax.QualifiedDatabaseID) error { + url := fmt.Sprintf("%s/database-number-of-workers", c.address.WithScheme(defaultScheme)) + c.logger.Debugf("Database number of workers: %s", url) + + // Get the request. + resp, err := c.httpClient.Get(url) + if err != nil { + return errors.Wrap(err, "getting number of workers") + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + b, _ := io.ReadAll(resp.Body) + return errors.Errorf("status code: %d: %s", resp.StatusCode, b) + } + + return nil +}