登录
首页 >  Golang >  Go问答

这段代码的goroutine有一点疑惑,求解

来源:SegmentFault

时间:2023-02-24 17:29:07 411浏览 收藏

来到golang学习网的大家,相信都是编程学习爱好者,希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《这段代码的goroutine有一点疑惑,求解》,介绍一下并发、go、concurrency,希望对大家的知识积累有所帮助,助力实战开发!

问题内容

func (client *Client) receive() {
    var err error
    for err == nil {
        var h codec.Header
        if err = client.cc.ReadHeader(&h); err != nil {
            break
        }
        call := client.removeCall(h.Seq)
        switch {
        case call == nil:
            // it usually means that Write partially failed
            // and call was already removed.
            err = client.cc.ReadBody(nil)
        case h.Error != "":
            call.Error = fmt.Errorf(h.Error)
            err = client.cc.ReadBody(nil)
            call.done()
        default:
            err = client.cc.ReadBody(call.Reply)
            if err != nil {
                call.Error = errors.New("reading body " + err.Error())
            }
            call.done()
        }
    }
    // error occurs, so terminateCalls pending calls
    client.terminateCalls(err)
}

func NewClient(conn net.Conn, opt *Option) (*Client, error) {
    f := codec.NewCodecFuncMap[opt.CodecType]
    if f == nil {
        err := fmt.Errorf("invalid codec type %s", opt.CodecType)
        log.Println("rpc client: codec error:", err)
        return nil, err
    }
    // send options with server
    if err := json.NewEncoder(conn).Encode(opt); err != nil {
        log.Println("rpc client: options error: ", err)
        _ = conn.Close()
        return nil, err
    }
    return newClientCodec(f(conn), opt), nil
}

func newClientCodec(cc codec.Codec, opt *Option) *Client {
    client := &Client{
        seq:     1, // seq starts with 1, 0 means invalid call
        cc:      cc,
        opt:     opt,
        pending: make(map[uint64]*Call),
    }
    go client.receive()
    return client
}

这是rpc的一段客户端的的一段代码,这里

newClientCodec()
返回后
go client.receive()
怎么能保证一直运行接受请求,主线程运行完coroutine也会终止啊

正确答案

这没啥问题吧?只要主线程生命周期比这个 goroutine 长就行了。

如果主线程代码是你自己写的,要么

sync.WaitGroup
去等 goroutine,要么阻塞掉主线程等信号就好了啊

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

func main() {
  newClientCodec(...)
  ch := make(chan os.Signal, 1)
  signal.Notify(ch, syscall.SIGINT)
  

要是主线程调用 listen 之类的网络 api 阻塞住了,就更不用管了。

除非知道主线程在特定情况下会退出,然后你又想主线程不要马上退出,先等 goroutine 处理完或保存好数据,也就是所谓的优雅停机 graceful shutdown 。

这种情况可以监听

SIGINT
SIGTERM
信号,
SIGINT
在键盘按下
ctrl+c
触发,
SIGTERM
会在
kill
命令、Docker 停止容器等情况下触发。在信号处理阶段用
context
sync.WaitGroup
让 goroutine 把手里的事情处理完再退出,完成 graceful shutdown。

文中关于golang的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《这段代码的goroutine有一点疑惑,求解》文章吧,也可关注golang学习网公众号了解相关技术文章。

声明:本文转载于:SegmentFault 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>