Bagaimana untuk menghantar WaitGroup ke panggilan fungsi berjujukan?

Kandungan soalan

Saya mempunyai fungsi yang boleh dipanggil secara berurutan atau serentak dalam goroutine yang berasingan.

Saya ingin memastikan fungsi dilaksanakan sepenuhnya sebelum goroutine utama selesai, jadi saya menghantar argumen *sync.waitgroup kepada fungsi tersebut. Sekarang, di beberapa tempat fungsi itu akan dipanggil secara berurutan.

Saya boleh menghantar nil waitgroup ke fungsi seperti ini:

func my_func(wg *sync.WaitGroup){
   if wg != nil{
   defer wg.Done()
   }
   // do the task
}

func main(){
my_func(nil) // sequential call

wg := sync.WaitGroup{}
wg.Add(1)
go my_func(&wg)  // concurrent call
wg.Wait()
}

Adakah cara yang lebih baik untuk mencapai matlamat ini?

Jawapan Betul

Yours my_func() 不应该知道/不应该关心它是如何执行的（无论是否在新的 goroutine 中）。因此，为此您不应该传递 wg. Jangan paksa penggunaan serentak atau bukan serentak api anda, biarkan pengguna pakej anda memutuskan cara mereka mahu memanggilnya.

Jika ada yang ingin menjalankannya secara serentak, dalam goroutine baru, ini boleh dilakukan dalam my_func() 的外部处理 my_func() seperti ini:

wg.add(1)
go func() {
    defer wg.done()
    my_func()
}()

Ini juga memungkinkan untuk meletakkan lebih banyak kod sebelum/selepas panggilan fungsi yang akan dilaksanakan sebelum wg.done() panggilan:

wg.add(1)
go func() {
    defer wg.done()

    // other code before
    my_func()
    // other code after
}()

Juga ambil perhatian bahawa jika anda mempunyai fungsi ini di banyak tempat, anda boleh mencipta fungsi pembantu yang menjaga pengendalian goroutine dan kumpulan tunggu:

func launchmyfunc(wg *sync.waitgroup) {
    go func() {
        defer wg.done()

        my_func()
    }()
}

Anda juga boleh mencipta pembantu yang menerima sebarang fungsi tanpa parameter dan tidak berbalas:

func launchfunc(wg *sync.waitgroup, f func()) {
    go func() {
        defer wg.done()

        f()
    }()
}

Menggunakan pembantu di atas, anda boleh melakukan ini:

wg.Add(1)
launchMyFunc(wg)

// or
wg.Add(1)
launchFunc(wg, my_func)

Atas ialah kandungan terperinci Bagaimana untuk menghantar WaitGroup ke panggilan fungsi berjujukan?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!