makara 0.3.5

data/lib/active_record/connection_adapters/makara_postgresql_adapter.rb

@@ -0,0 +1,44 @@
+require 'active_record/connection_adapters/makara_abstract_adapter'
+require 'active_record/connection_adapters/postgresql_adapter'
+
+if ActiveRecord::VERSION::MAJOR >= 4
+
+  module ActiveRecord
+    module ConnectionHandling
+      def makara_postgresql_connection(config)
+        ActiveRecord::ConnectionAdapters::MakaraPostgreSQLAdapter.new(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.makara_postgresql_connection(config)
+        ActiveRecord::ConnectionAdapters::MakaraPostgreSQLAdapter.new(config)
+      end
+    end
+  end
+
+end
+
+module ActiveRecord
+  module ConnectionAdapters
+    class MakaraPostgreSQLAdapter < ActiveRecord::ConnectionAdapters::MakaraAbstractAdapter
+
+      class << self
+        def visitor_for(*args)
+          ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.visitor_for(*args)
+        end
+      end
+
+      protected
+
+      def active_record_connection_for(config)
+        ::ActiveRecord::Base.postgresql_connection(config)
+      end
+
+    end
+  end
+end

data/lib/active_record/connection_adapters/mysql2_makara_adapter.rb

@@ -0,0 +1,44 @@
+require 'active_record/connection_adapters/makara_abstract_adapter'
+require 'active_record/connection_adapters/mysql2_adapter'
+
+if ActiveRecord::VERSION::MAJOR >= 4
+
+  module ActiveRecord
+    module ConnectionHandling
+      def mysql2_makara_connection(config)
+        ActiveRecord::ConnectionAdapters::MakaraMysql2Adapter.new(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.mysql2_makara_connection(config)
+        ActiveRecord::ConnectionAdapters::MakaraMysql2Adapter.new(config)
+      end
+    end
+  end
+
+end
+
+module ActiveRecord
+  module ConnectionAdapters
+    class MakaraMysql2Adapter < ActiveRecord::ConnectionAdapters::MakaraAbstractAdapter
+
+      class << self
+        def visitor_for(*args)
+          ActiveRecord::ConnectionAdapters::Mysql2Adapter.visitor_for(*args)
+        end
+      end
+
+      protected
+
+      def active_record_connection_for(config)
+        ::ActiveRecord::Base.mysql2_connection(config)
+      end
+
+    end
+  end
+end

data/lib/active_record/connection_adapters/postgresql_makara_adapter.rb

@@ -0,0 +1,44 @@
+require 'active_record/connection_adapters/makara_abstract_adapter'
+require 'active_record/connection_adapters/postgresql_adapter'
+
+if ActiveRecord::VERSION::MAJOR >= 4
+
+  module ActiveRecord
+    module ConnectionHandling
+      def postgresql_makara_connection(config)
+        ActiveRecord::ConnectionAdapters::MakaraPostgreSQLAdapter.new(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.postgresql_makara_connection(config)
+        ActiveRecord::ConnectionAdapters::MakaraPostgreSQLAdapter.new(config)
+      end
+    end
+  end
+
+end
+
+module ActiveRecord
+  module ConnectionAdapters
+    class MakaraPostgreSQLAdapter < ActiveRecord::ConnectionAdapters::MakaraAbstractAdapter
+
+      class << self
+        def visitor_for(*args)
+          ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.visitor_for(*args)
+        end
+      end
+
+      protected
+
+      def active_record_connection_for(config)
+        ::ActiveRecord::Base.postgresql_connection(config)
+      end
+
+    end
+  end
+end

data/lib/makara.rb

@@ -0,0 +1,25 @@
+require 'makara/version'
+require 'makara/railtie' if defined?(Rails)
+module Makara
+
+  autoload :Cache,              'makara/cache'
+  autoload :ConfigParser,       'makara/config_parser'
+  autoload :ConnectionWrapper,  'makara/connection_wrapper'
+  autoload :Context,            'makara/context'
+  autoload :ErrorHandler,       'makara/error_handler'
+  autoload :Middleware,         'makara/middleware'
+  autoload :Pool,               'makara/pool'
+  autoload :Proxy,              'makara/proxy'
+
+  module Errors
+    autoload :AllConnectionsBlacklisted,  'makara/errors/all_connections_blacklisted'
+    autoload :BlacklistConnection,        'makara/errors/blacklist_connection'
+    autoload :NoConnectionsAvailable,     'makara/errors/no_connections_available'
+  end
+
+  module Logging
+    autoload :Logger,     'makara/logging/logger'
+    autoload :Subscriber, 'makara/logging/subscriber'
+  end
+
+end

data/lib/makara/cache.rb

@@ -0,0 +1,53 @@
+require 'active_support/core_ext/object/try'
+
+# The Makara Cache should have access to your centralized cache store.
+# It serves the purpose of storing the Makara::Context across requests, servers, etc.
+
+module Makara
+  module Cache
+
+    autoload :MemoryStore, 'makara/cache/memory_store'
+    autoload :NoopStore,   'makara/cache/noop_store'
+
+    class << self
+
+      def store=(store)
+        @store = store
+      end
+
+      def read(key)
+        store.try(:read, key)
+      end
+
+      def write(key, value, ttl)
+        store.try(:write, key, value, :expires_in => ttl.to_i)
+      end
+
+      protected
+
+      def store
+        case @store
+        when :noop, :null
+          @store = Makara::Cache::NoopStore.new
+        when :memory
+          @store = Makara::Cache::MemoryStore.new
+        else
+          if defined?(Rails)
+
+            # in AR3 RAILS_CACHE may not be loaded if the full env is not present
+            # Rails.cache will throw an error because of it.
+            if ActiveRecord::VERSION::MAJOR < 4
+              @store ||= Rails.cache if defined?(RAILS_CACHE)
+            else
+              @store ||= Rails.cache if defined?(Rails)
+            end
+          end
+        end
+
+        @store
+      end
+
+    end
+
+  end
+end

data/lib/makara/cache/memory_store.rb

@@ -0,0 +1,28 @@
+module Makara
+  module Cache
+    class MemoryStore
+
+      def initialize
+        @data = {}
+      end
+
+      def read(key)
+        clean
+        @data[key].try(:[], 0)
+      end
+
+      def write(key, value, options = {})
+        clean
+        @data[key] = [value, Time.now.to_i + (options[:expires_in] || 5).to_i]
+        true
+      end
+
+      protected
+
+      def clean
+        @data.delete_if{|k,v| v[1] <= Time.now.to_i }
+      end
+
+    end
+  end
+end

data/lib/makara/cache/noop_store.rb

@@ -0,0 +1,15 @@
+module Makara
+  module Cache
+    class NoopStore
+
+      def read(key)
+        nil
+      end
+
+      def write(key, value, options = {})
+        nil
+      end
+
+    end
+  end
+end

data/lib/makara/config_parser.rb

@@ -0,0 +1,200 @@
+require 'digest/md5'
+require 'active_support/core_ext/hash/keys'
+require 'active_support/core_ext/hash/except'
+
+# Convenience methods to grab subconfigs out of the primary configuration.
+# Provides a way to generate a consistent ID based on a unique config.
+# Makara configs should be formatted like so:
+# --
+#   top_level: 'variable'
+#   another: 'top level variable'
+#   makara:
+#     master_ttl: 3
+#     blacklist_duration: 20
+#     connections:
+#       - role: 'master'
+#       - role: 'slave'
+#       - role: 'slave'
+#         name: 'slave2'
+
+module Makara
+  class ConfigParser
+
+    DEFAULTS = {
+      :master_ttl => 5,
+      :blacklist_duration => 30,
+      :sticky => true
+    }
+
+    # ConnectionUrlResolver is borrowed from Rails 4-2 since its location and implementation
+    # vary slightly among Rails versions, but the behavior is the same.  Thus, borrowing the
+    # class should be the most future-safe way to parse a database url.
+    #
+    # Expands a connection string into a hash.
+    class ConnectionUrlResolver # :nodoc:
+
+      # == Example
+      #
+      #   url = "postgresql://foo:bar@localhost:9000/foo_test?pool=5&timeout=3000"
+      #   ConnectionUrlResolver.new(url).to_hash
+      #   # => {
+      #     "adapter"  => "postgresql",
+      #     "host"     => "localhost",
+      #     "port"     => 9000,
+      #     "database" => "foo_test",
+      #     "username" => "foo",
+      #     "password" => "bar",
+      #     "pool"     => "5",
+      #     "timeout"  => "3000"
+      #   }
+      def initialize(url)
+        raise "Database URL cannot be empty" if url.blank?
+        @uri     = URI.parse(url)
+        @adapter = @uri.scheme.tr('-', '_')
+        @adapter = "postgresql" if @adapter == "postgres"
+
+        if @uri.opaque
+          @uri.opaque, @query = @uri.opaque.split('?', 2)
+        else
+          @query = @uri.query
+        end
+      end
+
+      # Converts the given URL to a full connection hash.
+      def to_hash
+        config = raw_config.reject { |_,value| value.blank? }
+        config.map { |key,value| config[key] = URI.unescape(value) if value.is_a? String }
+        config
+      end
+
+      private
+
+      def uri
+        @uri
+      end
+
+      # Converts the query parameters of the URI into a hash.
+      #
+      #   "localhost?pool=5&reaping_frequency=2"
+      #   # => { "pool" => "5", "reaping_frequency" => "2" }
+      #
+      # returns empty hash if no query present.
+      #
+      #   "localhost"
+      #   # => {}
+      def query_hash
+        Hash[(@query || '').split("&").map { |pair| pair.split("=") }]
+      end
+
+      def raw_config
+        if uri.opaque
+          query_hash.merge({
+            "adapter"  => @adapter,
+            "database" => uri.opaque })
+        else
+          query_hash.merge({
+            "adapter"  => @adapter,
+            "username" => uri.user,
+            "password" => uri.password,
+            "port"     => uri.port,
+            "database" => database_from_path,
+            "host"     => uri.host })
+        end
+      end
+
+      # Returns name of the database.
+      def database_from_path
+        if @adapter == 'sqlite3'
+          # 'sqlite3:/foo' is absolute, because that makes sense. The
+          # corresponding relative version, 'sqlite3:foo', is handled
+          # elsewhere, as an "opaque".
+
+          uri.path
+        else
+          # Only SQLite uses a filename as the "database" name; for
+          # anything else, a leading slash would be silly.
+
+          uri.path.sub(%r{^/}, "")
+        end
+      end
+    end
+
+    # NOTE: url format must be, e.g.
+    # url: mysql2://...
+    # NOT
+    # url: mysql2_makara://...
+    # since the '_' in the protocol (mysql2_makara) makes the URI invalid
+    # NOTE: Does not use ENV['DATABASE_URL']
+    def self.merge_and_resolve_default_url_config(config)
+      if ENV['DATABASE_URL']
+        Logging::Logger.log "Please rename DATABASE_URL to use in the database.yml", :warn
+      end
+      return config unless config.key?(:url)
+      url = config[:url]
+      url_config = ConnectionUrlResolver.new(url).to_hash
+      url_config = url_config.symbolize_keys
+      url_config.delete(:adapter)
+      config.delete(:url)
+      config.update(url_config)
+    end
+
+    attr_reader :makara_config
+
+    def initialize(config)
+      @config = config.symbolize_keys
+      @makara_config = DEFAULTS.merge(@config[:makara] || {})
+      @makara_config = @makara_config.symbolize_keys
+      @id = @makara_config[:id]
+    end
+
+
+    def id
+      @id ||= begin
+        sorted = recursive_sort(@config)
+        Digest::MD5.hexdigest(sorted.to_s)
+      end
+    end
+
+
+    def master_configs
+      all_configs.
+        select{|config| config[:role] == 'master' }.
+        map{|config| config.except(:role) }
+    end
+
+
+    def slave_configs
+      all_configs.
+        reject{|config| config[:role] == 'master' }.
+        map{|config| config.except(:role) }
+    end
+
+
+    protected
+
+
+    def all_configs
+      @makara_config[:connections].map do |connection|
+        base_config.merge(connection.symbolize_keys)
+      end
+    end
+
+
+    def base_config
+      @base_config ||= DEFAULTS.merge(@config).except(:makara)
+    end
+
+
+    def recursive_sort(thing)
+      return thing.to_s unless thing.include?(Enumerable)
+
+      thing.map do |part|
+        recursive_sort(part)
+      end
+
+      thing.sort_by(&:to_s)
+
+    end
+
+  end
+end