golang 控制goroutine调度顺序

Posted ningto.com

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了golang 控制goroutine调度顺序相关的知识,希望对你有一定的参考价值。

使用go关键字就可以很容易的启动一个goroutine,启动后他们的执行顺序是不能保证的。如果有多个goroutine,怎样按照我想要的顺序来执行呢?

如:

  • a1在b1和c1后面执行(b1和c1都执行完a1才能执行)
  • d1在a1后面执行(d1执行完才能执行a1)
  • e1在b1或者c1后面执行(b1或者c1有一个执行完就可以执行e1)

下面代码演示了怎样解决上面问题:
event/event.go

package event

import (
	"reflect"
	"sync"
	"sync/atomic"
)

type Event struct 
	fired int32
	c     chan struct
	once  sync.Once


func New() *Event 
	return &Eventc: make(chan struct)


func (ev *Event) Fire() int32 
	atomic.AddInt32(&ev.fired, 1)
	ev.once.Do(func() 
		close(ev.c)
	)
	return ev.fired


func (ev *Event) Wait() 
	<-ev.c


func (ev *Event) HasFired() bool 
	return atomic.LoadInt32(&ev.fired) > 0


func (ev *Event) Start(f func()) 
	go func() 
		defer ev.Fire()
		f()
	()


func (ev *Event) After(f func(), e *Event) 
	go func() 
		defer ev.Fire()
		e.Wait()
		f()
	()


func (ev *Event) Every(f func(), evs ...*Event) 
	go func() 
		defer ev.Fire()
		for _, e := range evs 
			e.Wait()
		
		f()
	()


func (ev *Event) Some(f func(), evs ...*Event) 
	var selectCase = make([]reflect.SelectCase, len(evs))
	for i, e := range evs 
		selectCase[i].Dir = reflect.SelectRecv
		selectCase[i].Chan = reflect.ValueOf(e.c)
	

	go func() 
		defer ev.Fire()
		reflect.Select(selectCase)
		f()
	()


main.go

package main

import (
	"fmt"
	"time"

	"godemo/event"
)

func main() 
	a1 := event.New()
	b1 := event.New()
	c1 := event.New()
	d1 := event.New()
	e1 := event.New()

	// a1在b1和c1后面
	a1.Every(func() 
		fmt.Println("a1")
	, b1, c1)

	b1.Start(func() 
		time.Sleep(100 * time.Millisecond)
		fmt.Println("b1")
	)

	c1.Start(func() 
		time.Sleep(1 * time.Second)
		fmt.Println("c1")
	)

	// d1在a1后面
	d1.After(func() 
		fmt.Println("d1")
	, a1)

	// e1在b1或c1后面
	e1.Some(func() 
		fmt.Println("e1")
	, b1, c1)

	time.Sleep(2 * time.Second)


再举个例子,如a, b, c, d按倒序执行:

package main

import (
	"fmt"

	"godemo/event"
)

func main() 
	a1 := event.New()
	b1 := event.New()
	c1 := event.New()
	d1 := event.New()

	a1.After(func() 
		fmt.Println("a1")
	, b1)

	b1.After(func() 
		fmt.Println("b1")
	, c1)

	c1.After(func() 
		fmt.Println("c1")
	, d1)

	d1.Start(func() 
		fmt.Println("d1")
	)

	a1.Wait()

以上是关于golang 控制goroutine调度顺序的主要内容,如果未能解决你的问题,请参考以下文章

Golang---GMP调度策略

golang的goroutine调度机制

golang Context for goroutines

Golang详解调度机制 抢占式调度

go语言之行--golang核武器goroutine调度原理channel详解

【golang详解】go语言GMP(GPM)原理和调度