关闭通道阻塞直到接收器读取它

问题描述 投票:0回答:1

我已经实现了一种通过关闭通道关闭我的服务器的方法,所以其他goroutines读取关闭的通道然后退出。关机后,我需要对服务器数据进行一些清理,如果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()
}
go concurrency channel goroutine
1个回答
4
投票

不。请使用sync.WaitGroup代替。

© www.soinside.com 2019 - 2024. All rights reserved.