pimon 0.1.11 → 0.2.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a8ca73f2a9b9b25e474fb1104f343eeba425dbba
+  data.tar.gz: 4fdee4b13a77a452c4023aed9e3701c23764d3f7
+SHA512:
+  metadata.gz: b489ca2b46048bd2e5f1b7d8d43ef02807c62212e43622cc84933499bfa83537c59c86f69c7e8cd90fe8672d7f6da4c9ff9f50594509e4ee0b3eb112a1ababce
+  data.tar.gz: 40dd2febfa8e037079beb66839abaaf61d0fe9aba177acb2ec9c41d9cc0d22a851761e66b5fc97658f66894b04c59565727d5c8f90f175024e2415f189eed515

data/bin/pimon

@@ -1,5 +1,4 @@
 #!/usr/bin/env ruby
-$: << "#{File.dirname(__FILE__)}/../lib"
 require 'optparse'
 
 options = {}
@@ -13,20 +12,20 @@ Options:
           'YAML configuration file for pimon') do |config|
     options[:pimon_config] = config
   end
-  
+
   opts.on('-d',
           '--daemonize',
           'Run Pimon daemonized in the background') do
     options[:daemonize] = true
   end
-  
+
   opts.on('-e ENVIRONMENT',
           '--environment ENVIRONMENT',
           'Application environment (default: "development", options: "development", "production")',
           /^development|production$/) do |environment|
     options[:environment] = environment
   end
-  
+
   opts.on('-i INTERFACE',
           '--interface INTERFACE',
           'Hostname or IP address of the interface to listen on (default: "localhost")') do |interface|
@@ -38,16 +37,16 @@ Options:
           'Port to use (default: 3000)') do |port|
     options[:port] = port
   end
-  
+
   opts.on('-P PIDFILE',
           '--pid PIDFILE',
           'File to store PID (default: /tmp/pimon.pid)') do |pid_file|
     options[:pid_file] = pid_file
   end
-  
+
   opts.on('-v',
           '--version',
-          'Display Pimon version') do |pid_file|
+          'Display Pimon version') do |_|
     ARGV[0] = 'version'
   end
 end
@@ -55,13 +54,13 @@ end
 begin
   option_parser.parse!
   if ARGV.empty?
-    puts "error: you must supply an action"
+    puts 'error: you must supply an action'
     puts option_parser.help
     exit 1
   end
-  
+
   pid_file = options[:pid_file] || '/tmp/pimon.pid'
-  
+
   case ARGV[0]
   when 'start'
     require 'thin'
@@ -72,27 +71,35 @@ begin
     ENV['PIMON_CONFIG'] = options[:pimon_config] if options[:pimon_config]
     puts "Pimon is starting at http://#{interface}:#{port}"
     puts "Running in #{ENV['RACK_ENV']} mode."
-    
+
     if ENV['PIMON_CONFIG']
       puts "Using configuration file #{ENV['PIMON_CONFIG']}"
     else
-      puts "Using default configuration"
+      puts 'Using default configuration'
     end
-    
-    server = Rack::Server.new(:config => config, :daemonize => options[:daemonize], :pid => pid_file, :Port => port, :server => 'thin', :Host => interface)
+
+    server_options = {
+      config: config,
+      daemonize: options[:daemonize],
+      pid: pid_file,
+      Port: port,
+      server: 'thin',
+      Host: interface
+    }
+
+    server = Rack::Server.new(server_options)
     server.start
   when 'stop'
     if File.file?(pid_file)
-      pid = `cat #{pid_file}`.to_i
-      Process.kill("KILL", pid)
-      File.delete(pid_file) if File.exists?(pid_file)
+      pid = File.read(pid_file).to_i
+      Process.kill('KILL', pid)
+      File.delete(pid_file) if File.exist?(pid_file)
       puts 'Pimon stopped...'
     else
       puts "Pid file not found at #{pid_file}, please supply a valid pid_file using -p or --pid"
     end
   when 'version'
-    require 'pimon/version'
-    puts "Pimon version is #{Pimon::VERSION}"
+    puts "Pimon version is #{Gem::Specification.load("#{File.dirname(__FILE__)}/../pimon.gemspec").version}"
   else
     STDERR.puts option_parser
   end

data/config/config.ru

@@ -1,4 +1,12 @@
-$: << "#{File.dirname(__FILE__)}/.."
-require 'lib/pimon'
+if ENV['RACK_ENV'] == 'development'
+  require 'rubygems'
+  require 'bundler'
 
-run Pimon
+  Bundler.require
+end
+
+require_relative '../lib/pimon'
+
+$PROGRAM_NAME = 'pimon'
+
+run Pimon::App

data/config/default.yml

