Patch to not stat user-agent or create session if the request is a he…

…alth-check.
wiecklabs · Mar 28, 2011 · e93fade · e93fade
1 parent 1766a6f
commit e93fade
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/harbor/contrib/stats.rb b/lib/harbor/contrib/stats.rb
@@ -78,7 +78,7 @@ def self.denied_user?(remote_ip, user_agent)
 
 Harbor::Session.register_event_handler(:session_created) do |event|
   if orm = Harbor::Contrib::Stats.orm
-    orm::UserAgent.create(event.session_id, event.remote_ip, event.user_agent) unless Harbor::Contrib::Stats.denied_user?(event.remote_ip, event.user_agent)
+    orm::UserAgent.create(event.session_id, event.remote_ip, event.user_agent) unless Harbor::Contrib::Stats.denied_user?(event.remote_ip, event.user_agent) || request.health_check?
   else    
     warn "Harbor::Contrib::Stats::orm must be set to generate statistics."
   end

diff --git a/lib/harbor/session.rb b/lib/harbor/session.rb
@@ -38,7 +38,11 @@ def initialize(request, key = nil)
       @cookie = request.cookies[key] || request.cookies[@options[:key]]
       @store = self.class.options[:store]
       @request = request
-      @data ||= @store.load_session(self, @cookie, @request)
+      if @request.health_check? then
+        @data ||= {}
+      else
+        @data ||= @store.load_session(self, @cookie, @request)
+      end
     end
 
     def session_created(session_id, remote_ip, user_agent_raw)