dd-vault 0.12.0

data/lib/vault/errors.rb

@@ -0,0 +1,72 @@
+module Vault
+  class VaultError < RuntimeError; end
+
+  class MissingTokenError < VaultError
+    def initialize
+      super <<-EOH
+Missing Vault token! I cannot make requests to Vault without a token. Please
+set a Vault token in the client:
+
+    Vault.token = "42d1dee5-eb6e-102c-8d23-cc3ba875da51"
+
+or authenticate with Vault using the Vault CLI:
+
+    $ vault auth ...
+
+or set the environment variable $VAULT_TOKEN to the token value:
+
+    $ export VAULT_TOKEN="..."
+
+Please refer to the documentation for more examples.
+EOH
+    end
+  end
+
+  class HTTPConnectionError < VaultError
+    attr_reader :address
+
+    def initialize(address, exception)
+      @address = address
+      @exception = exception
+
+      super <<-EOH
+The Vault server at `#{address}' is not currently
+accepting connections. Please ensure that the server is running and that your
+authentication information is correct.
+
+The original error was `#{exception.class}'. Additional information (if any) is
+shown below:
+
+    #{exception.message}
+
+Please refer to the documentation for more help.
+EOH
+    end
+
+    def original_exception
+      @exception
+    end
+  end
+
+  class HTTPError < VaultError
+    attr_reader :address, :response, :code, :errors
+
+    def initialize(address, response, errors = [])
+      @address, @response, @errors = address, response, errors
+      @code  = response.code.to_i
+      errors = errors.map { |error| "  * #{error}" }
+
+      super <<-EOH
+The Vault server at `#{address}' responded with a #{code}.
+Any additional information the server supplied is shown below:
+
+#{errors.join("\n").rstrip}
+
+Please refer to the documentation for help.
+EOH
+    end
+  end
+
+  class HTTPClientError < HTTPError; end
+  class HTTPServerError < HTTPError; end
+end

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