Sha256: 01f375db6860743c221d9e56cfce0b421dbef634931aa78a95b5add45da69359

Contents?: true

Size: 1.95 KB

Versions: 1

Compression:

Stored size: 1.95 KB

Contents

require 'mongo'
require 'facets'

module YogiBerra
  class Catcher
    extend Facets
    cattr_accessor :settings, :mongo_client, :connection

    class << self
      def load_db_settings(config_file = nil)
        if config_file
          database_config = config_file
        elsif defined?(Rails)
          database_config = "#{Rails.root}/config/yogi.yml"
        else
          YogiBerra::Logger.log("No config file specified!", :error)
        end
        if database_config
          begin
            File.open(database_config, 'r') do |f|
              yaml_file = YAML.load(f)
              environment = ENV["RAILS_ENV"] ? ENV["RAILS_ENV"] : ENV["YOGI_ENV"]
              YogiBerra::Logger.log("I get here! #{environment.inspect}", :info)
              @@settings = yaml_file["#{environment}"] if yaml_file
            end
          rescue
            YogiBerra::Logger.log("No such file: #{database_config}", :error)
          end
        end
      end

      def db_client(host, port)
        # :w => 0 set the default write concern to 0, this allows writes to be non-blocking
        # by not waiting for a response from mongodb
        @@mongo_client = Mongo::MongoClient.new(host, port, :w => 0)
      rescue
        YogiBerra::Logger.log("Couldn't connect to the mongo database on host: #{host} port: #{port}.", :error)
        nil
      end

      def quick_connection
        YogiBerra::Logger.log("settings::::::::::: #{settings.inspect}", :info)
        settings = @@settings || load_db_settings

        if settings
          host = settings["host"]
          port = settings["port"]
          client = db_client(host, port)
          if client
            @@connection = client[settings["database"]]
          else
            YogiBerra::Logger.log("Couldn't connect to the mongo database on host: #{host} port: #{port}.", :error)
          end
        else
          YogiBerra::Logger.log("Couldn't load the yogi.yml file.", :error)
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version Path
yogi_berra-0.0.2 lib/yogi_berra/catcher.rb