vault-kv 0.12.0

data/lib/vault/persistent/connection.rb

@@ -0,0 +1,42 @@
+##
+# A Net::HTTP connection wrapper that holds extra information for managing the
+# connection's lifetime.
+
+module Vault
+class PersistentHTTP::Connection # :nodoc:
+
+  attr_accessor :http
+
+  attr_accessor :last_use
+
+  attr_accessor :requests
+
+  attr_accessor :ssl_generation
+
+  def initialize http_class, http_args, ssl_generation
+    @http           = http_class.new(*http_args)
+    @ssl_generation = ssl_generation
+
+    reset
+  end
+
+  def finish
+    @http.finish
+  rescue IOError
+  ensure
+    reset
+  end
+
+  def reset
+    @last_use = PersistentHTTP::EPOCH
+    @requests = 0
+  end
+
+  def ressl ssl_generation
+    @ssl_generation = ssl_generation
+
+    finish
+  end
+
+end
+end

data/lib/vault/persistent/pool.rb

@@ -0,0 +1,48 @@
+module Vault
+class PersistentHTTP::Pool < Vault::ConnectionPool # :nodoc:
+
+  attr_reader :available # :nodoc:
+  attr_reader :key # :nodoc:
+
+  def initialize(options = {}, &block)
+    super
+
+    @available = PersistentHTTP::TimedStackMulti.new(@size, &block)
+    @key = :"current-#{@available.object_id}"
+  end
+
+  def checkin net_http_args
+    stack = Thread.current[@key][net_http_args]
+
+    raise ConnectionPool::Error, 'no connections are checked out' if
+      stack.empty?
+
+    conn = stack.pop
+
+    if stack.empty?
+      @available.push conn, connection_args: net_http_args
+    end
+
+    nil
+  end
+
+  def checkout net_http_args
+    stacks = Thread.current[@key] ||= Hash.new { |h, k| h[k] = [] }
+    stack  = stacks[net_http_args]
+
+    if stack.empty? then
+      conn = @available.pop connection_args: net_http_args
+    else
+      conn = stack.last
+    end
+
+    stack.push conn
+
+    conn
+  end
+
+end
+end
+
+require_relative 'timed_stack_multi'
+

data/lib/vault/persistent/timed_stack_multi.rb

@@ -0,0 +1,70 @@
+module Vault
+class PersistentHTTP::TimedStackMulti < ConnectionPool::TimedStack # :nodoc:
+
+  def initialize(size = 0, &block)
+    super
+
+    @enqueued = 0
+    @ques = Hash.new { |h, k| h[k] = [] }
+    @lru = {}
+    @key = :"connection_args-#{object_id}"
+  end
+
+  def empty?
+    (@created - @enqueued) >= @max
+  end
+
+  def length
+    @max - @created + @enqueued
+  end
+
+  private
+
+  def connection_stored? options = {} # :nodoc:
+    !@ques[options[:connection_args]].empty?
+  end
+
+  def fetch_connection options = {} # :nodoc:
+    connection_args = options[:connection_args]
+
+    @enqueued -= 1
+    lru_update connection_args
+    @ques[connection_args].pop
+  end
+
+  def lru_update connection_args # :nodoc:
+    @lru.delete connection_args
+    @lru[connection_args] = true
+  end
+
+  def shutdown_connections # :nodoc:
+    @ques.each_key do |key|
+      super connection_args: key
+    end
+  end
+
+  def store_connection obj, options = {} # :nodoc:
+    @ques[options[:connection_args]].push obj
+    @enqueued += 1
+  end
+
+  def try_create options = {} # :nodoc:
+    connection_args = options[:connection_args]
+
+    if @created >= @max && @enqueued >= 1
+      oldest, = @lru.first
+      @lru.delete oldest
+      @ques[oldest].pop
+
+      @created -= 1
+    end
+
+    if @created < @max
+      @created += 1
+      lru_update connection_args
+      return @create_block.call(connection_args)
+    end
+  end
+
+end
+end

data/lib/vault/request.rb

@@ -0,0 +1,43 @@
+module Vault
+  class Request
+    attr_reader :client
+
+    def initialize(client)
+      @client = client
+    end
+
+    # @return [String]
+    def to_s
+      "#<#{self.class.name}>"
+    end
+
+    # @return [String]
+    def inspect
+      "#<#{self.class.name}:0x#{"%x" % (self.object_id << 1)}>"
+    end
+
+    private
+
+    include EncodePath
+
+    # Removes the given header fields from options and returns the result. This
+    # modifies the given options in place.
+    #
+    # @param [Hash] options
+    #
+    # @return [Hash]
+    def extract_headers!(options = {})
+      extract = {
+        wrap_ttl: Vault::Client::WRAP_TTL_HEADER,
+      }
+
+      {}.tap do |h|
+        extract.each do |k,v|
+          if options[k]
+            h[v] = options.delete(k)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vault/response.rb

