halb 0.4.0 → 0.5.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: b634f023dedc0bea8e7800ee6dfe6db62eca5a77
+  data.tar.gz: 8eccaeb06b0ccc9e4268fe696eac390797272b88
+SHA512:
+  metadata.gz: 9a7eb04b4562fd23917938ab58e475953d4527c1bc39fbd8257f1c59acf7610e4d75e2fdb9e08e602f083135255ee3897883fae76a473a921874de23bb610c35
+  data.tar.gz: 70dc745da7e276dd49f0ff79ab4e7744923e943c4ae50c13e95586b99c1a9d9627ddb501873fb44f583fc8a068e943c2061a1521e8f115d61cc7e9b470f022a7

data/lib/halb.rb

@@ -1,2 +1,4 @@
 require 'halb/deploy'
-require 'halb/load_balancer'
+require 'halb/abstract_load_balancer'
+require 'halb/load_balancer'
+require 'halb/ha_proxy'

data/lib/halb/abstract_load_balancer.rb

@@ -0,0 +1,43 @@
+require 'net/ssh'
+
+module Halb
+  class AbstractLoadBalancer
+    def initialize(host, user, ssh_keys)
+      @host, @user, @ssh_keys = host, user, ssh_keys
+    end
+
+    def get_output_of(host=@host, command)
+      output=''
+      open_connection(host) do |ssh|
+        output=ssh.exec!(command)
+      end
+      output
+    end
+
+    def put_in_maintenance(machine)
+      service_endpoint=service_endpoint_for(machine)
+      perform(:command => in_maintenance_command(service_endpoint),
+              :exit_when => lambda { |ssh| !ssh.exec!(show_active_hosts_command).to_s.include?(service_endpoint) })
+    end
+
+    def remove_from_maintenance(host)
+      service_endpoint = service_endpoint_for(host)
+      perform(:command => out_of_maintenance_command(service_endpoint),
+              :exit_when => lambda { |ssh| ssh.exec!(show_active_hosts_command).to_s.include?(service_endpoint) })
+    end
+
+    def perform(params)
+      open_connection do |ssh|
+        ssh.exec!(params[:command])
+        loop do
+          break if params[:exit_when].call(ssh)
+          sleep(1)
+        end
+      end
+    end
+
+    def open_connection(host=@host, &block)
+      Net::SSH.start(host, @user, :keys => @ssh_keys, &block)
+    end
+  end
+end

data/lib/halb/ha_proxy.rb

@@ -0,0 +1,36 @@
+module Halb
+  class HAProxy < AbstractLoadBalancer
+    def initialize(host, user, ssh_keys, cluster_ip, proxy_names)
+      super(host, user, ssh_keys)
+      @cluster_ip, @proxy_names = cluster_ip, proxy_names
+    end
+
+    def show_active_hosts_command
+      proxy_names_expression = @proxy_names.join('|')
+      "echo 'show stat -1' | socat stdio /tmp/haproxy.sock | grep -P '(#{proxy_names_expression}).*,UP'"
+    end
+
+    def active?
+      get_output_of(@cluster_ip, 'hostname').strip == @host
+    end
+
+    def in_maintenance_command(machine)
+      maintenance_command_for(machine, 'disable')
+    end
+
+    def out_of_maintenance_command(machine)
+      maintenance_command_for(machine, 'enable')
+    end
+
+    def maintenance_command_for(machine, operation)
+      commands = @proxy_names.map do |proxy_name|
+        "#{operation} server #{proxy_name}/#{machine}"
+      end
+      "echo \"#{commands.join(' ; ')}\" | socat stdio /tmp/haproxy.sock"
+    end
+
+    def service_endpoint_for(machine)
+      machine
+    end
+  end
+end

data/lib/halb/load_balancer.rb

@@ -1,51 +1,23 @@
-require 'net/ssh'
-
 module Halb
