makara 0.3.5

data/gemfiles/ar31.gemfile

@@ -0,0 +1,29 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in makara.gemspec
+gemspec :path => '../'
+
+
+gem 'rake'
+gem 'activerecord', '3.1.8'
+gem 'rspec'
+gem 'rack', '1.6.0'
+gem 'i18n', '~> 0.6.0'
+gem 'mysql2', '~> 0.3.10', :platform => :ruby
+gem 'activerecord-jdbcmysql-adapter', :platform => :jruby
+gem 'activerecord-jdbcpostgresql-adapter', :platform => :jruby
+
+rmajor, rminor, rpatch = RUBY_VERSION.split(/[^\d]/)[0..2].map(&:to_i)
+
+if rmajor == 1 && rminor == 8
+  gem 'timecop', '0.5.9'
+else
+  gem 'timecop'
+end
+
+# 1.8 - 1.9.2
+if rmajor == 1 && (rminor == 8 || (rminor == 9 && rpatch < 3))
+  gem 'pg', '0.17.1', :platform => :ruby
+else
+  gem 'pg', :platform => :ruby
+end

data/gemfiles/ar32.gemfile

@@ -0,0 +1,29 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in makara.gemspec
+gemspec :path => '../'
+
+
+gem 'rake'
+gem 'activerecord', '3.2.6'
+gem 'rspec'
+gem 'rack', '1.6.0'
+gem 'i18n', '~> 0.6.0'
+gem 'mysql2', '~> 0.3.10', :platform => :ruby
+gem 'activerecord-jdbcmysql-adapter', :platform => :jruby
+gem 'activerecord-jdbcpostgresql-adapter', :platform => :jruby
+
+rmajor, rminor, rpatch = RUBY_VERSION.split(/[^\d]/)[0..2].map(&:to_i)
+
+if rmajor == 1 && rminor == 8
+  gem 'timecop', '0.5.9'
+else
+  gem 'timecop'
+end
+
+# 1.8 - 1.9.2
+if rmajor == 1 && (rminor == 8 || (rminor == 9 && rpatch < 3))
+  gem 'pg', '0.17.1', :platform => :ruby
+else
+  gem 'pg', :platform => :ruby
+end

data/gemfiles/ar40.gemfile

@@ -0,0 +1,15 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in makara.gemspec
+gemspec :path => '../'
+
+
+gem 'rake'
+gem 'activerecord', '4.0.0'
+gem 'rspec'
+gem 'rack', '1.6.0'
+gem 'timecop'
+gem 'mysql2', '~> 0.3.10', :platform => :ruby
+gem 'activerecord-jdbcmysql-adapter', :platform => :jruby
+gem 'pg', :platform => :ruby
+gem 'activerecord-jdbcpostgresql-adapter', :platform => :jruby

data/gemfiles/ar41.gemfile

@@ -0,0 +1,15 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in makara.gemspec
+gemspec :path => '../'
+
+
+gem 'rake'
+gem 'activerecord', '4.1.0'
+gem 'rspec'
+gem 'rack', '1.6.0'
+gem 'timecop'
+gem 'mysql2', '~> 0.3.10', :platform => :ruby
+gem 'activerecord-jdbcmysql-adapter', :platform => :jruby
+gem 'pg', :platform => :ruby
+gem 'activerecord-jdbcpostgresql-adapter', :platform => :jruby

data/gemfiles/ar42.gemfile

@@ -0,0 +1,15 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in makara.gemspec
+gemspec :path => '../'
+
+
+gem 'rake'
+gem 'activerecord', '~> 4.2.0'
+gem 'rspec'
+gem 'rack', '1.6.0'
+gem 'timecop'
+gem 'mysql2', '~> 0.3.10', :platform => :ruby
+gem 'activerecord-jdbcmysql-adapter', :platform => :jruby
+gem 'pg', :platform => :ruby
+gem 'activerecord-jdbcpostgresql-adapter', :platform => :jruby

data/lib/active_record/connection_adapters/jdbcmysql_makara_adapter.rb

