mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-15 11:21:04 +01:00
37 lines
520 B
Go
37 lines
520 B
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
reconnectingTimeout = 5 * time.Second
|
|
)
|
|
|
|
type Guard struct {
|
|
ctx context.Context
|
|
relayClient *Client
|
|
}
|
|
|
|
func NewGuard(context context.Context, relayClient *Client) *Guard {
|
|
g := &Guard{
|
|
ctx: context,
|
|
relayClient: relayClient,
|
|
}
|
|
|
|
return g
|
|
}
|
|
|
|
func (g *Guard) OnDisconnected() {
|
|
timeout := time.NewTimer(reconnectingTimeout)
|
|
defer timeout.Stop()
|
|
|
|
select {
|
|
case <-timeout.C:
|
|
_ = g.relayClient.Connect()
|
|
case <-g.ctx.Done():
|
|
return
|
|
}
|
|
}
|