hq-log-monitor-server 0.1.0 → 0.2.0

data/features/submit-event.feature

@@ -8,6 +8,12 @@ Feature: Log monitor server submit event via HTTP
       <log-monitor-server-config>
         <server port="${port}"/>
         <db host="${db-host}" port="${db-port}" name="${db-name}"/>
+        <icinga command-file="${command-file}">
+          <service name="service" icinga-host="host" icinga-service="service">
+            <type name="warning" level="warning"/>
+            <type name="critical" level="critical"/>
+          </service>
+        </icinga>
       </log-monitor-server-config>
       """
 
@@ -39,3 +45,7 @@ Feature: Log monitor server submit event via HTTP
         },
       }
       """
+    And icinga should receive:
+      """
+      PROCESS_SERVICE_CHECK_RESULT;host;service;1;WARNING 1 warning
+      """

data/features/support/env.rb

@@ -57,3 +57,11 @@ at_exit do
 	end
 
 end
+
+Before do
+	@command_file = Tempfile.new "cuke-"
+end
+
+After do
+	@command_file.unlink
+end

data/features/support/steps.rb

@@ -14,6 +14,7 @@ Given /^the log monitor server config:$/ do
 	config_string.gsub! "${db-host}", "localhost"
 	config_string.gsub! "${db-port}", "27017"
 	config_string.gsub! "${db-name}", mongo_db_name("logMonitorServer")
+	config_string.gsub! "${command-file}", @command_file.path
 
 	@log_monitor_server_config.write config_string
 	@log_monitor_server_config.flush
@@ -137,3 +138,14 @@ Then /^the (\d+(?:st|nd|rd|th)) summary should be:$/ do
 	end
 
 end
+
+Then /^icinga should receive:$/ do
+	|expected_command|
+
+	command_contents =
+		File.new(@command_file).to_a.map { |line| line.strip }
+
+	command_contents.should \
+		include expected_command
+
+end

data/lib/hq/log-monitor-server/script.rb

@@ -18,6 +18,7 @@ class Script
 
 	def initialize
 		@status = 0
+		@mutex = Mutex.new
 	end
 
 	def main
@@ -25,12 +26,14 @@ class Script
 		trap "INT" do
 			@web_server.shutdown
 		end
+		Thread.new { background }
 		run
 	end
 
 	def start
 		setup
 		Thread.new { run }
+		Thread.new { background }
 	end
 
 	def stop
@@ -48,6 +51,118 @@ class Script
 		@web_server.start
 	end
 
+	def background
+
+		@next_check = Time.now
+
+		loop do
+			sleep 1 while Time.now < @next_check
+			do_checks
+		end
+
+	end
+
+	def do_checks
+		@mutex.synchronize do
+
+			File.open @icinga_elem["command-file"], "a" do
+				|command_io|
+
+				summaries_by_service =
+					get_summaries_by_service
+
+				@icinga_elem.find("service").each do
+					|service_elem|
+
+					critical_count = 0
+					warning_count = 0
+					unknown_count = 0
+
+					summaries_by_service[service_elem["name"]]["types"].each do
+						|type_name, type_info|
+
+						type_elem =
+							service_elem.find_first("
+								type [@name = #{esc_xp type_name}]
+							")
+
+						if ! type_elem
+							unknown_count += 1
+						elsif type_elem["level"] == "critical"
+							critical_count += 1
+						elsif type_elem["level"] == "warning"
+							warning_count += 1
+						else
+							unknown_count += 1
+						end
+
+						status_int =
+							if critical_count > 0
+								2
+							elsif warning_count > 0
+								1
+							elsif unknown_count > 0
+								3
+							else
+								0
+							end
+
+						status_str =
+							if critical_count > 0
+								"CRITICAL"
+							elsif warning_count > 0
+								"WARNING"
+							elsif unknown_count > 0
+								"UNKNOWN"
+							else
+								"OK"
+							end
+
+						parts = []
+
+						if critical_count > 0
+							parts << "%d critical" % critical_count
+						end
+
+						if warning_count > 0
+							parts << "%d warning" % warning_count
+						end
+
+						if unknown_count > 0
+							parts << "%d unknown" % unknown_count
+						end
+
+						if parts.empty?
+							parts << "no new events"
+						end
+
+						message = "%s %s" % [
+							status_str,
+							parts.join(", "),
+						]
+
+						command = [
+							"PROCESS_SERVICE_CHECK_RESULT",
+							service_elem["icinga-host"],
+							service_elem["icinga-service"],
+							status_int,
+							message,
+						].join ";"
+
+						command_io.print "%s\n" % command
+
+					end
+
+				end
+
+			end
+
+			@next_check = Time.now + 60
+
+		end
+
+	end
+
 	def process_args
 
 		@opts, @args =
@@ -80,6 +195,9 @@ class Script
 		@db_elem =
 			@config_elem.find_first("db")
 
+		@icinga_elem =
+			@config_elem.find_first("icinga")
+
 	end
 
 	def connect_db
@@ -183,13 +301,17 @@ class Script
 
 		@db["summaries"].save summary
 
+		# perform checks
+
+		do_checks
+
 		# respond successfully
 
 		return 202, {}, []
 
 	end
 
-	def overview_page env
+	def get_summaries_by_service
 
 		summaries_by_service = {}
 
@@ -231,6 +353,15 @@ class Script
 
 		end
 
+		return summaries_by_service
+
+	end
+
+	def overview_page env
+
+		summaries_by_service =
+			get_summaries_by_service
+
 		headers = {}
 		html = []
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hq-log-monitor-server
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-04-02 00:00:00.000000000 Z
+date: 2013-04-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bson_ext