sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-04-28 21:26:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use conn id instead of peer id

Browse Source
This commit is contained in:
Zoltán Papp
2025-03-13 16:16:53 +01:00
parent 231a481c0f
commit 6107b394a0
2 changed files with 43 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
client/internal/lazyconn/inactivity/inactivity.go
View File

@@ -4,7 +4,7 @@ import (
"context"
"time"
log "github.com/sirupsen/logrus"
"github.com/netbirdio/netbird/client/internal/peer"
)
const (
@@ -12,21 +12,21 @@ const (
)
type InactivityMonitor struct {
peerID string
id peer.ConnID
timer *time.Timer
cancel context.CancelFunc
}
func NewInactivityMonitor(peerID string) *InactivityMonitor {
func NewInactivityMonitor(peerID peer.ConnID) *InactivityMonitor {
i := &InactivityMonitor{
peerID: peerID,
timer: time.NewTimer(0),
id: peerID,
timer: time.NewTimer(0),
}
i.timer.Stop()
return i
}
func (i *InactivityMonitor) Start(ctx context.Context, timeoutChan chan string) {
func (i *InactivityMonitor) Start(ctx context.Context, timeoutChan chan peer.ConnID) {
i.timer.Reset(inactivityThreshold)
defer i.timer.Stop()
@@ -36,7 +36,7 @@ func (i *InactivityMonitor) Start(ctx context.Context, timeoutChan chan string)
select {
case <-i.timer.C:
select {
case timeoutChan <- i.peerID:
case timeoutChan <- i.id:
case <-ctx.Done():
return
}
@@ -46,7 +46,6 @@ func (i *InactivityMonitor) Start(ctx context.Context, timeoutChan chan string)
}
func (i *InactivityMonitor) Stop() {
log.Info("--- cancel idle timer")
i.cancel()
}