Module: Switchman::ActiveRecord::Associations::BelongsToAssociation

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

Instance Method Summary collapse

Instance Method Details

#replace_keys(record, force: false) ⇒ Object 



41
42
43
44
45
46
47
48
 
# File 'lib/switchman/active_record/associations.rb', line 41

def replace_keys(record, force: false)
  if record&.class&.sharded_column?(reflection.association_primary_key(record.class))
    foreign_id = record[reflection.association_primary_key(record.class)]
    owner[reflection.foreign_key] = Shard.relative_id_for(foreign_id, record.shard, owner.shard)
  else
    super
  end
end

#shard ⇒ Object 



50
51
52
53
54
55
56
57
 
# File 'lib/switchman/active_record/associations.rb', line 50

def shard
  if @owner.class.sharded_column?(@reflection.foreign_key) &&
     (foreign_id = @owner[@reflection.foreign_key])
    Shard.shard_for(foreign_id, @owner.shard)
  else
    super
  end
end