問(wèn)題

go兩個(gè)協(xié)程使用了同一個(gè)文件句柄，其中一個(gè)協(xié)程關(guān)閉了這個(gè)文件句柄并退出了協(xié)程，如何及時(shí)通知另一個(gè)協(xié)程退出？

分析

當(dāng)一個(gè)協(xié)程關(guān)閉了這個(gè)文件對(duì)象后，底層文件的文件描述符就會(huì)被釋放。

這個(gè)時(shí)候，另一個(gè)協(xié)程還持有著同一個(gè)文件對(duì)象，但其實(shí)對(duì)應(yīng)的文件描述符已經(jīng)不存在了。

所以當(dāng)?shù)诙€(gè)協(xié)程通過(guò)這個(gè)文件對(duì)象再對(duì)文件進(jìn)行操作的時(shí)候，例如讀寫(xiě)文件等，就會(huì)發(fā)生異常，比如文件描述符不存在錯(cuò)誤等。

解決

為了避免這個(gè)問(wèn)題，共享文件對(duì)象的多個(gè)協(xié)程需要通過(guò)一個(gè)通道或鎖進(jìn)行協(xié)調(diào):

每個(gè)協(xié)程在使用文件對(duì)象前需要獲得鎖或從通道接收通知。

最后關(guān)閉文件對(duì)象的協(xié)程在關(guān)閉后，通過(guò)通道或解鎖來(lái)通知其他協(xié)程對(duì)象已經(jīng)無(wú)效。

其他協(xié)程收到通知后，就不再對(duì)這個(gè)已關(guān)閉的文件對(duì)象進(jìn)行操作。

1. 使用 channel 通道

在主協(xié)程中，可以定義一個(gè) channel，用來(lái)通知其它協(xié)程退出。協(xié)程在執(zhí)行時(shí)可以監(jiān)聽(tīng)這個(gè) channel，一旦接收到退出通知，就可以進(jìn)行清理工作，并退出協(xié)程。

quit := make(chan bool)
go func() {
    defer fmt.Println("Goroutine exit")
    for {
        select {
        case <-quit:
            return
        default:
            // ...
        }
    }
}()
// 主協(xié)程中發(fā)送退出通知
quit <- true

2. 使用 context 包

Go 語(yǔ)言標(biāo)準(zhǔn)庫(kù)中提供了 context 包，可以用來(lái)控制協(xié)程的生命周期。

在主協(xié)程中可以創(chuàng)建一個(gè) context 對(duì)象，并將其傳遞給協(xié)程，然后調(diào)用 cancel 方法，通知所有協(xié)程退出。

context.Background() 是 Go 上下文系統(tǒng)中一個(gè)重要的初始預(yù)定義上下文值，代表了一個(gè)沒(méi)有截止時(shí)間限制的空上下文環(huán)境。

1.使用WithCancel函數(shù)創(chuàng)建上下文和取消函數(shù)

ctx, cancel := context.WithCancel(context.Background())
go func(ctx context.Context) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println("exit")
            return
        default:
            // 執(zhí)行一些耗時(shí)的操作
            time.Sleep(1 * time.Second)
            fmt.Println("running")
        }
    }
}(ctx)
time.Sleep(5 * time.Second)
// cancel()
fmt.Println("cancelled")
time.Sleep(1 * time.Second)

2.使用Deadline超時(shí)結(jié)束協(xié)程

ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Second))

3.使用WithTimeout限定操作時(shí)間

ctx, cancel := context.WithTimeout(context.Background(), time.Second)

以上就是Go使用chan或context退出協(xié)程示例詳解的詳細(xì)內(nèi)容，更多關(guān)于Go chan context退出協(xié)程的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論