golang--使用channel來同步goroutine

在golang中同步goroutine有2種方法，要么使用channel，要么使用sync.WaitGroup，本文就是介紹如何通過channel來同步goroutine。先看代碼。

package main

import (
    "os"
    "os/signal"
    "runtime"
    "log"
    "syscall"
)

const NUM_OF_QUIT int = 100

func main() {
    runtime.GOMAXPROCS(runtime.NumCPU())
    done := make(chan bool)
    receive_channel := make(chan chan bool)
    finish := make(chan bool)


    for i := 0; i < NUM_OF_QUIT; i++ {
        go do_while_select(i, receive_channel, finish)
    }

    go handle_exit(done, receive_channel, finish)

    <-done
    os.Exit(0)

}
func handle_exit(done chan bool, receive_channel chan chan bool, finish chan bool) {
    sigs := make(chan os.Signal, 1)
    signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
    chan_slice := make([]chan bool, 0)
    for {
        select {
        case  <-sigs:
            for _, v := range chan_slice {
                v <- true
            }
            for i := 0; i < len(chan_slice); i++ {
                <-finish
            }
            done <- true
            runtime.Goexit()
        case single_chan := <-receive_channel:
            log.Println("the single_chan is ", single_chan)
            chan_slice = append(chan_slice, single_chan)
        }
    }
}
func do_while_select(num int, rece chan chan bool, done chan bool) {
    quit := make(chan bool)
    rece <- quit
    for {
        select {
        case <-quit:
            done <- true
            runtime.Goexit()
        default:
            //簡單輸出
            log.Println("the ", num, "is running")
        }
    }
}

上面的代碼保存為example.go，通過gotool編譯代碼：

go build example.go

在當前目錄下有example文件，在終端運行這個文件

2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
2013/03/19 21:17:14 the  0 is running
......

上面不斷輸出goroutine中的數字，等待退出信號。

新打開一個終端，通過ps找到這個進程名，通過kill工具干掉這個進程：

$ps aux | grep example
user  4026 77.9  0.0  39436  1716 pts/1    Sl+  21:19   0:17 ./example
$kill 4026

不久就可以看到在第一個終端里面不再打印，至此演示完畢。

代碼思想：

新建NUM_OF_QUIT個goroutine，這些個goroutine里面新建1個chan bool，通過這個channel來接受退出的信號，這些channel在新建的時候，已經發給了handle_exit。在handle_exit這個goroutine里面，1方面監控由系統發過來的退出信號，然后再通知其他的goroutin優雅地退出；另一方面通過slice收集其他goroutine發過來的channel。handle_exit通知其他的goroutine優雅退出后，再發信號給main進程主動退出。

可以修改NUM_OF_QUIT值，例如改為10000，這個時候，kill命令發出去后，要等待相當長的一段時間才能看到第一個終端停止打印。

參考：

Go by Example: Signals

https://gobyexample.com/signals

 轉貼請注明來自：格通