Remove code which generates a deprecation warning

use monotonic time for retry timeout check

.github/workflows/CI.yml

@@ -0,0 +1,38 @@
+name: CI
+
+on:
+  push:
+    branches:
+      - master
+  pull_request:
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        ruby:
+          - '2.5'
+          - '2.6'
+          - '2.7'
+          - '3.0'
+          - ruby-head
+    continue-on-error: ${{ matrix.ruby == 'ruby-head' }}
+    services:
+      memcached:
+        image: memcached
+        ports:
+          - 11211:11211
+      redis:
+        image: redis
+        ports:
+          - 6379:6379
+    steps:
+      - uses: actions/checkout@v2
+      - uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: ${{ matrix.ruby }}
+          bundler-cache: true
+      - run: |
+          bundle exec rake

.rubocop.yml

@@ -74,7 +74,7 @@ Style/SpaceInsideBrackets:
 Style/AndOr:
   Enabled: false
 
-Style/TrailingComma:
+Style/TrailingCommaInLiteral:
   Enabled: true
 
 Style/SpaceBeforeComma:
@@ -98,7 +98,7 @@ Style/SpaceAfterColon:
 Style/SpaceAfterComma:
   Enabled: true
 
-Style/SpaceAfterControlKeyword:
+Style/SpaceAroundKeyword:
   Enabled: true
 
 Style/SpaceAfterNot:
@@ -163,7 +163,7 @@ Style/StringLiterals:
   EnforcedStyle: double_quotes
 
 Metrics/CyclomaticComplexity:
-  Max: 8
+  Max: 10
 
 Metrics/LineLength:
   Max: 128
@@ -214,3 +214,6 @@ Metrics/ParameterLists:
 
 Metrics/PerceivedComplexity:
   Enabled: false
+
+Style/Documentation:
+  Enabled: false

.travis.yml

@@ -1,6 +0,0 @@
-language: ruby
-rvm:
-  - 2.2.0
-services:
-  - memcached
-  - redis-server

CHANGELOG.md

@@ -1,3 +1,55 @@
+## 0.4.0
+
+- Monotonic clock for locks, avoiding issues with DST (thanks @doits)
+- Pooled connection support (thanks @mlarraz)
+- Switch to Github actions for tests (thanks @mlarraz)
+- Update supported Ruby versions (thanks @mlarraz & @pat)
+
+## 0.3.4
+
+- Support for connection pooling when using memcached locks, via `with` blocks using Dalli (thanks to Lev).
+
+## 0.3.3
+
+- Default TTL for keys to allow for short-lived locking keys (thanks to Ian Remillard) without leaking memory.
+- Vastly improve initial lock acquisition, especially on Redis (thanks to Jeremy Wadscak).
+
+## 0.3.2
+
+- Custom lock tokens (thanks to avokhmin).
+
+## 0.3.1
+
+- Slight memory leak fix.
+
+## 0.3.0
+
+- Dramatically simplify the interface by forcing clients to specify the key & resources at lock initialization instead of every method call.
+
+## 0.2.3
+
+- Clarify documentation further with respect to semaphores.
+
+## 0.2.2
+
+- Fix bug with refresh - typo would've prevented real use.
+- Clean up code.
+- Improve documentation a bit.
+- 100% test coverage.
+
+## 0.2.1
+
+- Fix bug when dealing with real-world Redis error conditions.
+
+## 0.2.0
+
+- Refactor class methods into instance methods to simplify implementation.
+- Increase thread safety with Memcached implementation.
+
+## 0.1.3
+
+- Properly throw Suo::LockClientError when the connection itself fails (Memcache server not reachable, etc.)
+
 ## 0.1.2
 
 - Fix retry_timeout to properly use the full time (was being calculated incorrectly).
@@ -5,7 +57,7 @@
 
 ## 0.1.1
 
