golang每日一题:context、goroutine相关

24/11/28

题目描述

  1. 整体的超时控制
  2. A子协程发送数字 0-9
  3. B子协程计算a发来数字的平方
  4. 主线程打印输出最后的平方数
go 复制代码
package main

import (
    "context"
    "fmt"
    "testing"
    "time"
)

func TestGoroutine(t *testing.T) {
    // 设置超时
    ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)
    defer cancel() // 在函数结束时取消 context

    sendNum := make(chan int)
    resMul := make(chan int)

    // 启动第一个 goroutine,用于发送数据
    go func(ctx context.Context) {
        defer close(sendNum) // 在发送完数据后关闭 sendNum 通道
        for i := 0; i < 200; i++ {
            select {
            case sendNum <- i:
            case <-ctx.Done(): // 如果上下文被取消或超时,退出
                return
            }
        }
    }(ctx)

    // 启动第二个 goroutine,用于接收并处理数据
    go func(ctx context.Context) {
        defer close(resMul) // 在处理完数据后关闭 resMul 通道
        for {
            select {
            case num := <-sendNum:
                resMul <- num * num
            case <-ctx.Done(): // 如果上下文被取消或超时,退出
                return
            }
        }
    }(ctx)

    // 主 goroutine 处理结果
    for {
        select {
        case tem := <-resMul:
            fmt.Printf("resMul = %v\n", tem)
        case <-ctx.Done():
            fmt.Println(ctx.Err()) // 打印超时或取消的错误信息
            fmt.Println("time out...")
            return
        }
    }
}

改进:

  • defer cancel():在函数结束时调用 cancel() 来释放上下文资源。这样做可以确保 context 在不再需要时被清理。
  • 关闭通道:在发送数据完毕后,我们显式地关闭了 sendNum 通道。这有助于避免死锁,并且通知接收方数据已发送完毕。类似地,resMul 通道也应该在最后关闭。
  • 避免无限循环:通过在 select 中监听 ctx.Done(),在 context 超时或取消时退出 goroutine,而不需要使用额外的 flag 控制循环。
  • 更优雅的退出机制:通过 ctx.Done() 监听上下文取消,确保所有 goroutine 在超时或取消时能够正确退出。
相关推荐
pyengine39 分钟前
基于pandoc的MarkDown格式与word相互转换小工具开发(pyqt5)
开发语言·python·qt·word
YuSun_WK1 小时前
配置MambaIRv2: Attentive State Space Restoration的环境
开发语言·python
Nick_zcy1 小时前
开发基于python的商品推荐系统,前端框架和后端框架的选择比较
开发语言·python·前端框架·flask·fastapi
淬渊阁1 小时前
Go package
java·开发语言
篱笆院的狗1 小时前
MySQL 中 SQL 语句的详细执行过程
数据库·sql·mysql
冰茶_1 小时前
C#中常见的设计模式
java·开发语言·microsoft·设计模式·微软·c#·命令模式
Echo``1 小时前
2:QT联合HALCON编程—图像显示放大缩小
开发语言·c++·图像处理·qt·算法
IT成长日记2 小时前
【Hive入门】Hive基础操作与SQL语法:DDL操作全面指南
hive·hadoop·sql·ddl操作
JAVA学习通2 小时前
JAVA多线程(8.0)
java·开发语言
Luck_ff08102 小时前
【Python爬虫详解】第四篇:使用解析库提取网页数据——BeautifuSoup
开发语言·爬虫·python