flapjack 0.4.12 → 0.5.1

data/lib/flapjack/persistence/sqlite3.rb

@@ -0,0 +1,3 @@
+#!/usr/bin/env ruby 
+
+require File.join(File.dirname(__FILE__), 'sqlite3', 'sqlite3')

data/lib/flapjack/persistence/sqlite3/sqlite3.rb

@@ -0,0 +1,166 @@
+#!/usr/bin/env ruby 
+
+require 'sqlite3'
+
+module Flapjack
+  module Persistence
+    class Sqlite3
+      def initialize(options={})
+        @options = options
+        @config = OpenStruct.new(options)
+        @log = @config.log
+        connect
+      end
+
+      def any_parents_failed?(check_id)
+        rows = @db.execute(%(SELECT count(*) FROM "checks" INNER JOIN "related_checks" ON "related_checks".parent_id = "checks".id AND "related_checks".child_id = #{check_id};))
+        rows.flatten.first != "0"
+      end
+
+      def save_check(result)
+        if check = get_check(result[:id])
+          result[:updated_at] = Time.now
+          updates = result.map { |key, value| "#{key} = '#{value}'" }.join(', ')
+          statement = %(UPDATE "checks" SET #{updates} WHERE "id" = #{result[:id]};)
+          @db.execute(statement)
+        else
+          result[:created_at] = Time.now
+          columns, values = columns_and_values_for(result)
+          @db.execute(%(INSERT INTO "checks" #{columns} VALUES #{values}))
+        end
+        result
+      end
+
+      def get_check(id)
+        result = @db.execute2(%(SELECT * FROM "checks" WHERE id = "#{id}"))
+        return nil unless result[1]
+        hash = {}
+        result[0].each_with_index do |key, index|
+          hash[key] = result[1][index]
+        end
+        hash
+      end
+
+      def delete_check(id)
+        result = @db.execute2(%(DELETE FROM "checks" WHERE id = "#{id}"))
+      end
+
+      def all_checks
+        results = @db.execute2(%(SELECT * FROM "checks";))
+        
+        records = results[1..-1].map do |values|
+          hash = {}
+          values.each_with_index do |value, index|
+            hash[results[0][index]] = value
+          end
+          hash
+        end
+        
+        records
+      end
+
+      def save_check_relationship(attrs)
+        columns, values = columns_and_values_for(attrs)
+        @db.execute(%(INSERT INTO "related_checks" #{columns} VALUES #{values}))
+      end
+
+      def all_check_relationships
+        results = @db.execute2(%(SELECT * FROM "related_checks";))
+        
+        records = results[1..-1].map do |values|
+          hash = {}
+          values.each_with_index do |value, index|
+            hash[results[0][index]] = value
+          end
+          hash
+        end
+
+        records
+      end
+
+      # events
+      def all_events_for(id)
+        results = @db.execute2(%(SELECT * FROM "events" WHERE check_id = #{id}))
+
+        records = results[1..-1].map do |values|
+          hash = {}
+          values.each_with_index do |value, index|
+            hash[results[0][index]] = value
+          end
+          hash
+        end
+
+        records
+      end
+
+      def all_events
+        results = @db.execute2(%(SELECT * FROM "events";))
+
+        records = results[1..-1].map do |values|
+          hash = {}
+          values.each_with_index do |value, index|
+            hash[results[0][index]] = value
+          end
+          hash
+        end
+
+        records
+      end
+
+      def create_event(result)
+        @db.execute(%(INSERT INTO "events" ("check_id", "created_at") VALUES ("#{result.result.check_id}", "#{Time.now}")))
+        true
+      end
+
+      private
+      def connect
+        raise ArgumentError, "Database location wasn't specified" unless @config.database
+        @db = ::SQLite3::Database.new(@config.database)
+
+        auto_migrate if @config.auto_migrate
+      end
+
+      def auto_migrate
+        statements = [
+					%(DROP TABLE IF EXISTS "check_templates"),
+					%(DROP TABLE IF EXISTS "events"),
+					%(DROP TABLE IF EXISTS "related_checks"),
+					%(DROP TABLE IF EXISTS "checks"),
+					%(DROP TABLE IF EXISTS "nodes"),
+					%(PRAGMA table_info('nodes')),
+					%(SELECT sqlite_version(*)),
+					%(CREATE TABLE "nodes" ("fqdn" VARCHAR(50) NOT NULL, PRIMARY KEY("fqdn"))),
+					%(PRAGMA table_info('checks')),
+					%(CREATE TABLE "checks" ("created_at" DATETIME, "updated_at" DATETIME, "id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, "command" TEXT NOT NULL, "params" TEXT DEFAULT NULL, "name" VARCHAR(50) NOT NULL, "enabled" BOOLEAN DEFAULT 'f', "status" INTEGER DEFAULT 0, "deleted_at" DATETIME DEFAULT NULL, "node_fqdn" VARCHAR(50), "check_template_id" INTEGER)),
+					%(PRAGMA table_info('related_checks')),
+					%(CREATE TABLE "related_checks" ("parent_id" INTEGER, "child_id" INTEGER, "id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT)),
+					%(PRAGMA table_info('events')),
+					%(CREATE TABLE "events" ("created_at" DATETIME, "updated_at" DATETIME, "check_id" INTEGER NOT NULL, "id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, "deleted_at" DATETIME DEFAULT NULL)),
+					%(PRAGMA table_info('check_templates')),
+					%(CREATE TABLE "check_templates" ("id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, "command" TEXT NOT NULL, "name" VARCHAR(50) NOT NULL, "params" TEXT DEFAULT NULL))
+        ]
+
+        statements.each do |statement|
+          @db.execute(statement)
+        end
+      end
+
+      def columns_and_values_for(result)
+        @keys = []
+        @values = []
+        
+        result.each_pair do |k,v|
+          @keys << k
+          @values << v
+        end
+
+        keys   = "(\"" + @keys.join(%(", ")) + "\")"
+        values = "(\"" + @values.join(%(", ")) + "\")"
+
+        return keys, values
+      end
+
+    end
+  end
+end
+

data/lib/flapjack/transports/beanstalkd.rb

@@ -0,0 +1,33 @@
+#!/usr/bin/env ruby
+
+$: << File.expand_path(File.join(File.dirname(__FILE__), '..', '..'))
+require 'beanstalk-client'
+require 'flapjack/transports/result'
+
+module Flapjack
+  module Transport
+    class Beanstalkd
+      def initialize(options={})
+        @options = options
+        @config = OpenStruct.new(options)
+
+        unless @config.host && @config.port && @config.queue_name
+          raise ArgumentError, "You need to specify a beanstalkd host, port, and queue name to connect to."
+        end
+
+        @queue = Beanstalk::Pool.new(["#{@config.host}:#{@config.port}"], @config.queue_name)
+      end
+
+      def next
+        job = @queue.reserve # blocks
+        result = YAML::load(job.body)
+        Flapjack::Transport::Result.new(:job => job, :result => result)
+      end
+
+      def delete(result)
+        result.job.delete
+      end
+    end
+  end
+end
+

data/lib/flapjack/transports/result.rb

@@ -0,0 +1,58 @@
+#!/usr/bin/env ruby
+
+module Flapjack
+  module Transport
+    class Result
+
+      attr_accessor :job, :result
+
+      def initialize(options={})
+        @job = options[:job]
+        @result = OpenStruct.new(options[:result])
+      end
+  
+      # Whether a check returns an ok status. 
+      def ok?
+        @result.retval == 0
+      end
+    
+      # Whether a check has a warning status.
+      def warning?
+        @result.retval == 1
+      end
+    
+      # Whether a check has a critical status.
+      def critical?
+        @result.retval == 2
+      end
+    
+      # Human readable representation of the check's return value.
+      def status
+        case @result.retval
+        when 0 ; "ok"
+        when 1 ; "warning"
+        when 2 ; "critical"
+        end
+      end
+   
+      # FIXME: there is a *lot* of duplication here - implement a proxy
+      # object pattern?
+      def id
+        @result.check_id
+      end
+
+      def check_id
+        @result.check_id
+      end
+
+      def command
+        @result.command
+      end
+
+      def frequency
+        @result.frequency
+      end
+
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: flapjack
 version: !ruby/object:Gem::Version 
-  version: 0.4.12
+  version: 0.5.1
 platform: ruby
 authors: 
 - Lindsay Holmwood
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-11-22 00:00:00 +01:00
+date: 2010-02-28 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -40,7 +40,7 @@ dependencies:
     requirements: 
     - - "="
       - !ruby/object:Gem::Version 
-        version: 1.0.5
+        version: 1.1.5
     version: 
 - !ruby/object:Gem::Dependency 
   name: xmpp4r
@@ -63,68 +63,29 @@ dependencies:
         version: 1.2.3.1
     version: 
 - !ruby/object:Gem::Dependency 
-  name: dm-core
+  name: yajl-ruby
   type: :runtime
   version_requirement: 
   version_requirements: !ruby/object:Gem::Requirement 
     requirements: 
     - - "="
       - !ruby/object:Gem::Version 
-        version: 0.9.11
+        version: 0.6.4
     version: 
 - !ruby/object:Gem::Dependency 
-  name: dm-timestamps
+  name: sqlite3-ruby
   type: :runtime
   version_requirement: 
   version_requirements: !ruby/object:Gem::Requirement 
     requirements: 
     - - "="
       - !ruby/object:Gem::Version 
-        version: 0.9.11
-    version: 
-- !ruby/object:Gem::Dependency 
-  name: dm-types
-  type: :runtime
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
-    requirements: 
-    - - "="
-      - !ruby/object:Gem::Version 
-        version: 0.9.11
-    version: 
-- !ruby/object:Gem::Dependency 
-  name: dm-validations
-  type: :runtime
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
-    requirements: 
-    - - "="
-      - !ruby/object:Gem::Version 
-        version: 0.9.11
-    version: 
-- !ruby/object:Gem::Dependency 
-  name: data_objects
-  type: :runtime
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
-    requirements: 
-    - - "="
-      - !ruby/object:Gem::Version 
-        version: 0.9.12
-    version: 
-- !ruby/object:Gem::Dependency 
-  name: do_sqlite3
-  type: :runtime
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
-    requirements: 
-    - - "="
-      - !ruby/object:Gem::Version 
-        version: 0.9.12
+        version: 1.2.5
     version: 
 description: Flapjack is highly scalable and distributed monitoring system. It understands the Nagios plugin format, and can easily be scaled from 1 server to 1000.
 email: lindsay@holmwood.id.au
 executables: 
+- flapjack-benchmark
 - flapjack-notifier
 - flapjack-notifier-manager
 - flapjack-stats
@@ -140,6 +101,7 @@ files:
 - README.md
 - Rakefile
 - TODO.md
+- bin/flapjack-benchmark
 - bin/flapjack-notifier
 - bin/flapjack-notifier-manager
 - bin/flapjack-stats
@@ -149,30 +111,58 @@ files:
 - doc/CONFIGURING.md
 - doc/DEVELOPING.md
 - doc/INSTALL.md
+- doc/PACKAGING.md
 - etc/default/flapjack-notifier
 - etc/default/flapjack-workers
-- etc/flapjack/flapjack-notifier.yaml.example
-- etc/flapjack/recipients.yaml.example
+- etc/flapjack/flapjack-notifier.conf.example
+- etc/flapjack/recipients.conf.example
 - etc/init.d/flapjack-notifier
 - etc/init.d/flapjack-workers
+- features/flapjack-notifier-manager.feature
+- features/flapjack-worker-manager.feature
+- features/packaging-lintian.feature
+- features/persistence/couch.feature
+- features/persistence/sqlite3.feature
+- features/persistence/steps/couch_steps.rb
+- features/persistence/steps/generic_steps.rb
+- features/persistence/steps/sqlite3_steps.rb
+- features/steps/flapjack-notifier-manager_steps.rb
+- features/steps/flapjack-worker-manager_steps.rb
+- features/steps/packaging-lintian_steps.rb
+- features/support/env.rb
+- features/support/silent_system.rb
 - flapjack.gemspec
+- lib/flapjack/applications/notifier.rb
+- lib/flapjack/applications/worker.rb
 - lib/flapjack/checks/http_content
+- lib/flapjack/checks/ping
 - lib/flapjack/cli/notifier.rb
 - lib/flapjack/cli/notifier_manager.rb
 - lib/flapjack/cli/worker.rb
 - lib/flapjack/cli/worker_manager.rb
-- lib/flapjack/database.rb
-- lib/flapjack/models/check.rb
-- lib/flapjack/models/check_template.rb
-- lib/flapjack/models/node.rb
-- lib/flapjack/models/related_check.rb
-- lib/flapjack/notifier.rb
+- lib/flapjack/filters/any_parents_failed.rb
+- lib/flapjack/filters/ok.rb
+- lib/flapjack/inifile.rb
+- lib/flapjack/notifier_engine.rb
 - lib/flapjack/notifiers/mailer/init.rb
 - lib/flapjack/notifiers/mailer/mailer.rb
 - lib/flapjack/notifiers/xmpp/init.rb
 - lib/flapjack/notifiers/xmpp/xmpp.rb
 - lib/flapjack/patches.rb
-- lib/flapjack/result.rb
+- lib/flapjack/persistence/couch.rb
+- lib/flapjack/persistence/couch/connection.rb
+- lib/flapjack/persistence/couch/couch.rb
+- lib/flapjack/persistence/data_mapper.rb
+- lib/flapjack/persistence/data_mapper/data_mapper.rb
+- lib/flapjack/persistence/data_mapper/models/check.rb
+- lib/flapjack/persistence/data_mapper/models/check_template.rb
+- lib/flapjack/persistence/data_mapper/models/event.rb
+- lib/flapjack/persistence/data_mapper/models/node.rb
+- lib/flapjack/persistence/data_mapper/models/related_check.rb
+- lib/flapjack/persistence/sqlite3.rb
+- lib/flapjack/persistence/sqlite3/sqlite3.rb
+- lib/flapjack/transports/beanstalkd.rb
+- lib/flapjack/transports/result.rb
 has_rdoc: true
 homepage: http://flapjack-project.com
 licenses: []

data/etc/flapjack/flapjack-notifier.yaml.example

@@ -1,8 +0,0 @@
---- 
-:notifiers: 
-  :mailer: 
-    :from_address: notifications@my-domain.com
-  :xmpp: 
-    :jid: notifications@my-domain.com
-    :password: foo
-:database_uri: "sqlite3:///var/lib/flapjack/flapjack.db"