myriation/headscale
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Use request context to close when client disconnects

Browse source
This commit is contained in:
Juan Font Alonso 2022-06-20 21:47:02 +02:00
parent 5e9004c407
commit d404ba102d
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
poll.go
View file

@ -279,7 +279,7 @@ func (h *Headscale) PollNetMapStream(
keepAliveChan chan []byte, keepAliveChan chan []byte,
updateChan chan struct{}, updateChan chan struct{},
) { ) {
ctx := context.WithValue(context.Background(), machineNameContextKey, machine.Hostname) ctx := context.WithValue(r.Context(), machineNameContextKey, machine.Hostname)
ctx, cancel := context.WithCancel(ctx) ctx, cancel := context.WithCancel(ctx)
defer cancel() defer cancel()