hastats 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,9 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in hastats.gemspec
+gemspec
+
+group :development do
+  gem 'rake'
+end
+

data/README.rdoc

@@ -0,0 +1,14 @@
+== HAStats -- An HAProxy socket tool for Ruby
+
+HAStats communicates with a running HAProxy process through the "stats socket" (not enabled by default).  Through this socket you can query stats, reset counters, change server weightings, put servers in maintenance mode, and more.
+
+=== Using the command line tool
+
+  $ hastats info
+    Name: HAProxy
+    Version: 1.4.8
+    Release_date: 2010/06/16
+    Nbproc: 1
+    Process_num: 1
+    ...
+

data/Rakefile

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

data/bin/hastats

@@ -0,0 +1,70 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'hastats'
+require 'thor'
+
+class HAStats::Bin < Thor
+  map %w(-h --help) => :help
+  class_option :socket, :aliases => %w(-s), :default => '/usr/local/var/run/haproxy.sock'
+  class_option :json, :aliases => %w(-j), :type => :boolean, :default => false, :desc => "Return results in JSON for easy parsing"
+
+  desc 'info', 'Get information about the HAProxy process'
+  def info
+    print_result HAStats::Actions::Info.new(options[:socket]).run
+  end
+
+  desc 'stat', 'Get information about all frontends/backends/servers'
+  def stat
+    print_result HAStats::Actions::Stat.new(options[:socket]).run
+  end
+
+  desc 'disable [BACKEND] [SERVER]', 'Put a server into MAINT mode'
+  def disable(backend, server)
+    print_result HAStats::Actions::Disable.new(options[:socket]).run(backend, server)
+  end
+
+  desc 'enable [BACKEND] [SERVER]', 'Remove a server from MAINT mode'
+  def enable(backend, server)
+    print_result HAStats::Actions::Enable.new(options[:socket]).run(backend, server)
+  end
+
+  desc 'sess', '[FUTURE] Get current session information'
+  def sess
+    raise NotImplementedError
+  end
+
+  desc 'errors', '[FUTURE] Get current error information'
+  def errors
+    raise NotImplementedError
+  end
+
+  desc 'clear', '[FUTURE] Clear counter maximums (see --all option)'
+  method_option :all, :type => :boolean, :default => false, :desc => "Clear all counters"
+  def clear
+    raise NotImplementedError
+  end
+
+  desc 'set_weight [BACKEND] [SERVER] [WEIGHT]', '[FUTURE] Set a server\'s weight.  Weight is a number or percentage of current weight'
+  def set_weight(backend, server, weight)
+    raise NotImplementedError
+  end
+
+  desc 'get_weight [BACKEND]/[SERVER]', '[FUTURE] Get a server\'s weight'
+  def get_weight(backend, server)
+    raise NotImplementedError
+  end
+
+  protected
+
+  def print_result(result)
+    if options[:json]
+      puts result.to_json
+    else
+      puts result.raw
+    end
+  end
+end
+
+HAStats::Bin.start
+

data/hastats.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "hastats/version"
+
+Gem::Specification.new do |s|
+  s.name        = "hastats"
+  s.version     = HAStats::VERSION
+  s.authors     = ["Travis Petticrew"]
+  s.email       = ["bobo@petticrew.net"]
+  s.homepage    = ""
+  s.summary     = %q{A tool for interfacing with HAProxy through UNIX sockets}
+  s.description = %q{Allows reading and writing to the haproxy stats socket.  See the HAProxy documentation for available commands.}
+
+  s.rubyforge_project = "hastats"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  s.add_runtime_dependency "thor", '~>0.14.6'
+  s.add_runtime_dependency "json", '>=1'
+end

data/lib/hastats/actions/base.rb

@@ -0,0 +1,22 @@
+class HAStats::Actions::Base
+
+  attr_accessor :socket, :options
+
+  def initialize(socket, options={})
+    self.socket = socket.is_a?(HAStats::Socket) ? socket : HAStats::Socket.new(socket)
+    self.options = options
+  end
+
+  def run(*args)
+    response.new socket.run(command(*args))
+  end
+
+  def response
+    HAStats::Responses::Base
+  end
+
+  def command(*args)
+    raise NotImplementedError
+  end
+
+end

data/lib/hastats/actions/disable.rb

@@ -0,0 +1,12 @@
+class HAStats::Actions::Disable < HAStats::Actions::Base
+
+  def response
+    HAStats::Responses::BlankResult
+  end
+
+  def command(backend, server)
+    "disable server #{backend}/#{server}"
+  end
+
+end
+

data/lib/hastats/actions/enable.rb

@@ -0,0 +1,12 @@
+class HAStats::Actions::Enable < HAStats::Actions::Base
+
+  def response
+    HAStats::Responses::BlankResult
+  end
+
+  def command(backend, server)
+    "enable server #{backend}/#{server}"
+  end
+
+end
+

data/lib/hastats/actions/info.rb

@@ -0,0 +1,11 @@
+class HAStats::Actions::Info < HAStats::Actions::Base
+
+  def response
+    HAStats::Responses::Text
+  end
+
+  def command
+    "show info"
+  end
+
+end

data/lib/hastats/actions/stat.rb

