amqp-tools 0.0.2

data/amqp-tools.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "amqp-tools"
+  s.version     = "0.0.2"
+  s.date        = Date.today.to_s
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Paul Asmuth"]
+  s.email       = ["paul@paulasmuth.com"]
+  s.homepage    = "http://github.com/paulasmuth/amqp-tools"
+  s.summary     = %q{command line tools for AMQP/RabbitMQ}
+  s.description = %q{command line tools for AMQP/RabbitMQ}
+  s.licenses    = ["MIT"]
+
+  s.add_dependency "amqp", ">= 0.8.9"
+
+  s.bindir        = "bin"
+  s.executables   = ["amqp-send", "amqp-linux-health-monitor", "amqp-json-udp-acceptor"]
+  s.files         = `git ls-files`.split("\n") - [".gitignore", ".rspec", ".travis.yml"]
+  s.test_files    = `git ls-files -- spec/*`.split("\n")
+  s.require_paths = ["lib"]
+end

data/bin/amqp-json-udp-acceptor

@@ -0,0 +1,89 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+begin
+  require "amqp"
+rescue LoadError
+  puts "can't require amqp gem, please run"
+  puts
+  puts "    $ gem install amqp"
+  puts
+  exit 1
+end
+
+require "eventmachine"
+require "json"
+
+class UDPAcceptor < EventMachine::Connection
+
+  def self.run(opts = {})
+    EventMachine.run do
+      error_handler = proc do |*args|
+        log("[error] AMQP Error")
+        shutdown(1)
+      end
+
+      amqp_opts = {
+        :host => opts[:amqp_host],
+        :port => (opts[:amqp_port] || 5672),
+        :timeout => 1,
+        :on_possible_authentication_failure => error_handler,
+        :on_tcp_connection_failure => error_handler
+      }
+
+      begin
+        $amqp = AMQP.start(amqp_opts)
+      rescue AMQP::PossibleAuthenticationFailureError, AMQP::TCPConnectionFailed
+        shutdown(1)
+      end
+
+      EventMachine.open_datagram_socket(
+        "0.0.0.0", opts[:listen_port], self)
+    end
+  end
+
+  def self.log(msg)
+    puts "[#{Time.now.strftime("%F %T")}] #{msg}"
+  end
+
+  def self.shutdown(code = 0)
+    if EventMachine.reactor_running?
+      EventMachine.stop{ exit(code) }
+    end
+  end
+
+  def post_init
+    @amqp_channel  = AMQP::Channel.new($amqp)
+    @amqp_exchange = @amqp_channel.default_exchange
+    self.class.log("Listening...")
+  end
+
+  def receive_data(chunk)
+    if chunk[0] == '{'
+      channel = JSON.parse(chunk)["channel"] rescue nil
+      return unless channel
+      self.class.log "#{channel} => #{chunk}"
+      @amqp_exchange.publish chunk, :routing_key => channel
+    else
+      self.class.log "[error] not a json object: #{chunk[0..60]}"
+      return
+    end
+  end
+
+end
+
+unless ENV["LISTEN_PORT"] && ENV["AMQP_HOST"]
+  puts "usage: [ENV] #{$0}"
+  puts
+  puts "Environment Variables:"
+  puts "    AMQP_HOST=<addr>     amqp server address (mandatory)"
+  puts "    AMQP_PORT=<port>     amqp server port"
+  puts "    LISTEN_PORT=<port>   start udp listener on this port (mandatory)"
+  puts
+  exit 1
+end
+
+UDPAcceptor.run(
+  :amqp_host => "firehose",
+  :amqp_port => "5672",
+  :listen_port => 2323)

data/bin/amqp-linux-health-monitor

