summaryrefslogtreecommitdiff
path: root/vendor/github.com/sevlyar/go-daemon/command.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/sevlyar/go-daemon/command.go')
-rw-r--r--vendor/github.com/sevlyar/go-daemon/command.go99
1 files changed, 99 insertions, 0 deletions
diff --git a/vendor/github.com/sevlyar/go-daemon/command.go b/vendor/github.com/sevlyar/go-daemon/command.go
new file mode 100644
index 0000000..07d23c8
--- /dev/null
+++ b/vendor/github.com/sevlyar/go-daemon/command.go
@@ -0,0 +1,99 @@
+package daemon
+
+import (
+ "os"
+)
+
+// AddCommand is wrapper on AddFlag and SetSigHandler functions.
+func AddCommand(f Flag, sig os.Signal, handler SignalHandlerFunc) {
+ if f != nil {
+ AddFlag(f, sig)
+ }
+ if handler != nil {
+ SetSigHandler(handler, sig)
+ }
+}
+
+// Flag is the interface implemented by an object that has two state:
+// 'set' and 'unset'.
+type Flag interface {
+ IsSet() bool
+}
+
+// BoolFlag returns new object that implements interface Flag and
+// has state 'set' when var with the given address is true.
+func BoolFlag(f *bool) Flag {
+ return &boolFlag{f}
+}
+
+// StringFlag returns new object that implements interface Flag and
+// has state 'set' when var with the given address equals given value of v.
+func StringFlag(f *string, v string) Flag {
+ return &stringFlag{f, v}
+}
+
+type boolFlag struct {
+ b *bool
+}
+
+func (f *boolFlag) IsSet() bool {
+ if f == nil {
+ return false
+ }
+ return *f.b
+}
+
+type stringFlag struct {
+ s *string
+ v string
+}
+
+func (f *stringFlag) IsSet() bool {
+ if f == nil {
+ return false
+ }
+ return *f.s == f.v
+}
+
+var flags = make(map[Flag]os.Signal)
+
+// Flags returns flags that was added by the function AddFlag.
+func Flags() map[Flag]os.Signal {
+ return flags
+}
+
+// AddFlag adds the flag and signal to the internal map.
+func AddFlag(f Flag, sig os.Signal) {
+ flags[f] = sig
+}
+
+// SendCommands sends active signals to the given process.
+func SendCommands(p *os.Process) (err error) {
+ for _, sig := range signals() {
+ if err = p.Signal(sig); err != nil {
+ return
+ }
+ }
+ return
+}
+
+// ActiveFlags returns flags that has the state 'set'.
+func ActiveFlags() (ret []Flag) {
+ ret = make([]Flag, 0, 1)
+ for f := range flags {
+ if f.IsSet() {
+ ret = append(ret, f)
+ }
+ }
+ return
+}
+
+func signals() (ret []os.Signal) {
+ ret = make([]os.Signal, 0, 1)
+ for f, sig := range flags {
+ if f.IsSet() {
+ ret = append(ret, sig)
+ }
+ }
+ return
+}