haproxy 0.0.1

data/.gitignore

@@ -0,0 +1,3 @@
+pkg/*
+*.gem
+.bundle

data/Gemfile

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

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/haproxy.gemspec

@@ -0,0 +1,21 @@
+# -*- mode: ruby; encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "haproxy/version"
+
+Gem::Specification.new do |s|
+  s.name        = "haproxy"
+  s.version     = Haproxy::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Leandro López (inkel)"]
+  s.email       = ["inkel.ar@gmail.com"]
+  s.homepage    = "http://rubygems.org/gems/haproxy"
+  s.summary     = %q{HAProxy interface for reading statistics or managing servers (requires HAProxy 1.4+)}
+  s.description = %q{This gem is intended for use as a HAProxy interface when you need to read statistics or if you like to manage proxies thru Ruby}
+
+  s.rubyforge_project = "haproxy"
+
+  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"]
+end

data/lib/haproxy.rb

@@ -0,0 +1,21 @@
+libdir = File.dirname(__FILE__)
+$LOAD_PATH.unshift(libdir) unless $LOAD_PATH.include?(libdir)
+
+require 'uri'
+Dir["#{libdir}/haproxy/*.rb"].each do |file|
+  require file
+end
+
+module HAProxy
+
+  def self.read_stats(from)
+    uri = URI.parse(from)
+
+    if uri.is_a?(URI::Generic) and File.socket?(uri.path)
+      HAProxy::SocketReader.new(uri.path)
+    else
+      raise NotImplementedError, "Currently only sockets are implemented"
+    end
+  end
+
+end

data/lib/haproxy/csv_parser.rb

@@ -0,0 +1,32 @@
+module HAProxy
+  class CSVParser
+
+    # Uses CSV header from HAProxy 1.5, which is backwards compatible
+    COLUMNS = "pxname,svname,qcur,qmax,scur,smax,slim,stot,bin,bout,dreq,dresp,ereq,econ,eresp,wretr,wredis,status,weight,act,bck,chkfail,chkdown,lastchg,downtime,qlimit,pid,iid,sid,throttle,lbtot,tracked,type,rate,rate_lim,rate_max,check_status,check_code,check_duration,hrsp_1xx,hrsp_2xx,hrsp_3xx,hrsp_4xx,hrsp_5xx,hrsp_other,hanafail,req_rate,req_rate_max,req_tot,cli_abrt,srv_abrt".split(",").map(&:to_sym)
+    LIMIT = COLUMNS.length
+
+    def self.parse(line)
+      line.strip!
+      unless line.start_with? "#"
+        data = line.split(',')
+        pxname = data.shift
+
+        stats = { :pxname => pxname }
+
+        data.each_with_index do |value, i|
+          if i < LIMIT
+            stats[COLUMNS[i + 1]] = value
+          else
+            stats[:extra] = Array.new if stats[:extra].nil?
+            stats[:extra] << value
+          end
+        end
+
+        stats
+      else
+        raise ArgumentError, "Line is a comment"
+      end
+    end
+  end
+
+end

data/lib/haproxy/socket_reader.rb

@@ -0,0 +1,93 @@
+require 'socket'
+require 'haproxy/stats_reader'
+require 'haproxy/csv_parser'
+
+module HAProxy
+  class SocketReader < HAProxy::StatsReader
+
+    def initialize(path)
+      @path = path
+    end
+
+    def info
+      returning({}) do |info|
+        send_cmd "show info" do |line|
+          key, value = line.split(': ')
+          info[key.downcase.gsub('-', '_').to_sym] = value
+        end
+      end
+    end
+
+    def errors
+      returning("") do |errors|
+        send_cmd "show errors" do |line|
+          errors << line
+        end
+      end
+    end
+
+    def sessions
+      returning([]) do |sess|
+        send_cmd "show sess" do |line|
+          sess << line
+        end
+      end
+    end
+
+    TYPES = {
+      :frontend => 1,
+      :backend => 2,
+      :server => 4
+    }
+
+    def stats(types=[:all], options={})
+      params = {
+        :proxy => :all,
+        :server => :all
+      }.merge(options)
+
+      params[:proxy] = "-1" if params[:proxy].eql?(:all)
+      params[:server] = "-1" if params[:server].eql?(:all)
+      
+      types = [types] unless types.is_a?(Array)
+
+      params[:type] = case
+                      when types.eql?([:all])
+                        "-1"
+                      else
+                        types.map{ |type| TYPES[type] }.inject(:+)
+                      end
+
+      cmd = "show stat #{params[:proxy]} #{params[:type]} #{params[:server]}"
+
+      returning([]) do |stats|
+        send_cmd(cmd) do |line|
+          stats << CSVParser.parse(line) unless line.start_with?('#')
+        end
+      end
+    end
+
+    def frontends
+      stats :frontend, :proxy => :all, :server => :all
+    end
+
+    def backends
+      stats :frontend, :proxy => :all, :server => :all
+    end
+
+    def servers
+      stats :server, :proxy => :all, :server => :all
+    end
+
+    protected
+
+    def send_cmd(cmd, &block)
+      socket = UNIXSocket.new(@path)
+      socket.puts(cmd)
+      socket.each do |line|
+        yield(line.strip)
+      end
+    end
+
+  end
+end

data/lib/haproxy/stats_reader.rb

@@ -0,0 +1,51 @@
+libdir = File.dirname(__FILE__)
+$LOAD_PATH.unshift(libdir) unless $LOAD_PATH.include?(libdir)
+
+module HAProxy
+
+  class StatsReader
+
+    def info
+      raise NotImplementedError
+    end
+
+    def errors
+      raise NotImplementedError
+    end
+
+    def sessions
+      raise NotImplementedError
+    end
+
+    def stats
+      raise NotImplementedError
+    end
+
+    def frontends
+      raise NotImplementedError
+    end
+
+    def backends
+      raise NotImplementedError
+    end
+
+    def servers
+      raise NotImplementedError
+    end
+
+    protected
+
+    # Borrowed from Rails 3
+    def returning(value)
+      yield(value)
+      value
+    end
+
+    private
+
+    def initialize
+    end
+
+  end
+
+end

data/lib/haproxy/version.rb

@@ -0,0 +1,3 @@
+module Haproxy
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,73 @@
+--- !ruby/object:Gem::Specification 
+name: haproxy
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- "Leandro L\xC3\xB3pez (inkel)"
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-11-29 00:00:00 -03:00
+default_executable: 
+dependencies: []
+
+description: This gem is intended for use as a HAProxy interface when you need to read statistics or if you like to manage proxies thru Ruby
+email: 
+- inkel.ar@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Rakefile
+- haproxy.gemspec
+- lib/haproxy.rb
+- lib/haproxy/csv_parser.rb
+- lib/haproxy/socket_reader.rb
+- lib/haproxy/stats_reader.rb
+- lib/haproxy/version.rb
+has_rdoc: true
+homepage: http://rubygems.org/gems/haproxy
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: haproxy
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: HAProxy interface for reading statistics or managing servers (requires HAProxy 1.4+)
+test_files: []
+