join_cache 0.1.2 → 0.2.0

data/README.markdown

@@ -47,10 +47,12 @@ class Employee < ActiveRecord::Base
 end
 ```
 
+It also works with `has_many :through` associations!
+
 ## TODO
 
-* [Support has_many :through](https://github.com/KevinBongart/join_cache/issues/1)
 * [Support callbacks: after_add and after_remove](https://github.com/KevinBongart/join_cache/issues/2)
+* [Add tests](https://github.com/KevinBongart/join_cache/issues/3)
 
 [View the full list](https://github.com/KevinBongart/join_cache/issues)
 

data/VERSION

@@ -1 +1 @@
-0.1.2
+0.2.0

data/join_cache.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = "join_cache"
-  s.version = "0.1.2"
+  s.version = "0.2.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Kevin Bongart"]

data/lib/join_cache.rb

@@ -39,5 +39,48 @@ module JoinCache
         association.klass.where(id: send(cached_ids_name))
       end
     end
+
+    has_many_through_associations = reflect_on_all_associations(:has_many).select do |association|
+      association.is_a?(ActiveRecord::Reflection::ThroughReflection)
+    end
+
+    # Generates methods for faster retrieval of has_many :through associations
+    # by storing foreign keys in cache and avoiding large joins
+    #
+    # For example:
+    # class Physician < ActiveRecord::Base
+    #   has_many :appointments
+    #   has_many :patients, through: :appointments
+    #   include JoinCache
+    # end
+    #
+    # Physician.first.cached_patient_ids
+    # => [4, 8, 15, 16, 23, 42]
+    #
+    # Physician.first.cached_patients
+    # => Patient.where(id: [4, 8, 15, 16, 23, 42])
+    #
+    has_many_through_associations.each do |association|
+      singular_name   = association.name.to_s.singularize         # patient
+      plural_name     = association.plural_name                   # patients
+      cached_name     = "cached_#{plural_name}"                   # cached_patients
+      cached_ids_name = "cached_#{singular_name}_ids"             # cached_patient_ids
+      primary_key     = self.name.foreign_key                     # employee_id
+      foreign_key     = association.foreign_key                   # patient_id
+      join_table      = association.options[:through]             # appointments
+      join_model      = join_table.to_s.classify.constantize      # Appointment
+
+      # cached_patient_ids
+      define_method(cached_ids_name) do
+        Rails.cache.fetch("#{cache_key}/#{cached_ids_name}") do
+          join_model.where(primary_key => id).pluck(foreign_key.to_sym)
+        end
+      end
+
+      # cached_patients
+      define_method(cached_name) do
+        association.klass.where(id: send(cached_ids_name))
+      end
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: join_cache
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -109,7 +109,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -785509289355240113
+      hash: 1146460858880032511
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: