Contents

class Ridgepole::ExecuteExpander
  class Stub
    def method_missing(method_name, *args, &block)
      # Nothing to do
    end
  end

  cattr_accessor :noop,     :instance_writer => false, :instance_reader => false
  cattr_accessor :callback, :instance_writer => false, :instance_reader => false

  class << self
    def without_operation(callback = nil)
      begin
        self.noop = true
        self.callback = callback
        yield
      ensure
        self.noop = false
        self.callback = nil
      end
    end

    def expand_execute(connection)
      return if connection.respond_to?(:execute_with_noop)

      class << connection
        def execute_with_noop(sql, name = nil)
          if Ridgepole::ExecuteExpander.noop
            if (callback = Ridgepole::ExecuteExpander.callback)
              callback.call(sql, name)
            end

            Stub.new
          else
            execute_without_noop(sql, name)
          end
        end
        alias_method_chain :execute, :noop
      end
    end
  end # of class methods
end

require 'active_record/connection_adapters/abstract/schema_statements'

module ActiveRecord::ConnectionAdapters::SchemaStatements
  def add_index_with_noop(table_name, column_name, options = {})
    begin
      add_index_without_noop(table_name, column_name, options)
    rescue => e
      raise e unless Ridgepole::ExecuteExpander.noop
    end
  end
  alias_method_chain :add_index, :noop

  def remove_index_with_noop(table_name, options = {})
    begin
      remove_index_without_noop(table_name, options)
    rescue => e
      raise e unless Ridgepole::ExecuteExpander.noop
    end
  end
  alias_method_chain :remove_index, :noop
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
ridgepole-0.3.4	lib/ridgepole/execute_expander.rb
ridgepole-0.3.3	lib/ridgepole/execute_expander.rb
ridgepole-0.3.2	lib/ridgepole/execute_expander.rb
ridgepole-0.3.1	lib/ridgepole/execute_expander.rb
ridgepole-0.3.0	lib/ridgepole/execute_expander.rb