goroutine にシグナルを送信する

正確には goroutine へ直接シグナルを送る方法なんてありませんが、次のパッケージを使うことで似たようなことを実現することができます。(タイトルで釣ってしまって申し訳ないです!!)

Code-Hex/sigctx - GitHub　　

context with signal in golang

シグナル受信するための context を作成することができるパッケージです。

あるシグナルをプロセスが受け取ったら context cancel を行う

package main

import (
    "context"
    "fmt"
    "sync"
    "syscall"

    "github.com/Code-Hex/sigctx"
)

func main() {

    ctx := sigctx.WithCancelSignals(
        context.Background(),
        syscall.SIGINT,
        syscall.SIGTERM,
        syscall.SIGHUP,
    )

    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            for {
                select {
                // You can cancel by send signal.
                case <-ctx.Done():
                    fmt.Println(ctx.Err())
                    return
                }
            }
        }()
    }
    wg.Wait()
}

sigctx.WithCancelSignals の第一引数に context.Context を、その後ろに context キャンセルを行わせたい os.Signal を指定します。（この場合だと INT, TERM, HUP を受信すると context キャンセルを行います）
めっちゃシンプルですね！

受け取ったシグナルを context を使っている全ての goroutine へ通知する

package main

import (
    "context"
    "fmt"
    "sync"
    "syscall"

    "github.com/Code-Hex/sigctx"
)

func main() {

    ctx := sigctx.WithSignals(
        context.Background(),
        syscall.SIGINT,
        syscall.SIGTERM,
        syscall.SIGHUP,
    )

    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            for {
                select {
                // You can notify signal received.
                case <-sigctx.Recv(ctx):
                    signal, err := sigctx.Signal(ctx)
                    if err != nil {
                        fmt.Println(err.Error())
                        return
                    }
                    switch signal {
                    case syscall.SIGINT:
                        fmt.Println(signal.String())
                        return
                    default:
                        fmt.Println(signal.String())
                    }
                case <-ctx.Done():
                    fmt.Println(ctx.Err())
                    return
                }
            }
        }()
    }
    wg.Wait()
}

sigctx.WithSignals と sigctx.Recv を使うことで通知が可能になります。
sigctx.Recv の引数には「sigctx.WithSignals の返り値の context.Context」を渡すことで実現可能です。

また受信したシグナルを受け取るには sigctx.Signal を使うと受け取ることができます。sigctx.Signal の引数も「sigctx.WithSignals の返り値の context.Context」をわたうようにしてください。

この辺はもう少し工夫すればもっとシンプルになりそうですが、もっと良いアイデアお持ちの方がいましたら是非 PR を送っていただけると嬉しいです。
是非使ってみてください！