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
No known key found for this signature in database
GPG key ID: 09F62DC067465735
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(),
}