Tunggu fungsi selesai dalam golang

Di golang, menunggu fungsi selesai adalah keperluan pengaturcaraan biasa. Sama ada anda sedang menunggu goroutine untuk dilengkapkan atau menunggu data dalam saluran tiba, anda perlu menggunakan kaedah menunggu yang sesuai untuk mengendalikannya. Dalam artikel ini, kami akan memperkenalkan anda kepada beberapa kaedah dan teknik untuk menunggu penyiapan fungsi dalam golang. Sama ada anda seorang pemula atau pembangun berpengalaman, artikel ini akan memberikan anda panduan berguna dan kod contoh untuk membantu anda mengendalikan dengan lebih baik menunggu fungsi untuk menyelesaikan senario. Mari lihat lebih dekat!

Kandungan soalan

Saya mempunyai kod berikut dalam golang:

func A(){
  go print("hello")
}

func main() {
  A()
  // here I want to wait for the print to happen
  B()
}

Bagaimana untuk memastikan bahawa b() hanya dilaksanakan selepas pencetakan berlaku?

Penyelesaian

Gunakan sync.mutex

var l sync.mutex

func a() {
    go func() {
        print("hello")
        l.unlock()
    }()
}

func b() {
    print("world")
}

func testlock(t *testing.t) {
    l.lock()
    a()
    l.lock()
    // here i want to wait for the print to happen
    b()
    l.unlock()
}

Gunakan sync.waitgroup

var wg sync.waitgroup

func a() {
    go func() {
        print("hello")
        wg.done()
    }()
}

func b() {
    print("world")
}

func testlock(t *testing.t) {
    wg.add(1)
    a()
    wg.wait()
    // here i want to wait for the print to happen
    b()
}

Gunakan chan

func A() chan struct{} {
    c := make(chan struct{})
    go func() {     
        print("hello")
        c <- struct{}{}
    }()
    return c
}

func B() {
    print("world")
}

func TestLock(t *testing.T) {
    c := A()
    // here I want to wait for the print to happen
    <-c
    B()
}

Atas ialah kandungan terperinci Tunggu fungsi selesai dalam golang. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!