From b1ec1f4d18b2135f3f0d3b34a29f88eadf8ab928 Mon Sep 17 00:00:00 2001
From: Cloud Wu <cloudwu@gmail.com>
Date: Thu, 7 Jun 2018 16:10:57 +0800
Subject: [PATCH] token bugfix

---
 lualib/skynet.lua | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/lualib/skynet.lua b/lualib/skynet.lua
index 64d16a3e..a9a2b5c2 100644
--- a/lualib/skynet.lua
+++ b/lualib/skynet.lua
@@ -228,7 +228,8 @@ end
 function skynet.sleep(ti, token)
 	local session = c.intcommand("TIMEOUT",ti)
 	assert(session)
-	local succ, ret = suspend_sleep(session, token or coroutine.running())
+	token = token or coroutine.running()
+	local succ, ret = suspend_sleep(session, token)
 	sleep_session[token] = nil
 	if succ then
 		return
@@ -247,7 +248,7 @@ end
 function skynet.wait(token)
 	local session = c.genid()
 	token = token or coroutine.running()
-	local ret, msg = suspend_sleep(session, token or coroutine.running())
+	local ret, msg = suspend_sleep(session, token)
 	sleep_session[token] = nil
 	session_id_coroutine[session] = nil
 end