mirror of
https://github.com/netbirdio/netbird.git
synced 2026-07-18 20:49:56 +00:00
4d76fd3c80
Move the debug state dumper out of the peer package into client/internal/peer/state_dump with an exported StateDump type and NewStateDump constructor. Update conn.go, wg_watcher.go and the watcher test to reference it through the package.
125 lines
2.2 KiB
Go
125 lines
2.2 KiB
Go
package state_dump
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/peer/status"
|
|
)
|
|
|
|
type StateDump struct {
|
|
log *log.Entry
|
|
status *status.Recorder
|
|
key string
|
|
|
|
sentOffer int
|
|
remoteOffer int
|
|
remoteAnswer int
|
|
remoteCandidate int
|
|
p2pConnected int
|
|
switchToRelay int
|
|
wgCheckSuccess int
|
|
relayConnected int
|
|
localProxies int
|
|
|
|
mu sync.Mutex
|
|
}
|
|
|
|
func NewStateDump(key string, log *log.Entry, statusRecorder *status.Recorder) *StateDump {
|
|
return &StateDump{
|
|
log: log,
|
|
status: statusRecorder,
|
|
key: key,
|
|
}
|
|
}
|
|
|
|
func (s *StateDump) Start(ctx context.Context) {
|
|
ticker := time.NewTicker(10 * time.Minute)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
s.dumpState()
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *StateDump) RemoteOffer() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
s.remoteOffer++
|
|
}
|
|
|
|
func (s *StateDump) RemoteCandidate() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
s.remoteCandidate++
|
|
}
|
|
|
|
func (s *StateDump) SendOffer() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
s.sentOffer++
|
|
}
|
|
|
|
func (s *StateDump) dumpState() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
status := "unknown"
|
|
state, e := s.status.GetPeer(s.key)
|
|
if e == nil {
|
|
status = state.ConnStatus.String()
|
|
}
|
|
|
|
s.log.Infof("Dump stat: Status: %s, SentOffer: %d, RemoteOffer: %d, RemoteAnswer: %d, RemoteCandidate: %d, P2PConnected: %d, SwitchToRelay: %d, WGCheckSuccess: %d, RelayConnected: %d, LocalProxies: %d",
|
|
status, s.sentOffer, s.remoteOffer, s.remoteAnswer, s.remoteCandidate, s.p2pConnected, s.switchToRelay, s.wgCheckSuccess, s.relayConnected, s.localProxies)
|
|
}
|
|
|
|
func (s *StateDump) RemoteAnswer() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
s.remoteAnswer++
|
|
}
|
|
|
|
func (s *StateDump) P2PConnected() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
s.p2pConnected++
|
|
}
|
|
|
|
func (s *StateDump) SwitchToRelay() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
s.switchToRelay++
|
|
}
|
|
|
|
func (s *StateDump) WGcheckSuccess() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
s.wgCheckSuccess++
|
|
}
|
|
|
|
func (s *StateDump) RelayConnected() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
s.relayConnected++
|
|
}
|
|
|
|
func (s *StateDump) NewLocalProxy() {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
s.localProxies++
|
|
}
|