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 2021-09-17 00:25  Ron Ngai  阅读(409)  评论(0编辑  收藏  举报

导航