Changeset 1480

Timestamp:

04/08/08 12:20:21 (5 months ago)

Author:

dbryson

Message:

—

Files:

• hive/trunk/worker_bee/lib/rest_client.rb (deleted)
• hive/trunk/worker_bee/lib/worker_bee.rb (modified) (3 diffs)

hive/trunk/worker_bee/lib/worker_bee.rb

@@ -1 +1 @@
-module Hive
-  class WorkerBee
-, :client
+
-    
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-    end
-    
-    def setup_header
-      headers = {}
-'access_key'
+"access_key"
-      headers['User-Agent'] = "Hive-Worker-Bee"
-      headers['Date'] = Time.now.httpdate
-    end
-    
+    def get(url)
+      call_hive("get",url)
+    end
+   
+    def post(url,body)
+      call_hive("post",path,url)
+    end
+    
-    def ping?
-      begin
-
+
+
-        return true
-
+
+
-        return false
-      end
@@ -24 +50 @@
-    
-    # Need to test this
-end_client
+hip
-      # Grab data from the local DB, serialize
-      # and send to the Hive
-      job = BeeJob.get_job
+      
-      if job
-
-
-
-
-
+
-        
+        encoded_data = ActiveSupport::JSON.encode(job[1])
+        #form_data = "data=#{encoded_data}"
-        begin
-
+
+
+
-          job.destroy # Remove the job from the database
-        rescue Exception => e
@@ -49 +76 @@
-      loop do
-        if ping?
-end_client
+hip
-        end
-
+
+
-      end
+    end
+    
+    private
+   
+    def call_hive(verb,u,body=nil)
+      url = parse_url(u)
+      request = request_method(verb).new(url.path,{})
+      setup_headers!(request)
+      if body
+        request.set_form_data({'data'=>body})
+      end
+      Net::HTTP.new(url.host,url.port).start do |http| 
+        process_response http.request(request) 
+      end
+    end
+   
+    def process_response(res)
+      if %w(200 201).include? res.code
+                res.body
+            elsif res.code == "403"
+                raise "Unauthorized"
+            else
+                raise "HTTP Error. Status: #{res.code}  #{res.body}"
+            end
+    end
+    
+    def parse_url(url)
+            url = "http://#{url}" unless url.match(/^http/)
+            URI.parse(url)
+        end
+        
+    def request_method(verb)
+      Net::HTTP.const_get(verb.to_s.capitalize)
+    end
+   
+    def setup_headers!(req)
+      req['Authorization'] = @key
+      req['User-Agent'] = "Hive-Worker-Bee"
+      req['Date'] = Time.now.httpdate
-    end
-