Sha256: 6c76a1f96aea55cfa118b386a295e182e86f817af2a7f1e8fff65cfa07b45a63

Contents?: true

Size: 1.6 KB

Versions: 21

Compression:

Stored size: 1.6 KB

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.6.39 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.38 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.37 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.36 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.35 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.34 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.33 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.32 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.31 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.30 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.29 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.28 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.27 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.26 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.25 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.24 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.6.23 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.5.5 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.5.4 lib/flapjack/persistence/data_mapper/data_mapper.rb
flapjack-0.5.3 lib/flapjack/persistence/data_mapper/data_mapper.rb