patriot-workflow-scheduler 0.8.3 → 0.8.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a2f52963b03deb11f61726fe49e77f6712b98dbe
-  data.tar.gz: b05887e2a2fe307e8201d79e342547223dcafae1
+  metadata.gz: 171fc8719f6d43b4923df7c7f6441d9953aba4c9
+  data.tar.gz: 1c21f08807b6bf8b5ffee473184cbc972323d75e
 SHA512:
-  metadata.gz: 9203fe98d1ab635950a4d8883be8af94e8ca33328acb12fbc4728898a0dfc074ccb5b9418c7f89331fe030a956f4dd10cf1d9c59d02c2869b516988f5de884f3
-  data.tar.gz: 8e3b4677a6ba120cec68f445632a5534c63c64330da1d444bf0949ff5fa72dad601106f3cf8e446ee71a3a123e55c9a0b54d9e9cf5c9ad14eaed4b4284b3d0f8
+  metadata.gz: e6cf0a2084f683b914c01107e3884e61ad79ee0038f93dd5febe848e11a181d11692837e791ee5b441cd22fe98f4de6242db7379a29b753bca643f097e5221a1
+  data.tar.gz: f56e71de2572befe76194efe4a89e2f8874dcac4443ef12c5495ebbfef007a16df17710722e81b15bb2e688755b27f2a7709c3183c5143c56be92b79e5b73e1f

data/lib/patriot/command/post_processor/http_notification.rb

@@ -0,0 +1,66 @@
+require 'rest_client'
+require 'json'
+
+module Patriot
+  module Command
+    module PostProcessor
+      class HttpNotification < Patriot::Command::PostProcessor::Base
+
+        CALLBACK_URL = :callback_url
+        ON_PROP_KEY = :on
+
+        declare_post_processor_name :http_notification
+
+        def validate_props(props)
+          raise "#{CALLBACK_URL} is not specified" unless props.has_key?(CALLBACK_URL)
+          raise "#{CALLBACK_URL} is not a correct URL" unless valid_url?(props.fetch(CALLBACK_URL, ""))
+          raise "#{ON_PROP_KEY} is not specified" unless props.has_key?(ON_PROP_KEY)
+        end
+
+        def process(cmd, worker, job_ticket)
+          on = @props[ON_PROP_KEY]
+          on = [on] unless on.is_a?(Array)
+          on = on.map{|o| Patriot::Command::ExitCode.value_of(o)}
+          exit_code = job_ticket.exit_code
+          return unless on.include?(exit_code)
+
+          callback_url = @props[CALLBACK_URL]
+          case exit_code
+          when Patriot::Command::ExitCode::SUCCEEDED then send_callback(job_ticket.job_id, callback_url, "SUCCEEDED")
+          when Patriot::Command::ExitCode::FAILED then send_callback(job_ticket.job_id, callback_url, "FAILED")
+          end
+        end
+
+        def send_callback(job_id, callback_url, state)
+          retries = 0
+          begin
+            return RestClient.post(
+              callback_url,
+              {'job_id' => job_id, 'state' => state }.to_json,
+              :content_type => 'application/json'
+            )
+          rescue RestClient::Exception => error
+            retries += 1
+            if retries < 3
+              retry
+            else
+              raise error
+            end
+          end
+        end
+
+        def valid_url?(url)
+          begin
+             URI.parse(url)
+          rescue URI::InvalidURIError
+            return false
+          end
+          return true
+        end
+
+      end
+    end
+  end
+end
+
+

data/lib/patriot/command/post_processor.rb

@@ -9,6 +9,7 @@ module Patriot
       require 'patriot/command/post_processor/discard_on_fail'
       require 'patriot/command/post_processor/retrial'
       require 'patriot/command/post_processor/mail_notification'
+      require 'patriot/command/post_processor/http_notification'
     end
   end
 end

data/lib/patriot/version.rb

@@ -1,3 +1,3 @@
 module Patriot
-  VERSION = "0.8.3"
+  VERSION = "0.8.4"
 end