12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970 |
- // Copyright 2015 Daniel Theophanes.
- // Use of this source code is governed by a zlib-style
- // license that can be found in the LICENSE file.package service
- //+build windows
- package minwinsvc
- import (
- "os"
- "sync"
- "golang.org/x/sys/windows/svc"
- )
- var (
- onExit func()
- guard sync.Mutex
- )
- func init() {
- interactive, err := svc.IsAnInteractiveSession()
- if err != nil {
- panic(err)
- }
- if interactive {
- return
- }
- go func() {
- _ = svc.Run("", runner{})
- guard.Lock()
- f := onExit
- guard.Unlock()
- // Don't hold this lock in user code.
- if f != nil {
- f()
- }
- // Make sure we exit.
- os.Exit(0)
- }()
- }
- func setOnExit(f func()) {
- guard.Lock()
- onExit = f
- guard.Unlock()
- }
- type runner struct{}
- func (runner) Execute(args []string, r <-chan svc.ChangeRequest, changes chan<- svc.Status) (bool, uint32) {
- const cmdsAccepted = svc.AcceptStop | svc.AcceptShutdown
- changes <- svc.Status{State: svc.StartPending}
- changes <- svc.Status{State: svc.Running, Accepts: cmdsAccepted}
- for {
- c := <-r
- switch c.Cmd {
- case svc.Interrogate:
- changes <- c.CurrentStatus
- case svc.Stop, svc.Shutdown:
- changes <- svc.Status{State: svc.StopPending}
- return false, 0
- }
- }
- return false, 0
- }
|