84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
package pidfile
|
|
|
|
// based on original work found here, github.com/soellman/pidfile
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"syscall"
|
|
)
|
|
|
|
var (
|
|
ErrProcessRunning = errors.New("process is running")
|
|
ErrFileStale = errors.New("pidfile exists but process is not running")
|
|
ErrFileInvalid = errors.New("pidfile has invalid contents")
|
|
)
|
|
|
|
// Remove a pidfile
|
|
func Remove(filename string) error {
|
|
return os.RemoveAll(filename)
|
|
}
|
|
|
|
// Write writes a pidfile, returning an error
|
|
// if the process is already running or pidfile is orphaned
|
|
func Write(filename string) (int, error) {
|
|
return WriteControl(filename, os.Getpid(), false)
|
|
}
|
|
|
|
func WriteControl(filename string, pid int, overwrite bool) (int, error) {
|
|
// Check for existing pid
|
|
oldpid, err := pidfileContents(filename)
|
|
if err != nil && !os.IsNotExist(err) {
|
|
return oldpid, err
|
|
}
|
|
|
|
// We have a pid
|
|
if err == nil {
|
|
if pidIsRunning(oldpid) {
|
|
return oldpid, ErrProcessRunning
|
|
}
|
|
if !overwrite {
|
|
return -1, ErrFileStale
|
|
}
|
|
}
|
|
|
|
// We're clear to (over)write the file
|
|
return pid, os.WriteFile(filename, []byte(fmt.Sprintf("%d\n", pid)), 0644)
|
|
}
|
|
|
|
func pidfileContents(filename string) (int, error) {
|
|
contents, err := os.ReadFile(filename)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
pid, err := strconv.Atoi(strings.TrimSpace(string(contents)))
|
|
if err != nil {
|
|
return 0, ErrFileInvalid
|
|
}
|
|
|
|
return pid, nil
|
|
}
|
|
|
|
func pidIsRunning(pid int) bool {
|
|
process, err := os.FindProcess(pid)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
err = process.Signal(syscall.Signal(0))
|
|
|
|
if err != nil && err.Error() == "no such process" {
|
|
return false
|
|
}
|
|
|
|
if err != nil && err.Error() == "os: process already finished" {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|