Added Hook for Windows Service

This commit is contained in:
Siva 2018-06-18 14:49:59 -04:00
parent 6e9cbeecd0
commit 411b24d95d
2 changed files with 44 additions and 0 deletions

1
service_os/service.go Normal file
View File

@ -0,0 +1 @@
package service_os

View File

@ -0,0 +1,43 @@
//+build windows
package service_os
import (
"os"
wsvc "golang.org/x/sys/windows/svc"
)
type serviceWindows struct {}
func init() {
interactive, err := wsvc.IsAnInteractiveSession()
if err != nil {
panic(err)
}
if interactive {
return
}
go func() {
_ = wsvc.Run("", serviceWindows{})
os.Exit(0)
}()
}
func (serviceWindows) Execute(args []string, r <-chan wsvc.ChangeRequest, s chan<- wsvc.Status) (svcSpecificEC bool, exitCode uint32) {
const accCommands = wsvc.AcceptStop | wsvc.AcceptShutdown
s <- wsvc.Status{State: wsvc.StartPending}
s <- wsvc.Status{State: wsvc.Running, Accepts: accCommands}
for {
c := <-r
switch c.Cmd {
case wsvc.Interrogate:
s <- c.CurrentStatus
case wsvc.Stop, wsvc.Shutdown:
s <- wsvc.Status{State: wsvc.StopPending}
return false, 0
}
}
return false, 0
}