-- Use [MessagePack](https://github.com/msgpack/msgpack-ruby) for semaphore serialization.
+- Use [MessagePack](https://github.com/msgpack/msgpack-ruby) for lock serialization.
 
 ## 0.1.0
 

README.md

@@ -1,8 +1,8 @@
-# Suo [![Build Status](https://travis-ci.org/nickelser/suo.svg?branch=master)](https://travis-ci.org/nickelser/suo) [![Gem Version](https://badge.fury.io/rb/suo.svg)](http://badge.fury.io/rb/suo)
+# Suo [![Build Status](https://github.com/nickelser/suo/workflows/CI/badge.svg)](https://github.com/nickelser/suo/actions?query=workflow%3ACI) [![Code Climate](https://codeclimate.com/github/nickelser/suo/badges/gpa.svg)](https://codeclimate.com/github/nickelser/suo) [![Gem Version](https://badge.fury.io/rb/suo.svg)](http://badge.fury.io/rb/suo)
 
 :lock: Distributed semaphores using Memcached or Redis in Ruby.
 
-Suo provides a very performant distributed lock solution using Compare-And-Set (`CAS`) commands in Memcached, and `WATCH/MULTI` in Redis.
+Suo provides a very performant distributed lock solution using Compare-And-Set (`CAS`) commands in Memcached, and `WATCH/MULTI` in Redis. It allows locking both single exclusion (like a mutex - sharing one resource), as well as multiple resources.
 
 ## Installation
 
@@ -18,34 +18,75 @@ gem 'suo'
 
 ```ruby
 # Memcached
-suo = Suo::Client::Memcached.new(connection: "127.0.0.1:11211")
+suo = Suo::Client::Memcached.new("foo_resource", connection: "127.0.0.1:11211")
 
 # Redis
-suo = Suo::Client::Redis.new(connection: {host: "10.0.1.1"})
+suo = Suo::Client::Redis.new("baz_resource", connection: {host: "10.0.1.1"})
 
 # Pre-existing client
-suo = Suo::Client::Memcached.new(client: some_dalli_client)
+suo = Suo::Client::Memcached.new("bar_resource", client: some_dalli_client)
 
-suo.lock("some_key") do
+suo.lock do
   # critical code here
   @puppies.pet!
 end
 
-2.times do
+# The resources argument is the number of resources the semaphore will allow to lock (defaulting to one - a mutex)
-  Thread.new do
+suo = Suo::Client::Memcached.new("bar_resource", client: some_dalli_client, resources: 2)
-    # second argument is the number of resources - so this will run twice
+
-    suo.lock("other_key", 2, timeout: 0.5) { puts "Will run twice!" }
+Thread.new { suo.lock { puts "One"; sleep 2 } }
+Thread.new { suo.lock { puts "Two"; sleep 2 } }
+Thread.new { suo.lock { puts "Three" } }
+
+# will print "One" "Two", but not "Three", as there are only 2 resources
+
+# custom acquisition timeouts (time to acquire)
+suo = Suo::Client::Memcached.new("protected_key", client: some_dalli_client, acquisition_timeout: 1) # in seconds
+
+# manually locking/unlocking
+# the return value from lock without a block is a unique token valid only for the current lock
+# which must be unlocked manually
+token = suo.lock
+foo.baz!
+suo.unlock(token)
+
+# custom stale lock expiration (cleaning of dead locks)
+suo = Suo::Client::Redis.new("other_key", client: some_redis_client, stale_lock_expiration: 60*5)
+```
+
+### Stale locks
+
+"Stale locks" - those acquired more than `stale_lock_expiration` (defaulting to 3600 or one hour) ago - are automatically cleared during any operation on the key (`lock`, `unlock`, `refresh`). The `locked?` method will not return true if only stale locks exist, but will not modify the key itself.
+
+To re-acquire a lock in the middle of a block, you can use the refresh method on client.
+
+```ruby
+suo = Suo::Client::Redis.new("foo")
+
+# lock is the same token as seen in the manual example, above
+suo.lock do |token|
+  5.times do
+    baz.bar!
+    suo.refresh(token)
   end
 end
 ```
 
+### Time To Live
+
+```ruby
+Suo::Client::Redis.new("bar_resource", ttl: 60) #ttl in seconds
+```
+
+A key representing a set of lockable resources is removed once the last resource lock is released and the `ttl` time runs out. When another lock is acquired and the key has been removed the key has to be recreated.
+
+
 ## TODO
- - better stale key handling (refresh blocks)
  - more race condition tests
 
 ## History
 
-View the [changelog](https://github.com/nickelser/suo/blob/master/CHANGELOG.md)
+View the [changelog](https://github.com/nickelser/suo/blob/master/CHANGELOG.md).
 
 ## Contributing
 

lib/suo.rb

@@ -1,2 +1,15 @@
+require "securerandom"
+require "monitor"
+
+require "dalli"
+
+require "redis"
+
+require "msgpack"
+
 require "suo/version"
-require "suo/clients"
+
+require "suo/errors"
+require "suo/client/base"
+require "suo/client/memcached"
+require "suo/client/redis"

lib/suo/client/base.rb

@@ -2,195 +2,177 @@ module Suo
   module Client
     class Base
       DEFAULT_OPTIONS = {
-        retry_timeout: 0.1,
+        acquisition_timeout: 0.1,
-        retry_delay: 0.01,
+        acquisition_delay: 0.01,
-        stale_lock_expiration: 3600
+        stale_lock_expiration: 3600,
+        resources: 1,
+        ttl: 60,
       }.freeze
 
-      def initialize(options = {})
+      BLANK_STR = "".freeze
-        @options = self.class.merge_defaults(options)
+
+      attr_accessor :client, :key, :resources, :options
+
+      include MonitorMixin
+
+      def initialize(key, options = {})
+        fail "Client required" unless options[:client]
+
+        @options = DEFAULT_OPTIONS.merge(options)
+        @retry_count = (@options[:acquisition_timeout] / @options[:acquisition_delay].to_f).ceil
+        @client = @options[:client]
+        @resources = @options[:resources].to_i
+        @key = key
+
+        super() # initialize Monitor mixin for thread safety
       end
 
-      def lock(key, resources = 1, options = {})
+      def lock(custom_token = nil)
-        options = self.class.merge_defaults(@options.merge(options))
+        token = acquire_lock(custom_token)
-        token = self.class.lock(key, resources, options)
 
-        if token
+        if block_given? && token
           begin
-            yield if block_given?
+            yield
           ensure
-            self.class.unlock(key, token, options)
+            unlock(token)
           end
-
-          true
         else
-          false
+          token
         end
       end
 
-      def locked?(key, resources = 1)
+      def locked?
-        self.class.locked?(key, resources, @options)
+        locks.size >= resources
       end
 
-      class << self
+      def locks
-        def lock(key, resources = 1, options = {})
+        val, _ = get
-          options = merge_defaults(options)
+        cleared_locks = deserialize_and_clear_locks(val)
-          acquisition_token = nil
-          token = SecureRandom.base64(16)
 
-          retry_with_timeout(key, options) do
+        cleared_locks
-            val, cas = get(key, options)
-
-            if val.nil?
-              set_initial(key, options)
-              next
       end
 
-            locks = deserialize_and_clear_locks(val, options)
+      def refresh(token)
+        retry_with_timeout do
+          val, cas = get
 
-            if locks.size < resources
+          cas = initial_set if val.nil?
-              add_lock(locks, token)
 
-              newval = serialize_locks(locks)
+          cleared_locks = deserialize_and_clear_locks(val)
 
-              if set(key, newval, cas, options)
+          refresh_lock(cleared_locks, token)
-                acquisition_token = token
-                break
-              end
-            end
-          end
 
-          acquisition_token
+          break if set(serialize_locks(cleared_locks), cas, expire: cleared_locks.empty?)
-        end
-
-        def locked?(key, resources = 1, options = {})
-          locks(key, options).size >= resources
-        end
-
-        def locks(key, options)
-          options = merge_defaults(options)
-          val, _ = get(key, options)
-          locks = deserialize_locks(val)
-
-          locks
-        end
-
-        def refresh(key, acquisition_token, options = {})
-          options = merge_defaults(options)
-
-          retry_with_timeout(key, options) do
-            val, cas = get(key, options)
-
-            if val.nil?
-              set_initial(key, options)
-              next
-            end
-
-            locks = deserialize_and_clear_locks(val, options)
-
-            refresh_lock(locks, acquisition_token)
-
-            break if set(key, serialize_locks(locks), cas, options)
         end
       end
 
-        def unlock(key, acquisition_token, options = {})
+      def unlock(token)
-          options = merge_defaults(options)
+        return unless token
 
-          return unless acquisition_token
+        retry_with_timeout do
-
+          val, cas = get
-          retry_with_timeout(key, options) do
-            val, cas = get(key, options)
 
           break if val.nil?
 
-            locks = deserialize_and_clear_locks(val, options)
+          cleared_locks = deserialize_and_clear_locks(val)
 
-            acquisition_lock = remove_lock(locks, acquisition_token)
+          acquisition_lock = remove_lock(cleared_locks, token)
 
           break unless acquisition_lock
-            break if set(key, serialize_locks(locks), cas, options)
+          break if set(serialize_locks(cleared_locks), cas, expire: cleared_locks.empty?)
         end
-        rescue FailedToAcquireLock => _ # rubocop:disable Lint/HandleExceptions
+      rescue LockClientError => _ # rubocop:disable Lint/HandleExceptions
         # ignore - assume success due to optimistic locking
       end
 
-        def clear(key, options = {}) # rubocop:disable Lint/UnusedMethodArgument
+      def clear
         fail NotImplementedError
       end
 
-        def merge_defaults(options = {})
-          options = self::DEFAULT_OPTIONS.merge(options)
-
-          fail "Client required" unless options[:client]
-
-          options[:retry_count] = (options[:retry_timeout] / options[:retry_delay].to_f).ceil
-
-          options
-        end
-
       private
 
-        def get(key, options) # rubocop:disable Lint/UnusedMethodArgument
+      attr_accessor :retry_count
+
+      def acquire_lock(token = nil)
+        token ||= SecureRandom.base64(16)
+
+        retry_with_timeout do
+          val, cas = get
+
+          cas = initial_set if val.nil?
+
+          cleared_locks = deserialize_and_clear_locks(val)
+
+          if cleared_locks.size < resources
+            add_lock(cleared_locks, token)
+
+            newval = serialize_locks(cleared_locks)
+
+            return token if set(newval, cas)
+          end
+        end
+
+        nil
+      end
+
+      def get
         fail NotImplementedError
       end
 
-        def set(key, newval, oldval, options) # rubocop:disable Lint/UnusedMethodArgument
+      def set(newval, cas) # rubocop:disable Lint/UnusedMethodArgument
         fail NotImplementedError
       end
 
-        def set_initial(key, options) # rubocop:disable Lint/UnusedMethodArgument
+      def initial_set(val = BLANK_STR) # rubocop:disable Lint/UnusedMethodArgument
         fail NotImplementedError
       end
 
-        def synchronize(key, options)
+      def synchronize
-          yield(key, options)
+        mon_synchronize { yield }
       end
 
-        def retry_with_timeout(key, options)
+      def retry_with_timeout
-          start = Time.now.to_f
+        start = Process.clock_gettime(Process::CLOCK_MONOTONIC)
 
-          options[:retry_count].times do
+        retry_count.times do
-            if options[:retry_timeout]
+          elapsed = Process.clock_gettime(Process::CLOCK_MONOTONIC) - start
-              now = Time.now.to_f
+          break if elapsed >= options[:acquisition_timeout]
-              break if now - start > options[:retry_timeout]
-            end
 
-            synchronize(key, options) do
+          synchronize do
             yield
           end
 
-            sleep(rand(options[:retry_delay] * 1000).to_f / 1000)
+          sleep(rand(options[:acquisition_delay] * 1000).to_f / 1000)
         end
       rescue => _
-          raise FailedToAcquireLock
+        raise LockClientError
       end
 
       def serialize_locks(locks)
         MessagePack.pack(locks.map { |time, token| [time.to_f, token] })
       end
 
-        def deserialize_and_clear_locks(val, options)
+      def deserialize_and_clear_locks(val)
-          clear_expired_locks(deserialize_locks(val), options)
+        clear_expired_locks(deserialize_locks(val))
       end
 
       def deserialize_locks(val)
-          unpacked = (val.nil? || val == "") ? [] : MessagePack.unpack(val)
+        unpacked = (val.nil? || val == BLANK_STR) ? [] : MessagePack.unpack(val)
 
         unpacked.map do |time, token|
           [Time.at(time), token]
         end
-        rescue EOFError => _
+      rescue EOFError, MessagePack::MalformedFormatError => _
         []
       end
 
-        def clear_expired_locks(locks, options)
+      def clear_expired_locks(locks)
         expired = Time.now - options[:stale_lock_expiration]
         locks.reject { |time, _| time < expired }
       end
 
-        def add_lock(locks, token)
+      def add_lock(locks, token, time = Time.now.to_f)
-          locks << [Time.now.to_f, token]
+        locks << [time, token]
       end
 
       def remove_lock(locks, acquisition_token)
@@ -200,8 +182,7 @@ module Suo
 
       def refresh_lock(locks, acquisition_token)
         remove_lock(locks, acquisition_token)
-          add_lock(locks, token)
+        add_lock(locks, acquisition_token)
-        end
       end
     end
   end

lib/suo/client/errors.rb

@@ -1,7 +0,0 @@
-module Suo
-  module Client
-    module Errors
-      class FailedToAcquireLock < StandardError; end
-    end
-  end
-end

lib/suo/client/memcached.rb

@@ -1,29 +1,34 @@
 module Suo
   module Client
     class Memcached < Base
-      def initialize(options = {})
+      def initialize(key, options = {})
         options[:client] ||= Dalli::Client.new(options[:connection] || ENV["MEMCACHE_SERVERS"] || "127.0.0.1:11211")
         super
       end
 
-      class << self
+      def clear
-        def clear(key, options = {})
+        @client.with { |client| client.delete(@key) }
-          options = merge_defaults(options)
-          options[:client].delete(key)
       end
 
       private
 
-        def get(key, options)
+      def get
-          options[:client].get_cas(key)
+        @client.with { |client| client.get_cas(@key) }
       end
 
-        def set(key, newval, cas, options)
+      def set(newval, cas, expire: false)
-          options[:client].set_cas(key, newval, cas)
+        if expire
+          @client.with { |client| client.set_cas(@key, newval, cas, @options[:ttl]) }
+        else
+          @client.with { |client| client.set_cas(@key, newval, cas) }
+        end
       end
 
-        def set_initial(key, options)
+      def initial_set(val = BLANK_STR)
-          options[:client].set(key, "")
+        @client.with do |client|
+          client.set(@key, val)
+          _val, cas = client.get_cas(@key)
+          cas
         end
       end
     end

lib/suo/client/redis.rb

@@ -1,42 +1,54 @@
 module Suo
   module Client
     class Redis < Base
-      def initialize(options = {})
+      OK_STR = "OK".freeze
+
+      def initialize(key, options = {})
         options[:client] ||= ::Redis.new(options[:connection] || {})
         super
       end
 
-      class << self
+      def clear
-        def clear(key, options = {})
+        with { |r| r.del(@key) }
-          options = merge_defaults(options)
-          options[:client].del(key)
       end
 
       private
 
-        def get(key, options)
+      def with(&block)
-          [options[:client].get(key), nil]
+        if @client.respond_to?(:with)
+          @client.with(&block)
+        else
+          yield @client
+        end
       end
 
-        def set(key, newval, _, options)
+      def get
-          ret = options[:client].multi do |multi|
+        [with { |r| r.get(@key) }, nil]
-            multi.set(key, newval)
       end
 
-          ret[0] == "OK"
+      def set(newval, _, expire: false)
+        ret = with do |r|
+          r.multi do |rr|
+            if expire
+              rr.setex(@key, @options[:ttl], newval)
+            else
+              rr.set(@key, newval)
+            end
+          end
         end
 
-        def synchronize(key, options)
+        ret && ret[0] == OK_STR
-          options[:client].watch(key) do
-            yield
       end
+
+      def synchronize
+        with { |r| r.watch(@key) { yield } }
       ensure
-          options[:client].unwatch
+        with { |r| r.unwatch }
       end
 
-        def set_initial(key, options)
+      def initial_set(val = BLANK_STR)
-          options[:client].set(key, "")
+        set(val, nil)
-        end
+        nil
       end
     end
   end

lib/suo/clients.rb

@@ -1,14 +0,0 @@
-require "securerandom"
-require "monitor"
-
-require "dalli"
-require "dalli/cas/client"
-
-require "redis"
-
-require "msgpack"
-
-require "suo/client/errors"
-require "suo/client/base"
-require "suo/client/memcached"
-require "suo/client/redis"

lib/suo/errors.rb

@@ -0,0 +1,3 @@
+module Suo
+  class LockClientError < StandardError; end
+end

lib/suo/version.rb

@@ -1,3 +1,3 @@
 module Suo
-  VERSION = "0.1.2"
+  VERSION = "0.4.0".freeze
 end

suo.gemspec

@@ -9,23 +9,25 @@ Gem::Specification.new do |spec|
   spec.authors       = ["Nick Elser"]
   spec.email         = ["nick.elser@gmail.com"]
 
-  spec.summary       = %q(Distributed semaphores using Memcached or Redis.)
+  spec.summary       = %q(Distributed locks (mutexes & semaphores) using Memcached or Redis.)
-  spec.description   = %q(Distributed semaphores using Memcached or Redis.)
+  spec.description   = %q(Distributed locks (mutexes & semaphores) using Memcached or Redis.)
   spec.homepage      = "https://github.com/nickelser/suo"
   spec.license       = "MIT"
 
   spec.files         = `git ls-files -z`.split("\x0")
   spec.bindir        = "bin"
-  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
+  spec.required_ruby_version = ">= 2.5"
+
   spec.add_dependency "dalli"
   spec.add_dependency "redis"
   spec.add_dependency "msgpack"
 
-  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "bundler"
-  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rake", "~> 13.0"
-  spec.add_development_dependency "rubocop", "~> 0.30.0"
+  spec.add_development_dependency "rubocop", "~> 0.49.0"
   spec.add_development_dependency "minitest", "~> 5.5.0"
+  spec.add_development_dependency "codeclimate-test-reporter", "~> 0.4.7"
 end

test/client_test.rb

@@ -3,119 +3,361 @@ require "test_helper"
 TEST_KEY = "suo_test_key".freeze
 
 module ClientTests
-  def test_requires_client
+  def client(options = {})
-    exception = assert_raises(RuntimeError) do
+    @client.class.new(options[:key] || TEST_KEY, options.merge(client: @client.client))
-      @klass.lock(TEST_KEY, 1)
   end
 
-    assert_equal "Client required", exception.message
+  def test_throws_failed_error_on_bad_client
+    assert_raises(Suo::LockClientError) do
+      client = @client.class.new(TEST_KEY, client: {})
+      client.lock
+    end
   end
 
-  def test_class_single_resource_locking
+  def test_single_resource_locking
-    lock1 = @klass.lock(TEST_KEY, 1, client: @klass_client)
+    lock1 = @client.lock
     refute_nil lock1
 
-    locked = @klass.locked?(TEST_KEY, 1, client: @klass_client)
+    locked = @client.locked?
     assert_equal true, locked
 
-    lock2 = @klass.lock(TEST_KEY, 1, client: @klass_client)
+    lock2 = @client.lock
     assert_nil lock2
 
-    @klass.unlock(TEST_KEY, lock1, client: @klass_client)
+    @client.unlock(lock1)
 
-    locked = @klass.locked?(TEST_KEY, 1, client: @klass_client)
+    locked = @client.locked?
 
     assert_equal false, locked
   end
 
-  def test_class_multiple_resource_locking
+  def test_lock_with_custom_token
-    lock1 = @klass.lock(TEST_KEY, 2, client: @klass_client)
+    token = 'foo-bar'
+    lock  = @client.lock token
+    assert_equal lock, token
+  end
+
+  def test_empty_lock_on_invalid_data
+    @client.send(:initial_set, "bad value")
+    assert_equal false, @client.locked?
+  end
+
+  def test_clear
+    lock1 = @client.lock
     refute_nil lock1
 
-    locked = @klass.locked?(TEST_KEY, 2, client: @klass_client)
+    @client.clear
-    assert_equal false, locked
 
-    lock2 = @klass.lock(TEST_KEY, 2, client: @klass_client)
+    assert_equal false, @client.locked?
+  end
+
+  def test_multiple_resource_locking
+    @client = client(resources: 2)
+
+    lock1 = @client.lock
+    refute_nil lock1
+
+    assert_equal false, @client.locked?
+
+    lock2 = @client.lock
     refute_nil lock2
 
-    locked = @klass.locked?(TEST_KEY, 2, client: @klass_client)
+    assert_equal true, @client.locked?
-    assert_equal true, locked
 
-    @klass.unlock(TEST_KEY, lock1, client: @klass_client)
+    @client.unlock(lock1)
 
-    locked = @klass.locked?(TEST_KEY, 1, client: @klass_client)
+    assert_equal false, @client.locked?
-    assert_equal true, locked
 
-    @klass.unlock(TEST_KEY, lock2, client: @klass_client)
+    assert_equal 1, @client.locks.size
 
-    locked = @klass.locked?(TEST_KEY, 1, client: @klass_client)
+    @client.unlock(lock2)
-    assert_equal false, locked
+
+    assert_equal false, @client.locked?
+    assert_equal 0, @client.locks.size
   end
 
-  def test_instance_single_resource_locking
+  def test_block_single_resource_locking
     locked = false
 
-    @client.lock(TEST_KEY, 1) { locked = true }
+    @client.lock { locked = true }
 
     assert_equal true, locked
   end
 
-  def test_instance_unlocks_on_exception
+  def test_block_unlocks_on_exception
     assert_raises(RuntimeError) do
-      @client.lock(TEST_KEY, 1) { fail "Test" }
+      @client.lock{ fail "Test" }
     end
 
-    locked = @klass.locked?(TEST_KEY, 1, client: @klass_client)
+    assert_equal false, @client.locked?
-    assert_equal false, locked
   end
 
-  def test_instance_multiple_resource_locking
+  def test_readme_example
+    output = Queue.new
+    @client = client(resources: 2)
+    threads = []
+
+    threads << Thread.new { @client.lock { output << "One"; sleep 0.5 } }
+    threads << Thread.new { @client.lock { output << "Two"; sleep 0.5 } }
+    sleep 0.1
+    threads << Thread.new { @client.lock { output << "Three" } }
+
+    threads.each(&:join)
+
+    ret = []
+
+    ret << (output.size > 0 ? output.pop : nil)
+    ret << (output.size > 0 ? output.pop : nil)
+
+    ret.sort!
+
+    assert_equal 0, output.size
+    assert_equal %w(One Two), ret
+    assert_equal false, @client.locked?
+  end
+
+  def test_block_multiple_resource_locking
     success_counter = Queue.new
     failure_counter = Queue.new
 
+    @client = client(acquisition_timeout: 0.9, resources: 50)
+
     100.times.map do |i|
       Thread.new do
-        success = @client.lock(TEST_KEY, 50, retry_timeout: 0.5) do
+        success = @client.lock do
-          sleep(2)
+          sleep(3)
           success_counter << i
         end
 
         failure_counter << i unless success
       end
-    end.map(&:join)
+    end.each(&:join)
 
     assert_equal 50, success_counter.size
     assert_equal 50, failure_counter.size
+    assert_equal false, @client.locked?
   end
 
-  def test_instance_multiple_resource_locking_longer_timeout
+  def test_block_multiple_resource_locking_longer_timeout
     success_counter = Queue.new
     failure_counter = Queue.new
 
+    @client = client(acquisition_timeout: 3, resources: 50)
+
     100.times.map do |i|
       Thread.new do
-        success = @client.lock(TEST_KEY, 50, retry_timeout: 2) do
+        success = @client.lock do
           sleep(0.5)
           success_counter << i
         end
 
         failure_counter << i unless success
       end
-    end.map(&:join)
+    end.each(&:join)
 
     assert_equal 100, success_counter.size
     assert_equal 0, failure_counter.size
+    assert_equal false, @client.locked?
+  end
+
+  def test_unstale_lock_acquisition
+    success_counter = Queue.new
+    failure_counter = Queue.new
+
+    @client = client(stale_lock_expiration: 0.5)
+
+    t1 = Thread.new { @client.lock { sleep 0.6; success_counter << 1 } }
+    sleep 0.3
+    t2 = Thread.new do
+      locked = @client.lock { success_counter << 1 }
+      failure_counter << 1 unless locked
+    end
+
+    [t1, t2].each(&:join)
+
+    assert_equal 1, success_counter.size
+    assert_equal 1, failure_counter.size
+    assert_equal false, @client.locked?
+  end
+
+  def test_stale_lock_acquisition
+    success_counter = Queue.new
+    failure_counter = Queue.new
+
+    @client = client(stale_lock_expiration: 0.5)
+
+    t1 = Thread.new { @client.lock { sleep 0.6; success_counter << 1 } }
+    sleep 0.55
+    t2 = Thread.new do
+      locked = @client.lock { success_counter << 1 }
+      failure_counter << 1 unless locked
+    end
+
+    [t1, t2].each(&:join)
+
+    assert_equal 2, success_counter.size
+    assert_equal 0, failure_counter.size
+    assert_equal false, @client.locked?
+  end
+
+  def test_refresh
+    @client = client(stale_lock_expiration: 0.5)
+
+    lock1 = @client.lock
+
+    assert_equal true, @client.locked?
+
+    @client.refresh(lock1)
+
+    assert_equal true, @client.locked?
+
+    sleep 0.55
+
+    assert_equal false, @client.locked?
+
+    lock2 = @client.lock
+
+    @client.refresh(lock1)
+
+    assert_equal true, @client.locked?
+
+    @client.unlock(lock1)
+
+    # edge case with refresh lock in the middle
+    assert_equal true, @client.locked?
+
+    @client.clear
+
+    assert_equal false, @client.locked?
+
+    @client.refresh(lock2)
+
+    assert_equal true, @client.locked?
+
+    @client.unlock(lock2)
+
+    # now finally unlocked
+    assert_equal false, @client.locked?
+  end
+
+  def test_block_refresh
+    success_counter = Queue.new
+    failure_counter = Queue.new
+
+    @client = client(stale_lock_expiration: 0.5)
+
+    t1 = Thread.new do
+      @client.lock do |token|
+        sleep 0.6
+        @client.refresh(token)
+        sleep 1
+        success_counter << 1
+      end
+    end
+
+    t2 = Thread.new do
+      sleep 0.8
+      locked = @client.lock { success_counter << 1 }
+      failure_counter << 1 unless locked
+    end
+
+    [t1, t2].each(&:join)
+
+    assert_equal 1, success_counter.size
+    assert_equal 1, failure_counter.size
+    assert_equal false, @client.locked?
+  end
+
+  def test_refresh_multi
+    success_counter = Queue.new
+    failure_counter = Queue.new
+
+    @client = client(stale_lock_expiration: 0.5, resources: 2)
+
+    t1 = Thread.new do
+      @client.lock do |token|
+        sleep 0.4
+        @client.refresh(token)
+        success_counter << 1
+        sleep 0.5
+      end
+    end
+
+    t2 = Thread.new do
+      sleep 0.55
+      locked = @client.lock do
+        success_counter << 1
+        sleep 0.5
+      end
+
+      failure_counter << 1 unless locked
+    end
+
+    t3 = Thread.new do
+      sleep 0.75
+      locked = @client.lock { success_counter << 1 }
+      failure_counter << 1 unless locked
+    end
+
+    [t1, t2, t3].each(&:join)
+
+    assert_equal 2, success_counter.size
+    assert_equal 1, failure_counter.size
+    assert_equal false, @client.locked?
+  end
+
+  def test_increment_reused_client
+    i = 0
+
+    threads = 2.times.map do
+      Thread.new do
+        @client.lock { i += 1 }
+      end
+    end
+
+    threads.each(&:join)
+
+    assert_equal 2, i
+    assert_equal false, @client.locked?
+  end
+
+  def test_increment_new_client
+    i = 0
+
+    threads = 2.times.map do
+      Thread.new do
+        # note this is the method that generates a *new* client
+        client.lock { i += 1 }
+      end
+    end
+
+    threads.each(&:join)
+
+    assert_equal 2, i
+    assert_equal false, @client.locked?
   end
 end
 
 class TestBaseClient < Minitest::Test
   def setup
-    @klass = Suo::Client::Base
+    @client = Suo::Client::Base.new(TEST_KEY, client: {})
   end
 
   def test_not_implemented
     assert_raises(NotImplementedError) do
-      @klass.send(:get, TEST_KEY, {})
+      @client.send(:get)
+    end
+
+    assert_raises(NotImplementedError) do
+      @client.send(:set, "", "")
+    end
+
+    assert_raises(NotImplementedError) do
+      @client.send(:initial_set)
+    end
+
+    assert_raises(NotImplementedError) do
+      @client.send(:clear)
     end
   end
 end
@@ -124,13 +366,13 @@ class TestMemcachedClient < Minitest::Test
   include ClientTests
 
   def setup
-    @klass = Suo::Client::Memcached
+    @dalli = Dalli::Client.new("127.0.0.1:11211")
-    @client = @klass.new
+    @client = Suo::Client::Memcached.new(TEST_KEY)
-    @klass_client = Dalli::Client.new("127.0.0.1:11211")
+    teardown
   end
 
   def teardown
-    @klass_client.delete(TEST_KEY)
+    @dalli.delete(TEST_KEY)
   end
 end
 
@@ -138,13 +380,13 @@ class TestRedisClient < Minitest::Test
   include ClientTests
 
   def setup
-    @klass = Suo::Client::Redis
+    @redis = Redis.new
-    @client = @klass.new
+    @client = Suo::Client::Redis.new(TEST_KEY)
-    @klass_client = Redis.new
+    teardown
   end
 
   def teardown
-    @klass_client.del(TEST_KEY)
+    @redis.del(TEST_KEY)
   end
 end
 

test/test_helper.rb

@@ -1,9 +1,13 @@
 $LOAD_PATH.unshift File.expand_path("../../lib", __FILE__)
 
+if ENV["CODECLIMATE_REPO_TOKEN"]
+  require "codeclimate-test-reporter"
+  CodeClimate::TestReporter.start
+end
+
 require "suo"
 require "thread"
 require "minitest/autorun"
 require "minitest/benchmark"
 
 ENV["SUO_TEST"] = "true"
-