golang监控goroutine异常退出

在golang中，我们可以很轻易产生数以万计的goroutine，不过这也带来了麻烦：在运行中某一个goroutine异常退出，怎么办？

在erlang中，有link原语，2个进程可以链接在一起，一个在异常退出的时候，向另一个进程呼喊崩溃的原因，然后由另一个进程处理这些信号，包括是否重启这个进程。在这方面，erlang的确做得很好，估计以后这个特性会在golang中得到实现。

由此得到启发，我写了一个简单的程序，监控goroutine异常退出。

package main

import (
    "log"
    "runtime"
    "math/rand"
    "time"
)

type message struct {
    normal bool //true means exit normal, otherwise
    state map[string]interface{} //goroutine state
}

func main() {
    runtime.GOMAXPROCS(runtime.NumCPU())
    mess := make(chan message, 10)
    for i := 0; i < 100; i++ {
        go worker(mess)
    }
    supervisor(mess)
}

func worker(mess chan message) {
    defer func() {
        exit_message := message{state:make(map[string] interface{})}
        i := recover()
        if i != nil {
            exit_message.normal = false
        } else {
            exit_message.normal = true
        }
        mess <- exit_message
    }()
    now := time.Now()
    seed := now.UnixNano()
    rand.Seed(seed)
    num := rand.Int63()
    if num % 2 != 0 {
        panic("not evening")
    } else {
        runtime.Goexit()
    }
}


func supervisor(mess chan message) {
    for i := 0; i < 100; i++ {
        m := <- mess 
        switch m.normal {
        case true :
            log.Println("exit normal, nothing serious!")
        case false:
            log.Println("exit abnormal, something went wrong")
        }

    }
}

原本是想把goroutine异常退出时的状态，也就是某些变量保存到message中的state，然后在另一个goroutine里面处理这些数据，不过能力有限，暂时做不出来，以后会完善的。这里只验证goroutine在异常退出的时候，向另一个goroutine发送信号。

在worker函数里面，会判断随机数是不是偶数，不是的话会使用panic异常退出，然后使用defer向supervisor发送信号，supervisor接受到就输出结果。

下面看一下效果：

我们用了100个goroutine，也就有100个退出信号。由图可以得到我们的程序运行良好。

转贴请注明来自：格通