Help us understand the problem. What is going on with this article?

goroutine にシグナルを送信する

More than 3 years have passed since last update.

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

Code-Hex/sigctx - GitHub  GitHub stars

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.WithSignalssigctx.Recv を使うことで通知が可能になります。
sigctx.Recv の引数には「sigctx.WithSignals の返り値の context.Context」を渡すことで実現可能です。

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

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

codehex
IQは4億、偏差値は3億、合計7億のアカウントです。
http://codehex.hateblo.jp/
mercari
フリマアプリ「メルカリ」を、グローバルで開発しています。
https://tech.mercari.com/
Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away
Comments
No comments
Sign up for free and join this conversation.
If you already have a Qiita account
Why do not you register as a user and use Qiita more conveniently?
You need to log in to use this function. Qiita can be used more conveniently after logging in.
You seem to be reading articles frequently this month. Qiita can be used more conveniently after logging in.
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away
ユーザーは見つかりませんでした