keep the key never overrides

lib/resty/lock.lua

@@ -94,6 +94,7 @@ function _M.new(_, dict_name, opts)
         step = opts.step
         ratio = opts.ratio
         max_step = opts.max_step
+        safe_add = opts.safe_add
     end
 
     if not exptime then
@@ -107,6 +108,7 @@ function _M.new(_, dict_name, opts)
     local self = {
         cdata = cdata,
         dict = dict,
+        dict_add = safe_add and dict.safe_add or dict.add,
         timeout = timeout or 5,
         exptime = exptime,
         step = step or 0.001,
@@ -128,7 +130,7 @@ function _M.lock(self, key)
         return nil, "locked"
     end
     local exptime = self.exptime
-    local ok, err = dict:add(key, true, exptime)
+    local ok, err = self.dict_add(dict, key, true, exptime)
     if ok then
         cdata.key_id = ref_obj(key)
         if not shdict_mt then
@@ -154,7 +156,7 @@ function _M.lock(self, key)
         elapsed = elapsed + step
         timeout = timeout - step
 
-        local ok, err = dict:add(key, true, exptime)
+        local ok, err = self.dict_add(dict, key, true, exptime)
         if ok then
             cdata.key_id = ref_obj(key)
             if not shdict_mt then

t/sanity.t

@@ -468,3 +468,34 @@ lock 2: unlock: nil, unlocked
 --- no_error_log
 [error]
 
+
+
+=== TEST 14: serial lock and unlock with safe_add
+--- http_config eval: $::HttpConfig
+--- config
+    location = /t {
+        content_by_lua '
+            local lock = require "resty.lock"
+            for i = 1, 2 do
+                local lock = lock:new("cache_locks", {safe_add = true})
+                local elapsed, err = lock:lock("foo")
+                ngx.say("lock: ", elapsed, ", ", err)
+                local ok, err = lock:unlock()
+                if not ok then
+                    ngx.say("failed to unlock: ", err)
+                end
+                ngx.say("unlock: ", ok)
+            end
+        ';
+    }
+--- request
+GET /t
+--- response_body
+lock: 0, nil
+unlock: 1
+lock: 0, nil
+unlock: 1
+
+--- no_error_log
+[error]
+