Go语学习笔记 – 处理超时问题 – Context使用

411次阅读
没有评论

学习笔记,写到哪是哪。

最近看了看go语言Context的使用,觉着很神奇也很便捷。

说到context,我在学习golang的时候看到很多库都用到了,简而言之,context可以处理多个goroutine之间的交互问题。类比于其他语言,你可能需要自己定义一个线程安全对象,通过线程安全对象来实现多个线程间的交互操作。golang直接有个默认包context,可以省掉每次定义的过程,还是很方便的。

具体关于Context的使用,我就不细说了,网上很多。

本文主要将我用context来实现一种超时场景的处理。

demo1

先使用context.WithTimeout方法来实现超时处理。

代码如下:

package main

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

func handle() {
	//构建超时上下文
	_ctx, _cancel := context.WithTimeout(context.Background(), 5*time.Second)
	go work(_ctx)
	time.Sleep(6 * time.Second)
	_cancel()
}


//工作
func work(ctx context.Context) {
	for {
		time.Sleep(1 * time.Second)
		select {
		case <-ctx.Done():
			fmt.Println("work done")
		default:
			fmt.Println("working")
		}
	}
}


func main() {
	handle()
}

代码说明

1、 构建一个5秒超时的上下文传入goroutine,work方法轮询上下文状态。

执行结果

Go语学习笔记 

demo2

先使用context.WithDeadline方法来实现超时处理。

代码如下:

package main

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


func handle1() {
	//构建超时上下文
	_ctx, _cancel := context.WithDeadline(context.Background(), time.Now().Add(5*time.Second))
	go work1(_ctx)
	time.Sleep(6 * time.Second)
	_cancel()
}

//工作1
func work1(ctx context.Context) {
	for {
		time.Sleep(1 * time.Second)
		if _deadline, _a := ctx.Deadline(); _a {
			if time.Now().After(_deadline) {
				fmt.Println("after deadline")
				break
			}
		}
		select {
		case <-ctx.Done():
			fmt.Println("work done")
		default:
			fmt.Println("working")
		}
	}
}

func main() {
	handle1()
}

代码说明

1、注意WithDeadline后面的时间参数方式,和WithTimeout不同。

2、这里在轮训前判断一下是否当前时间已经超过deadline,如果超过了直接跳出。

执行结果

Go语学习笔记

小结

context还有很多用法,需要在项目中多使用,以后分享更多感受。

Go语学习笔记

 

神龙|纯净稳定代理IP免费测试>>>>>>>>天启|企业级代理IP免费测试>>>>>>>>IPIPGO|全球住宅代理IP免费测试

相关文章:

版权声明:Python教程2022-10-19发表,共计1310字。
新手QQ群:570568346,欢迎进群讨论 Python51学习