Module: Switchman::ActiveRecord::Relation

Defined in:
lib/switchman/active_record/relation.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(klass) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
 
# File 'lib/switchman/active_record/relation.rb', line 4

def self.included(klass)
  klass::SINGLE_VALUE_METHODS.concat [ :shard, :shard_source ]

  %w{exec_queries update_all delete_all}.each do |method|
    klass.alias_method_chain(method, :deshackles)
  end

  %w{initialize explain to_a update_all delete_all new create create!}.each do |method|
    klass.alias_method_chain(method, :sharding)
  end
end

Instance Method Details

#activate(&block) ⇒ Object 



87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
 
# File 'lib/switchman/active_record/relation.rb', line 87

def activate(&block)
  shards = all_shards
  if (Array === shards && shards.length == 1)
    if shards.first == DefaultShard || shards.first == Shard.current(klass.shard_category)
      yield(self, shards.first)
    else
      shards.first.activate(klass.shard_category) { yield(self, shard_value) }
    end
  else
    # TODO: implement local limit to avoid querying extra shards
    Shard.with_each_shard(shards, [klass.shard_category]) do
      shard(Shard.current(klass.shard_category), :to_a).activate(&block)
    end
  end
end

#create_with_sharding(*args, &block) ⇒ Object 



35
36
37
 
# File 'lib/switchman/active_record/relation.rb', line 35

def create_with_sharding(*args, &block)
  primary_shard.activate(klass.shard_category) { create_without_sharding(*args, &block) }
end

#create_with_sharding!(*args, &block) ⇒ Object 



39
40
41
 
# File 'lib/switchman/active_record/relation.rb', line 39

def create_with_sharding!(*args, &block)
  primary_shard.activate(klass.shard_category) { create_without_sharding!(*args, &block) }
end

#exec_queries_with_deshackles(*args) ⇒ Object 



43
44
45
46
47
48
49
50
51
 
# File 'lib/switchman/active_record/relation.rb', line 43

def exec_queries_with_deshackles(*args)
  if self.lock_value
    db = Shard.current(shard_category).database_server
    if ::Shackles.environment != db.shackles_environment
      return db.unshackle { exec_queries_without_deshackles(*args) }
    end
  end
  exec_queries_without_deshackles(*args)
end

#explain_with_sharding ⇒ Object 



53
54
55
 
# File 'lib/switchman/active_record/relation.rb', line 53

def explain_with_sharding
  self.activate { |relation| relation.send(:explain_without_sharding) }
end

#initialize_with_sharding(*args) ⇒ Object 



16
17
18
19
20
 
# File 'lib/switchman/active_record/relation.rb', line 16

def initialize_with_sharding(*args)
  initialize_without_sharding(*args)
  self.shard_value = Shard.current(klass.try(:shard_category) || :default) unless shard_value
  self.shard_source_value = :implicit unless shard_source_value
end

#merge(*args) ⇒ Object 



22
23
24
25
26
27
28
29
 
# File 'lib/switchman/active_record/relation.rb', line 22

def merge(*args)
  relation = super
  if relation.shard_value != self.shard_value && relation.shard_source_value == :implicit
    relation.shard_value = self.shard_value
    relation.shard_source_value = self.shard_source_value
  end
  relation
end

#new_with_sharding(*args, &block) ⇒ Object 



31
32
33
 
# File 'lib/switchman/active_record/relation.rb', line 31

def new_with_sharding(*args, &block)
  primary_shard.activate(klass.shard_category) { new_without_sharding(*args, &block) }
end

#to_a_with_sharding ⇒ Object 



57
58
59
60
61
62
63
64
65
66
 
# File 'lib/switchman/active_record/relation.rb', line 57

def to_a_with_sharding
  return @records if loaded?
  results = self.activate { |relation| relation.send(:to_a_without_sharding) }
  case shard_value
  when Array, ::ActiveRecord::Relation, ::ActiveRecord::Base
    @records = results
    @loaded = true
  end
  results
end