lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-24 12:20:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

queue lock request in socket, fix issue #51

Browse Source
This commit is contained in:
云风
2013-10-28 13:35:28 +08:00
parent 95584a0a88
commit 3327bb7f82
1 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
lualib/socket.lua
View File

@@ -230,16 +230,15 @@ function socket.lock(id)
local s = socket_pool[id]
assert(s)
local lock_set = s.lock
local co = coroutine.running()
if not lock_set then
lock_set = {}
s.lock = lock_set
lock_set[co] = true
elseif next(lock_set) == nil then
lock_set[co] = true
end
local co = coroutine.running()
if #lock_set == 0 then
lock_set[1] = co
else
assert(lock_set[co] == nil)
lock_set[co] = true
table.insert(lock_set, co)
skynet.wait()
end
end
@@ -250,18 +249,11 @@ function socket.unlock(id)
local lock_set = s.lock
assert(lock_set)
local co = coroutine.running()
assert(lock_set[co])
lock_set[co] = nil
while true do
co = next(lock_set)
if co == nil then
break
end
if skynet.wakeup(co) then
break
else
lock_set[co] = nil
end
assert(lock_set[1] == co)
table.remove(lock_set,1)
co = lock_set[1]
if co then
skynet.wakeup(co)
end
end