@@ -0,0 +1,25 @@
+require 'active_record/connection_adapters/makara_abstract_adapter'
+require 'active_record/connection_adapters/jdbcmysql_adapter'
+require 'active_record/connection_adapters/mysql2_makara_adapter'
+
+if ActiveRecord::VERSION::MAJOR >= 4
+
+  module ActiveRecord
+    module ConnectionHandling
+      def jdbcmysql_makara_connection(config)
+        mysql2_makara_connection(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.jdbcmysql_makara_connection(config)
+        self.mysql2_makara_connection(config)
+      end
+    end
+  end
+
+end

data/lib/active_record/connection_adapters/jdbcpostgresql_makara_adapter.rb

@@ -0,0 +1,25 @@
+require 'active_record/connection_adapters/makara_abstract_adapter'
+require 'active_record/connection_adapters/jdbcpostgresql_adapter'
+require 'active_record/connection_adapters/postgresql_makara_adapter'
+
+if ActiveRecord::VERSION::MAJOR >= 4
+
+  module ActiveRecord
+    module ConnectionHandling
+      def jdbcpostgresql_makara_connection(config)
+        postgresql_makara_connection(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.jdbcpostgresql_makara_connection(config)
+        self.postgresql_makara_connection(config)
+      end
+    end
+  end
+
+end

data/lib/active_record/connection_adapters/makara_abstract_adapter.rb

@@ -0,0 +1,209 @@
+require 'active_record'
+
+module ActiveRecord
+  module ConnectionAdapters
+    class MakaraAbstractAdapter < ::Makara::Proxy
+
+
+      class ErrorHandler < ::Makara::ErrorHandler
+
+
+        HARSH_ERRORS = [
+          'ActiveRecord::RecordNotUnique',
+          'ActiveRecord::InvalidForeignKey',
+          'Makara::Errors::BlacklistConnection'
+        ].map(&:freeze).freeze
+
+
+        CONNECTION_MATCHERS = [
+          /(closed|lost|no|terminating|terminated)\s?([^\s]+)?\sconnection/,
+          /gone away/,
+          /connection[^:]+refused/,
+          /could not connect/,
+          /can\'t connect/,
+          /cannot connect/,
+          /connection[^:]+closed/,
+          /can\'t get socket descriptor/,
+          /connection to [a-z0-9.]+:[0-9]+ refused/,
+          /timeout expired/,
+          /could not translate host name/,
+          /timeout waiting for a response/,
+          /the database system is (starting|shutting)/
+        ].map(&:freeze).freeze
+
+
+        def handle(connection)
+
+          yield
+
+        rescue Exception => e
+          # do it via class name to avoid version-specific constant dependencies
+          case e.class.name
+          when *harsh_errors
+            harshly(e)
+          else
+            if connection_message?(e) || custom_error_message?(connection, e)
+              gracefully(connection, e)
+            else
+              harshly(e)
+            end
+          end
+
+        end
+
+
+        def harsh_errors
+          HARSH_ERRORS
+        end
+
+
+        def connection_matchers
+          CONNECTION_MATCHERS
+        end
+
+
+        def connection_message?(message)
+          message = message.to_s.downcase
+
+          case message
+          when *connection_matchers
+            true
+          else
+            false
+          end
+        end
+
+
+        def custom_error_message?(connection, message)
+          custom_error_matchers = connection._makara_custom_error_matchers
+          return false if custom_error_matchers.empty?
+
+          message = message.to_s
+
+          custom_error_matchers.each do |matcher|
+
+            if matcher.is_a?(String)
+
+              # accept strings that look like "/.../" as a regex
+              if matcher =~ /^\/(.+)\/([a-z])?$/
+
+                options = $2 ? (($2.include?('x') ? Regexp::EXTENDED : 0) |
+                          ($2.include?('i') ? Regexp::IGNORECASE : 0) |
+                          ($2.include?('m') ? Regexp::MULTILINE : 0)) : 0
+
+                matcher = Regexp.new($1, options)
+              end
+            end
+
+            return true if matcher === message
+          end
+
+          false
+        end
+
+
+      end
+
+
+      hijack_method :execute, :select_rows, :exec_query, :transaction
+      send_to_all :connect, :reconnect!, :verify!, :clear_cache!, :reset!
+
+      SQL_MASTER_MATCHERS           = [/\A\s*select.+for update\Z/i, /select.+lock in share mode\Z/i].map(&:freeze).freeze
+      SQL_SLAVE_MATCHERS            = [/\A\s*select\s/i].map(&:freeze).freeze
+      SQL_ALL_MATCHERS              = [/\A\s*set\s/i].map(&:freeze).freeze
+      SQL_SKIP_STICKINESS_MATCHERS  = [/\A\s*show\s([\w]+\s)?(field|table|database|schema|view|index)(es|s)?/i, /\A\s*(set|describe|explain|pragma)\s/i].map(&:freeze).freeze
+
+
+      def sql_master_matchers
+        SQL_MASTER_MATCHERS
+      end
+
+
+      def sql_slave_matchers
+        SQL_SLAVE_MATCHERS
+      end
+
+
+      def sql_all_matchers
+        SQL_ALL_MATCHERS
+      end
+
+
+      def sql_skip_stickiness_matchers
+        SQL_SKIP_STICKINESS_MATCHERS
+      end
+
+
+      def initialize(config)
+        @error_handler = ::ActiveRecord::ConnectionAdapters::MakaraAbstractAdapter::ErrorHandler.new
+        super(config)
+      end
+
+
+      protected
+
+
+      def appropriate_connection(method_name, args)
+        if needed_by_all?(method_name, args)
+
+          handling_an_all_execution(method_name) do
+            # slave pool must run first.
+            @slave_pool.provide_each do |con|
+              hijacked do
+                yield con
+              end
+            end
+
+            @master_pool.provide_each do |con|
+              hijacked do
+                yield con
+              end
+            end
+          end
+
+        else
+
+          super(method_name, args) do |con|
+            yield con
+          end
+
+        end
+      end
+
+
+      def should_stick?(method_name, args)
+        sql = args.first.to_s
+        return false if sql_skip_stickiness_matchers.any?{|m| sql =~ m }
+        super
+      end
+
+
+      def needed_by_all?(method_name, args)
+        sql = args.first.to_s
+        return true if sql_all_matchers.any?{|m| sql =~ m }
+        false
+      end
+
+
+      def needs_master?(method_name, args)
+        sql = args.first.to_s
+        return true if sql_master_matchers.any?{|m| sql =~ m }
+        return false if sql_slave_matchers.any?{|m| sql =~ m }
+        true
+      end
+
+
+      def connection_for(config)
+        config = Makara::ConfigParser.merge_and_resolve_default_url_config(config)
+        active_record_connection_for(config)
+      end
+
+
+      def active_record_connection_for(config)
+        raise NotImplementedError
+      end
+
+
+    end
+  end
+end

data/lib/active_record/connection_adapters/makara_jdbcmysql_adapter.rb

@@ -0,0 +1,25 @@
+require 'active_record/connection_adapters/makara_abstract_adapter'
+require 'active_record/connection_adapters/jdbcmysql_adapter'
+require 'active_record/connection_adapters/makara_mysql2_adapter'
+
+if ActiveRecord::VERSION::MAJOR >= 4
+
+  module ActiveRecord
+    module ConnectionHandling
+      def makara_jdbcmysql_connection(config)
+        makara_mysql2_connection(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.makara_jdbcmysql_connection(config)
+        self.makara_mysql2_connection(config)
+      end
+    end
+  end
+
+end

data/lib/active_record/connection_adapters/makara_jdbcpostgresql_adapter.rb

@@ -0,0 +1,25 @@
+require 'active_record/connection_adapters/makara_abstract_adapter'
+require 'active_record/connection_adapters/jdbcpostgresql_adapter'
+require 'active_record/connection_adapters/makara_postgresql_adapter'
+
+if ActiveRecord::VERSION::MAJOR >= 4
+
+  module ActiveRecord
+    module ConnectionHandling
+      def makara_jdbcpostgresql_connection(config)
+        makara_postgresql_connection(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.makara_jdbcpostgresql_connection(config)
+        self.makara_postgresql_connection(config)
+      end
+    end
+  end
+
+end

data/lib/active_record/connection_adapters/makara_mysql2_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 makara_mysql2_connection(config)
+        ActiveRecord::ConnectionAdapters::MakaraMysql2Adapter.new(config)
+      end
+    end
+  end
+
+else
+
+  module ActiveRecord
+    class Base
+      def self.makara_mysql2_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