whichdigital/active-rest-client

View on GitHub
lib/active_rest_client/connection_manager.rb

Summary

Maintainability
A
0 mins
Test Coverage
module ActiveRestClient
  class ConnectionManager
    def self.reset!
      Thread.current[:_connections]={}
    end

    def self.get_connection(base_url)
      raise Exception.new("Nil base URL passed to ConnectionManager.get_connection") if base_url.nil?
      Thread.current[:_connections] ||= {}
      Thread.current[:_connections][base_url] ||= Connection.new(base_url)
      Thread.current[:_connections][base_url]
    end

    def self.find_connection_for_url(url)
      Thread.current[:_connections] ||= {}
      found = Thread.current[:_connections].keys.detect {|key| url[0,key.length] == key}
      Thread.current[:_connections][found] if found
    end

    def self.in_parallel(base_url)
      begin
        require 'typhoeus'
        require 'typhoeus/adapters/faraday'
      rescue LoadError
        raise MissingOptionalLibraryError.new("To call '::ActiveRestClient::ConnectionManager.in_parallel' you must include the gem 'Typhoeus' in your Gemfile.")
      end
      session = ConnectionManager.get_connection(base_url).session
      session.in_parallel do
        yield
      end
    end

  end

  class MissingOptionalLibraryError < StandardError ; end
end