From 2e55095f6fc1ec2799bbdfd1143d487d2de1d7a3 Mon Sep 17 00:00:00 2001
From: David Heinemeier Hansson <david@loudthinking.com>
Date: Sat, 17 Feb 2007 18:16:44 +0000
Subject: Added that rendering will automatically insert the etag header on 200
 OK responses. The etag is calculated using MD5 of the response body. If a
 request comes in that has a matching etag, the response will be changed to a
 304 Not Modified and the response body will be set to an empty string. [DHH]

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@6158 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
---
 actionpack/lib/action_controller/base.rb    | 18 ++++++++++++++++++
 actionpack/lib/action_controller/request.rb |  4 ++++
 2 files changed, 22 insertions(+)

(limited to 'actionpack/lib')

diff --git a/actionpack/lib/action_controller/base.rb b/actionpack/lib/action_controller/base.rb
index 3c855a9346..e870dc3045 100755
--- a/actionpack/lib/action_controller/base.rb
+++ b/actionpack/lib/action_controller/base.rb
@@ -7,6 +7,7 @@ require 'action_controller/url_rewriter'
 require 'action_controller/status_codes'
 require 'drb'
 require 'set'
+require 'md5'
 
 module ActionController #:nodoc:
   class ActionControllerError < StandardError #:nodoc:
@@ -648,6 +649,12 @@ module ActionController #:nodoc:
       # <tt>render_partial(partial_path = default_template_name, object = nil, local_assigns = {})</tt> and
       # <tt>render_partial_collection(partial_name, collection, partial_spacer_template = nil, local_assigns = {})</tt>.
       #
+      # == Automatic etagging
+      #
+      # Rendering will automatically insert the etag header on 200 OK responses. The etag is calculated using MD5 of the
+      # response body. If a request comes in that has a matching etag, the response will be changed to a 304 Not Modified
+      # and the response body will be set to an empty string.
+      #
       # === Rendering a template
       #
       # Template rendering works just like action rendering except that it takes a path relative to the template root.
@@ -870,6 +877,17 @@ module ActionController #:nodoc:
         else
           response.body = text
         end
+
+        if response.headers['Status'] == "200 OK" && response.body.size > 0
+          response.headers['Etag'] = "\"#{MD5.new(text).to_s}\""
+          
+          if request.headers['HTTP_IF_NONE_MATCH'] == response.headers['Etag']
+            response.headers['Status'] = "304 Not Modified"
+            response.body = ''
+          end
+        end
+        
+        response.body
       end
 
       def render_javascript(javascript, status = nil, append_response = true) #:nodoc:
diff --git a/actionpack/lib/action_controller/request.rb b/actionpack/lib/action_controller/request.rb
index 9f75b32f06..cb5a95b4e1 100755
--- a/actionpack/lib/action_controller/request.rb
+++ b/actionpack/lib/action_controller/request.rb
@@ -51,6 +51,10 @@ module ActionController
       @env['REQUEST_METHOD'].downcase.to_sym == :head
     end
 
+    def headers
+      @env
+    end
+
     # Determine whether the body of a HTTP call is URL-encoded (default)
     # or matches one of the registered param_parsers. 
     #
-- 
cgit v1.2.3