haproxy_manager 0.1.0

data/.gitignore

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

data/.rvmrc

@@ -0,0 +1 @@
+rvm ruby-1.9.2@haproxy_man --create

data/Gemfile

@@ -0,0 +1,3 @@
+source "http://rubygems.org"
+
+gemspec

data/Rakefile

@@ -0,0 +1,10 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+
+desc "Run all RSpec tests"
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec
+task :test => [:spec]

data/Readme.md

@@ -0,0 +1,21 @@
+HaProxy Manager
+===============
+
+Haproxy provides ways to add and remove servers on the fly and a lot of other things. This gem lets you use that via a nice ruby API, so that it can be used in applications for deployments testing and gathering statistics.
+
+This was extracted out of our rolling deployment scheme.
+
+Prerequsites
+=============
+
+For this to work haproxy 
+
+* Version 1.4 and above
+
+* Should be running with the following configuration.
+
+  `stats socket /home/ubuntu/haproxysock level admin`
+
+  It means that haproxy will open a socket at /home/ubuntu/haproxysock. You can specify the level. We use `admin`.
+
+

data/haproxy_manager.gemspec

@@ -0,0 +1,27 @@
+require File.expand_path("../lib/haproxy_manager/version.rb", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name = "haproxy_manager"
+  s.version = HAProxyManager::VERSION
+  s.platform = Gem::Platform::RUBY
+  s.authors = ["Sreekanth(sreeix)", "Smita Bhat(sbhat)"]
+  s.email = ["gabbar@activesphere.com"]
+  s.homepage = "https://github.com/althea/haproxy-manager"
+  s.summary = 'HAproxy manager for controlling haproxy'
+  s.description = 'Manages haproxy farms and servers'
+
+  s.rubyforge_project = "haproxy_manager"
+  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.default_executable = "couchup"
+
+  s.require_paths = ["lib"]
+
+  s.add_dependency 'couchrest'
+
+  s.add_development_dependency 'rspec'
+  s.add_development_dependency 'rake'
+  s.add_development_dependency 'mocha'
+  s.add_development_dependency 'bundler'
+end

data/lib/haproxy_manager.rb

@@ -0,0 +1 @@
+require 'haproxy_manager/instance'

data/lib/haproxy_manager/instance.rb

@@ -0,0 +1,64 @@
+require 'socket'
+module HAProxyManager
+  class Instance
+    def initialize(socket)
+      @socket = HAPSocket.new(socket)
+      @print_response = Proc.new {|response| puts response}
+      backends = @socket.execute( "show stat -1 4 -1" )[1..-1].collect{|item| item.split(",")[0..1]}
+      @backends  = backends.inject({}){|hash, items| (hash[items[0]] ||=[]) << items[1]; hash}
+    end
+
+    # Diables a server in the server in a backend for maintenance.
+    # If backend is not specified then all the backends in which the serverid exists are disabled.
+
+    def disable(serverid, backend = nil)
+      all_servers(serverid, backend).each do |item|
+        @socket.exec "disable server #{item[0]}/#{item[1]}", &@print_response
+      end
+    end
+
+    # Enables a server in the server in a backend.
+    # If backend is not specified then all the backends in which the serverid exists are enabled.
+    def enable(serverid, backend = nil)
+      all_servers(serverid, backend).each do |item|
+        @socket.exec "enable server #{item[0]}/#{item[1]}", &@print_response
+      end
+    end
+
+    def backends
+      @backends.keys
+    end
+
+    def servers(backend = nil)
+      backend.nil? ? @backends.values.flatten : @backends[backend]
+    end
+
+    private
+    def all_servers(serverid, backend)
+      if(backend.nil?)
+        items = @backends.collect{|a, b| [a, serverid] if b.include?(serverid)}.compact
+      else
+        items = [[backend, serverid]]
+      end
+    end
+  end
+
+  class HAPSocket
+    def initialize(file)
+      @file = file
+    end
+
+    def execute(cmd, &block)
+      socket = UNIXSocket.new(@file)
+      socket.write("#{cmd};")
+      response = []
+      socket.each do |line|
+        data = line.strip
+        next if data.empty?
+        response << data
+      end
+      yield response if block_given?
+      response
+    end
+  end
+end

data/lib/haproxy_manager/version.rb

@@ -0,0 +1,3 @@
+module HAProxyManager
+  VERSION = "0.1.0"
+end

data/spec/instance_spec.rb

@@ -0,0 +1,59 @@
+require File.expand_path('../spec_helper', __FILE__)
+module HAProxyManager
+  describe Instance do
+    before(:all) do
+      @stat_response = ["# 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,",
+        "foo-farm,preprod-app,0,9,0,60,60,137789,34510620,3221358490,,0,,3,720,0,0,UP,12,1,0,562,143,45394,255790,,1,1,1,,113890,,2,0,,88,L7OK,200,20,0,134660,2028,147,230,0,0,,,,20,6,",
+       "foo-farm,preprod-bg,0,0,0,3,30,31,14333,380028,,0,,0,9,4,2,DOWN,5,1,0,4,10,2453494,4518397,,1,1,2,,6,,2,0,,2,L4CON,,0,0,16,0,0,0,0,0,,,,1,0,",
+       "foo-farm,preprod-test,0,0,0,0,30,0,0,0,,0,,0,0,0,0,DOWN,5,1,0,0,1,5017534,5017534,,1,1,3,,0,,2,0,,0,L4CON,,0,0,0,0,0,0,0,0,,,,0,0,",
+       "foo-https-farm,preprod-app,0,0,0,3,60,6219,2577996,71804141,,0,,1,30,3,0,UP,12,1,0,559,137,45394,255774,,1,2,1,,1948,,2,0,,2,L7OK,200,109,0,5912,181,11,29,0,0,,,,501,0,",
+       "foo-https-farm,preprod-bg,0,0,0,0,30,0,0,0,,0,,0,0,0,0,DOWN,5,1,0,4,4,2453494,4518368,,1,2,2,,0,,2,0,,0,L4CON,,0,0,0,0,0,0,0,0,,,,0,0,", 
+       "foo-https-farm,preprod-test,0,0,0,0,30,0,0,0,,0,,0,0,0,0,DOWN,5,1,0,0,1,5017532,5017532,,1,2,3,,0,,2,0,,0,L4CON,,0,0,0,0,0,0,0,0,,,,0,0,"]
+    end
+    before(:each) do
+      HAPSocket.any_instance.expects(:execute).returns(@stat_response)
+      @instance = Instance.new("foo")
+    end
+
+    describe "creation" do
+      it "parses stats and lists backends" do
+        @instance.backends.size.should == 2
+        @instance.backends.should include "foo-farm"
+        @instance.backends.should include "foo-https-farm"
+      end
+
+      it "parses stats and lists servers" do
+        @instance.servers('foo-farm').size.should == 3
+      end
+      it "understands servers without backend are all servers" do
+        @instance.servers.size.should == 6
+        @instance.servers.should include "preprod-bg"
+        @instance.servers.should include "preprod-test"
+      end
+    end
+    
+    describe "enables servers" do
+      it "enables a server" do
+        HAPSocket.any_instance.expects(:exec).with('enable server foo-farm/preprod-bg')
+        @instance.enable("preprod-bg", "foo-farm")
+      end
+
+      it "enables a all servers in multiple backends" do
+        HAPSocket.any_instance.expects(:exec).with('enable server foo-farm/preprod-bg')
+        HAPSocket.any_instance.expects(:exec).with('enable server foo-https-farm/preprod-bg')
+        @instance.enable("preprod-bg")
+      end
+
+      it "disables a server" do
+        HAPSocket.any_instance.expects(:exec).with('disable server foo-farm/preprod-bg')
+        @instance.disable("preprod-bg", "foo-farm")
+      end
+
+      it "disables a server in all backends" do
+        HAPSocket.any_instance.expects(:exec).with('disable server foo-farm/preprod-bg')
+        HAPSocket.any_instance.expects(:exec).with('disable server foo-https-farm/preprod-bg')
+        @instance.disable("preprod-bg")
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require 'rspec'
+require 'mocha'
+require File.expand_path('../../lib/haproxy_manager', __FILE__)
+
+RSpec.configure do |config|
+  config.mock_framework = :mocha
+end

metadata

@@ -0,0 +1,145 @@
+--- !ruby/object:Gem::Specification
+name: haproxy_manager
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Sreekanth(sreeix)
+- Smita Bhat(sbhat)
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: couchrest
+  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: rspec
+  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'
+- !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'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  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'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  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: Manages haproxy farms and servers
+email:
+- gabbar@activesphere.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rvmrc
+- Gemfile
+- Rakefile
+- Readme.md
+- haproxy_manager.gemspec
+- lib/haproxy_manager.rb
+- lib/haproxy_manager/instance.rb
+- lib/haproxy_manager/version.rb
+- spec/instance_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/althea/haproxy-manager
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1069670280984611263
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1069670280984611263
+requirements: []
+rubyforge_project: haproxy_manager
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: HAproxy manager for controlling haproxy
+test_files:
+- spec/instance_spec.rb
+- spec/spec_helper.rb