From 3300bf6d0a7914d1e0914f90fdd5d968fcbfe010 Mon Sep 17 00:00:00 2001
From: Benoit Germain <bnt.germain@gmail.com>
Date: Sat, 16 Nov 2013 10:39:34 +0100
Subject: lanes.genlock() improvements and fix

   * Fixed function returned by lanes.genlock() not handling numeric keys properly when release lock
   * Enable lanes.genlock() to attempt lock with an optional "try" mode
---
 src/lanes.lua | 54 ++++++++++++++++++++++++++++++++++--------------------
 1 file changed, 34 insertions(+), 20 deletions(-)

(limited to 'src')

diff --git a/src/lanes.lua b/src/lanes.lua
index 0e27ea6..1a1c252 100644
--- a/src/lanes.lua
+++ b/src/lanes.lua
@@ -614,28 +614,42 @@ end -- settings.with_timers
 -- acquire (+M) and release (-M). For binary locks, use M==1.
 --
 -- PUBLIC LANES API
-local function genlock( linda, key, N )
-    linda:limit(key,N)
-    linda:set(key,nil)  -- clears existing data
+local genlock = function( linda, key, N)
+	linda:limit( key, N)
+	linda:set( key, nil)  -- clears existing data
 
-    --
-    -- [true [, ...]= trues(uint)
-    --
-    local function trues(n)
-        if n>0 then return true,trues(n-1) end
-    end
+	--
+	-- [true [, ...]= trues(uint)
+	--
+	local function trues( n)
+		if n > 0 then
+			return true, trues( n - 1)
+		end
+	end
 
-    return
-    function(M)
-        if M>0 then
-            -- 'nil' timeout allows 'key' to be numeric
-            linda:send( nil, key, trues(M) )    -- suspends until been able to push them
-        else
-            for i=1,-M do
-                linda:receive( key )
-            end
-        end
-    end
+	-- use an optimized version for case N == 1
+	return (N == 1) and
+	function( M, mode_)
+		local timeout = (mode_ == "try") and 0 or nil
+		if M > 0 then
+			-- 'nil' timeout allows 'key' to be numeric
+			return linda:send( timeout, key, true)    -- suspends until been able to push them
+		else
+			local k = linda:receive( nil, key)
+			return k and true or false
+		end
+	end
+	or
+	function( M, mode_)
+		local timeout = (mode_ == "try") and 0 or nil
+		if M > 0 then
+			-- 'nil' timeout allows 'key' to be numeric
+			return linda:send( timeout, key, trues(M))    -- suspends until been able to push them
+		else
+			local k = linda:receive( nil, linda.batched, key, -M)
+			return k and true or false
+		end
+	end
 end
 
 
-- 
cgit v1.2.3-55-g6feb