golang函数通信管道超时处理策略

使用管道进行通信时，为防止管道接收端一直阻塞， golang 提供两种超时处理策略：使用 context 设置时间限制或使用 select 监听多个管道，当管道接收端没有收到数据时，这两个策略都会超时。

Golang 函数通信管道超时处理策略

管道是 Golang 中进行进程间通信的一种常见方式。但是，当管道接收端收不到数据时，它会一直阻塞。为了防止这种阻塞，我们可以使用带有超时的管道接收操作。

超时处理策略

立即学习“go语言免费学习笔记（深入）”；

万知

万知: 你的个人AI工作站

下载

有两种主要的超时处理策略：

• 使用 Context: 使用 context.Context 类型，我们可以设置一个时间限制，在这个时间限制内没有收到数据，操作就会超时。
• 使用 select： 使用 select 语句，我们可以同时监听多个管道。当管道接收端没有收到数据时，select 会超时。

实战案例

下面是一个使用 Context 超时处理策略的管道接收操作示例：

package main

import (
    "context"
    "fmt"
    "log"
    "sync/atomic"
    "time"
)

func main() {
    // 创建一个管道
    ch := make(chan int)
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    // 并发地将数据发送到管道
    go func() {
        for i := 0; i < 10; i++ {
            ch <- i
        }
    }()

    // 使用 Context 超时接收数据
    go func() {
        var total uint64
        for {
            select {
            case <-ctx.Done():
                fmt.Println("Timeout reached!")
                return
            case value := <-ch:
                total += uint64(value)
            }
        }
    }()

    log.Printf("Total: %d", total)
}

使用 select 超时处理策略的管道接收操作示例：

package main

import (
    "fmt"
    "log"
    "sync/atomic"
    "time"
)

func main() {
    // 创建一个管道
    ch := make(chan int)

    // 创建一个 select 语句来监听管道和超时
    var total uint64
    go func() {
        for {
            select {
            case value := <-ch:
                total += uint64(value)
            case <-time.After(5 * time.Second):
                fmt.Println("Timeout reached!")
                return
            }
        }
    }()

    // 并发地将数据发送到管道
    go func() {
        for i := 0; i < 10; i++ {
            ch <- i
        }
    }()

    log.Printf("Total: %d", total)
}