1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
package main
import (
"flag"
"log"
"os"
"os/signal"
"strings"
"sync"
"sync/atomic"
"syscall"
"time"
"github.com/xlgmokha/minit/pkg/procfile"
)
var (
pidMutex sync.Mutex
pids []int
procfilePath *string
)
func init() {
procfilePath = flag.String("f", "Procfile", "path to Procfile")
flag.Parse()
log.SetFlags(0)
}
func addPid(pid int) {
pidMutex.Lock()
defer pidMutex.Unlock()
pids = append(pids, pid)
}
func removePid(pid int) {
pidMutex.Lock()
defer pidMutex.Unlock()
for i, p := range pids {
if p == pid {
pids = append(pids[:i], pids[i+1:]...)
break
}
}
}
func forwardSignalToAll(sig os.Signal) {
pidMutex.Lock()
defer pidMutex.Unlock()
signal := sig.(syscall.Signal)
for _, pid := range pids {
syscall.Kill(-pid, signal)
}
}
func main() {
var wg sync.WaitGroup
var shutdown int32
for _, path := range strings.Split(*procfilePath, ",") {
procs, err := procfile.ParseFile(path)
if err != nil {
log.Fatalln(err)
}
for _, proc := range procs {
wg.Add(1)
go func(proc *procfile.Proc) {
defer wg.Done()
for atomic.LoadInt32(&shutdown) == 0 {
cmd := proc.NewCommand()
if cmd.Start() != nil {
time.Sleep(2 * time.Second)
continue
}
addPid(cmd.Process.Pid)
cmd.Wait()
removePid(cmd.Process.Pid)
time.Sleep(time.Second)
}
}(proc)
}
}
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan)
go func() {
for sig := range sigChan {
if sig == syscall.SIGINT || sig == syscall.SIGTERM {
atomic.StoreInt32(&shutdown, 1)
forwardSignalToAll(sig)
return
}
forwardSignalToAll(sig)
}
}()
wg.Wait()
}
|
