0%

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原本协程。
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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

输出:

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