是否关闭通道块,直到接收器读取它

时间:2018-09-03 09:19:03

标签: go concurrency channel goroutine

我已经实现了一种通过关闭通道来关闭服务器的方法,因此其他goroutine会先读取已关闭的通道,然后退出。关闭之后,我需要对服务器数据进行一些清理,如果close()阻塞,直到所有其他goroutine都读取了关闭的通道,我就可以不加锁地访问数据。所以问题是:关闭通道是否阻塞,直到接收者从中读取?下面是示例代码:

package main

type server struct {
    chStop chan struct{}
    data map[int]interface{}
}

func newServer() *server {
    return &server {
        chStop: make(chan struct{})
    }
}

func (s *server) stop() {
    close(s.chStop)
    // do something with s.data
    ...
    // if other goroutines already read closed s.chStop and exited,
    // we can access s.data without lock
}

func (s *server) run2() {
    ...
    for {
        select{
        case <-s.chStop:
            return
        case <- other channel:
        // access s.data with lock
        }
    }
}

func (s *server) run() {
    ch := make(chan struct{})
    ...
    for {
        select{
        case <-s.chStop:
            return
        case <- ch:
            // access s.data with lock
        }
    }
}

func main() {
    s := newServer()
    go s.run()
    go s.run2()
    s.stop()
}

1 个答案:

答案 0 :(得分:4)

不。请改用sync.WaitGroup