Bagaimana untuk Membunuh Berbilang Goroutine Secara Serentak dalam Go?

Cara Membunuh Berbilang Goroutine Sekaligus dalam Go

Dalam Go, anda mungkin menghadapi senario di mana anda perlu menamatkan berbilang goroutine secara serentak. Ini amat berguna apabila penyiapan satu goroutine seharusnya mencetuskan penamatan yang lain.

Pertimbangkan coretan kod berikut:

func main() {

  go func() {
    ...
    if err != nil {
      return
    }
  }()

  go func() {
    ...
    if err != nil {
      return
    }
  }()

}

Dalam kes ini, anda ingin memastikan bahawa apabila salah satu goroutine kembali, yang lain juga harus keluar. Pendekatan biasa untuk mencapai ini ialah menggunakan saluran untuk memberi isyarat. Walau bagaimanapun, ini boleh menyebabkan panik menulis kepada saluran tertutup.

Penyelesaian yang boleh dipercayai untuk masalah ini ialah memanfaatkan mekanisme konteks Go. Konteks membolehkan anda mewujudkan komunikasi antara rutin dan isyarat apabila operasi selesai atau dibatalkan. Berikut ialah contoh:

package main

import (
    "context"
    "sync"
)

func main() {

    ctx, cancel := context.WithCancel(context.Background())
    wg := sync.WaitGroup{}
    wg.Add(3)
    go func() {
        defer wg.Done()
        for {
            select {
            // msg from other goroutine finish
            case <-ctx.Done():
                // end
            }
        }
    }()

    go func() {
        defer wg.Done()
        for {
            select {
            // msg from other goroutine finish
            case <-ctx.Done():
                // end
            }
        }
    }()

    go func() {
        defer wg.Done()
        // your operation
        // call cancel when this goroutine ends
        cancel()
    }()
    wg.Wait()
}

Dalam contoh ini, kami mencipta konteks yang boleh dihantar kepada goroutine untuk menyemak isyarat penamatan. Apabila goroutine ketiga menyelesaikan operasinya, ia memanggil cancel(), yang menghantar isyarat kepada kedua-dua goroutine menunggu untuk siap. Akibatnya, semua goroutine ditamatkan dengan baik.

Atas ialah kandungan terperinci Bagaimana untuk Membunuh Berbilang Goroutine Secara Serentak dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!