Class: Nosql::Connection

Inherits:
Object
  • Object
show all
Defined in:
lib/nosql.rb

Constant Summary collapse

EXEC_METHODS = 
[:execute, :exec_query]

Class Method Summary collapse

Class Method Details

.disable! ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/nosql.rb', line 24

def disable!
  #cleanup stubbed functions
  connection = ActiveRecord::Base.connection
  EXEC_METHODS.each do |method_name|
    original_method_name = "original_#{method_name}"
    next unless connection.respond_to?(method_name) && connection.respond_to?(original_method_name)
    connection.class_eval do
      alias_method method_name, original_method_name
    end
  end
end

.enable!(&block) ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/nosql.rb', line 11

def enable!(&block)
  connection = ActiveRecord::Base.connection
  EXEC_METHODS.each do |method_name|
    next unless connection.respond_to?(method_name)
    connection.class_eval do
      alias_method "original_#{method_name}", method_name
      define_method(method_name) do |*args|
        raise Nosql::Error.new(args)
      end
    end
  end
end