文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

golang设置超时

2023-05-16 14:14

关注

Golang是一种高效的编程语言,被广泛应用于各种领域的开发中。在一些特定场景下,我们需要设置超时来避免程序的死锁或卡死。在本文中,我将详细介绍Golang如何设置超时,以及几种设置超时的方式。

一、使用channel

在Golang中,可以使用channel实现超时的功能。我们可以定义一个带有timeout参数的channel,并在程序执行后按照timeout时间取出数据。

例如,以下是一个简单的示例代码:

package main

import (
    "fmt"
    "time"
)

func main() {
    ch := make(chan string, 1)

    go func() {
        time.Sleep(time.Second * 3)
        ch <- "task completed"
    }()

    select {
    case res := <-ch:
        fmt.Println(res)
    case <-time.After(time.Second * 2):
        fmt.Println("timeout")
    }
}

我们可以看到,创建了一个带有timeout参数的channel,程序执行后会开启一个并发协程,该协程执行任务需要3秒钟的时间。在select语句中,我们监听了两个channel。如果在timeout时间内收到了数据,即任务执行完毕,程序会打印出"task completed"。如果在timeout时间内没有收到数据,程序会打印出"timeout"。

二、使用context

在Golang中,context包提供了一种用来传递请求范围数据、取消信号和超时时间的方法。我们可以使用WithTimeout函数来创建一个携带超时时间的Context对象。

例如,以下是一个简单的示例代码:

package main

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

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
    defer cancel()

    go func(ctx context.Context) {
        time.Sleep(time.Second * 3)
        select {
        case <-ctx.Done():
            fmt.Println(ctx.Err())
        default:
            fmt.Println("task completed")
        }
    }(ctx)

    select {
    case <-ctx.Done():
        fmt.Println(ctx.Err())
    }
}

在该示例中,我们使用WithTimeout函数创建了一个超时时间为2秒的Context对象。在开启的并发协程中,我们阻塞等待2秒钟,如果在2秒钟内任务执行完毕,协程会打印出"task completed",否则会打印出context超时错误信息。在select语句中,我们监听了Context对象的Done() channel。如果收到了Done() channel的信号,程序就会取消任务并输出错误信息。

三、使用cancel

在一些特定情况下,我们可能需要通过设置cancel函数来取消某个任务的执行。例如,在从网络请求数据的场景中,可能需要设置一个超时时间,如果在超时时间内没有请求到数据,我们需要取消请求。

例如,以下是一个简单的示例代码:

package main

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

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
    defer cancel()

    ch := make(chan string, 1)

    go func(ctx context.Context) {
        time.Sleep(time.Second * 3)
        ch <- "task completed"
    }(ctx)

    select {
    case res := <-ch:
        fmt.Println(res)
    case <-ctx.Done():
        fmt.Println(ctx.Err())
    }

    cancel()
}

在该示例中,我们创建了一个携带超时时间的Context对象,并开启了一个并发协程执行任务。在select语句中,我们对任务和Context对象的Done() channel进行监听,如果任务执行完毕,则会输出"task completed",否则输出超时错误信息。在程序最后调用cancel()函数,用以取消任务执行。如果任务在超时时间内完成,程序会正常退出。否则,在cancel()函数被执行前,程序一直阻塞在select语句中。

四、使用deadline

除了使用WithTimeout函数,context包还提供了另一种超时的方式,即使用WithDeadline函数。与WithTimeout函数不同的是,WithDeadline函数的参数是一个截止时间,而不是超时时间。在截止时间之前任务执行完毕,程序会正常退出。

例如,以下是一个简单的示例代码:

package main

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

func main() {
    ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Second*2))
    defer cancel()

    go func(ctx context.Context) {
        time.Sleep(time.Second * 3)
        select {
        case <-ctx.Done():
            fmt.Println(ctx.Err())
        default:
            fmt.Println("task completed")
        }
    }(ctx)

    select {
    case <-ctx.Done():
        fmt.Println(ctx.Err())
    }
}

在该示例中,我们使用WithDeadline函数创建了一个截止时间为2秒后的Context对象。如果在截止时间之前任务执行完毕,程序会输出"task completed",否则输出超时错误信息。在select语句中,我们同样监听了Context对象的Done() channel。如果收到Done() channel的信号,程序会输出错误信息。

总结:

Golang中提供了多种设置超时的方式。通过使用channel、context、cancel和deadline等方法,我们可以避免程序死锁或卡死的问题,并保证程序运行的高效性和稳定性。在实际开发中,根据不同的需求和场景,我们可以选择合适的方法来设置超时。

以上就是golang设置超时的详细内容,更多请关注编程网其它相关文章!

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