@@ -0,0 +1,69 @@
+#!/bin/bash
+
+# This scripts sends a JSON message containing system health
+# information to an AMQP broker.
+#
+# Usage:
+#   $ ./linux_health_monitor.sh [amqp-send options]
+# e.g.
+#   $ ./linux_health_monitor.sh -h localhost --routing-key serverhealth
+#
+# The JSON body will look like this:
+# {
+#   "hostname": "my-hostname",
+#   "uptime": 1231523,
+#   "load_avg": [0.32, 0.16, 0.13],
+#   "mem": { "total": 4069, "free": 1096 },
+#   "swap": { "total": 0, "free": 0 },
+#   "disk_usage": "43%"
+# }
+
+if [[ $# -lt 1 ]]; then
+  echo "usage: $0 [amqp-send options]"
+  exit 1
+fi
+
+for cmd in df wc nc sed grep amqp-send; do
+  which $cmd &> /dev/null
+
+  if [ $? != 0 ]; then
+    echo "error: command not found: $cmd"
+    exit 1
+  fi
+done
+
+data=$(
+
+  # hostname
+  echo "\"hostname\": \"$(hostname)\","
+
+  # uptime
+  cat /proc/uptime | \
+    sed -e 's/^\([0-9]\+\.[0-9]\+\).*/"uptime": \1,/'
+
+  # load_avg
+  cat /proc/loadavg | \
+    sed -e 's/^\([0-9]\+\.[0-9]\+\) \([0-9]\+\.[0-9]\+\) \([0-9]\+\.[0-9]\+\).*/"load_avg": [\1, \2, \3],/'
+
+  # mem usage
+  cat /proc/meminfo | \
+    grep -E '^(MemTotal|MemFree)' | \
+    sed -e 's/Mem//' -e 's/Total:[^0-9]\+\([0-9]\+\).*/"mem": { "total": \1,/' \
+        -e 's/Free:[^0-9]\+\([0-9]\+\).*/"free": \1 },/'
+
+  # swap_usage
+  cat /proc/meminfo | \
+    grep -E '^(SwapTotal|SwapFree)' | \
+    sed -e 's/Swap//' -e 's/Total:[^0-9]\+\([0-9]\+\).*/"swap": { "total": \1,/' \
+        -e 's/Free:[^0-9]\+\([0-9]\+\).*/"free": \1 },/'
+
+  # disk usage
+    df | grep -E '\/$' | \
+      sed -e 's/.* \([0-9]\+%\) .*/"disk_usage": "\1"/'
+
+)
+
+echo "{ $data }" | sed -e ':x;N;$!bx;s/\n/ /g' | \
+  amqp-send $@
+
+exit $?

data/bin/amqp-send

@@ -0,0 +1,124 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+
+begin
+  require "amqp"
+rescue LoadError
+  puts "can't require amqp gem, please run"
+  puts
+  puts "    $ gem install amqp"
+  puts
+  exit 1
+end
+
+require "eventmachine"
+require 'optparse'
+
+opts = {
+  :host => ENV["AMQP_HOST"],
+  :port => ENV["AMQP_PORT"]
+}
+
+opt_parser = OptionParser.new do |base|
+
+  base.on("-h", "--host HOST", "AMQP server host (mandatory)") do |arg|
+    opts[:host] = arg
+  end
+
+  base.on("-p", "--port PORT", "AMQP server port") do |arg|
+    opts[:port] = arg
+  end
+
+  base.on("--routing-key KEY", "AMQP routing key (mandatory)") do |arg|
+    opts[:routing_key] = arg
+  end
+
+  #base.on("--exchange EXCHANGE", "AMQP exchange") do |arg|
+  #  opts[:exchange] = arg
+  #end
+
+  base.on("-f", "--file FILE", "read payload from file") do |arg|
+    opts[:file] = arg
+  end
+
+end
+
+opt_parser.parse!
+
+examples = %Q{
+    $ echo "fnord" | amqpsend -h localhost --routing-key my_channel
+    $ amqpsend -h localhost --routing-key my_channel -f /tmp/somefile
+}
+
+mandatory_opts = %w(host routing_key)
+mandatory_opts.each do |opt|
+  unless opts[opt.to_sym]
+    puts "[error] missing option: #{opt}\n\n"
+    puts "Usage: #{$0} [options]\n\nOptions:\n"
+    puts opt_parser.summarize
+    puts "\nThe following environment variables may also be set:"
+    puts "    AMQP_HOST, AMQP_PORT"
+    puts "\nIf no -f option is specified the payload will be read from STDIN\n\n"
+    puts "Examples: #{examples}"
+    exit 1
+  end
+end
+
+error_handler = proc do |*args|
+  puts("[error] AMQP Error")
+  exit 1
+end
+
+amqp_opts = {
+  :host => opts[:host],
+  :port => opts[:port] || 5672,
+  :timeout => 1,
+  :on_possible_authentication_failure => error_handler,
+  :on_tcp_connection_failure => error_handler
+}
+
+publish_opts = {
+  :routing_key => opts[:routing_key],
+  :auto_delete => true
+}
+
+payload = if opts[:file]
+  IO.read(opts[:file])
+else
+  unless IO.select([STDIN], [], [], 0.01)
+    puts "Reading from STDIN..."
+  end
+
+  STDIN.read
+end
+
+if !payload || payload.size == 0
+  puts("[error] no payload (read from STDIN)")
+  exit 1
+end
+
+EventMachine.run do
+  amqp          = AMQP.start(amqp_opts)
+  amqp_channel  = AMQP::Channel.new(amqp)
+  amqp_exchange = amqp_channel.default_exchange
+
+  amqp_exchange.publish(payload, publish_opts) do
+    amqp.close { exit 0 }
+  end
+end
+
+unless ENV["LISTEN_PORT"] && ENV["AMQP_HOST"]
+  puts "usage: [ENV] #{$0}"
+  puts
+  puts "Environment Variables:"
+  puts "    AMQP_HOST=<addr>       amqp server address (mandatory)"
+  puts "    AMQP_PORT=<port>       amqp server port"
+  puts "    AMQP_RKEY=<key>        amqp routing key (mandatory)"
+  puts
+  exit 1
+end
+
+UDPAcceptor.run(
+  :amqp_host => "firehose",
+  :amqp_port => "5672",
+  :listen_port => 2323)

metadata

@@ -0,0 +1,64 @@
+--- !ruby/object:Gem::Specification
+name: amqp-tools
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Paul Asmuth
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: amqp
+  requirement: &13890980 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.8.9
+  type: :runtime
+  prerelease: false
+  version_requirements: *13890980
+description: command line tools for AMQP/RabbitMQ
+email:
+- paul@paulasmuth.com
+executables:
+- amqp-send
+- amqp-linux-health-monitor
+- amqp-json-udp-acceptor
+extensions: []
+extra_rdoc_files: []
+files:
+- amqp-tools.gemspec
+- bin/amqp-json-udp-acceptor
+- bin/amqp-linux-health-monitor
+- bin/amqp-send
+homepage: http://github.com/paulasmuth/amqp-tools
+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.17
+signing_key: 
+specification_version: 3
+summary: command line tools for AMQP/RabbitMQ
+test_files: []