cwinfo/yggdrasil-go
5
0
Fork 0
mirror of https://github.com/cwinfo/yggdrasil-go.git synced 2024-11-22 10:40:27 +00:00
Code Releases Activity

Merge pull request #19 from neilalexander/sigterm

Browse Source 
Catch SIGTERM and Windows service termination
This commit is contained in:
Arceliar 2018-02-18 15:35:47 -06:00 committed by GitHub
commit a1afebc0df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
yggdrasil.go
View File

@ -9,6 +9,7 @@ import "io/ioutil"
import "net" import "net"
import "os" import "os"
import "os/signal" import "os/signal"
import "syscall"
import "time" import "time"
import "regexp" import "regexp"
@ -21,7 +22,7 @@ import "golang.org/x/net/ipv6"
import . "yggdrasil" import . "yggdrasil"
import _ "github.com/kardianos/minwinsvc" import "github.com/kardianos/minwinsvc"
/** /**
* This is a very crude wrapper around src/yggdrasil * This is a very crude wrapper around src/yggdrasil
@ -306,7 +307,13 @@ func main() {
} }
// Catch interrupt to exit gracefully // Catch interrupt to exit gracefully
c := make(chan os.Signal, 1) c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt, os.Kill) signal.Notify(c, os.Interrupt, syscall.SIGTERM)
// Create a function to capture the service being stopped on Windows
winTerminate := func() {
c <- os.Interrupt
}
minwinsvc.SetOnExit(winTerminate)
// Wait for the terminate/interrupt signal
<-c <-c
logger.Println("Stopping...") logger.Println("Stopping...")
} }