mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 18:00:49 +01:00
0c039274a4
This update adds new relay integration for NetBird clients. The new relay is based on web sockets and listens on a single port. - Adds new relay implementation with websocket with single port relaying mechanism - refactor peer connection logic, allowing upgrade and downgrade from/to P2P connection - peer connections are faster since it connects first to relay and then upgrades to P2P - maintains compatibility with old clients by not using the new relay - updates infrastructure scripts with new relay service
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
reconnectingTimeout = 5 * time.Second
|
|
)
|
|
|
|
// Guard manage the reconnection tries to the Relay server in case of disconnection event.
|
|
type Guard struct {
|
|
ctx context.Context
|
|
relayClient *Client
|
|
}
|
|
|
|
// NewGuard creates a new guard for the relay client.
|
|
func NewGuard(context context.Context, relayClient *Client) *Guard {
|
|
g := &Guard{
|
|
ctx: context,
|
|
relayClient: relayClient,
|
|
}
|
|
return g
|
|
}
|
|
|
|
// OnDisconnected is called when the relay client is disconnected from the relay server. It will trigger the reconnection
|
|
// todo prevent multiple reconnection instances. In the current usage it should not happen, but it is better to prevent
|
|
func (g *Guard) OnDisconnected() {
|
|
ticker := time.NewTicker(reconnectingTimeout)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
err := g.relayClient.Connect()
|
|
if err != nil {
|
|
log.Errorf("failed to reconnect to relay server: %s", err)
|
|
continue
|
|
}
|
|
return
|
|
case <-g.ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|