GOLANG使用Context管理关联goroutine

原文:https://gocn.io/article/333

一般一个业务很少不用到goroutine的,因为很多方法是需要等待的,例如http.Server.ListenAndServe这个就是等待的,除非关闭了Server或Listener,否则是不会返回的。除非是一个API服务器,否则肯定需要另外起goroutine发起其他的服务,而且对于API服务器来说,在http.Handler的处理函数中一般也需要起goroutine,如何管理这些goroutine,在GOLANG1.7提供context.Context

先看一个简单的,如果启动两个goroutine,一个是HTTP,还有个信号处理的收到退出信号做清理:

wg := sync.WaitGroup{}
defer wg.Wait()

wg.Add(1)
go func() {
    defer wg.Done()

    ss := make(os.Signal,0)
    signal.Notify(ss,syscall.SIGINT,syscall.SIGTERM)
    for s := ss {
        fmt.Println("Got signal",s)
        break
    }
}()

wg.Add(1)
go func() {
    defer wg.Done()

    svr := &http.Server{ Addr:":8080",Handler:nil,}
    fmt.Println(svr.ListenAndServe())
}

很清楚,起了两个goroutine,然后用WaitGroup等待它们退出。如果它们之间没有交互,不互相影响,那真的是蛮简单的,可惜这样是不行的,因为信号的goroutine收到退出信号后,应该通知server退出。暴力一点的是直接调用svr.Close(),但是如果有些请求还需要取消怎么办呢?最好用Context了:

wg := sync.WaitGroup{}
defer wg.Wait()

ctx,cancel := context.WithCancel(context.Background())

wg.Add(1)
go func() {
    defer wg.Done()

    ss := make(os.Signal, 0)
    signal.Notify(ss,syscall.SIGINT,syscall.SIGTERM)
    for {
        select {
        case <- ctx.Done():
            return
        case s := <- ss:
            fmt.Println("Got signal",s)
            cancel() // 取消请求,通知用到ctx的所有goroutine
            return
        }
    }
}()

wg.Add(1)
go func() {
    defer wg.Done()
    defer cancel()

    svr := &http.Server{ Addr:":8080",Handler:nil,}

    go func(){
        select {
        case <- ctx.Done():
            svr.Close()
        }
    }

    fmt.Println(svr.ListenAndServe())
}

这个方式可以在新开goroutine时继续使用,譬如新加一个goroutine,里面读写了UDPConn:

wg.Add(1)
go func() {
    defer wg.Done()
    defer cancel()

    var conn *net.UDPConn
    if conn,err = net.Dial("udp","127.0.0.1:1935"); err != nil {
        fmt.Println("Dial UDP server Failed,err is",err)
        return
    }

    fmt.Println(UDPRead(ctx,conn))
}()

UDPRead = func(ctx context.Context,conn *net.UDPConn) (err error) {
    wg := sync.WaitGroup{}
    defer wg.Wait()

    ctx,cancel := context.WithCancel(ctx)

    wg.Add(1)
    go func() {
        defer wg.Done()
        defer cancel()

        for {
            b := make([]byte,core.MTUSize)
            size,_,err := conn.ReadFromUDP(b)
            // 处理UDP包 b[:size]
        }
    }()

    select {
    case <-ctx.Done():
        conn.Close()
    }
    return
}

如果只是用到HTTP Server,可以这么写:

func run(ctx contex.Context) {
    server := &http.Server{Addr: addr,Handler: nil}
    go func() {
        select {
        case <-ctx.Done():
            server.Close()
        }
    }()

    http.HandleFunc("/api",func(w http.ResponseWriter,r *http.Request) {
    })

    fmt.Println(server.ListenAndServe())
}

如果需要提供一个API来让服务器退出,可以这么写:

func run(ctx contex.Context) {
    server := &http.Server{Addr: addr,Handler: nil}

    ctx,cancel := context.WithCancel(ctx)
    http.HandleFunc("/quit",r *http.Request) {
        cancel() // 使用局部的ctx和cancel
    })

    go func() {
        select {
        case <-ctx.Done():
            server.Close()
        }
    }()

    fmt.Println(server.ListenAndServe())
}

使用局部的ctx和cancel,可以避免cancel传入的ctx,只是影响当前的ctx。

相关文章

程序目录结构 简单实现,用户登录后返回一个jwt的token,下次请求带上token请求用户信息接口并返回信息...
本篇博客的主要内容是用go写一个简单的Proof-of-Work共识机制,不涉及到网络通信环节,只是一个本地的简...
简介 默克尔树(MerkleTree)是一种典型的二叉树结构,其主要特点为: 最下面的叶节点包含存储数据或其...
接下来学习并发编程, 并发编程是go语言最有特色的地方, go对并发编程是原生支持. goroutine是go中最近本...
先普及一下, 什么是广度优先搜索 广度优先搜索类似于树的层次遍历。从图中的某一顶点出发,遍历每一个顶...
第一天: 接口的定义和实现 第二天: 一. go语言是面向接口编程. 在学习继承的时候说过, go语言只有封装,...