更新

1 files changed, 49 insertions, 2 deletions

diff --git a/server-beta/src/task.lua b/server-beta/src/task.lua
index fa44b7c6..23b6ae2f 100644
--- a/server-beta/src/task.lua
+++ b/server-beta/src/task.lua
@@ -1,7 +1,54 @@
-local thread = require 'bee.thread'
+local timer  = require 'timer'
 
 ---@class task
 local m = {}
 m.type = 'task'
 
-return m
+--- 设置错误处理器
+---@param errHandle function {comment = '当有错误发生时，会以错误堆栈为参数调用该函数'}
+function m.setErrorHandle(errHandle)
+    m.errorHandle = errHandle
+end
+
+--- 创建一个任务
+function m.create(callback)
+    coroutine.create(callback)
+end
+
+--- 休眠一段时间
+---@param time number
+function m.sleep(time)
+    local co, main = coroutine.running()
+    if main then
+        if m.errorHandle then
+            m.errorHandle(debug.traceback('Cant sleep in main thread'))
+        end
+        return
+    end
+    timer.wait(time, function ()
+        local suc, err = coroutine.resume(co)
+        if not suc and m.errHandle then
+            m.errHandle(debug.traceback(co, err))
+        end
+    end)
+    coroutine.yield()
+end
+
+--- 等待直到唤醒
+---@param waker function
+function m.wait(waker)
+    local co, main = coroutine.running()
+    if main then
+        if m.errorHandle then
+            m.errorHandle(debug.traceback('Cant wait in main thread'))
+        end
+        return
+    end
+    waker(function ()
+        local suc, err = coroutine.resume(co)
+        if not suc and m.errHandle then
+            m.errHandle(debug.traceback(co, err))
+        end
+    end)
+    coroutine.yield()
+end