-  class LoadBalancer
-    SHOW_ACTIVE_HOSTS_COMMAND='ipvsadm -l -n'
-
-    def initialize(host, user, ssh_keys)
-      @host, @user, @ssh_keys = host, user, ssh_keys
+  class LoadBalancer < AbstractLoadBalancer
+    def show_active_hosts_command
+      'ipvsadm -l -n'
     end
 
     def active?
-      !/TCP[ ]+\d{1,3}.\d{1,3}.\d{1,3}.\d{1,3}/.match(get_output_of(SHOW_ACTIVE_HOSTS_COMMAND)).nil?
-    end
-
-    def get_output_of(command)
-      output=''
-      open_connection do |ssh|
-        output=ssh.exec!(command)
-      end
-      output
+      !/TCP[ ]+\d{1,3}.\d{1,3}.\d{1,3}.\d{1,3}/.match(get_output_of(show_active_hosts_command)).nil?
     end
 
-    def put_in_maintenance(ip_address)
-      service_endpoint=service_endpoint_for(ip_address)
-      perform(:command => "touch /etc/ha.d/maintenance/#{service_endpoint}", :exit_when => lambda { |ssh| !ssh.exec!(SHOW_ACTIVE_HOSTS_COMMAND).include?(service_endpoint) })
+    def in_maintenance_command(service_endpoint)
+      "touch /etc/ha.d/maintenance/#{service_endpoint}"
     end
 
-    def remove_from_maintenance(ip_address)
-      service_endpoint = service_endpoint_for(ip_address)
-      perform(:command => "rm /etc/ha.d/maintenance/#{service_endpoint}", :exit_when => lambda { |ssh| ssh.exec!(SHOW_ACTIVE_HOSTS_COMMAND).include?(service_endpoint) })
+    def out_of_maintenance_command(service_endpoint)
+      "rm /etc/ha.d/maintenance/#{service_endpoint}"
     end
 
     def service_endpoint_for(ip_address)
       "#{ip_address}:80"
     end
-
-    def perform(params)
-      open_connection do |ssh|
-        ssh.exec!(params[:command])
-        loop do
-          break if params[:exit_when].call(ssh)
-          sleep(1)
-        end
-      end
-    end
-
-    def open_connection(&block)
-      Net::SSH.start(@host, @user, :keys => @ssh_keys, &block)
-    end
   end
 end

metadata

@@ -1,91 +1,103 @@
 --- !ruby/object:Gem::Specification
 name: halb
 version: !ruby/object:Gem::Version
-  version: 0.4.0
-  prerelease: 
+  version: 0.5.0
 platform: ruby
 authors:
 - TeamIguana
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-20 00:00:00.000000000 Z
+date: 2014-04-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &16970280 !ruby/object:Gem::Requirement
-    none: false
+  requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *16970280
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: net-ssh
-  requirement: &16968600 !ruby/object:Gem::Requirement
-    none: false
+  requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *16968600
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: mocha
-  requirement: &16967380 !ruby/object:Gem::Requirement
-    none: false
+  requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *16967380
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: test-unit
-  requirement: &16966040 !ruby/object:Gem::Requirement
-    none: false
+  requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ! '>='
+    - - ">="
       - !ruby/object:Gem::Version
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *16966040
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: 
 email: 
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/halb.rb
+- lib/halb/abstract_load_balancer.rb
 - lib/halb/deploy.rb
+- lib/halb/ha_proxy.rb
 - lib/halb/load_balancer.rb
-- lib/halb.rb
 homepage: https://github.com/TeamIguana/halb
 licenses: []
+metadata: {}
 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.15
+rubygems_version: 2.2.1
 signing_key: 
-specification_version: 3
-summary: Manage deploys behind HA.D/LDirector/LVS Load Balancers
+specification_version: 4
+summary: Manage deploys behind HA.D/LDirector/LVS/HAProxy Load Balancers
 test_files: []