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

Question

Максим Федоров @Maksclub

maksfedorov.ru

Go

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

Запущено 20 горутин такой функции.

func (w *Writer) Work(ctx context.Context) {
	// start

	f, err := os.Create("./tmp/" + filename)
	if err != nil {
		panic(err)
	}

	defer f.Close()


	// Запись шапки в файл

	for {
		select {
		case <-ctx.Done():
			// Как  выполнить тут работу в конце работы горутины? сейчас не работает
		case it := <-*w.Channel:
			// Запись элементов в файл
		}
	}

	// Или может тут  выполнить  работу в конце работы горутины ? тоже не работает
}

В цикле я каждой из них (через канал для каждой, 20 горутин — значит 20 каналов) отправляю сообщения, а каждая пишет в свой файл

ctx := context.Background()
	ctx, cancel := context.WithCancel(ctx)

defer func() {
		fmt.Println("Finish items loading!") // это сообщеие выводится
		cancel()
	}()

// тут стартанули нашу пачку горутин
 
	for {
		res, err := esScroller.Do(ctx)
		
		for _, hit := range res.Hits.Hits {

    		// ... билд item

    		//  отправляем во все каналы элемент
    		for _, w := range f.Writers {
        		*w.Channel <- item
   		 }
	    }
}

Когда закончились сообщения — как завершить все и проделать в каждой из них полезную финальную работу?

Вопрос задан более трёх лет назад
391 просмотр

Комментировать

Подписаться 1 Средний Комментировать

Answer 1 · 2021-05-26 12:44:00

Сработал такой вариант со SO с применением еще канала, который не позволит завершить выполнение main()

func main() {
    forever := make(chan struct{})
    ctx, cancel := context.WithCancel(context.Background())

    go func(ctx context.Context) {
        for {
            select {
            case <-ctx.Done():  // if cancel() execute
                forever <- struct{}{}
                return
            default:
                fmt.Println("for loop")
            }

            time.Sleep(500 * time.Millisecond)
        }
    }(ctx)

    go func() {
        time.Sleep(3 * time.Second)
        cancel()
    }()

    <-forever
    fmt.Println("finish")
}

Src: https://stackoverflow.com/questions/6807590/how-to...

Вариант 2: Можно через WaitGroup завершить или через отдельную горутину с буферезированным каналом (на число горутин-воркеров)

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

Войдите на сайт