@@ -1,14 +1,8 @@
-chart:
-  cpu:
-    color: '#D2691E'
-  disk:
-    color: '#CDC673'
-  mem:
-    color: '#87CEFA'
-  temp:
-    color: '#FF9B04'
-  swap:
-    color: '#3CB371'
+colors:
+  cpu:  '#D2691E'
+  disk: '#CDC673'
+  mem:  '#87CEFA'
+  temp: '#FF9B04'
+  swap: '#3CB371'
 stats_collector:
-  number_of_stats: 6
-  time_period_in_secs: 30
+  time_period_in_secs: 10

data/config/test.yml

@@ -1,15 +1,9 @@
-chart:
-  cpu:
-    color: '#D2691E'
-  disk:
-    color: '#CDC673'
-  mem:
-    color: '#87CEFA'
-  temp:
-    color: '#FF9B04'
-  swap:
-    color: '#3CB371'
+colors:
+  cpu:  '#D2691E'
+  disk: '#CDC673'
+  mem:  '#87CEFA'
+  temp: '#FF9B04'
+  swap: '#3CB371'
 hostname: 'test_hostname'
 stats_collector:
-  number_of_stats: 6
   time_period_in_min: 10

data/config/test_broken.yml

@@ -8,5 +8,4 @@ chart:
   swap:
     color: '#3CB371'
 stats_collector:
-  number_of_stats: 6
   time_period_in_min: 10

data/lib/pimon.rb

@@ -1,71 +1,4 @@
-$: << File.dirname(__FILE__)
+require 'pimon/app'
 
-if File.file?('Gemfile')
-  require 'bundler/setup'
-  Bundler.require(:default, ENV['RACK_ENV'])
-end
-require 'eventmachine'
-require 'json'
-require 'haml'
-require 'pimon/pimon_config'
-require 'pimon/stats_collector'
-require 'sinatra'
-require 'sinatra-websocket'
-
-class Pimon < Sinatra::Base
-  set :public_folder, "#{File.dirname(__FILE__)}/pimon/public"
-  set :views,         "#{File.dirname(__FILE__)}/pimon/views"
-  set :sockets, []
-
-  configure :development, :production do
-    filename = "#{File.dirname(__FILE__)}/../config/default.yml"
-    config = PimonConfig.create_new(ENV['PIMON_CONFIG'] || filename)
-    
-    EventMachine::next_tick do
-      settings.timer = EventMachine::add_periodic_timer(config.stats[:time_period_in_secs]) do
-        settings.stats_checker.collect_stats
-        @stats = settings.stats_checker.show_stats
-        
-        settings.sockets.each{ |s| s.send(@stats) }
-      end
-    end
-    
-    set :config, config
-    set :stats_checker, StatsCollector.new(config)
-    set :timer, nil
-    
-    settings.stats_checker.collect_stats
-    @stats = settings.stats_checker.show_stats
-  end
-  
-  configure :test do
-    config = PimonConfig.create_new("#{File.dirname(__FILE__)}/../config/test.yml")
-    
-    set :config, config
-    set :stats_checker, StatsCollector.new(config)
-    set :timer, nil
-  end
-  
-  get '/' do
-    if !request.websocket?
-      last_update = settings.stats_checker.last_update
-      last_modified(last_update) if ENV['RACK_ENV'] != 'development' && last_update
-      
-      haml :index
-    else
-      request.websocket do |ws|
-        ws.onopen do
-          settings.sockets << ws
-          @stats ||= settings.stats_checker.show_stats
-          ws.send(@stats)
-        end
-        ws.onmessage do |msg|
-          puts "message received!! -> #{msg}"
-        end
-        ws.onclose do
-          settings.sockets.delete(ws)
-        end
-      end
-    end
-  end
+module Pimon
 end

data/lib/pimon/app.rb

