当前位置:首页 > 文章列表 > Golang > Go问答 > 如果通道已满或在特定时间后从 Go 缓冲通道读取

如果通道已满或在特定时间后从 Go 缓冲通道读取

来源:stackoverflow 2024-04-19 22:27:34 0浏览 收藏

来到golang学习网的大家,相信都是编程学习爱好者,希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《如果通道已满或在特定时间后从 Go 缓冲通道读取》,介绍一下,希望对大家的知识积累有所帮助,助力实战开发!

问题内容

如果通道已满或经过一定时间,我想从通道进行处理。我的用例与现有问题类似,我尝试修改现有答案

我的代码位于 https://go.dev/play/p/hagz9hhqj0i,

package main

import (
    "fmt"
    "sync"
    "time"
)

type audit struct {
    id int
}

const batchsize = 5

var batch = make([]audit, 0, batchsize)

func upsertbigqueryaudits(audits []audit) {
    fmt.printf("processing batch of %d\n", len(audits))
    for _, a := range audits {
        fmt.printf("%d ", a.id)
    }
    fmt.println()
    batch = []audit{}
}
func process(full <-chan struct{}) {
    ticker := time.newticker(1 * time.nanosecond)

    for {
        select {
        case <-full:
            fmt.println("from full")
            upsertbigqueryaudits(batch)
        case <-ticker.c:
            fmt.println("from ticker")
            if len(batch) > 0 {
                fmt.println("from ticker sending batch")
                upsertbigqueryaudits(batch)
            }
        }
    }
}
func processaudits(audits <-chan audit, full chan<- struct{}, batchsize int) {
    for audit := range audits {
        batch = append(batch, audit)
        if len(batch) == cap(batch) {
            // upsertbigqueryaudits(batch)
            fmt.println("sending full")
            full <- struct{}{}

        }
    }

}

func produceaudits(x int, to chan audit) {
    for i := 0; i < x; i++ {
        to <- audit{
            id: i,
        }
    }
}

func main() {
    var wg sync.waitgroup
    audits := make(chan audit)
    full := make(chan struct{})
    wg.add(1)
    go func() {
        defer wg.done()
        process(full)
    }()
    wg.add(1)
    go func() {
        defer wg.done()
        processaudits(audits, full, batchsize)
    }()
    wg.add(1)
    go func() {
        defer wg.done()
        produceaudits(25, audits)
        close(audits)
    }()
    wg.wait()
    fmt.println("complete")
}

在这里,我想在批处理充满 batchsize 元素时或在经过一定时间(例如 1 纳秒)时处理批处理。

编辑:内联添加了我的担忧。 这是示例输出。

timeout running program
Sending full <--- this line comes twice
Sending full
From full
Processing batch of 10 <--- instead of 5 it is processing 10 items
0 1 2 3 4 5 6 7 8 9 
From full
Processing batch of 1
10 
Sending full
Sending full
From full
Processing batch of 1 <-- instead of 5 it is processing 1 items. It does not process more than 1 item.
11 
From full
Processing batch of 0

...
...
Sending full
Sending full
From full
Processing batch of 1 <-- instead of 5 it is processing 1 items
24 
From full
Processing batch of 0

From ticker <-- this line comes only after consuming 25 items.  
From ticker
From ticker
From ticker

正确答案


我在https://elliotchance.medium.com/batch-a-channel-by-size-or-time-in-go-92fa3098f65找到了答案

package main

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

func BatchStringsCtx[T any](ctx context.Context, values <-chan T, maxItems int, maxTimeout time.Duration) chan []T {
    batches := make(chan []T)

    go func() {
        defer close(batches)

        for keepGoing := true; keepGoing; {
            var batch []T
            expire := time.After(maxTimeout)
            for {
                select {
                case <-ctx.Done():
                    keepGoing = false
                    goto done

                case value, ok := <-values:
                    if !ok {
                        keepGoing = false
                        goto done
                    }

                    batch = append(batch, value)
                    if len(batch) == maxItems {
                        goto done
                    }

                case <-expire:
                    goto done
                }
            }

        done:
            if len(batch) > 0 {
                batches <- batch
            }
        }
    }()

    return batches
}

func main() {
    strings := make(chan string)
    go func() {
        strings <- "hello"
        strings <- "there" // hit limit of 2

        strings <- "how"
        time.Sleep(15 * time.Millisecond) // hit timeout

        strings <- "are"
        strings <- "you" // hit limit of 2

        // A really long time without any new values.
        // The context was cancelled around 300ms,
        // before this sleep finished.
        time.Sleep(500 * time.Millisecond)

        strings <- "doing?" // never read

        close(strings)
    }()

    //ctx, cancel := context.WithTimeout(context.Background(), 300 * time.Millisecond)
    ctx := context.Background()

    start := time.Now()
    batches := BatchStringsCtx[string](ctx, strings, 2, 10*time.Millisecond)
    for batch := range batches {
        fmt.Println(time.Now().Sub(start), batch)
    }
}

今天带大家了解了的相关知识,希望对你有所帮助;关于Golang的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~

版本声明
本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
“godef:必须至少有一个包含该文件的包”,即使该文件确实有一个包?“godef:必须至少有一个包含该文件的包”,即使该文件确实有一个包?
上一篇
“godef:必须至少有一个包含该文件的包”,即使该文件确实有一个包?
捷途山海T1即将亮相,延续硬派越野风格并搭载混动技术
下一篇
捷途山海T1即将亮相,延续硬派越野风格并搭载混动技术
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    543次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    512次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    499次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    484次学习
查看更多
AI推荐
  • 千音漫语:智能声音创作助手,AI配音、音视频翻译一站搞定!
    千音漫语
    千音漫语,北京熠声科技倾力打造的智能声音创作助手,提供AI配音、音视频翻译、语音识别、声音克隆等强大功能,助力有声书制作、视频创作、教育培训等领域,官网:https://qianyin123.com
    984次使用
  • MiniWork:智能高效AI工具平台,一站式工作学习效率解决方案
    MiniWork
    MiniWork是一款智能高效的AI工具平台,专为提升工作与学习效率而设计。整合文本处理、图像生成、营销策划及运营管理等多元AI工具,提供精准智能解决方案,让复杂工作简单高效。
    943次使用
  • NoCode (nocode.cn):零代码构建应用、网站、管理系统,降低开发门槛
    NoCode
    NoCode (nocode.cn)是领先的无代码开发平台,通过拖放、AI对话等简单操作,助您快速创建各类应用、网站与管理系统。无需编程知识,轻松实现个人生活、商业经营、企业管理多场景需求,大幅降低开发门槛,高效低成本。
    971次使用
  • 达医智影:阿里巴巴达摩院医疗AI影像早筛平台,CT一扫多筛癌症急慢病
    达医智影
    达医智影,阿里巴巴达摩院医疗AI创新力作。全球率先利用平扫CT实现“一扫多筛”,仅一次CT扫描即可高效识别多种癌症、急症及慢病,为疾病早期发现提供智能、精准的AI影像早筛解决方案。
    989次使用
  • 智慧芽Eureka:更懂技术创新的AI Agent平台,助力研发效率飞跃
    智慧芽Eureka
    智慧芽Eureka,专为技术创新打造的AI Agent平台。深度理解专利、研发、生物医药、材料、科创等复杂场景,通过专家级AI Agent精准执行任务,智能化工作流解放70%生产力,让您专注核心创新。
    969次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码