@@ -0,0 +1,11 @@
+class HAStats::Actions::Stat < HAStats::Actions::Base
+
+  def response
+    HAStats::Responses::CSV
+  end
+
+  def command
+    "show stat"
+  end
+
+end

data/lib/hastats/actions.rb

@@ -0,0 +1,8 @@
+module HAStats::Actions
+  autoload :Base, 'hastats/actions/base'
+  autoload :Info, 'hastats/actions/info'
+  autoload :Stat, 'hastats/actions/stat'
+  autoload :Enable, 'hastats/actions/enable'
+  autoload :Disable, 'hastats/actions/disable'
+end
+

data/lib/hastats/responses/base.rb

@@ -0,0 +1,20 @@
+class HAStats::Responses::Base
+
+  attr_accessor :raw, :parsed
+
+  def initialize(raw)
+    self.raw = raw
+    parse
+  end
+
+  # The base class does no parsing
+  def parse
+    self.parsed = self.raw
+  end
+
+  def to_json
+    JSON.dump(self.parsed)
+  end
+
+end
+

data/lib/hastats/responses/blank_result.rb

@@ -0,0 +1,11 @@
+class HAStats::Responses::BlankResult < HAStats::Responses::Base
+
+  def parse
+    if self.raw.strip.empty?
+      self.parsed = { :success => true }
+    else
+      self.parsed = { :success => false, :message => self.raw.strip }
+    end
+  end
+
+end

data/lib/hastats/responses/csv.rb

@@ -0,0 +1,24 @@
+require 'csv'
+
+class HAStats::Responses::CSV < HAStats::Responses::Base
+
+  def parse
+    self.parsed = []
+    header = nil
+    self.raw.each do |row|
+      row = CSV.parse(row.gsub(/#*\s*/, '').strip).first
+      if row && header
+        result = {}
+        header.each_index do |i|
+          result[header[i]] = row[i]
+        end
+        self.parsed << result
+      elsif row
+        header = row
+      end
+    end
+    self.parsed
+  end
+
+end
+

data/lib/hastats/responses/text.rb

@@ -0,0 +1,12 @@
+class HAStats::Responses::Text < HAStats::Responses::Base
+
+  def parse
+    self.parsed = {}
+    self.raw.split("\n").each do |line|
+      name, value = line.split(/:/, 2)
+      self.parsed[name.strip.downcase.to_sym] = value.strip if name && value
+    end
+  end
+
+end
+

data/lib/hastats/responses.rb

@@ -0,0 +1,7 @@
+module HAStats::Responses
+  autoload :Base, 'hastats/responses/base'
+  autoload :Text, 'hastats/responses/text'
+  autoload :CSV, 'hastats/responses/csv'
+  autoload :BlankResult, 'hastats/responses/blank_result'
+end
+

data/lib/hastats/socket.rb

@@ -0,0 +1,27 @@
+class HAStats::Socket
+
+  attr_accessor :socket_addr
+
+  def initialize(socket_addr)
+    @socket_addr = socket_addr
+  end
+
+  def socket
+    @socket ||= UNIXSocket.new(socket_addr)
+  end
+
+  def reset
+    @socket = nil
+  end
+
+  def run(cmd)
+    reset
+    socket.puts(cmd)
+    result = ""
+    while out = socket.gets
+      result += out
+    end
+    result
+  end
+
+end

data/lib/hastats/text_response.rb

@@ -0,0 +1 @@
+

data/lib/hastats/version.rb

@@ -0,0 +1,4 @@
+module HAStats
+  VERSION = "0.0.1"
+end
+

data/lib/hastats.rb

@@ -0,0 +1,11 @@
+require 'socket'
+require 'json'
+
+require "hastats/version"
+
+module HAStats
+  autoload :Socket, 'hastats/socket'
+  autoload :Actions, 'hastats/actions'
+  autoload :Responses, 'hastats/responses'
+end
+

metadata

@@ -0,0 +1,115 @@
+--- !ruby/object:Gem::Specification 
+name: hastats
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Travis Petticrew
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-10-04 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  prerelease: false
+  type: :runtime
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 43
+        segments: 
+        - 0
+        - 14
+        - 6
+        version: 0.14.6
+  name: thor
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  prerelease: false
+  type: :runtime
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 1
+        segments: 
+        - 1
+        version: "1"
+  name: json
+  version_requirements: *id002
+description: Allows reading and writing to the haproxy stats socket.  See the HAProxy documentation for available commands.
+email: 
+- bobo@petticrew.net
+executables: 
+- hastats
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.rdoc
+- Rakefile
+- bin/hastats
+- hastats.gemspec
+- lib/hastats.rb
+- lib/hastats/actions.rb
+- lib/hastats/actions/base.rb
+- lib/hastats/actions/disable.rb
+- lib/hastats/actions/enable.rb
+- lib/hastats/actions/info.rb
+- lib/hastats/actions/stat.rb
+- lib/hastats/responses.rb
+- lib/hastats/responses/base.rb
+- lib/hastats/responses/blank_result.rb
+- lib/hastats/responses/csv.rb
+- lib/hastats/responses/text.rb
+- lib/hastats/socket.rb
+- lib/hastats/text_response.rb
+- lib/hastats/version.rb
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: hastats
+rubygems_version: 1.8.9
+signing_key: 
+specification_version: 3
+summary: A tool for interfacing with HAProxy through UNIX sockets
+test_files: []
+