1、透過Channel傳遞退出訊號
channel作為go的一種基本資料型別，它有3種基本狀態：nil、 open、closed。
透過Channel共享數據，而不是透過共享記憶體共享數據。主流程可以透過channel向任何goroutine發送停止訊號，就像下面這樣：
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、建立一個Waitgroup的實例，假設這裡我們叫它wg
#2、在每個goroutine啟動的時候，呼叫wg.Add(1)，這個操作可以在goroutine啟動之前調用，也可以在goroutine裡面調用。當然，也可以在創建n個goroutine前調用wg.Add(n)
3、當每個goroutine完成任務後，呼叫wg.Done()
4、在等待所有goroutine的地方呼叫wg.Wait()，它在所有執行了wg.Add(1)的goroutine都調用完wg.Done()前阻塞，當所有goroutine都調用完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()
}更多golang知識請關注golang教學欄位。
以上是golang協程如何關閉的詳細內容。更多資訊請關注PHP中文網其他相關文章！
