lib/alephant/broker.rb in alephant-broker-1.0.5 vs lib/alephant/broker.rb in alephant-broker-1.1.0

@@ -1,12 +1,14 @@
 require 'alephant/broker/version'
 require 'alephant/broker/request'
 require 'alephant/broker/environment'
 require 'alephant/broker'
+require 'ostruct'
 
 module Alephant
   module Broker
+    @@poll = true
 
     def self.handle(env)
       Request::Handler.process env
     end
 
@@ -16,17 +18,29 @@
 
     def self.config=(c)
       @@configuration = c
     end
 
+    def self.poll?
+      @@poll
+    end
+
+    def self.poll=(state)
+      @@poll = state
+    end
+
     class Application
       def initialize(c = nil)
         Broker.config = c unless c.nil?
       end
 
       def call(env)
-        send response_for(environment_for(env))
+        if ::Alephant::Broker.poll?
+          send response_for(environment_for(env))
+        else
+          send stop_poll_response
+        end
       end
 
       def environment_for(env)
         Environment.new env
       end
@@ -40,12 +54,28 @@
           response.status,
           {
             "Content-Type" => response.content_type,
             "X-Version"    => response.version.to_s,
             "X-Cached"     => response.cached.to_s
-          },
+          }.merge(response.headers),
           [ response.content.to_s ]
         ]
+      end
+
+      private
+
+      def stop_poll_response
+        response = OpenStruct.new(
+          :status  => 420,
+          :content => "Stopped polling",
+          :cached  => false,
+          :version => 0,
+          :headers => {
+            "Content-Type"   => "plain/text",
+            "X-Cached"       => "false",
+            "X-Stop-Polling" => "true"
+          }
+        )
       end
     end
   end
 end