config.rb

require 'sinatra/base'

module Sinatra
  module Config
    def self.registered(app)
      app.configure do
        app.set :wsrep_state_dir, '/etc/mysql/wsrep'

        app.set(:hostname) { `hostname`.chomp }

        logger = ::Logger.new('log/check.log')
        logger.level = ::Logger::INFO
        app.set :check_logger, logger

        error_logger = ::File.new(::File.join('log', 'error.log'), 'a+')
        error_logger.sync = true
        app.set :error_logger, error_logger

        app.set :checkers, {}

        app.set :max_staleness, 10
        app.set :check_interval, 1

        app.set :statsd_host, '127.0.0.1'
        app.set :statsd_port, 8125

        app.set :services, [:xdb, :zk]
      end

      app.configure :test do
        app.set :connection_settings, {
          host: 'localhost',
          username: 'root',
          database: 'health_check',
          reconnect: true
        }
        app.set :zk_connection_settings, {
          host: 'localhost',
          port: 2181,
          timeout: 5
        }
        app.set :zk_outstanding_threshold, 60
        app.set :hostname, 'test.local'

        logger = Logger.new('/dev/null')
        logger.level = Logger::DEBUG
        app.set :check_logger, logger

        app.set :statsd_host, '0.0.0.0'
      end

      app.configure :development do
        app.set :wsrep_state_dir, 'spec/data/3_node_cluster_synced'
        app.set :connection_settings, {
          host: 'localhost',
          username: 'root',
          database: 'health_check',
          reconnect: true
        }

        app.set :zk_connection_settings, {
          host: 'localhost',
          port: 2181,
          timeout: 5
        }
        app.set :zk_outstanding_threshold, 60

        app.set :hostname, 'dev.local'

        logger = Logger.new(STDOUT)
        logger.level = Logger::DEBUG
        app.set :check_logger, logger
      end

      app.configure :production do
        config = YAML.load_file ENV['WHAZZUP_CONFIG']
        app.set :services, config['services'].map(&:to_sym)

        app.set :check_interval, config['check_interval'].to_i if config['check_interval']

        if config['connection_settings']
          app.set :connection_settings, {
            host: 'localhost',
            username: config['connection_settings']['username'],
            password: config['connection_settings']['password'],
            database: 'health_check',
            reconnect: true
          }
        end

        if config['zk']
          app.set :zk_connection_settings, {
            host: config['zk']['host'],
            port: config['zk']['port'],
            timeout: config['zk']['timeout']
          }
          app.set :zk_outstanding_threshold, config['zk']['outstanding_threshold']
        end
      end
    end
  end
end