Apply suggestions from code review

Co-authored-by: JmPotato <[email protected]> Signed-off-by: okJiang <[email protected]>
tikv · May 23, 2024 · 09bc239 · 09bc239
1 parent 27a74a5
commit 09bc239
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/client/http/interface.go b/client/http/interface.go
@@ -338,6 +338,7 @@ func (c *client) SetStoreLabels(ctx context.Context, storeID int64, storeLabels
 		WithBody(jsonInput))
 }
 
+// GetHealthStatus gets the health status of the cluster.
 func (c *client) GetHealthStatus(ctx context.Context) ([]Health, error) {
 	var healths []Health
 	err := c.request(ctx, newRequestInfo().

diff --git a/tools/pd-ctl/pdctl/command/health_command.go b/tools/pd-ctl/pdctl/command/health_command.go
@@ -32,7 +32,7 @@ func NewHealthCommand() *cobra.Command {
 }
 
 func showHealthCommandFunc(cmd *cobra.Command, _ []string) {
-	health, err := PDCli.GetHealthStatus(context.Background())
+	health, err := PDCli.GetHealthStatus(cmd.Context())
 	if err != nil {
 		cmd.Println(err)
 		return