Contents

require 'sequel'
require 'sqlite3'

module Ring
class SQA

  class Database
    def add record
      record[:time]    = Time.now.utc.to_i
      record[:latency] = nil
      record[:result]  = 'no response'
      ping = Ping.new(record).save
      Log.debug "added '#{record}' to database with id #{ping.id}" if CFG.debug?
      return ping
    end

    def update record_id, result, latency=nil
      if record = Ping[record_id]
        Log.debug "updating record_id '#{record_id}' with result '#{result}' and latency '#{latency}'" if CFG.debug?
        record.update(:result=>result, :latency=>latency)
      else
        Log.error "wanted to update record_id #{record_id}, but it does not exist"
      end
    end

    def nodes_down first_id
      max_id = (Ping.max(:id) or first_id)
      [max_id, id_range(first_id, max_id).exclude(:result => 'ok')]
    end

    def up_since? id, peer
      Ping.where{id > id}.where(:peer=>peer).count > 0
    end

    def purge older_than=3600
      Ping.where{time < (Time.now.utc-older_than).to_i}.delete
    end

    def id_range first, last
      Ping.distinct.where(:id=>first..last)
    end

    def table_exists?
      @db.table_exists?(:pings)
    end

    private

    def initialize
      sequel_opts = { max_connections: 1, pool_timout: 60 }
      if CFG.ram_database?
        @db = Sequel.sqlite sequel_opts
      else
        file = '%s.db' % CFG.afi
        file = File.join CFG.directory, file
        File.unlink file rescue nil # delete old database
        @db = Sequel.sqlite file, sequel_opts
      end
      create_db
      require_relative 'database/model.rb'
    end

    def create_db
      @db.create_table?(:pings) do
        primary_key :id
        Fixnum      :time
        String      :peer
        Fixnum      :latency
        String      :result
      end
    end
  end

end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
ring-sqa-0.4.4	lib/ring/sqa/database.rb
ring-sqa-0.4.3	lib/ring/sqa/database.rb