go的singleflight学习

方法

Do
DoChan
Forget

使用示例

sg "golang.org/x/sync/singleflight"func TestDo(t *testing.T) {var g sg.Groupv, err, _ := g.Do("key", func() (interface{}, error) {return "bar", nil})if got, want := fmt.Sprintf("%v (%T)", v, v), "bar (string)"; got != want {t.Errorf("Do = %v; want %v", got, want)}if err != nil {t.Errorf("Do error = %v", err)}
}func TestDoErr(t *testing.T) {var g sg.GroupsomeErr := errors.New("Some error")v, err, _ := g.Do("key", func() (interface{}, error) {return nil, someErr})if err != someErr {t.Errorf("Do error = %v; want someErr %v", err, someErr)}if v != nil {t.Errorf("unexpected non-nil value %#v", v)}
}func TestDoDupSuppress(t *testing.T) {var g sg.Groupvar wg1, wg2 sync.WaitGroupc := make(chan string, 1)var calls int32// 原子读取calls,如果为1则将wg1 -1,同时从chan中读取内容,然后再写回,// sleep 10mills 后,返回 chan 中的内容fn := func() (interface{}, error) {if atomic.AddInt32(&calls, 1) == 1 {fmt.Println("[fn] first call")// First invocation.// 第一次调用wg1.Done()}fmt.Println("[fn] call")v := <-cc <- v // pump; make available for any future calls// 模拟耗时的下游请求,让更多的协程等待sg的Do结果time.Sleep(10 * time.Millisecond) // let more goroutines enter Doreturn v, nil}const n = 10wg1.Add(1)for i := 0; i < n; i++ {curI := iwg1.Add(1)wg2.Add(1)go func() {fmt.Println("[TestDoDupSuppress] curI", curI)defer wg2.Done() // 维持协程的结束wg1.Done()       // 维持协程的开始// 位置1(不是下一行的标识)v, err, _ := g.Do("key", fn) // 位置1的下一个位置if err != nil {t.Errorf("Do error: %v", err)return}if s, _ := v.(string); s != "bar" {t.Errorf("Do = %T %v; want %q", v, v, "bar")}}()}wg1.Wait() // 全部协程都就绪了,保证所有协程进入位置1// 此时只有一个协程进入了Do中,等待在chan上c <- "bar"wg2.Wait()// 此时所有的协程都处理结束// 此时的calls一定处于[1, n)got := atomic.LoadInt32(&calls)fmt.Println("got", got)if got <= 0 || got >= n {t.Errorf("number of calls = %d; want over 0 and less than %d", got, n)}
}// Test that singleflight behaves correctly after Forget called.
// See https://github.com/golang/go/issues/31420
// 忘记一个key,forget about a key
// 下一次调用时,可以继续使用Do
// Future calls to Do for this key will call the function
// 而不是等待一个更早的请求去直接拿结果
// rather than waiting for an earlier call to complete
func TestForget(t *testing.T) {var g sg.Groupvar (firstStarted  = make(chan struct{}) // 没有缓冲的通道unblockFirst  = make(chan struct{})firstFinished = make(chan struct{}))go func() {g.Do("key", func() (i interface{}, e error) {// 关闭一个已关闭的channel会导致panic,所以在关闭之前,需要确认channel是否已被关闭// 或者处理可能出现的panic// 注意// 关闭channel并不是必须的// 当不会再向channel写入数据时,可以关闭channel// 当其他goroutine试图从该channel接收时,它们会立即接收到一个零值而不再阻塞close(firstStarted)<-unblockFirstclose(firstFinished)return})}()<-firstStarted  // 第一次请求卡在这里等待结果g.Forget("key") // 第一次请求的Do还卡在<-unblockFirst上,此时执行ForgetunblockSecond := make(chan struct{}) // 第二次请求,卡在 <-unblockSecond 中secondResult := g.DoChan("key", func() (i interface{}, e error) {<-unblockSecondreturn 2, nil})close(unblockFirst) // 通知第一次请求的Do中执行<-unblockFirst<-firstFinished     // 等待第一次请求中的Do执行完成// 第三次请求,返回3thirdResult := g.DoChan("key", func() (i interface{}, e error) {return 3, nil})close(unblockSecond) // 通知第二次请求的Do执行完成r2 := <-secondResult // 读取第二次的结果if r2.Val != 2 {t.Errorf("We should receive result produced by second call, expected: 2, got %d", r2.Val)}r3 := <-thirdResult // 读取第三次的结果if r3.Val != 2 {t.Errorf("We should receive result produced by second call, expected: 2, got %d", r3.Val)}
}// TestDoChan 结果存储到 Chan 中
func TestDoChan(t *testing.T) {var g sg.Groupch := g.DoChan("key", func() (interface{}, error) {return "bar", nil})res := <-chv := res.Valerr := res.Errif got, want := fmt.Sprintf("%v (%T)", v, v), "bar (string)"; got != want {t.Errorf("Do = %v; want %v", got, want)}if err != nil {t.Errorf("Do error = %v", err)}
}// Test singleflight behaves correctly after Do panic.
// See https://github.com/golang/go/issues/41133,这个issues有意思
// 如果panic,则所有协程都panic
func TestPanicDo(t *testing.T) {var g sg.Groupfn := func() (interface{}, error) {panic("invalid memory address or nil pointer dereference")}const n = 5waited := int32(n)panicCount := int32(0) // 如果panic,则所有协程都panicdone := make(chan struct{})for i := 0; i < n; i++ {go func() {defer func() {// defer中先panicCount、再waited// 主函数判断的时候是先判断 waited,再判断 panicCountif err := recover(); err != nil {t.Logf("Got panic: %v\n%s", err, debug.Stack())atomic.AddInt32(&panicCount, 1)}if atomic.AddInt32(&waited, -1) == 0 {close(done)}}()g.Do("key", fn)}()}select {case <-done: // 关闭通道if panicCount != n {t.Errorf("Expect %d panic, but got %d", n, panicCount)}case <-time.After(time.Second): // 1秒后还没有执行完成t.Fatalf("Do hangs")}
}// runtime.Goexit() 用于终止调用它的 goroutine
// 该函数不返回任何值。当执行此函数时,goroutine 将立即停止执行
// 但是并不会影响其他的 goroutine
// 随后,程序将恢复执行其他的 goroutine
// 如果主 goroutine 执行了这个函数,所有的 goroutine 将停止
//
// 这个函数通常在需要提前结束一个 goroutine 或者在特定条件下需要关闭 goroutine 时使用
// 但通常情况下,通过 return 或者到达函数尾部来结束一个 goroutine
// 使用 runtime.Goexit() 非常罕见
//
func TestGoexitDo(t *testing.T) {var g sg.Groupfn := func() (interface{}, error) {fmt.Println("[fn] before goexit")runtime.Goexit()return nil, nil}const n = 5waited := int32(n)done := make(chan struct{})for i := 0; i < n; i++ {go func() {var err errordefer func() {if err != nil {t.Errorf("Error should be nil, but got: %v", err)}tmp := atomic.AddInt32(&waited, -1)fmt.Println("waited", tmp)if tmp == 0 {close(done)}}()// g中使用匿名函数执行fn,所以当fn中直接runtime.Goexit(),则该协程被kill// 将不执行下一行打印,而是直接执行deferval, err, shared := g.Do("key", fn)fmt.Printf("val %v, err %v, shared %v\n", jsonx.ToString(val), err, shared)}()}select {case <-done:fmt.Println("finish")case <-time.After(time.Second):t.Fatalf("Do hangs")}
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/734616.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

文件操作上(c语言)

目录 1. 文件的作用2. 什么是文件2.1 程序文件2.2 数据文件2.3 文件名 3. 二进制文件和文本文件4. 文件的打开和关闭4.1 流和标准流4.1.1 流4.1.2 标准流 4.2 文件指针4.3 文件的打开与关闭4.3.1 文件的打开模式4.3.2 实例代码 1. 文件的作用 使用文件可以将数据进行持久化的保…

C++ 理解“引用”以及在编程中使用时的注意事项

引言&#xff1a; 在编程中&#xff0c;“引用”是一个变量或者内存地址的别名。它允许我们通过不同的名称来访问同一内存位置。引用的使用可以提高代码的可读性和灵活性&#xff0c;但也带来了一些潜在的问题。在这篇博客中&#xff0c;我们将深入理解引用的概念&#xff0c;并…

openssl调试记录

openssl不能直接解密16进制密文&#xff0c;需要把密文转化成base64格式才能解密 调试记录如下&#xff1a;

Qt 中Json文件的操作

Json文件的读取 QFile file("data.json"); //准备好的文件file.open(QIODevice::ReadOnly|QIODevice::Text);QByteArray arr file.readAll();QJsonDocument jsonDoc QJsonDocument::fromJson(arr);QJsonObject jsonObj jsonDoc.object();qint32 id jsonObj["…

Dubbo-记录

1.概念 Apache Dubbo 是一款 RPC 服务开发框架&#xff0c;用于解决微服务架构下的服务治理与通信问题&#xff0c;官方提供了 Java、Golang 等多语言 SDK 实现。使用 Dubbo 开发的微服务原生具备相互之间的远程地址发现与通信能力&#xff0c; 利用 Dubbo 提供的丰富服务治理…

Qt QListwidget与QStackedWidget或QTableWidget实现多界面切换的效果

文章目录 效果图使用QStackedWidget实现使用QTableWidget实现总结 效果图 使用QStackedWidget实现 QStackedWidget提供了一种堆栈式的界面布局方式。功能&#xff1a;QStackedWidget允许开发者在一个固定区域内显示多个子窗口或页面&#xff0c;但同时只显示其中一个子窗口&am…

windows下安装python3.8

一、从官网下载安装包 官网地址&#xff1a;https://www.python.org/downloads/ 华为云地址&#xff1a;https://mirrors.huaweicloud.com/python/ 第三方镜像&#xff1a;https://registry.npmmirror.com/binary.html?pathpython/ 注意&#xff1a;从python3.8.10版本开始…

PostgreSQL索引篇 | Hash索引

Hash索引 PostgreSQL版本为8.4.1 &#xff08;本文为《PostgreSQL数据库内核分析》一书的总结笔记&#xff0c;需要电子版的可私信我&#xff09; 在实际的数据库系统中&#xff0c;除了B-Tree外&#xff0c;还有多种数据结构可做索引&#xff0c;Hash表就是其中的一种。通过…

HTML_CSS_盒子模型

盒子模型组成 内容区域&#xff08;comtent&#xff09;内边距区域&#xff08;padding&#xff09;边框区域&#xff08;border&#xff09;外边距区域&#xff08;margin&#xff09; 布局标签 标签&#xff1a;<div> </div> 和 <span> …

Leetcode笔记——二叉树的迭代遍历

中序遍历&#xff1a; 定义一个 保存中间量的栈 和一个 结果数组 1. 模板写法 注释版&#xff1a; 背诵版&#xff1a; 前序遍历 1. 中 右 左 的顺序遍历 2. 模板写法&#xff0c;按中 左 右 的顺序遍历 后序遍历 1. 模板写法&#xff0c;按左 右 中 的顺序遍历 注释版&am…

FreeRTOS操作系统学习——同步互斥与通信

同步&#xff08;Synchronization&#xff09; 同步是一种机制&#xff0c;用于确保多个任务能够按照特定的顺序协调执行或共享数据。当一个任务需要等待其他任务完成某个操作或满足某个条件时&#xff0c;同步机制可以帮助任务进行协调和等待。 在FreeRTOS中&#xff0c;常见…

深入了解 Java 方法和参数的使用方法

Java 方法 简介 方法是一块仅在调用时运行的代码。您可以将数据&#xff08;称为参数&#xff09;传递到方法中。方法用于执行特定的操作&#xff0c;它们也被称为函数。 使用方法的原因 重用代码&#xff1a;定义一次代码&#xff0c;多次使用。提高代码的结构化和可读性。…

Spring官网中查看MongoDB的API文档的详细步骤

目录 Spring官网中查看MongoDB的API文档的详细步骤1、进入 Spring 官网2、选择 Mongodb的文档介绍3、点击API文档4、进入文档查询页面 Spring官网中查看MongoDB的API文档的详细步骤 1、进入 Spring 官网 首先进入Spring的官网&#xff0c;然后点击【Spring Data】 2、选择 Mon…

Java二叉树 (2)

&#x1f435;本篇文章将对二叉树的一些基础操作进行梳理和讲解 一、操作简述 int size(Node root); // 获取树中节点的个数int getLeafNodeCount(Node root); // 获取叶子节点的个数int getKLevelNodeCount(Node root,int k); // 获取第K层节点的个数int getHeight(Node r…

hadoop 总结

1.hadoop 配置文件 core-site hdfs-site yarn-site.xml worker hdfs-site.xml <?xml version"1.0" encoding"UTF-8"?> <?xml-stylesheet type"text/xsl" href"configuration.xsl"?> <configuration><pr…

P1958 上学路线

难度&#xff1a;普及- 题目描述 你所在城市的街道好像一个棋盘&#xff0c;有 a 条南北方向的街道和 b 条东西方向的街道。南北方向的 a 条街道从西到东依次编号为 1 到 a&#xff0c;而东西方向的 b 条街道从南到北依次编号为 1 到 b&#xff0c;南北方向的街道 i 和东西方…

Java中常用的函数式接口

Java中常用的函数式接口 在Java中&#xff0c;函数式接口&#xff08;Functional Interface&#xff09;是一种只有一个抽象方法的接口。Java 8引入了函数式接口和Lambda表达式&#xff0c;使得编写简洁、易读的代码成为可能。Java中常用的函数式接口包括&#xff1a; Runnab…

单数码管(arduino)

1.连接方法 挨个点亮每个灯 #include <Arduino.h>int pin_list[] {4, 5, 19, 21, 22, 2, 15, 18}; int num_pins sizeof(pin_list) / sizeof(pin_list[0]); // 计算数组中的元素数量void setup() {// 设置每个引脚为输出for(int i 0; i < num_pins; i) {pinMode(p…

C语言:ctype和string库中的部分常用函数的应用和实现

在编程过程中&#xff0c;我们经常要处理字符和字符串&#xff0c;C语言标准库中就提供了一系列的库函数&#xff0c;便于我们操作库函数。 字符分类函数 C语⾔中有⼀系列的函数是专⻔做字符分类的&#xff0c;也就是⼀个字符是属于什么类型的字符的。这些函数的使⽤都需要包含…

Springboot 集成kafka 消费者实现ssl方式连接监听消息实现消费

证书准备&#xff1a;springboot集成kafka 消费者实现 如何配置是ssl方式连接的时候需要进行证书的转换。原始的证书是pem, 或者csr方式 和key方式的时候需要转换&#xff0c;因为kafka里面是jks 需要通过openssl进行转换。 证书处理&#xff1a; KeyStore 用于存储客户端的证…