71fb0a723e
* copyright headers for agent folder
47 lines
924 B
Go
47 lines
924 B
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
//go:build windows
|
|
// +build windows
|
|
|
|
package service_os
|
|
|
|
import (
|
|
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{})
|
|
}()
|
|
}
|
|
|
|
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:
|
|
chanGraceExit <- 1
|
|
s <- wsvc.Status{State: wsvc.StopPending}
|
|
return false, 0
|
|
}
|
|
}
|
|
|
|
return false, 0
|
|
}
|