Contents

# Encoding: utf-8

require 'timeout'

module Etcd
  module Mod
    # implement etcd lock module
    module Lock
      def mod_lock_endpoint
        '/mod/v2/lock'
      end

      def acquire_lock(key, ttl, opts = {})
        path = mod_lock_endpoint + key + "?ttl=#{ttl}"
        timeout = opts[:timeout] || 60
        Timeout.timeout(timeout) do
          api_execute(path, :post, params: opts)
        end
      end

      def renew_lock(key, ttl, opts = {})
        unless opts.key?(:index) || opts.key?(:value)
          fail ArgumentError, 'You mast pass index or value'
        end
        path = mod_lock_endpoint + key + "?ttl=#{ttl}"
        timeout = opts[:timeout] || 60
        Timeout.timeout(timeout) do
          api_execute(path, :put, params: opts).body
        end
      end

      def get_lock(key, opts = {})
        api_execute(mod_lock_endpoint + key, :get, params: opts).body
      end

      def delete_lock(key, opts = {})
        unless opts.key?(:index) || opts.key?(:value)
          fail ArgumentError, 'You must pass index or value'
        end
        api_execute(mod_lock_endpoint + key, :delete, params: opts)
      end

      # rubocop:disable RescueException
      def lock(key, ttl, opts = {})
        acquire_lock('/' + key, ttl, opts)
        index = get_lock('/' + key, field: index)
        begin
          yield key
        rescue Exception => e
          raise e
        ensure
          delete_lock(key, index: index)
        end
      end
      # rubocop:enable RescueException

      alias_method :retrive_lock, :get_lock
      alias_method :release_lock, :delete_lock
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
etcd-0.2.1	lib/etcd/mod/lock.rb
etcd-0.2.0.beta.1	lib/etcd/mod/lock.rb