build_status_server 0.7 → 0.8

data/lib/build_status_server/server.rb

@@ -27,16 +27,20 @@ is listening at the same port (#{udp_server["address"]}:#{udp_server["port"]}
 
       puts "Listening on UDP #{udp_server["address"]}:#{udp_server["port"]}" if config.verbose
 
-      while true
-        data, addr = sock.recvfrom(2048)
+      begin
+        while true
+          data, addr = sock.recvfrom(2048)
 
-        if process_job(data)
-          status = process_all_statuses
-          notify(status)
+          if process_job(data)
+            status = process_all_statuses
+            notify(status)
+          end
         end
+      rescue Interrupt
+        puts "Good bye."
+        sock.close
+        exit
       end
-
-      sock.close
     end
 
     private

data/lib/build_status_server/server.rb~

@@ -0,0 +1,162 @@
+# TODO
+# move all configuration stuff to Config
+# and just call config[:blaj] instead of if blah
+module BuildStatusServer
+  class Server
+    attr_reader :config, :store
+
+    def initialize(options = {})
+      @config = Config.new
+      config.load(options)
+    end
+
+    def listen
+      sock = UDPSocket.new
+      udp_server = config.udp_server
+
+      begin
+        sock.bind(udp_server["address"], udp_server["port"])
+      rescue Errno::EADDRINUSE
+        STDERR.puts <<-EOT
+There appears that another instance is running, or another process
+is listening at the same port (#{udp_server["address"]}:#{udp_server["port"]}
+
+        EOT
+        exit
+      end
+
+      puts "Listening on UDP #{udp_server["address"]}:#{udp_server["port"]}" if config.verbose
+
+      begin
+        while true
+          data, addr = sock.recvfrom(2048)
+
+          if process_job(data)
+            status = process_all_statuses
+            notify(status)
+          end
+        end
+      rescue Interrupt
+        puts "Good bye."
+        exit
+      end
+
+      sock.close
+    end
+
+    private
+
+    def load_store
+      @store = begin
+                 YAML.load_file(config.store_file)
+               rescue
+                 {}
+               end
+      @store = {} unless store.class == Hash
+    end
+
+
+    def process_job(data = "{}")
+      job = JSON.parse(data)
+
+      build_name = job["name"]
+
+      unless should_process_build(build_name)
+        STDOUT.puts "Ignoring #{build_name} (#{config.mask["regex"]}--#{config.mask["policy"]})" if config.verbose
+        return false
+      end
+
+      if job.class != Hash or
+        job["build"].class != Hash
+        STDERR.puts "Pinged with an invalid payload"
+        return false
+      end
+
+      phase  = job["build"]["phase"]
+      status = job["build"]["status"]
+
+      if phase == "FINISHED"
+        STDOUT.puts "Got #{status} for #{build_name} on #{Time.now} [#{job.inspect}]" if config.verbose
+        case status
+        when "SUCCESS", "FAILURE"
+          load_store
+          store[build_name] = status
+          File.open(config.store_file, "w") { |file| YAML.dump(store, file) }
+          return true
+        end
+      else
+        STDOUT.puts "Started for #{build_name} on #{Time.now} [#{job.inspect}]" if config.verbose
+      end
+
+      return false
+    end
+
+    def should_process_build(build_name)
+      # If mask exists, then ...
+      ! (
+        !!config.mask &&
+        !!config.mask["regex"] &&
+        ((config.mask["policy"] == "include" && build_name !~ config.mask["regex"]) ||
+         (config.mask["policy"] != "include" && build_name =~ config.mask["regex"])
+      ))
+    end
+
+    def process_all_statuses
+      pass = true
+
+      @store.values.each do |val|
+        pass &&= (val == "pass" || val == "SUCCESS")
+      end
+
+      pass
+    end
+
+    def notify(status)
+      tcp_client = config.tcp_client
+
+      attempts = 0
+      light  = status ? tcp_client["pass"] : tcp_client["fail"]
+
+      begin
+        timeout(5) do
+          attempts += 1
+          client = TCPSocket.new(tcp_client["host"], tcp_client["port"])
+          client.print "GET #{light} HTTP/1.0\n\n"
+          answer = client.gets(nil)
+          STDOUT.puts answer if config.verbose
+          client.close
+        end
+      rescue Timeout::Error => ex
+        STDERR.puts "Error: #{ex} while trying to send #{light}"
+        retry unless attempts > 2
+      rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH => ex
+        STDERR.puts "Error: #{ex} while trying to send #{light}"
+        STDERR.puts "Will wait for 2 seconds and try again..."
+        sleep 2
+        retry unless attempts > 2
+      end
+    end
+  end
+end
+
+__END__
+
+Example payload:
+{
+  "name":"test",
+  "url":"job/test/",
+  "build":{
+    "full_url":"http://cronus.local:3001/job/test/20/",
+    "number":20,
+    "phase":"FINISHED",
+    "status":"SUCCESS",
+    "url":"job/test/20/"
+  }
+}
+
+We're getting this error once in a while:
+/usr/local/lib/ruby/1.8/timeout.rb:64:in `notify': execution expired (Timeout::Error)
+        from /home/jcmuller/build_notifier/lib/server.rb:102:in `notify'
+        from /home/jcmuller/build_notifier/lib/server.rb:33:in `listen'
+        from bin/server:5
+

data/lib/build_status_server/version.rb

@@ -1,3 +1,3 @@
 module BuildStatusServer
-  VERSION = "0.7"
+  VERSION = "0.8"
 end

data/lib/build_status_server/version.rb~

@@ -1,3 +1,3 @@
 module BuildStatusServer
-  VERSION = "0.6"
+  VERSION = "0.7"
 end

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: build_status_server
 version: !ruby/object:Gem::Version 
-  hash: 5
+  hash: 27
   prerelease: 
   segments: 
   - 0
-  - 7
-  version: "0.7"
+  - 8
+  version: "0.8"
 platform: ruby
 authors: 
 - Juan C. Muller
@@ -83,6 +83,7 @@ extra_rdoc_files: []
 files: 
 - lib/build_status_server/config.rb
 - lib/build_status_server/server.rb
+- lib/build_status_server/server.rb~
 - lib/build_status_server/version.rb
 - lib/build_status_server/version.rb~
 - lib/build_status_server.rb