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

Add a sleep to reduce the impact of #727

Browse source
This commit is contained in:
Juan Font Alonso 2022-08-11 13:37:25 +02:00
parent 8e56d8b425
commit e1e3feb6a8
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
api.go
View file

@ -21,6 +21,7 @@ import (
) )
const ( const (
registrationHoldoff = time.Second * 5 // TODO(juan): remove this once https://github.com/juanfont/headscale/issues/727 is fixed
reservedResponseHeaderSize = 4 reservedResponseHeaderSize = 4
RegisterMethodAuthKey = "authkey" RegisterMethodAuthKey = "authkey"
RegisterMethodOIDC = "oidc" RegisterMethodOIDC = "oidc"
@ -237,11 +238,17 @@ func (h *Headscale) RegistrationHandler(
Str("Followup", registerRequest.Followup). Str("Followup", registerRequest.Followup).
Msg("Machine is waiting for interactive login") Msg("Machine is waiting for interactive login")
ticker := time.NewTicker(registrationHoldoff)
select {
case <-req.Context().Done():
return
case <-ticker.C:
h.handleMachineRegistrationNew(writer, req, machineKey, registerRequest) h.handleMachineRegistrationNew(writer, req, machineKey, registerRequest)
return return
} }
} }
}
log.Info(). log.Info().
Caller(). Caller().