fosrl/newt
1
0
Fork 0
mirror of https://github.com/fosrl/newt.git synced 2025-05-13 05:30:39 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle termination message

Browse source
This commit is contained in:
Owen Schwartz 2024-12-08 21:45:02 -05:00
parent 49580f1a3c
commit b9a3632a1e
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD
2 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
main.go
View file

@ -220,6 +220,17 @@ func main() {
var connected bool var connected bool
var wgData WgData var wgData WgData
client.RegisterHandler("newt/terminate", func(msg websocket.WSMessage) {
logger.Info("Received terminate message")
if pm != nil {
pm.Stop()
}
if dev != nil {
dev.Close()
}
client.Close()
})
// Register handlers for different message types // Register handlers for different message types
client.RegisterHandler("newt/wg/connect", func(msg websocket.WSMessage) { client.RegisterHandler("newt/wg/connect", func(msg websocket.WSMessage) {
logger.Info("Received registration message") logger.Info("Received registration message")

4
websocket/client.go
View file

@ -86,6 +86,10 @@ func (c *Client) Close() error {
if c.conn != nil { if c.conn != nil {
return c.conn.Close() return c.conn.Close()
} }
// stop the ping monitor
c.setConnected(false)
return nil return nil
} }