ddtop 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in ddtop.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Genki Sugawara
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,47 @@
+# ddtop
+
+A tool that send to Datadog the data while execute `top`.
+
+[![Gem Version](https://badge.fury.io/rb/ddtop.png)](http://badge.fury.io/rb/ddtop)
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'ddtop'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install ddtop
+
+## Usage
+
+```sh
+$ export DD_API_KEY=...
+$ ddtop # The outputted data is transmitted to Datadog.
+top - 14:12:24 up  4:23,  4 users,  load average: 2.57, 2.11, 1.43
+Tasks: 116 total,   3 running, 113 sleeping,   0 stopped,   0 zombie
+Cpu(s): 27.0%us,  6.9%sy,  0.0%ni, 65.9%id,  0.0%wa,  0.0%hi,  0.2%si,  0.0%st
+Mem:  15343072k total,  1582800k used, 13760272k free,    19544k buffers
+Swap:        0k total,        0k used,        0k free,  1124444k cached
+
+  PID USER      PR  NI  VIRT  RES  SHR S %CPU %MEM    TIME+  COMMAND
+16901 ec2-user  20   0  125m 6000 1984 R 99.8  0.0   0:03.86 perl
+19501 ec2-user  20   0 4871m 232m 4536 S 69.8  1.5   5:48.33 ruby
+25238 ec2-user  20   0  107m 2024 1568 S  3.3  0.0   0:13.79 bash
+...
+```
+
+![datadog-metrics.png](https://bitbucket.org/winebarrel/ddtop/downloads/datadog-metrics.png)
+
+## Contributing
+
+1. Fork it ( http://github.com/<my-github-username>/ddtop/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/bin/ddtop

@@ -0,0 +1,13 @@
+#!/usr/bin/env ruby
+$: << File.expand_path("#{File.dirname __FILE__}/../lib")
+require 'ddtop'
+
+civis = system('tput civis')
+
+trap(:INT) do
+  puts "\e[0m"
+  system('tput cnorm') if civis
+  exit
+end
+
+Ddtop.run

data/ddtop.gemspec

@@ -0,0 +1,24 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ddtop/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'ddtop'
+  spec.version       = Ddtop::VERSION
+  spec.authors       = ['Genki Sugawara']
+  spec.email         = ['sgwr_dts@yahoo.co.jp']
+  spec.summary       = %q{A tool that send to Datadog the data while execute top}
+  spec.description   = %q{A tool that send to Datadog the data while execute top}
+  spec.homepage      = 'https://bitbucket.org/winebarrel/ddtop'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'dogapi'
+  spec.add_development_dependency 'bundler', '~> 1.5'
+  spec.add_development_dependency 'rake'
+end

data/lib/ddtop/ddtop.rb

@@ -0,0 +1,176 @@
+class Ddtop::Ddtop
+  TOP_CMD = 'top -b'
+  PREFIX = 'ddtop'
+
+  def initialize
+    api_key = ENV['DD_API_KEY']
+    host = %x[hostname -f 2> /dev/null].strip
+    host = Socket.gethostname if host.empty?
+    @dog = Dogapi::Client.new(api_key, nil, host)
+  end
+
+  def run
+    summary, fields = nil
+    values = []
+
+    head_hook = proc do
+      if summary and not values.empty?
+        emit_points(summary, fields, values)
+        values.clear
+      end
+    end
+
+    top_each_lien(head_hook) do |top, line|
+      if head?(line)
+        summary, fields = parse_header(top, line)
+      elsif not line.empty?
+        values << line.split(/\s+/, fields.length)
+      end
+    end
+  end
+
+  private
+
+  def top_each_lien(head_hook)
+    cmd = [TOP_CMD, ARGV].flatten.join(' ')
+
+    IO.popen(cmd) do |top|
+      while line = gets_with_print(top, &head_hook)
+        yield(top, line)
+      end
+    end
+  end
+
+  def gets_with_print(top)
+    line = top.gets
+
+    if head?(line)
+      @win_rows = win_rows
+      @curr_row = 0
+      yield if block_given?
+      clear_screen
+    end
+
+    @curr_row += 1
+
+    if @curr_row < @win_rows
+      print line.sub(/\s+\Z/, "\n")
+    elsif @curr_row == @win_rows
+      print line.sub(/\s+\Z/, '')
+    end
+
+    return line.strip
+  end
+
+  def head?(line)
+    line =~ /\Atop\b/
+  end
+
+  def parse_header(top, line)
+    summary_lines = [line] + (1..4).map { gets_with_print(top) }
+    gets_with_print(top)
+    fields_line = gets_with_print(top)
+
+    summary = parse_summary(summary_lines)
+    fields = fields_line.split(/\s+/)
+
+    [summary, fields]
+  end
+
+  def parse_summary(lines)
+    summary = {}
+    first_line = lines[0].split(/[\s,]+/)
+    summary[:time] = parse_time(first_line[2])
+    load_avg = first_line.values_at(-3, -2, -1)
+    summary[:load_avg] = Hash[*['1m', '5m', '15m'].zip(load_avg).flatten]
+    summary[:tasks]    = line_to_items(lines[1]) {|i| i.each_slice(2).map {|i, j| [j, i] } }
+    summary[:cpu]      = line_to_items(lines[2]) {|i| i.map {|j| j.split('%').values_at(1, 0) } }
+    summary[:mem]      = line_to_items(lines[3]) {|i| i.each_slice(2).map {|i, j| [j, i] } }
+    summary[:swap]     = line_to_items(lines[4]) {|i| i.each_slice(2).map {|i, j| [j, i] } }
+    return summary
+  end
+
+  def parse_time(time)
+    now = Time.now
+    Time.mktime(now.year, now.mon, now.day, *time.split(':').map {|i| i.to_i })
+  end
+
+  def line_to_items(line)
+    items = yield(line.split(/[\s,:]+/).slice(1..-1))
+    Hash[*items.flatten]
+  end
+
+  def emit_points(summary, names, values)
+    time = summary.delete(:time)
+    fields = aggregate_fields(names, values)
+    emit_points0(time, summary)
+    emit_points0(time, fields)
+  end
+
+  def emit_points0(time, metrics)
+    metrics.each do |category, items|
+      items.each do |name, value|
+        metric_name = [PREFIX, category, name].join('.')
+
+        Thread.start do
+          @dog.emit_point(metric_name, expand_value(value), :timstamp => time)
+        end
+      end
+    end
+  rescue Timeout::Error
+    # nothing to do
+  end
+
+  def aggregate_fields(names, values)
+    aggregated = {}
+
+    values.map {|v|
+      h = Hash[*names.zip(v).flatten]
+      %w(PID USER PR NI S).each {|k| h.delete(k) }
+      h['TIME+'] = parse_time_plus(h['TIME+'])
+      h
+    }.sort_by {|h|
+      h['COMMAND']
+    }.chunk {|h|
+      h['COMMAND']
+    }.each {|cmd, hs|
+      total = {}
+
+      hs.each do |h|
+        h.delete('COMMAND')
+
+        h.each do |k, v|
+          total[k] ||= 0.0
+          total[k] += v.to_f
+        end
+      end
+
+      aggregated['process.' + cmd] = total
+    }
+
+    return aggregated
+  end
+
+  def parse_time_plus(time)
+    t = time.split(/[.:]/).map {|i| i.to_f }
+    t[0] * 60 + t[1] + t[2] / 100
+  end
+
+  def win_rows
+    `stty -a`.split("\n").first.split(';')[1].strip.split(/\s+/, 2)[1].to_i
+  end
+
+  def clear_screen
+    print "\e[2J\n"
+  end
+
+  def expand_value(value)
+    n = 0
+
+    case value
+    when /m\Z/ then n = 1
+    end
+
+    value.to_f * (1024 ** n)
+  end
+end

data/lib/ddtop/version.rb

@@ -0,0 +1,3 @@
+module Ddtop
+  VERSION = '0.0.1'
+end

data/lib/ddtop.rb

@@ -0,0 +1,12 @@
+module Ddtop
+  def self.run
+    ::Ddtop::Ddtop.new.run
+  end
+end
+
+require 'dogapi'
+require 'socket'
+require 'thread'
+
+require 'ddtop/version'
+require 'ddtop/ddtop'

metadata

@@ -0,0 +1,105 @@
+--- !ruby/object:Gem::Specification
+name: ddtop
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Genki Sugawara
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-02-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: dogapi
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A tool that send to Datadog the data while execute top
+email:
+- sgwr_dts@yahoo.co.jp
+executables:
+- ddtop
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/ddtop
+- ddtop.gemspec
+- lib/ddtop.rb
+- lib/ddtop/ddtop.rb
+- lib/ddtop/version.rb
+homepage: https://bitbucket.org/winebarrel/ddtop
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: A tool that send to Datadog the data while execute top
+test_files: []