mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-16 07:16:38 +00:00
feature: add stop handling for engine
This commit is contained in:
@@ -53,7 +53,7 @@ func SetupCloseHandler() {
|
||||
go func() {
|
||||
for range c {
|
||||
fmt.Println("\r- Ctrl+C pressed in Terminal")
|
||||
os.Exit(0)
|
||||
stopUP <- 0
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
@@ -39,6 +39,7 @@ var (
|
||||
iFaceBlackList[config.IFaceBlackList[i]] = struct{}{}
|
||||
}
|
||||
engine := connection.NewEngine(signalClient, config.StunTurnURLs, config.WgIface, config.WgAddr, iFaceBlackList)
|
||||
defer engine.Stop()
|
||||
|
||||
err = engine.Start(myKey, config.Peers)
|
||||
if err != nil {
|
||||
@@ -48,8 +49,9 @@ var (
|
||||
//signalClient.WaitConnected()
|
||||
|
||||
SetupCloseHandler()
|
||||
<-stopUP
|
||||
code := <-stopUP
|
||||
log.Println("Receive signal to stop running")
|
||||
os.Exit(code)
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user