mongoid-sorted-relations 0.0.1 → 0.0.2

data/lib/mongoid/sorted_relations.rb

@@ -4,24 +4,35 @@ module Mongoid
 
     def self.included(base)
       base.extend(ClassMethods)
+      base.after_initialize :freeze_relation_ids
     end
 
     module ClassMethods
       def init_sorted_relations
         self.relations.each do |k,v|
-          if [:references_many, :references_and_referenced_in_many].include? v.macro
-            send(:define_method, "sorted_#{v.name}") { sorted_relation v }
-          end
+          send(:define_method, "sorted_#{v.name}") { sorted_relation v } if [:references_many, :references_and_referenced_in_many].include? v.macro
         end
       end
     end
 
-    def sorted_relation(relation)
-      ids = self.send(relation.key).map{ |rid| rid.to_s }
+    def freeze_relation_ids
+      @cache_sorted_documents = {}
+      @cache_sorted_ids = {}
 
-      documents = self.send(relation.name).sort_by { |x| ids.index(x.id.to_s) }
+      self.relations.each do |k,v|
+        if [:references_many, :references_and_referenced_in_many].include? v.macro
+          @cache_sorted_ids[v.key] = [self.send(v.key)].flatten.map{ |rid| rid.to_s }
+        end
+      end
+    end
+
+    def sorted_relation(relation)
+      if not @cache_sorted_documents[relation.name]
+        documents = self.send(relation.name).sort_by { |x| @cache_sorted_ids[relation.key].index(x.id.to_s) }
+        @cache_sorted_documents[relation.name] = Mongoid::FakeCriteria.new(documents)
+      end
 
-      return Mongoid::FakeCriteria.new(documents)
+      return @cache_sorted_documents[relation.name]
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongoid-sorted-relations
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-05-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mongoid
-  requirement: &2168955220 !ruby/object:Gem::Requirement
+  requirement: &2160751500 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '2.0'
   type: :runtime
   prerelease: false
-  version_requirements: *2168955220
+  version_requirements: *2160751500
 - !ruby/object:Gem::Dependency
   name: bson_ext
-  requirement: &2168680680 !ruby/object:Gem::Requirement
+  requirement: &2160750920 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -32,10 +32,10 @@ dependencies:
         version: '1.6'
   type: :development
   prerelease: false
-  version_requirements: *2168680680
+  version_requirements: *2160750920
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &2168678240 !ruby/object:Gem::Requirement
+  requirement: &2160750080 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: 0.8.7
   type: :development
   prerelease: false
-  version_requirements: *2168678240
+  version_requirements: *2160750080
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &2168675460 !ruby/object:Gem::Requirement
+  requirement: &2160539920 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: '2.0'
   type: :development
   prerelease: false
-  version_requirements: *2168675460
+  version_requirements: *2160539920
 - !ruby/object:Gem::Dependency
   name: rspec-aspic
-  requirement: &2168674340 !ruby/object:Gem::Requirement
+  requirement: &2160538360 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,7 +65,7 @@ dependencies:
         version: 0.0.2
   type: :development
   prerelease: false
-  version_requirements: *2168674340
+  version_requirements: *2160538360
 description: Preserve the order of the mongoid documents in a referenced one-to-many
   or many-to-many relationship.
 email: