Module: ActiveRecordShards::Model

Defined in:
lib/active_record_shards/model.rb

Defined Under Namespace

Modules: InstanceMethods

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.extended(base) ⇒ Object 



64
65
66
67
 
# File 'lib/active_record_shards/model.rb', line 64

def self.extended(base)
  base.send(:include, InstanceMethods)
  base.after_initialize :initialize_shard_and_replica
end

Instance Method Details

#is_sharded? ⇒ Boolean

rubocop:disable Naming/PredicateName

Returns:

  • (Boolean) 


13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/active_record_shards/model.rb', line 13

def is_sharded? # rubocop:disable Naming/PredicateName
  return @_ars_model_is_sharded unless @_ars_model_is_sharded.nil?

  @_ars_model_is_sharded =
    if self == ActiveRecord::Base
      sharded != false && supports_sharding?
    elsif self == base_class
      if sharded.nil?
        ActiveRecord::Base.is_sharded?
      else
        sharded != false
      end
    else
      base_class.is_sharded?
    end
end

#not_sharded ⇒ Object 



5
6
7
8
9
10
11
 
# File 'lib/active_record_shards/model.rb', line 5

def not_sharded
  if self != ActiveRecord::Base && self != base_class
    raise "You should only call not_sharded on direct descendants of ActiveRecord::Base"
  end

  self.sharded = false
end

#on_replica_by_default=(value) ⇒ Object 



41
42
43
44
45
46
47
 
# File 'lib/active_record_shards/model.rb', line 41

def on_replica_by_default=(value)
  if self == ActiveRecord::Base
    raise ArgumentError, "Cannot set on_replica_by_default on ActiveRecord::Base"
  else
    base_class.instance_variable_set(:@on_replica_by_default, value)
  end
end

#on_replica_by_default? ⇒ Boolean

Returns:

  • (Boolean) 


30
31
32
33
34
35
36
37
38
39
 
# File 'lib/active_record_shards/model.rb', line 30

def on_replica_by_default?
  if self == ActiveRecord::Base
    false
  else
    base = base_class
    if base.instance_variable_defined?(:@on_replica_by_default)
      base.instance_variable_get(:@on_replica_by_default)
    end
  end
end