summaryrefslogtreecommitdiffstats
path: root/svc_windows.go
diff options
context:
space:
mode:
Diffstat (limited to 'svc_windows.go')
-rw-r--r--svc_windows.go70
1 files changed, 70 insertions, 0 deletions
diff --git a/svc_windows.go b/svc_windows.go
new file mode 100644
index 0000000000..91e2b6a4dd
--- /dev/null
+++ b/svc_windows.go
@@ -0,0 +1,70 @@
+// 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
+}