Как заставить горутины запускаться в порядке очередности?

Question

[DollyPapper]

Добрый день!

Есть следующий код

package main

import (
	"fmt"
	"sync"
)

func main() {
	wg := &sync.WaitGroup{}
	ch := make(chan string)
	mu := sync.Mutex{}

	for i := 0; i < 5; i++ {
		wg.Add(1)
		go func(group *sync.WaitGroup, i int) {
			defer group.Done()

			mu.Lock()
			ch <- fmt.Sprintf("Goroutine %d", i)
			mu.Unlock()
		}(wg, i)
	}

	for i := 0; i < 5; i++ {
		select {
		case s := <-ch:
			fmt.Println(s)
		}
	}

	wg.Wait()
}

Есть проблема, что значения Goroutine N печатаются не в порядке очереди, а в порядке очереди запуска горутин. Всю голову уже сломал, но не могу придумать, как обеспечить гарантию порядка в данном случае?

Comments

[Александр Павлюк]

Не использовать горутины. Сделать простой цикл и он у вас будет запускать и обрабатывать вещи по-порядку.

Вопрос в том, что вы хотите достичь? Может вам все же не важен порядок запуска, но важно потом результаты сложить по-порядку? Такое можно сделать.

Опишите подробнее, где допустим параллелизм, а где нет для вашей задачи.

[DollyPapper]

Александр Павлюк, задача с собеса, думаю наличие параллелизма тут должно остаться как есть, а вот сложить результаты по порядку это как раз то что нужно. Как этого можно достичь?

[Александр Павлюк]

DollyPapper, в данном случае нам известно количество данных (5 штук), значит можем создать слайс на 5 элементов и горутины пусть кладут результат каждая в i-ю ячейку. Просто и эффективно.

[Александр Павлюк]

Кстати, лочить мьютекс при записи в канал не нужно

[d'Ivan]

У горутин нет гарантированного порядка выполнения.

Answer 1

[d'Ivan]

Сначала запускаем воркеры, потом создаем задачи. По окончанию задач надо закрыть канал.

package main

import (
	"fmt"
	"sync"
	"time"
)

const (
	jobsNum    = 5
	workersNum = 2
)

type Job struct {
	id int
}

func worker(id int, jobChan <-chan Job, wg *sync.WaitGroup) {
	for job := range jobChan {
		process(id, job, wg)
	}
}

func process(id int, job Job, wg *sync.WaitGroup) {
	defer wg.Done()
	fmt.Printf("Worker %d, Job start: %d\n", id, job.id)
	time.Sleep(1 * time.Second)
	fmt.Printf("Worker %d, Job end: %d\n", id, job.id)
}

func main() {
	fmt.Println("Start")
	jobChan := make(chan Job, 10)

	var wg sync.WaitGroup

	wg.Add(jobsNum)

	// start the workers
	for i := 0; i < workersNum; i++ {
		go worker(i, jobChan, &wg)
	}

	// enqueue jobs
	for i := 0; i < jobsNum; i++ {
		jobChan <- Job{
			id: i,
		}
	}
	close(jobChan)

	wg.Wait()
	fmt.Println("Done")
}

вывод

Start
Worker 1, Job start: 1
Worker 0, Job start: 0
Worker 0, Job end: 0
Worker 0, Job start: 2
Worker 1, Job end: 1
Worker 1, Job start: 3
Worker 0, Job end: 2
Worker 0, Job start: 4
Worker 1, Job end: 3
Worker 0, Job end: 4
Done

Answer 2

[ykalchevskiy]

Если этот код является лишь шаблоном задачи с собеседования, который предполагалось расширить, то можно добавить условную переменную и обычную переменную для текущего счетчика:

package main

import (
	"fmt"
	"sync"
)

func main() {
	wg := &sync.WaitGroup{}
	ch := make(chan string)
	mu := sync.Mutex{}
	s := 0
	c := sync.NewCond(&mu)

	for i := 0; i < 5; i++ {
		wg.Add(1)
		go func(group *sync.WaitGroup, i int) {
			defer group.Done()

			mu.Lock() // c.L.Lock()
			for s != i {
				c.Wait()
			}
			ch <- fmt.Sprintf("Goroutine %d", i)
			s += 1
			c.Broadcast()
			mu.Unlock() // c.L.Unlock()
		}(wg, i)
	}

	c.Broadcast()

	for i := 0; i < 5; i++ {
		select {
		case s := <-ch:
			fmt.Println(s)
		}
	}

	wg.Wait()
}

Comments

[ykalchevskiy]

Горутины продолжают выполнять свою работу раздельно, но результаты добавляются в канал строго по очереди.
Пример 1 (обратный порядок, ждем 4, 3, 2, 1, 0, потом выводим все): https://play.golang.com/p/G6KIJzyjNxf
Пример 2 (прямой порядок, 0 выполняется и сразу выводим, потом 1 и т.д.): https://play.golang.com/p/ikrlu4xv3s7

[ykalchevskiy]

DollyPapper, это не решение задачи?