Merge a783bd8 into 2e31a14

active_model_serializers.gemspec

@@ -21,5 +21,6 @@ Gem::Specification.new do |gem|
   gem.required_ruby_version = ">= 1.9.3"
 
   gem.add_dependency "activemodel", ">= 3.2"
+  gem.add_dependency "activesupport", ">= 3.2"
   gem.add_development_dependency "rails", ">= 3.2"
 end

lib/active_model/serializer.rb

@@ -9,14 +9,22 @@ module ActiveModel
   class Serializer
     include Serializable
 
+    class_attribute :cache
+    class_attribute :perform_caching
+
     @mutex = Mutex.new
 
     class << self
       def inherited(base)
         base._root = _root
+        base.perform_caching = false
         base._attributes = (_attributes || []).dup
         base._associations = (_associations || {}).dup
       end
+
+      def cached(value = true)
+        self.perform_caching = value
+      end
 
       def setup
         @mutex.synchronize do
@@ -122,11 +130,45 @@ def json_key
     end
 
     def attributes
-      filter(self.class._attributes.dup).each_with_object({}) do |name, hash|
-        hash[name] = send(name)
+      if perform_cached?
+        cache_key = expand_cache_key([self.class.to_s.underscore, cache_key, 'attributes-json', digest])
+        p "Fetch Cache: #{cache_key}" if cache.read(cache_key)        
+        cache.fetch cache_key do
+          p "Write Cache: #{cache_key}"
+          filter(self.class._attributes.dup).each_with_object({}) do |name, hash|
+            hash[name] = send(name)
+          end
+        end
+      else
+        filter(self.class._attributes.dup).each_with_object({}) do |name, hash|
+          hash[name] = send(name)
+        end
       end
     end
 
+    def perform_cached?
+      perform_caching && cache && respond_to?(:cache_key)
+    end
+
+    def expand_cache_key(*args)
+      ActiveSupport::Cache.expand_cache_key(args)
+    end
+
+    def digest
+      class_source = File.read(default_file_path) rescue File.read(find_file_path)
+      Digest::MD5.hexdigest(class_source)
+    end
+
+    def default_file_path
+      Rails.root.join("app/serializers", "#{self.class.to_s.tableize.singularize}.rb").to_s
+    end
+
+    def find_file_path
+      self.class.instance_methods(false).map { |m|
+        self.class.instance_method(m).source_location.first
+      }.select{|f| f.match(Rails.root.to_s) }.last
+    end
+
     def associations
       associations = self.class._associations
       included_associations = filter(associations.keys)

lib/active_model/serializer/railtie.rb

@@ -6,5 +6,9 @@ class Railtie < Rails::Railtie
       require 'active_model/serializer/generators/serializer/scaffold_controller_generator'
       require 'active_model/serializer/generators/resource_override'
     end
+
+    initializer "caching.active_model_serializer" do |app|
+      ActiveModel::Serializer.cache = Rails.cache
+    end
   end
 end

lib/active_model_serializers.rb

@@ -1,4 +1,6 @@
 require 'active_model'
+require 'active_support'
+require 'active_support/core_ext'
 require 'active_model/serializer'
 require 'active_model/serializer_support'
 require 'active_model/serializer/version'
@@ -7,7 +9,7 @@
 begin
   require 'action_controller'
   require 'action_controller/serialization'
-
+  
   ActiveSupport.on_load(:action_controller) do
     include ::ActionController::Serialization
   end