@@ -0,0 +1,81 @@
+require 'eventmachine'
+require 'json'
+require 'haml'
+require 'sinatra'
+require 'faye/websocket'
+
+require 'pimon/config'
+require 'pimon/stats_collector'
+
+Faye::WebSocket.load_adapter('thin')
+
+module Pimon
+  class App < Sinatra::Base
+    set :public_folder, "#{File.dirname(__FILE__)}/public"
+    set :views,         "#{File.dirname(__FILE__)}/views"
+    set :connected_websockets, []
+
+    configure :development, :production do
+      filename = "#{File.dirname(__FILE__)}/../../config/default.yml"
+      config = Pimon::Config.load(ENV['PIMON_CONFIG'] || filename)
+
+      set :config, config
+      set :stats_collector, Pimon::StatsCollector.new(config)
+      set :timer, nil
+    end
+
+    configure :test do
+      config = Pimon::Config.load("#{File.dirname(__FILE__)}/../../config/test.yml")
+
+      set :config, config
+      set :stats_collector, Pimon::StatsCollector.new(config)
+      set :timer, nil
+    end
+
+    def initialize(*args, &bk)
+      super
+      start_collecting_stats
+      self
+    end
+
+    get '/' do
+      locals = {
+        charts: settings.stats_collector.charts,
+        single_stats: settings.stats_collector.single_stats,
+        hostname: settings.config.hostname
+      }
+
+      haml :index, locals: locals
+    end
+
+    get '/stats' do
+      if Faye::WebSocket.websocket?(request.env)
+        websocket = Faye::WebSocket.new(request.env)
+
+        websocket.on(:open) do
+          settings.connected_websockets << websocket
+          websocket.send(settings.stats_collector.show_stats.to_json)
+        end
+
+        websocket.on(:close) do
+          settings.connected_websockets.delete(websocket)
+        end
+
+        websocket.rack_response
+      else
+        halt 404
+      end
+    end
+
+    def start_collecting_stats
+      return if settings.test?
+      settings.stats_collector.collect_stats
+      EventMachine.next_tick do
+        settings.timer = EventMachine.add_periodic_timer(settings.config.stats['time_period_in_secs']) do
+          settings.stats_collector.collect_stats
+          settings.connected_websockets.each { |s| s.send(settings.stats_collector.show_stats.to_json) }
+        end
+      end
+    end
+  end
+end

data/lib/pimon/config.rb

@@ -0,0 +1,34 @@
+require 'yaml'
+
+module Pimon
+  class Config
+    attr_reader :config
+    attr_reader :hostname
+
+    def self.load(filename)
+      new(filename)
+    end
+
+    def colors
+      config['colors']
+    end
+
+    def stats
+      config['stats_collector']
+    end
+
+    def hostname
+      @hostname ||= config['hostname'].nil? ? `hostname` : config['hostname']
+    end
+
+    private
+
+    def initialize(filename)
+      @config = YAML.load_file(filename)
+      @config.freeze
+    rescue StandardError => e
+      puts "Error while loading config file: #{filename}"
+      raise e
+    end
+  end
+end

data/lib/pimon/probe.rb

@@ -0,0 +1,15 @@
+require 'pimon/probe/cpu_freq'
+require 'pimon/probe/cpu_usage'
+require 'pimon/probe/disk_usage'
+require 'pimon/probe/memory_usage'
+require 'pimon/probe/swap_usage'
+require 'pimon/probe/system_memory'
+require 'pimon/probe/temperature'
+require 'pimon/probe/uptime'
+
+# Monkeypatching OpenStruct to render correct json in a probe sample
+class OpenStruct
+  def to_json(options)
+    table.to_json(options)
+  end
+end

data/lib/pimon/probe/cpu_freq.rb

@@ -0,0 +1,14 @@
+module Pimon
+  module Probe
+    class CpuFreq
+      def self.check(date = Time.now)
+        value = `cat /sys/devices/system/cpu/cpu0/cpufreq/scaling_cur_freq`.to_i / 1000
+        OpenStruct.new(date: date.strftime('%Y-%m-%d %H:%M:%S'), probe_name: 'cpu_freq', value: value, unit: unit)
+      end
+
+      def self.unit
+        'Mhz'
+      end
+    end
+  end
+end

data/lib/pimon/probe/cpu_usage.rb

@@ -1,11 +1,14 @@
-require 'pimon/probe/probe'
+module Pimon
+  module Probe
+    class CpuUsage
+      def self.check(date = Time.now)
+        value = 100 - `vmstat 1 2`.split(/\n/)[3].split(' ')[14].to_i
+        OpenStruct.new(date: date.strftime('%Y-%m-%d %H:%M:%S'), probe_name: 'cpu', value: value, unit: unit)
+      end
 
-class Probe::CpuUsage < Probe
-  def self.check
-    100 - `vmstat 1 2`.split(/\n/)[3].split(" ")[14].to_i
-  end
-  
-  def self.symbol
-    :cpu
+      def self.unit
+        '%'
+      end
+    end
   end
 end

data/lib/pimon/probe/disk_usage.rb

@@ -1,11 +1,14 @@
-require 'pimon/probe/probe'
+module Pimon
+  module Probe
+    class DiskUsage
+      def self.check(date = Time.now)
+        value = `df`.split(/\n/)[1].split(' ')[4].to_i
+        OpenStruct.new(date: date.strftime('%Y-%m-%d %H:%M:%S'), probe_name: 'disk', value: value, unit: unit)
+      end
 
-class Probe::DiskUsage < Probe
-  def self.check
-    `df`.split(/\n/)[1].split(" ")[4].to_i
-  end
-  
-  def self.symbol
-    :disk
+      def self.unit
+        '%'
+      end
+    end
   end
 end