Go语学习笔记 - 处理超时问题 - Context使用 | 从零开始Go语言

学习笔记，写到哪是哪。

最近看了看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方法轮询上下文状态。

执行结果

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，如果超过了直接跳出。

执行结果

小结

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