@@ -0,0 +1,89 @@
+module Vault
+  class Response
+    # Defines a new field. This is designed to be used by the subclass as a
+    # mini-DSL.
+    #
+    # @example Default
+    #   field :data
+    #
+    # @example With a mutator
+    #   field :present, as: :present?
+    #
+    # @param n [Symbol] the name of the field
+    # @option opts [Symbol] :as alias for method name
+    #
+    # @!visibility private
+    def self.field(n, opts = {})
+      self.fields[n] = opts
+
+      if opts[:as].nil?
+        attr_reader n
+      else
+        define_method(opts[:as]) do
+          instance_variable_get(:"@#{n}")
+        end
+      end
+    end
+
+    # Returns the list of fields defined on this subclass.
+    # @!visibility private
+    def self.fields
+      @fields ||= {}
+    end
+
+    # Decodes the given object (usually a Hash) into an instance of this class.
+    #
+    # @param object [Hash<Symbol, Object>]
+    def self.decode(object)
+      self.new(object)
+    end
+
+    def initialize(opts = {})
+      # Initialize all fields as nil to start
+      self.class.fields.each do |k, _|
+        instance_variable_set(:"@#{k}", nil)
+      end
+
+      # For each supplied option, set the instance variable if it was defined
+      # as a field.
+      opts.each do |k, v|
+        if self.class.fields.key?(k)
+          opts = self.class.fields[k]
+
+          if (m = opts[:load]) && !v.nil?
+            v = m.call(v)
+          end
+
+          if opts[:freeze]
+            v = v.freeze
+          end
+
+          instance_variable_set(:"@#{k}", v)
+        end
+      end
+    end
+
+    # Create a hash-bashed representation of this response.
+    #
+    # @return [Hash]
+    def to_h
+      self.class.fields.inject({}) do |h, (k, opts)|
+        if opts[:as].nil?
+          h[k] = self.public_send(k)
+        else
+          h[k] = self.public_send(opts[:as])
+        end
+
+        if !h[k].nil? && !h[k].is_a?(Array) && h[k].respond_to?(:to_h)
+          h[k] = h[k].to_h
+        end
+
+        h
+      end
+    end
+
+    def ==(other)
+      self.to_h == other.to_h
+    end
+  end
+end

data/lib/vault/vendor/connection_pool.rb

@@ -0,0 +1,150 @@
+require_relative 'connection_pool/version'
+require_relative 'connection_pool/timed_stack'
+
+
+# Generic connection pool class for e.g. sharing a limited number of network connections
+# among many threads.  Note: Connections are lazily created.
+#
+# Example usage with block (faster):
+#
+#    @pool = ConnectionPool.new { Redis.new }
+#
+#    @pool.with do |redis|
+#      redis.lpop('my-list') if redis.llen('my-list') > 0
+#    end
+#
+# Using optional timeout override (for that single invocation)
+#
+#    @pool.with(:timeout => 2.0) do |redis|
+#      redis.lpop('my-list') if redis.llen('my-list') > 0
+#    end
+#
+# Example usage replacing an existing connection (slower):
+#
+#    $redis = ConnectionPool.wrap { Redis.new }
+#
+#    def do_work
+#      $redis.lpop('my-list') if $redis.llen('my-list') > 0
+#    end
+#
+# Accepts the following options:
+# - :size - number of connections to pool, defaults to 5
+# - :timeout - amount of time to wait for a connection if none currently available, defaults to 5 seconds
+#
+module Vault
+class ConnectionPool
+  DEFAULTS = {size: 5, timeout: 5}
+
+  class Error < RuntimeError
+  end
+
+  def self.wrap(options, &block)
+    Wrapper.new(options, &block)
+  end
+
+  def initialize(options = {}, &block)
+    raise ArgumentError, 'Connection pool requires a block' unless block
+
+    options = DEFAULTS.merge(options)
+
+    @size = options.fetch(:size)
+    @timeout = options.fetch(:timeout)
+
+    @available = TimedStack.new(@size, &block)
+    @key = :"current-#{@available.object_id}"
+  end
+
+if Thread.respond_to?(:handle_interrupt)
+
+  # MRI
+  def with(options = {})
+    Thread.handle_interrupt(Exception => :never) do
+      conn = checkout(options)
+      begin
+        Thread.handle_interrupt(Exception => :immediate) do
+          yield conn
+        end
+      ensure
+        checkin
+      end
+    end
+  end
+
+else
+
+  # jruby 1.7.x
+  def with(options = {})
+    conn = checkout(options)
+    begin
+      yield conn
+    ensure
+      checkin
+    end
+  end
+
+end
+
+  def checkout(options = {})
+    conn = if stack.empty?
+      timeout = options[:timeout] || @timeout
+      @available.pop(timeout: timeout)
+    else
+      stack.last
+    end
+
+    stack.push conn
+    conn
+  end
+
+  def checkin
+    conn = pop_connection # mutates stack, must be on its own line
+    @available.push(conn) if stack.empty?
+
+    nil
+  end
+
+  def shutdown(&block)
+    @available.shutdown(&block)
+  end
+
+  private
+
+  def pop_connection
+    if stack.empty?
+      raise ConnectionPool::Error, 'no connections are checked out'
+    else
+      stack.pop
+    end
+  end
+
+  def stack
+    ::Thread.current[@key] ||= []
+  end
+
+  class Wrapper < ::BasicObject
+    METHODS = [:with, :pool_shutdown]
+
+    def initialize(options = {}, &block)
+      @pool = ::ConnectionPool.new(options, &block)
+    end
+
+    def with(&block)
+      @pool.with(&block)
+    end
+
+    def pool_shutdown(&block)
+      @pool.shutdown(&block)
+    end
+
+    def respond_to?(id, *args)
+      METHODS.include?(id) || with { |c| c.respond_to?(id, *args) }
+    end
+
+    def method_missing(name, *args, &block)
+      with do |connection|
+        connection.send(name, *args, &block)
+      end
+    end
+  end
+end
+end