Contents

#!/usr/bin/env ruby 

require 'dm-core'
require 'dm-validations'
require 'dm-types'
require 'dm-timestamps'

Dir.glob(File.join(File.dirname(__FILE__), 'models', '*.rb')).each do |model|
  require model
end

module Flapjack
  module Persistence
    class DataMapper
      def initialize(options={})
        @options = options
        @config = OpenStruct.new(options)
        @log = @config.log
        connect
      end

      def any_parents_failed?(result)
        check = Check.get(result.check_id)
        if check
          check.worst_parent_status != 0
        else
          @log.warning("Check with id #{result.check_id} doesn't exist!")
          false # this will notify
        end
      end

      def save(result)
        check = Check.get(result.check_id)
        if check
          check.status = result.status
          check.save
        else
          @log.warning("Check with id #{result.check_id} doesn't exist!")
          true
        end
      end

      def create_event(result)
        event = Event.new(:check_id => result.check_id)
        event.save
      end

      private 
      def connect
        raise ArgumentError, "Database URI wasn't specified" unless @config.uri
        ::DataMapper.setup(:default, @config.uri)
        validate_structure
      end

      def validate_structure
        begin
          ::DataMapper.repository(:default).adapter.execute("SELECT 'id' FROM 'checks';")
        rescue Sqlite3Error => e
          @log.warning("The specified database doesn't appear to have any structure!")
          @log.warning("Exiting.")
          raise
        end
      end
    end
  end
end

Version data entries

21 entries across 21 versions & 1 rubygems

Version	Path
flapjack-0.5.1	lib/flapjack/persistence/data_mapper/data_mapper.rb