Minta Go untuk menjalankan semua goroutine sebelum meneruskan

Apabila melakukan pengaturcaraan serentak, kita sering menghadapi situasi di mana kita perlu menunggu semua goroutine selesai sebelum meneruskan pelaksanaan. Dalam bahasa Go, kita boleh mencapai matlamat ini dengan menggunakan WaitGroup. WaitGroup ialah semafor mengira yang boleh digunakan untuk menunggu penyiapan kumpulan goroutine. Sebelum meneruskan, kita perlu memanggil kaedah Tunggu WaitGroup untuk memastikan semua goroutine telah menyelesaikan tugas mereka. Dalam artikel ini, kami akan memperkenalkan cara menggunakan WaitGroup dengan betul untuk mengurus susunan pelaksanaan goroutine.

Kandungan soalan

Saya memerlukan penjadual golang untuk menjalankan semua goroutine sebelum meneruskan, runtime.gosched() tidak dapat menyelesaikannya.

Masalahnya ialah rutin pergi berjalan begitu pantas sehingga "select" dalam start() berjalan selepas "select" dalam stopstream() dan kemudian "case <-chanstopstream:" penerima tidak bersedia untuk pengirim "case retchan <-benar:". Ternyata apabila ini berlaku, hasilnya adalah kelakuan yang sama seperti ketika stopstream() digantung

Jalankan kod ini https://go.dev/play/p/dq85xqju2q_z Banyak kali anda akan melihat kedua-dua respons ini Sambutan yang dijangkakan apabila tidak digantung:

2009/11/10 23:00:00 start
2009/11/10 23:00:00 receive chan
2009/11/10 23:00:03 end

Tindak balas yang dijangkakan apabila digantung, tetapi tidak begitu pantas apabila digantung:

2009/11/10 23:00:00 start
2009/11/10 23:00:00 default
2009/11/10 23:00:01 timer
2009/11/10 23:00:04 end

Kod

package main

import (
    "log"
    "runtime"
    "sync"
    "time"
)

var wg sync.WaitGroup

func main() {
    wg.Add(1)
    //run multiples routines on a huge system
    go start()
    wg.Wait()
}
func start() {
    log.Println("Start")
    chanStopStream := make(chan bool)
    go stopStream(chanStopStream)
    select {
    case <-chanStopStream:
        log.Println("receive chan")
    case <-time.After(time.Second): //if stopStream hangs do not wait more than 1 second
        log.Println("TIMER")
        //call some crash alert
    }
    time.Sleep(3 * time.Second)
    log.Println("end")
    wg.Done()
}

func stopStream(retChan chan bool) {
    //do some work that can be faster then caller or not
    runtime.Gosched()
    //time.Sleep(time.Microsecond) //better solution then runtime.Gosched()
    //time.Sleep(2 * time.Second) //simulate when this routine hangs more than 1 second
    select {
    case retChan <- true:
    default: //select/default is used because if the caller times out this routine will hangs forever
        log.Println("default")
    }
}

Penyelesaian

Tiada cara untuk menjalankan semua goroutine lain sebelum meneruskan pelaksanaan goroutine semasa.

Selesaikan masalah dengan memastikan goroutine tidak tersekat stopstream:

Pilihan 1: Tukar chanstopstream kepada saluran buffer chanstopstream 更改为缓冲通道。这确保了 stopstream. Ini memastikan bahawa

func start() {
    log.println("start")
    chanstopstream := make(chan bool, 1) // <--- buffered channel
    go stopstream(chanstopstream)
    ...
}

func stopstream(retchan chan bool) {
    ...
    // always send. no select/default needed.
    retchan <- true
}

Pilihan 2

func start() {
    log.Println("Start")
    chanStopStream := make(chan bool) // buffered channel not required
    go stopStream(chanStopStream)
    ...

func stopStream(retChan chan bool) {
    ...
    close(retChan)
}

Atas ialah kandungan terperinci Minta Go untuk menjalankan semua goroutine sebelum meneruskan. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!