fix: add context timeout to health check

Signed-off-by: Gaius <gaius.qi@gmail.com>
dragonflyoss · Dec 15, 2023 · f15da08 · f15da08
1 parent 1303d77
commit f15da08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/rpc/health/client/client.go b/pkg/rpc/health/client/client.go
@@ -77,7 +77,7 @@ func Check(ctx context.Context, target string, opts ...grpc.DialOption) error {
 	}
 	defer healthClient.Close()
 
-	if err := healthClient.Check(context.Background(), &healthpb.HealthCheckRequest{}); err != nil {
+	if err := healthClient.Check(ctx, &healthpb.HealthCheckRequest{}); err != nil {
 		return err
 	}