pid.go 725 B

1234567891011121314151617181920212223242526272829303132
  1. package ncutils
  2. import (
  3. "fmt"
  4. "os"
  5. "strconv"
  6. )
  7. // PIDFILE - path/name of pid file
  8. const PIDFILE = "/var/run/netclient.pid"
  9. // SavePID - saves the pid of running program to disk
  10. func SavePID() error {
  11. pid := os.Getpid()
  12. if err := os.WriteFile(PIDFILE, []byte(fmt.Sprintf("%d", pid)), 0644); err != nil {
  13. return fmt.Errorf("could not write to pid file %w", err)
  14. }
  15. return nil
  16. }
  17. // ReadPID - reads a previously saved pid from disk
  18. func ReadPID() (int, error) {
  19. bytes, err := os.ReadFile(PIDFILE)
  20. if err != nil {
  21. return 0, fmt.Errorf("could not read pid file %w", err)
  22. }
  23. pid, err := strconv.Atoi(string(bytes))
  24. if err != nil {
  25. return 0, fmt.Errorf("pid file contents invalid %w", err)
  26. }
  27. return pid, nil
  28. }