haproxyctl 0.0.1

data/haproxyctl

@@ -0,0 +1,164 @@
+#!/usr/bin/env ruby
+#
+# HAProxy control script to start, stop, restart, configcheck, etc, as
+# well as communicate to the stats socket.
+#
+# See https://github.com/flores/haproxyctl/README
+#
+# This line here is just for Redhat users who like "service haproxyctl blah"
+# chkconfig: 2345 80 30
+# description: HAProxy is a fast and reliable load balancer for UNIX systems
+# HAProxyctl is an easy way to do init shit and talk to its stats socket
+#
+
+require 'pathname'
+lib = File.join(File.dirname(Pathname.new(__FILE__).realpath), '../lib')
+$:.unshift lib unless $:.include?(lib)
+
+require 'haproxyctl'
+include HAProxyCTL
+
+argument = ARGV.join(' ')
+
+unless has_exec?
+  puts usage if argument =~ /help/ || ARGV.length < 1
+
+  raise "Cannot find haproxy executable. Please ensure it is in your $PATH, or set $HAPROXY_BIN environment variable."
+end
+
+display_usage! if argument =~ /help/ || ARGV.length < 1
+
+begin
+  case argument
+  when "start"
+    if pidof
+      raise("haproxy is already running on pid #{pidof}!")
+    else
+      start
+    end
+  when "stop"
+    stop(check_running)
+  when "restart"
+    if pidof
+      stop(pidof)
+      stillpidof = check_running()
+      while stillpidof == pidof
+        puts "still haven't killed old pid.  waiting 3s for existing connections to die... (ctrl+c to stop)"
+        sleep 3
+        stillpidof = check_running() || 0
+      end
+      start()
+    else
+      puts "haproxy was not running.  starting..."
+      start()
+    end
+  when "reload"
+    if ( pidof )
+      reload(pidof)
+    else
+      puts "haproxy not running.  starting..."
+      start()
+    end
+  when "status"
+    if ( pidof )
+      puts "haproxy is running on pid #{pidof}.\nthese ports are used and guys are connected:"
+      system("lsof -ln -i |awk \'$2 ~ /#{pidof}/ {print $8\" \"$9}\'")
+    else
+      puts "haproxy is not running"
+    end
+  when "configcheck"
+    puts `#{exec} -c -f #{config_path}`
+  when "nagios"
+    if ( pidof )
+      puts "OK"
+      exit
+    else
+      puts "CRITICAL: HAProxy is not running!"
+      exit(2)
+    end
+  when "cloudkick"
+    if ( pidof )
+      puts "status ok haproxy is running"
+      conn = `lsof -ln -i |grep -c #{pidof}`.chomp.to_i
+      # removes the listener
+      conn = conn - 1
+      puts "metric connections int #{conn}"
+      status=unixsock("show stat")
+      status.each do |line|
+        line = line.split(',')
+        if (line[0] !~ /^#/)
+          host = "#{line[0]}_#{line[1]}"
+          puts "metric #{host}_request_rate int #{line[47]}" if line[47].to_i > 0
+          puts "metric #{host}_total_requests gauge #{line[49]}" if line[49].to_i > 0
+          puts "metric #{host}_health_check_duration int #{line[35]}" if line[35].to_i > 0
+          puts "metric ${host}_current_queue int #{line[3]}" if line[3].to_i > 0
+        end
+      end
+    else
+      puts "status err haproxy is not running!"
+    end
+  when "show health"
+    status=unixsock("show stat")
+    status.each do |line|
+      data = line.split(',')
+      printf "%-30s %-30s %-7s %3s\n", data[0], data[1], data[17], data[18]
+    end
+  when /show backend(s?)/
+    status=unixsock("show stat").grep(/BACKEND/)
+    status.each do |line|
+      data = line.split(',')
+      printf "%-30s %-30s %-7s %3s\n", data[0], data[1], data[17], data[18]
+    end
+  when /disable all EXCEPT (.+)/
+    servername=$1
+    status=unixsock("show stat")
+    backend = status.grep(/#{servername}/)
+    backend.each do |line|
+      backend_group = line.split(',')
+      status.each do |pool|
+        data = pool.split(',')
+        if ( (data[0] == backend_group[0]) && ( data[1] !~ /#{servername}|BACKEND|FRONTEND/ ) && ( data[17] = 'UP' ) )
+          unixsock("disable server #{data[0]}/#{data[1]}")
+        end
+      end
+    end
+  when /disable all (.+)/
+    servername=$1
+    status=unixsock("show stat")
+    status.each do |line|
+      data = line.split(',')
+      if ( ( data[1] = servername ) && ( data[17] = 'UP' ) )
+        unixsock("disable server #{data[0]}/#{servername}")
+      end
+    end
+  when /enable all EXCEPT (.+)/
+    servername=$1
+    status=unixsock("show stat")
+    backend = status.grep(/#{servername}/)
+    backend.each do |line|
+      backend_group = line.split(',')
+      status.each do |pool|
+        data = pool.split(',')
+        if ( (data[0] == backend_group[0]) && ( data[1] !~ /#{servername}|BACKEND|FRONTEND/ ) && ( data[17] =~ /Down|MAINT/i ) )
+          unixsock("enable server #{data[0]}/#{data[1]}")
+        end
+      end
+    end
+  when /enable all (.+)/
+    servername=$1
+    status=unixsock("show stat")
+    status.each do |line|
+      data = line.split(',')
+      if ( ( data[1] = servername ) && ( data[17] =~ /Down|MAINT/i ) )
+        unixsock("enable server #{data[0]}/#{servername}")
+      end
+    end
+  when 'version'
+    version
+  else
+    puts unixsock(argument)
+  end
+rescue Errno::ENOENT => e
+  STDERR.puts e
+  exit 1
+end

data/haproxyctl.gemspec

@@ -0,0 +1,18 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/haproxyctl/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Carlo Flores"]
+  gem.email         = ["github@petalphile.com"]
+  gem.description   = %q{This is a simple wrapper to make life with HAProxy a little more convenient. Acts as an init script for start, stop, reload, restart, etc. Leverages 'socket' to enable and disable servers on the fly. Formats server weight and backends in a readable way. Provides Nagios and Cloudkick health checks. Compatible with RHEL chkconfig/service.}
+  gem.summary       = %q{Wrapper to talk to the HAProxy socket, as well as regular init (start|stop|status|etc)}
+  gem.homepage      = "https://github.com/flores/haproxyctl"
+  gem.rubyforge_project = "haproxyctl"
+  gem.license       = "MIT"
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "haproxyctl"
+  gem.require_paths = ["lib"]
+  gem.version       = HAProxyCTL::VERSION
+end

data/install-haproxy/haproxy_src_install.sh

@@ -0,0 +1,82 @@
+#!/bin/bash -ex
+#
+# This installs latest HAProxy from source along with HAProxyCTL
+#
+# It will clobber files and stuff and is only meant as a very
+# quick and dirty (but sometimes handy) installer.
+#
+
+STARTINGDIR=$PWD
+
+# make sure we have make, pcre and junk
+if [ -e /etc/redhat-release ]; then
+	OS=redhat;
+elif [ -e /etc/debian_version ]; then
+	OS=debian;
+fi
+
+if [ $OS ]; then
+	if [ $OS = 'redhat' ]; then
+		yum install -y pcre-devel make gcc libgcc git;
+	elif [ $OS = 'debian' ]; then
+		apt-get update;
+		apt-get install -y libpcre3 libpcre3-dev build-essential libgcc1 git;
+	fi
+else
+	echo -e "I only understand Debian/RedHat/CentOS and this box does not appear to be any.\nExiting.\n- love, $0.";
+	exit 2;
+fi
+
+# grab last stable.  HAProxy's site versions nicely - these will still be here after the next update
+mkdir /usr/local/src || echo "Oops, /usr/local/src exists!"
+cd /usr/local/src || exit 2
+wget http://haproxy.1wt.eu/download/1.4/src/haproxy-1.4.20.tar.gz
+
+# get rid of an existing haproxy
+if [ -e /usr/local/haproxy ]; then
+  rm -fr /usr/local/haproxy
+fi
+
+# check the checksum
+MD5CHECK=`md5sum /usr/local/src/haproxy-1.4.20.tar.gz |awk '{print $1}'`
+if [ "$MD5CHECK" != "0cd3b91812ff31ae09ec4ace6355e29e" ] ; then
+        echo -e "MD5s do not match!\nBailing.";
+        exit 2;
+fi
+
+tar xvfz haproxy-1.4.20.tar.gz
+cd haproxy-1.4.20
+
+if uname -a | grep x86_64 ; then
+	make TARGET=linux26 CPU=x86_64 USE_PCRE=1
+else 
+	make TARGET=linux26 CPU=686 USE_PCRE=1
+fi
+
+make install
+
+if [ -e /usr/sbin/haproxy ]; then
+  rm -f /usr/sbin/haproxy
+fi
+  
+ln -s /usr/local/sbin/haproxy /usr/sbin/haproxy
+
+# grab carlo's haproxyctl script/init
+cd /usr/local
+if [ -e /usr/local/haproxy ]; then
+  cd haproxy;
+  git pull;
+else
+  git clone https://github.com/flores/haproxyctl.git
+  ln -s /usr/local/haproxyctl/haproxyctl /etc/init.d/haproxyctl
+fi
+
+# remove make and gcc
+if [ $OS = 'redhat' ]; then
+	chkconfig --add haproxyctl;
+	yum remove -y gcc make
+elif [ $OS = 'debian' ]; then
+	apt-get purge -y build-essential
+fi
+
+cd $STARTINGDIR

data/lib/haproxyctl.rb

@@ -0,0 +1,122 @@
+require 'haproxyctl/version'
+require 'haproxyctl/environment'
+require 'socket'
+
+module HAProxyCTL
+  include Environment
+
+  def start
+    puts "starting haproxy..."
+    system("#{exec} -f #{config_path} -D -p #{pidfile}")
+    newpid = check_running()
+    if ( newpid =~ /^\d+$/ )
+      puts "haproxy is running on pid #{newpid}"
+      return true
+    else
+      puts "error.  haproxy did not start!"
+      return nil
+    end
+  end
+
+  def stop(pid)
+    if pid
+      puts "stopping haproxy on pid #{pid}..."
+      system("kill #{pid}") || system("kill -9 #{pid}")
+      puts "... stopped"
+    else
+      puts "haproxy is not running!"
+    end
+  end
+
+  def reload(pid)
+    if ( pid )
+      puts "gracefully stopping connections on pid #{pid}..."
+      system("#{exec} -f #{config_path} -sf #{pid}")
+      puts "checking if connections still alive on #{pid}..."
+      nowpid = check_running()
+      while ( pid == nowpid )
+        puts "still haven't killed old pid.
+                            waiting 2s for existing connections to die...
+                            (ctrl+c to stop this check)"
+        sleep 2
+        nowpid = check_running() || 0
+      end
+      puts "reloaded haproxy on pid #{nowpid}"
+    else
+      puts "haproxy is not running!"
+    end
+  end
+
+  def unixsock(command)
+
+    output=[]
+    runs = 0
+      
+    begin 
+      ctl=UNIXSocket.open(socket)
+      if (ctl)
+        ctl.write "#{command}\r\n"
+      else
+        puts "cannot talk to #{socket}"
+      end
+    rescue Errno::EPIPE
+      ctl.close
+      sleep 0.5
+      runs += 1
+      if ( runs < 4 )
+        retry
+      else
+        puts "the unix socket at #{socket} closed before we could complete this request"
+        exit
+      end
+    end
+    while (line = ctl.gets) do
+      unless ( line =~ /Unknown command/ )
+        output << line
+      end
+    end
+    ctl.close
+
+    return output
+  end
+
+  def display_usage!
+    puts usage
+    exit
+  end
+
+  def usage
+    <<-USAGE
+usage: #{$0} <argument>
+  where <argument> can be:
+    start                       : start haproxy unless it is already running
+    stop                        : stop an existing haproxy
+    restart                     : immediately shutdown and restart
+    reload                      : gracefully terminate existing connections, reload #{config_path}
+    status                      : is haproxy running?  on what ports per lsof?
+    configcheck                 : check #{config_path}
+    nagios                      : nagios-friendly status for running process and listener
+    cloudkick                   : cloudkick.com-friendly status and metric for connected users
+    show health                 : show status of all frontends and backend servers
+    show backends               : show status of backend pools of servers
+    enable all <server>         : re-enable a server previously in maint mode on multiple backends
+    disable all <server>        : disable a server from every backend it exists
+    enable all EXCEPT <server>  : like 'enable all', but re-enables every backend except for <server>
+    disable all EXCEPT <server> : like 'disable all', but disables every backend except for <server>
+    clear counters              : clear max statistics counters (add 'all' for all counters)
+    help                        : this message
+    prompt                      : toggle interactive mode with prompt
+    quit                        : disconnect
+    show info                   : report information about the running process
+    show stat                   : report counters for each proxy and server
+    show errors                 : report last request and response errors for each proxy
+    show sess [id]              : report the list of current sessions or dump this session
+    get weight                  : report a server's current weight
+    set weight                  : change a server's weight
+    set timeout                 : change a timeout setting
+    disable server              : set a server in maintenance mode
+    enable server               : re-enable a server that was previously in maintenance mode
+    version                     : version of this script
+USAGE
+  end
+end

data/lib/haproxyctl/environment.rb

@@ -0,0 +1,71 @@
+module HAProxyCTL
+  module Environment
+    attr_accessor :pidof, :config_path, :config, :exec
+
+    def version
+      puts "HAProxyCTL #{HAProxyCTL::VERSION}"
+    end
+
+    def config_path
+      @config_path ||= ENV['HAPROXY_CONFIG'] || '/etc/haproxy/haproxy.cfg'
+    end
+
+    def config
+      @config ||= File.read(config_path)
+    end
+
+    def has_exec?
+      !exec.nil?
+    end
+
+    def exec
+      return(@exec) if @exec
+
+      @exec   = ENV['HAPROXY_BIN']
+      @exec ||= `which haproxy`.chomp
+
+      if @exec.empty?
+        begin
+          `haproxy -v 2>/dev/null`
+          @exec = 'haproxy'
+        rescue Errno::ENOENT => e
+          @exec = nil
+        end
+      end
+
+      return(@exec)
+    end
+
+    def socket
+      @socket ||= begin
+        config.match /stats socket \s*([^\s]*)/
+        $1 || raise("Expecting 'stats socket <UNIX_socket_path>' in #{config_path}")
+      end
+    end
+
+    def pidfile
+      if config.match(/pidfile \s*([^\s]*)/)
+        @pidfile = $1
+      else
+        std_pid = "/var/run/haproxy.pid"
+        if File.exists?(std_pid)
+          @pidfile = std_pid
+        else
+          raise("Expecting 'pidfile <pid_file_path>' in #{config_path} or a pid file in #{std_pid}")
+        end
+      end
+    end
+
+    # @return [String, nil] Returns the PID of HAProxy as a string, if running. Nil otherwise.
+    def check_running
+      pid = File.read(pidfile)
+      pid.strip!
+
+      # verify this pid exists and is haproxy
+      if pid =~ /^\d+$/ and `ps -p #{pid} -o cmd=` =~ /#{exec}/
+        return pid
+      end
+    end
+    alias :pidof :check_running
+  end
+end