mirror of
https://github.com/binwiederhier/ntfy.git
synced 2025-05-12 21:20:36 +01:00
Compare commits
2 commits
b68990cbcc
...
2691acfd08
Author | SHA1 | Date | |
---|---|---|---|
![]() |
2691acfd08 | ||
![]() |
feb6647f23 |
1 changed files with 70 additions and 0 deletions
70
cmd/serve_windows.go
Normal file
70
cmd/serve_windows.go
Normal file
|
@ -0,0 +1,70 @@
|
|||
// Copyright 2015 Daniel Theophanes.
|
||||
// Use of this source code is governed by a zlib-style
|
||||
// license that can be found at: https://github.com/kardianos/minwinsvc
|
||||
|
||||
//go:build windows && !noserver
|
||||
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"golang.org/x/sys/windows/svc"
|
||||
)
|
||||
|
||||
var (
|
||||
onExit func()
|
||||
guard sync.Mutex
|
||||
)
|
||||
|
||||
func init() {
|
||||
isService, err := svc.IsWindowsService()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if !isService {
|
||||
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
|
||||
}
|
Loading…
Reference in a new issue