halb 0.1.0 → 0.2.0

data/lib/halb.rb

@@ -0,0 +1,2 @@
+require 'halb/deploy'
+require 'halb/load_balancer'

data/lib/halb/deploy.rb

@@ -0,0 +1,23 @@
+require 'socket'
+
+module Halb
+  class Deploy
+    def initialize(load_balancers, machines)
+      @load_balancers=load_balancers
+      @this_computer=Socket.gethostname.downcase
+      @machines = machines
+    end
+
+    def active_balancer
+      @load_balancers.detect { |lb| lb.active? }
+    end
+
+    def remove_from_production
+      active_balancer.put_in_maintenance(@machines[@this_computer])
+    end
+
+    def put_into_production
+      active_balancer.remove_from_maintenance(@machines[@this_computer])
+    end
+  end
+end

data/lib/halb/load_balancer.rb

@@ -0,0 +1,51 @@
+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
+    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
+    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) })
+    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) })
+    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, 'root', :keys => @ssh_keys, &block)
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: halb
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-07-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
-  requirement: &11187500 !ruby/object:Gem::Requirement
+  requirement: &16720700 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,21 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *11187500
+  version_requirements: *16720700
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: &16719360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *16719360
 - !ruby/object:Gem::Dependency
   name: mocha
-  requirement: &11130580 !ruby/object:Gem::Requirement
+  requirement: &16717980 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +43,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *11130580
+  version_requirements: *16717980
 - !ruby/object:Gem::Dependency
   name: test-unit
-  requirement: &11129640 !ruby/object:Gem::Requirement
+  requirement: &17139380 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,13 +54,15 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *11129640
+  version_requirements: *17139380
 description: 
 email: 
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/halb/deploy.rb
+- lib/halb/load_balancer.rb
 - lib/halb.rb
 homepage: https://github.com/TeamIguana/halb
 licenses: []