Move context per cure's suggestion

This commit is contained in:
Kristoffer Dalby
2021-10-31 16:26:51 +00:00
parent 3f30bf1e33
commit 12f2a7cee0
2 changed files with 5 additions and 9 deletions

View File

@@ -316,11 +316,7 @@ func getHeadscaleApp() (*headscale.Headscale, error) {
return h, nil
}
func getHeadscaleGRPCClient() (apiV1.HeadscaleServiceClient, *grpc.ClientConn) {
// TODO(kradalby): Make configurable
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
func getHeadscaleGRPCClient(ctx context.Context) (apiV1.HeadscaleServiceClient, *grpc.ClientConn) {
grpcOptions := []grpc.DialOption{
grpc.WithBlock(),
}