Module: RedisOrm::Associations::HasMany

Included in:
Base
Defined in:
lib/redis_orm/associations/has_many.rb

Instance Method Summary collapse

Instance Method Details

#has_many(foreign_models, options = {}) ⇒ Object

user.avatars => user:1:avatars => [1, 22, 234] => Avatar.find([1, 22, 234]) options

*:dependant* key: either *destroy* or *nullify* (default)



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/redis_orm/associations/has_many.rb', line 7

def has_many(foreign_models, options = {})
  class_associations = class_variable_get(:"@@associations")
  class_associations[model_name] << {:type => :has_many, :foreign_models => foreign_models, :options => options}
  
  foreign_models_name = options[:as] ? options[:as].to_sym : foreign_models.to_sym

  define_method foreign_models_name.to_sym do
    Associations::HasManyProxy.new(model_name, id, foreign_models, options)
  end

  # user = User.find(1)
  # user.avatars = Avatar.find(23) => user:1:avatars => [23]
  define_method "#{foreign_models_name}=" do |records|
    if !options[:as]
      # clear old assocs from related models side
      old_records = self.send(foreign_models).to_a
      if !old_records.empty?
        # cache here which association with current model have old record's model
        has_many_assoc = old_records[0].get_associations.detect do |h|
          h[:type] == :has_many && h[:foreign_models] == model_name.pluralize.to_sym
        end
        
        has_one_or_belongs_to_assoc = old_records[0].get_associations.detect do |h|
          [:has_one, :belongs_to].include?(h[:type]) && h[:foreign_model] == model_name.to_sym
        end
        
        old_records.each do |record|
          if has_many_assoc
            $redis.zrem "#{record.model_name}:#{record.id}:#{model_name.pluralize}", id
          elsif has_one_or_belongs_to_assoc
            $redis.del "#{record.model_name}:#{record.id}:#{model_name}"
          end
        end
      end
      
      # clear old assocs from this model side
      $redis.zremrangebyscore "#{model_name}:#{id}:#{foreign_models}", 0, Time.now.to_f
    end

    records.to_a.each do |record|
      # we use here *foreign_models_name* not *record.model_name.pluralize* because of the :as option
      key = "#{model_name}:#{id}:#{foreign_models_name}"
      $redis.zadd(key, Time.now.to_f, record.id)
      set_expire_on_reference_key(key)
      
      record.get_indices.each do |index|
        save_index_for_associated_record(index, record, [model_name, id, record.model_name.pluralize]) # record.model_name.pluralize => foreign_models_name
      end

      # article.comments = [comment1, comment2] 
      # iterate through the array of comments and create backlink
      # check whether *record* object has *has_many* declaration and it states *self.model_name* in plural
      if assoc = class_associations[record.model_name].detect{|h| h[:type] == :has_many && h[:foreign_models] == model_name.pluralize.to_sym} #&& !$redis.zrank("#{record.model_name}:#{record.id}:#{model_name.pluralize}", id)#record.model_name.to_s.camelize.constantize.find(id).nil?
        assoc_foreign_models_name = assoc[:options][:as] ? assoc[:options][:as] : model_name.pluralize
        key = "#{record.model_name}:#{record.id}:#{assoc_foreign_models_name}"
        $redis.zadd(key, Time.now.to_f, id) if !$redis.zrank(key, id)
        set_expire_on_reference_key(key)
      end
        
      # check whether *record* object has *has_one* declaration and it states *self.model_name*
      if assoc = record.get_associations.detect{|h| [:has_one, :belongs_to].include?(h[:type]) && h[:foreign_model] == model_name.to_sym}
        foreign_model_name = assoc[:options][:as] ? assoc[:options][:as] : model_name
        key = "#{record.model_name}:#{record.id}:#{foreign_model_name}"
        # overwrite assoc anyway so we don't need to check record.send(model_name.to_sym).nil? here
        $redis.set(key, id)              
        set_expire_on_reference_key(key)
      end
    end
  end
end