Class: Octopus::ScopeProxy

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

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(shard, klass) ⇒ ScopeProxy

Returns a new instance of ScopeProxy.



4
5
6
7
 
# File 'lib/octopus/scope_proxy.rb', line 4

def initialize(shard, klass)
  @shard = shard
  @klass = klass
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(method, *args, &block) ⇒ Object 



27
28
29
30
31
32
33
34
 
# File 'lib/octopus/scope_proxy.rb', line 27

def method_missing(method, *args, &block)
  @klass.connection.run_queries_on_shard(@shard) do
    @klass = @klass.send(method, *args, &block)
  end

  return @klass if @klass.is_a?(ActiveRecord::Base) or @klass.is_a?(Array) or @klass.is_a?(Fixnum) or @klass.nil?
  return self
end

Instance Attribute Details

#klass ⇒ Object

Returns the value of attribute klass.



2
3
4
 
# File 'lib/octopus/scope_proxy.rb', line 2

def klass
  @klass
end

#shard ⇒ Object

Returns the value of attribute shard.



2
3
4
 
# File 'lib/octopus/scope_proxy.rb', line 2

def shard
  @shard
end

Instance Method Details

#==(other) ⇒ Object 



36
37
38
39
 
# File 'lib/octopus/scope_proxy.rb', line 36

def ==(other)
  @shard == other.shard
  @klass == other.klass
end

#connection ⇒ Object 



22
23
24
25
 
# File 'lib/octopus/scope_proxy.rb', line 22

def connection
  @klass.connection().current_shard = @shard
  @klass.connection()
end

#transaction(options = {}, &block) ⇒ Object

Transaction Method send all queries to a specified shard.



16
17
18
19
20
 
# File 'lib/octopus/scope_proxy.rb', line 16

def transaction(options = {}, &block)
  @klass.connection.run_queries_on_shard(@shard) do
    @klass = @klass.connection().transaction(options, &block)
  end      
end

#using(shard) ⇒ Object 



9
10
11
12
13
 
# File 'lib/octopus/scope_proxy.rb', line 9

def using(shard)
  raise "Nonexistent Shard Name: #{shard}" if @klass.connection.instance_variable_get(:@shards)[shard].nil?
  @shard = shard
  return self
end