golang使用context关闭go routine的方法

本篇内容主要讲解“golang使用context关闭go routine的方法”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“golang使用context关闭go routine的方法”吧!

go中的context

go语言提供了context包，支持在多个go routine之间传递信号，可以使用它优雅的取消任务，从而优雅的结束go routine，节约系统资源。

实践

今天遇到了这个任务，有多科考试，需要比对一下各个学科的考试学生名单是否一致，考虑到假如有好几科，然后每科有几千个学生，如果只用循环那么需要循环几万次，于是就开了协程结合context进行处理。话不多说，直接上代码：

    var baseStus []examstudent.StudentViewModel
	linq.From(stuMap[int(sexams[0].SeID)]).OrderBy(func(i interface{}) interface{} {
		return i.(examstudent.StudentViewModel).ExamCode
	}).ToSlice(&baseStus)
    //定义一个支持取消的context
	ctx, cancel := context.WithCancel(context.Background())
    //标志，用来区分是一致还是不一致
	sch := make(chan int, len(sexams)-1)
	for i, _ := range sexams {
		if i == 0 {
			continue
		}
        //开启协程
		go func(c context.Context, index int) {
			var compareStus []examstudent.StudentViewModel
			linq.From(stuMap[int(sexams[i].SeID)]).OrderBy(func(x interface{}) interface{} {
				return x.(examstudent.StudentViewModel).ExamCode
			}).ToSlice(&compareStus)
			for x, vv := range baseStus {
				select {
				case <-c.Done():
					return
				default:
					if vv.ExamCode != compareStus[x].ExamCode {
						cancel() //如果不一致就取消任务
						return
					}
				}
			}
			sch <- index
		}(ctx, i)

	}
	defer cancel()
	defer close(sch)
	for {
		select {//select语句
		case <-ctx.Done():
			return false, nil  //执行了这个地方就表示不一致了
		case <-sch:
			if len(sch) == 0 {
				return true, nil
			}
		}
	}

到此，相信大家对“golang使用context关闭go routine的方法”有了更深的了解，不妨来实际操作一番吧！这里是亿速云网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！