skynet超时机制实现

skynet.call在设计的时候,是没有考虑超时的问题的。云风的解释是,加入之后会使得在其构建的系统增加不必要的复杂度。这个解释也是合理的,但是还是免不了有需求要使用到超时的机制。

举个简单例子,skynet实现的web服务,一些http请求的时候,如果没有超时机制,http请求就会一直占用。

云风给出了一种解决方案,就是增加一个代理服务,让其负责转发,然后代理服务增加超时的检查。也是一种不错的方式。

skynet 是支持协程的,应该说snlua服务,都是基于协程实现的。基于协程的特性,本身就能实现超时检测,逻辑也很简单。

执行函数的时候,调用 do_func_or_timeout 传入 msecfunc 。函数就会启用(fork)两个协程,并且让当前线程wait:

  • 第一个协程,sleep msec,之后检测 is_wait 如果为false,结束;否则wakeup原本协程;
  • 第二个线程则是执行 func函数,正常返回后,检查is_wait 如果为false,结束;否则wakeup原本协程。
local Skynet = require "skynet"
local M = {}

local STATUS_SUCC = 0
local STATUS_TIMEOUT = 1

function M.do_func_or_timeout(msec, func, ...)
    local args = ...
    local ctx = {
        is_wait = true,
        corout  = coroutine.running(),
    }

    Skynet.fork(function()
        Skynet.sleep(msec)  -- 超时等待msec 1/00秒
        if not ctx.is_wait then
            return
        end

        ctx.is_wait = false
        ctx.status = STATUS_TIMEOUT
        Skynet.wakeup(ctx.corout)
    end)

    Skynet.fork(function()
        local status = STATUS_SUCC
        local ret = func(args)

        if not ctx.is_wait then
            return
        end

        ctx.is_wait = false
        ctx.status  = status
        ctx.ret     = ret

        Skynet.wakeup(ctx.corout)
    end)
    Skynet.wait()
    return ctx.status, ctx.ret
end

print("== start test ==", Skynet.time())
local status, ret = M.do_func_or_timeout(200, function()
    Skynet.sleep(300)
    return "ok"
end)
print("== status ret 1 ==", Skynet.time(), status, ret)

local status, ret = M.do_func_or_timeout(600, function()
    Skynet.sleep(300)
    return "ok"
end)
print("== status ret 2 ==", Skynet.time(), status, ret)

return M

输出:

== start test ==        1631807810.97
== status ret 1 ==      1631807812.97   1       nil
== status ret 2 ==      1631807815.97   0       ok

posted on   Ron Ngai  阅读(542)  评论(0编辑  收藏  举报

编辑推荐:
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 提示词工程——AI应用必不可少的技术
· Open-Sora 2.0 重磅开源!
· 周边上新:园子的第一款马克杯温暖上架

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

点击右上角即可分享
微信分享提示