使用goroutine开启一个协程

package main

import (
	"fmt"
	"sync"
	"time"
)
var wg sync.WaitGroup
// 初始的例子
func worker() {
	defer wg.Done()
	for {
		fmt.Println("worker...")
		time.Sleep(time.Second)
	}
	// 如何接收外部命令实现退出
}
func main() {
	wg.Add(1)
	go worker()
	// 如何优雅的实现结束子goroutine
	wg.Wait() // 等待...
	fmt.Println("over")
}

上述例子会一直等待worker运行，死循环不会主动退出，那我们如何让worker协程退出呢？主要由以下一些办法。

1 全局变量方式

package main
import (
	"fmt"
	"sync"
	"time"
)
var wg sync.WaitGroup
// 全局变量的方式
var exit bool
func worker() {
	defer wg.Done()
	for {
		fmt.Println("worker...")
		time.Sleep(time.Second)
		// 如何接收外部命令实现退出
		if exit {
			break
		}
	}
}
func main() {
	wg.Add(1)
	go worker()
	// 如何优雅的实现结束子goroutine
	time.Sleep(time.Second * 5)
	exit = true // 通过修改全局变量的方式通知子goroutine退出
	wg.Wait()   // 等待...
	fmt.Println("over")
}

主协程等待5s后改变exit全局变量，worker中每次循环判断exit，为true时退出

2 使用channel传递变量

package main
import (
	"fmt"
	"sync"
	"time"
)
var wg sync.WaitGroup
// 使用channel的方式实现
func worker(ch <-chan struct{}) {
	defer wg.Done()
LABEL:
	for {
		fmt.Println("worker...")
		time.Sleep(time.Second)
		select {
		case <-ch:
			break LABEL
		default:
		}
	}
	// 如何接收外部命令实现退出
}
func main() {
	var exitChan = make(chan struct{})
	wg.Add(1)
	go worker(exitChan)
	// 如何优雅的实现结束子goroutine
	time.Sleep(time.Second * 5)
	exitChan <- struct{}{}
	wg.Wait() // 等待...
	fmt.Println("over")
}

for循环中使用select，当<-ch能够执行时，即有程序向ch写入数据时，worker才会退出

3 使用context实现

package main
import (
	"context"
	"fmt"
	"sync"
	"time"
)
var wg sync.WaitGroup

func worker(ctx context.Context) {
	defer wg.Done()
	LABEL:
	for {
		fmt.Println("worker...")
		time.Sleep(time.Second)
		select {
			case <-ctx.Done():
				break LABEL
		default:

		}
	}
	// 如何接收外部命令实现退出
}
func main() {
	ctx, cancel := context.WithCancel(context.Background())
	wg.Add(1)
	go worker(ctx)
	// 如何优雅的实现结束子goroutine
	time.Sleep(time.Second*5)
	cancel() // 调用cancel函数告诉子goroutine退出
	wg.Wait() // 等待...
	fmt.Println("cancel...")
	time.Sleep(time.Second*5)
	fmt.Println("over")
}

Context类型，专门用来简化对于处理单个请求的多个 goroutine 之间与请求域的数据、取消信号、截止时间等相关操作，这些操作可能涉及多个 API 调用。
Done方法需要返回一个Channel，这个Channel会在当前工作完成或者上下文被取消之后关闭，多次调用Done方法会返回同一个Channel；
WithCancel会返回ctx和cancel，调用cancel时会向包含ctx的协程发送信号。当调用返回的cancel函数或当关闭父上下文的Done通道时，将关闭返回上下文的Done通道，无论先发生什么情况。
Context是线程安全的，可以放心的在多个goroutine中传递。
参考链接

文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/133473.html