golangでシグナルを拾ってgracefulにgoroutineを停めたい・改

これの続き。
http://qiita.com/arc279/items/c44d4a18a851ff454c64

ちょっとまとめてみた。名前は適当。
使い方は main() を見られたし。

main.go

package main

import (
	"fmt"
	"os"
	"os/signal"
	"sync"
	"syscall"
	"time"
)

type Gracefully struct {
	trapSig  []os.Signal
	callback func(os.Signal) bool
	procs    []func()
}

func NewGracefully(callback func(os.Signal) bool, sig ...os.Signal) *Gracefully {
	self := Gracefully{
		trapSig:  sig,
		callback: callback,
		procs:    make([]func(), 0),
	}
	return &self
}

func (self *Gracefully) Append(proc func()) {
	self.procs = append(self.procs, proc)
}

func (self *Gracefully) Run() {
	var wg sync.WaitGroup
	wg.Add(len(self.procs))

	exitChan := make([]chan int, len(self.procs))
	for i, _ := range self.procs {
		exitChan[i] = make(chan int, 1)
	}

	for i, proc := range self.procs {
		go func(ch chan int, proc func()) {
			defer wg.Done()
			for {
				select {
				case <-ch:
					return
				default:
					// なんらかの並行処理するやつ
					proc()
				}
			}
		}(exitChan[i], proc)
	}

	signalChan := make(chan os.Signal)
	signal.Notify(signalChan, self.trapSig...)

	go func() {
		for {
			sig := <-signalChan
			for _, trap := range self.trapSig {
				if sig == trap {
					if self.callback(sig) {
						// callbackがtrueを返したら抜ける
						for _, c := range exitChan {
							c <- 1
						}
					}
				}
			}
		}
	}()

	wg.Wait()
}

func main() {

	// HUP と USR2 を拾って、HUPの時は抜けたいコールバック
	g := NewGracefully(func(sig os.Signal) bool {
		fmt.Println(sig)
		return sig == syscall.SIGHUP
	}, syscall.SIGHUP, syscall.SIGUSR2)

	procs := []string{
		"proc1",
		"proc2",
		"proc3",
		"proc4",
		"proc5",
		"proc6",
		"proc7",
	}

	for _, p := range procs {
		// javascriptみたいｗ
		g.Append(func(s string) func() {
			return func() {
				fmt.Println(s)
				time.Sleep(500 * time.Millisecond)
			}
		}(p))
	}

	g.Run()

	fmt.Println("stop gracefully")
	os.Exit(0)
}

ビルドして実行

$ go build -o sample main.go && ./sample

シグナルを送る

HUPだと停まる

$ kill -SIGHUP `pgrep -f sample`

USR2だと停まらない

$ kill -SIGUSR2 `pgrep -f sample`

それ以外は拾わない

$ kill -SIGUSR1 `pgrep -f sample`