使用goroutine开启一个协程

package mainimport ("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()// 如何优雅的实现结束子goroutinewg.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()// 如何优雅的实现结束子goroutinetime.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 LABELdefault:}}// 如何接收外部命令实现退出
}
func main() {var exitChan = make(chan struct{})wg.Add(1)go worker(exitChan)// 如何优雅的实现结束子goroutinetime.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.WaitGroupfunc worker(ctx context.Context) {defer wg.Done()LABEL:for {fmt.Println("worker...")time.Sleep(time.Second)select {case <-ctx.Done():break LABELdefault:}}// 如何接收外部命令实现退出
}
func main() {ctx, cancel := context.WithCancel(context.Background())wg.Add(1)go worker(ctx)// 如何优雅的实现结束子goroutinetime.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中传递。
参考链接