how to stop groutine with channel

2348 ワード
Go Go テキストリンク

overview

バックグラウンド等で動かすgoroutineを使ったforループプログラムで、シグナルを受け取った際に終了させる処理例

example1

ここではシグナルの代わりに適当なチャンネルを使う。

package main

import (
    "fmt"
    "time"
)

func main() {
    c := make(chan string, 1)
    sig := make(chan string, 1)
    go func(){
        for{
            fmt.Println("Hello")

            select{
            case s := <-c:
                fmt.Println(s)
                sig<-"FINISH"
                return
            default:
            }
            time.Sleep(1 * time.Second) //
        }
    }()

    time.Sleep(10 * time.Second)
    c<-"done"
    s:= <-sig
    fmt.Println(s)
}

example2

実際にシグナルを使った例

package main

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

func Loop(c chan os.Signal, done chan int) {
    go func() {
        for {
            fmt.Println("do something....")
            select {
            case s := <-c:
                fmt.Println("Got signal:", s)
                close(c)
                done <- 0
                return
            default:
            }
            time.Sleep(1 * time.Second)
        }
    }()
}
func main() {
    sig := make(chan os.Signal, 1)
    signal.Notify(sig, syscall.SIGTERM, syscall.SIGINT)

    done := make(chan int)
    defer close(done)

    Loop(sig, done)

    <-done
    fmt.Println("Exit")
    os.Exit(0)
}

実行結果

  • mainではdoneチャンネル受信待ちにする。
  • シグナルを送るとチャンネルに送信される。
  • 所定の処理が終わったあとにチャネルを受信する。シグナルが来ていなければ処理を継続。シグナルが来ている場合はdoneチャンネルに送信する、mainでdoneチャンネルを受信し処理を終了する。
$ go run signal.go
do something....
do something....
do something....
do something....
do something....
do something....
do something....
do something....
do something....
do something....
do something....
^Cdo something....
Got signal: interrupt
Exit
phpによるmongodbパケットソート
runtime属性の追加