skynet超时机制实现
skynet.call
在设计的时候,是没有考虑超时的问题的。云风的解释是,加入之后会使得在其构建的系统增加不必要的复杂度。这个解释也是合理的,但是还是免不了有需求要使用到超时的机制。
举个简单例子,skynet实现的web服务,一些http请求的时候,如果没有超时机制,http请求就会一直占用。
云风给出了一种解决方案,就是增加一个代理服务,让其负责转发,然后代理服务增加超时的检查。也是一种不错的方式。
skynet
是支持协程的,应该说snlua服务,都是基于协程实现的。基于协程的特性,本身就能实现超时检测,逻辑也很简单。
执行函数的时候,调用 do_func_or_timeout
传入 msec
和 func
。函数就会启用(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
作者:Ron Ngai
出处:http://rondsny.github.io
关于作者:断码码农一枚。
欢迎转载,但未经作者同意须在文章页面明显位置给出原文连接
如有问题,可以通过rondsny#gmail.com 联系我,非常感谢。