feature: add stop handling for engine

This commit is contained in:
braginini
2021-06-21 11:18:03 +02:00
parent 74355a2292
commit 8088c7a591
4 changed files with 26 additions and 2 deletions

View File

@@ -53,7 +53,7 @@ func SetupCloseHandler() {
go func() {
for range c {
fmt.Println("\r- Ctrl+C pressed in Terminal")
os.Exit(0)
stopUP <- 0
}
}()
}

View File

@@ -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)
},
}
)