怎么關(guān)閉golang協(xié)程

這篇文章給大家分享的是關(guān)閉golang協(xié)程的方法。小編覺得挺實用的，因此分享給大家做個參考。一起跟隨小編過來看看吧。

1、通過Channel傳遞退出信號

channel作為go的一種基本數(shù)據(jù)類型，它有3種基本狀態(tài)：nil、open、closed。

通過Channel共享數(shù)據(jù)，而不是通過共享內(nèi)存共享數(shù)據(jù)。主流程可以通過channel向任何goroutine發(fā)送停止信號，就像下面這樣：

func run(done chan int) {
        for {
                select {
                case <-done:
                        fmt.Println("exiting...")
                        done <- 1
                        break
                default:
                }
 
                time.Sleep(time.Second * 1)
                fmt.Println("do something")
        }
}
 
func main() {
        c := make(chan int)
 
        go run(c)
 
        fmt.Println("wait")
        time.Sleep(time.Second * 5)
 
        c <- 1
        <-c
 
        fmt.Println("main exited")
}

2、使用waitgroup

通常情況下，我們像下面這樣使用waitgroup:

1、創(chuàng)建一個Waitgroup的實例，假設(shè)此處我們叫它wg

2、在每個goroutine啟動的時候，調(diào)用wg.Add(1)，這個操作可以在goroutine啟動之前調(diào)用，也可以在goroutine里面調(diào)用。當然，也可以在創(chuàng)建n個goroutine前調(diào)用wg.Add(n)

3、當每個goroutine完成任務(wù)后，調(diào)用wg.Done()

4、在等待所有g(shù)oroutine的地方調(diào)用wg.Wait()，它在所有執(zhí)行了wg.Add(1)的goroutine都調(diào)用完wg.Done()前阻塞，當所有g(shù)oroutine都調(diào)用完wg.Done()之后它會返回。

示例：

type Service struct {
        // Other things
 
        ch        chan bool
        waitGroup *sync.WaitGroup
}
 
func NewService() *Service {
	s := &Service{
                // Init Other things
                ch:        make(chan bool),
                waitGroup: &sync.WaitGroup{},
	}
 
	return s
}
 
func (s *Service) Stop() {
        close(s.ch)
        s.waitGroup.Wait()
}
 
func (s *Service) Serve() {
        s.waitGroup.Add(1)
        defer s.waitGroup.Done()
 
        for {
                select {
                case <-s.ch:
                        fmt.Println("stopping...")
                        return
                default:
                }
                s.waitGroup.Add(1)
                go s.anotherServer()
	}
}
func (s *Service) anotherServer() {
        defer s.waitGroup.Done()
        for {
                select {
                case <-s.ch:
                        fmt.Println("stopping...")
                        return
                default:
                }
 
                // Do something
        }
}
 
func main() {
 
        service := NewService()
        go service.Serve()
 
        // Handle SIGINT and SIGTERM.
        ch := make(chan os.Signal)
        signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
        fmt.Println(<-ch)
 
        // Stop the service gracefully.
        service.Stop()
}

關(guān)于關(guān)閉golang協(xié)程的方法就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學(xué)到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。