hawatel_tlb 0.1.0

data/lib/hawatel_tlb/version.rb

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

data/lib/hawatel_tlb/watchdog.rb

@@ -0,0 +1,38 @@
+require 'socket'
+require 'timeout'
+require 'bigdecimal'
+
+module HawatelTlb
+  module WatchDog
+
+      # Check node status
+      #
+      def watcher
+        @group.each do |node|
+          if node.state == 'enable'
+            start = Time.now()
+            port_open?(node) ? port_state = 'online' : port_state = 'offline'
+            stop  = Time.now() - start
+            node.status = {:time => Time.now.to_i, :state => port_state, :respond_time => stop.round(4)}
+          end
+        end
+      end
+
+      private
+      # Check if port is open
+      #
+      def port_open?(node)
+        Timeout::timeout(node.timeout) do
+          begin
+            TCPSocket.new(node.host, node.port).close
+            true
+          rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
+            false
+          end
+        end
+      rescue Timeout::Error, SocketError
+        false
+      end
+
+  end
+end

data/spec/client_spec.rb

@@ -0,0 +1,76 @@
+require 'spec_helper'
+
+describe HawatelTlb::Client do
+
+ let(:client) { HawatelTlb::Client.new }
+
+
+ describe 'Validate_host_settings?' do
+
+   it 'Add valid host' do
+     result = client.add({:host => '8.8.8.3', :port => 443, :weight => 0})
+     expect(result).to eq('success')
+   end
+
+   it 'Add host with invalid ip' do
+    result = client.add({:host => '8.256.2.3', :port => 80, :weight => 0})
+    expect(result).to eq('incorrect host name or ip address, ')
+   end
+
+   it 'Add host with invalid port number' do
+    result = client.add({:host => '8.3.2.3', :port => 0, :weight => 0})
+    expect(result).to eq('incorrect port number, ')
+   end
+
+   it 'Add host with invalid domainame' do
+    result = client.add({:host => 'example.23sd', :port => 80, :weight => 0})
+    expect(result).to eq('incorrect host name or ip address, ')
+   end
+
+   it 'Duplicate host' do
+     client.add({:host => 'example.com', :port => 80, :weight => 0})
+     result_2 = client.add({:host => 'example.com', :port => 80, :weight => 0})
+     expect(result_2).to eq('host with the same configuration already exists in the group, ')
+   end
+
+ end
+
+ describe 'Delete host' do
+
+   it 'Delete non exist id' do
+     result = client.del(22)
+     expect(result).to eq('invalid host id')
+   end
+
+   it 'String instead Fixnumer' do
+     result = client.del('asdasd')
+     expect(result).to eq('invalid value')
+   end
+
+   it 'Delete host' do
+     id = client.add({:host => 'example.com', :port => 80, :weight => 0})
+     expect(client.del(1)).to eq('host successful deleted')
+   end
+
+ end
+
+
+ describe 'Configure group' do
+
+   it 'Configure' do
+     result = client.configure({:mode => 'RR'})
+     expect(result).to eq('invalid mode')
+   end
+
+ end
+
+ it "Get current list" do
+   client.add({:host => 'example.com', :port => 443, :weight => 0})
+   client.add({:host => 'example2.com', :port => 443, :weight => 0})
+   client.add({:host => 'example3.com', :port => 443, :weight => 0})
+   expect(client.list[0][:id]).to eq(1)
+   expect(client.list[0][:host]).to eq('example.com')
+   expect(client.list[0][:port]).to eq(443)
+   expect(client.list[0][:weight]).to eq(0)
+ end
+end

data/spec/modes/dynamicratio_spec.rb

@@ -0,0 +1,43 @@
+require 'spec_helper'
+
+describe HawatelTlb::Mode::DynamicRatio do
+
+  let(:client) { HawatelTlb::Client.new }
+  before do
+    stub_const("HawatelTlb::Mode::DynamicRatio::RECALC_WEIGHT_INTERVAL", 1)
+  end
+
+  context 'dynamic weights' do
+    let(:loop_count) { 1000 }
+    it 'weight is not in init state' do
+      setup_dynamicratio_mode
+      exec_node_method(loop_count)
+      verify_weights(client.list)
+      verify_weights(client.list)
+
+    end
+  end
+
+  private
+
+  def verify_weights(nodes)
+    nodes.each do |node|
+      expect(node.weight).to be > 1
+    end
+  end
+
+  def setup_dynamicratio_mode
+    client.add({:host => 'example.com', :port => 80})
+    client.add({:host => 'example2.com', :port => 80})
+    client.add({:host => 'example3.com', :port => 80})
+    client.configure(:mode => 'dynamicratio')
+    client.mode.debug = 0
+  end
+
+  def exec_node_method(count)
+    (0..count-1).each do
+      client.node
+    end
+  end
+
+end

data/spec/modes/fastest_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+
+describe HawatelTlb::Mode::RoundRobin do
+  let(:client) { HawatelTlb::Client.new }
+
+  it 'Fastest respond server' do
+    client.add({:host => 'example.com', :port => 443, :weight => 0})
+    client.add({:host => 'google.com', :port => 443, :weight => 0})
+    client.add({:host => 'hawatel.com', :port => 443, :weight => 0})
+    client.configure({:mode => 'fastest', :interval => 5})
+    expect(client.node[:host]).to be_a_kind_of(String)
+    expect(client.node[:port]).to be_a_kind_of(Fixnum)
+  end
+
+  it 'Lack of statistics for one node' do
+    client.add({:host => 'example.com', :port => 443, :weight => 0})
+    client.configure({:mode => 'fastest'})
+    client.add({:host => 'google.com', :port => 443, :weight => 0})
+    expect(client.node[:host]).to eq('example.com')
+    expect(client.node[:port]).to eq(443)
+  end
+
+
+  it 'Available only one node' do
+    client.add({:host => 'example.com', :port => 443, :weight => 0,})
+    client.add({:host => 'thisdomainshouldexist2.com', :port => 443, :weight => 0 })
+    client.configure({:mode => 'fastest'})
+    expect(client.node[:host]).to eq('example.com')
+    expect(client.node[:port]).to eq(443)
+  end
+
+  it 'All nodes are disabled' do
+    client.add({:host => 'example.com', :port => 443, :weight => 0, :state => 'disable'})
+    client.add({:host => 'google.com', :port => 443, :weight => 0, :state => 'disable'})
+    client.configure({:mode => 'fastest'})
+    expect(client.node).to eq(false)
+  end
+
+  it 'All nodes are offline' do
+    client.add({:host => 'thisdomainshouldexist1.com', :port => 443, :weight => 0,})
+    client.add({:host => 'thisdomainshouldexist2.com', :port => 443, :weight => 0 })
+    client.configure({:mode => 'fastest'})
+    expect(client.node).to eq(false)
+  end
+
+end

data/spec/modes/ratio_spec.rb

@@ -0,0 +1,74 @@
+require 'spec_helper'
+
+describe HawatelTlb::Mode::Ratio do
+
+  let(:client) { HawatelTlb::Client.new }
+
+  context "the most important node" do
+    let(:loop_count) { 1000 }
+
+    it "the first one" do
+      setup_ratio_mode(2,1,1)
+      exec_node_method(loop_count)
+
+      nodes = client.list
+
+      expect(nodes[0].ratio[:traffic]).to eq(500)
+      expect(nodes[1].ratio[:traffic]).to eq(250)
+      expect(nodes[2].ratio[:traffic]).to eq(250)
+
+      expect(sum_traffic(nodes)).to eq(loop_count)
+    end
+
+    it "the second one" do
+      setup_ratio_mode(2,40,10)
+      exec_node_method(loop_count)
+
+      nodes = client.list
+
+      expect(nodes[0].ratio[:traffic]).to eq(39)
+      expect(nodes[1].ratio[:traffic]).to eq(769)
+      expect(nodes[2].ratio[:traffic]).to eq(192)
+
+      expect(sum_traffic(nodes)).to eq(loop_count)
+    end
+
+    it "the third one" do
+      setup_ratio_mode(0,60,1000)
+      exec_node_method(loop_count)
+
+      nodes = client.list
+
+      expect(nodes[0].ratio[:traffic]).to eq(1)
+      expect(nodes[1].ratio[:traffic]).to eq(57)
+      expect(nodes[2].ratio[:traffic]).to eq(942)
+
+      expect(sum_traffic(nodes)).to eq(loop_count)
+    end
+  end
+
+  private
+
+  def setup_ratio_mode(w1, w2, w3)
+    client.add({:host => 'example.com', :port => 80, :weight => w1})
+    client.add({:host => 'example2.com', :port => 80, :weight => w2})
+    client.add({:host => 'example3.com', :port => 80, :weight => w3})
+    client.configure(:mode => 'ratio')
+    client.mode.debug = 0
+  end
+
+  def exec_node_method(count)
+    (0..count-1).each do
+      client.node
+    end
+  end
+
+  def sum_traffic(nodes)
+    traffic = 0
+    nodes.each do |node|
+      traffic += node.ratio[:traffic]
+    end
+    traffic
+  end
+
+end

data/spec/modes/roundrobin_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+describe HawatelTlb::Mode::RoundRobin do
+  let(:client) { HawatelTlb::Client.new }
+
+  it 'all nodes are enabled' do
+    client.add({:host => 'example.com', :port => 443, :weight => 0})
+    client.add({:host => 'example2.com', :port => 443, :weight => 0})
+    client.configure({:mode => 'roundrobin'})
+    expect(client.node[:host]).to eq('example.com')
+  end
+
+  it 'all nodes are disabled' do
+    client.add({:host => 'example.com', :port => 443, :weight => 0, :state => 'disable'})
+    client.add({:host => 'example2.com', :port => 443, :weight => 0, :state => 'disable'})
+    client.configure({:mode => 'roundrobin'})
+    expect(client.node).to eq(false)
+  end
+
+  it 'overloop nodes' do
+    client.add({:host => 'example.com', :port => 443, :weight => 0, :state => 'enable'})
+    client.add({:host => 'example2.com', :port => 443, :weight => 0, :state => 'enable'})
+    client.add({:host => 'github.com', :port => 80, :weight => 0, :state => 'enable'})
+    client.configure({:mode => 'roundrobin'})
+    ip_1 = client.node[:host]
+    ip_2 = client.node[:host]
+    ip_3 = client.node[:host]
+    ip_4 = client.node[:host]
+    expect(ip_1).to eq('example.com')
+    expect(ip_2).to eq('example2.com')
+    expect(ip_3).to eq('github.com')
+    expect(ip_4).to eq('example.com')
+  end
+
+  it 'all nodes are offline' do
+    client.add({:host => 'thisdomainshouldexist1.com', :port => 443, :weight => 0,})
+    client.add({:host => 'thisdomainshouldexist2.com', :port => 443, :weight => 0 })
+    client.configure({:mode => 'roundrobin'})
+    expect(client.node).to eq(false)
+  end
+
+end

data/spec/modes/weighted_spec.rb

@@ -0,0 +1,40 @@
+ require 'spec_helper'
+
+describe HawatelTlb::Mode::Weighted do
+  let(:client) { HawatelTlb::Client.new }
+
+  it 'Find host with highest weight' do
+    client.add({:host => 'example.com', :port => 80, :weight => 0})
+    client.add({:host => 'wp.pl', :port => 80, :weight => 5})
+    client.add({:host => 'onet.pl', :port => 80, :weight => 3})
+    client.configure({:mode => 'weighted'})
+    stat = client.node
+    expect(stat[:host]).to eq('wp.pl')
+    expect(stat[:port]).to eq(80)
+  end
+
+  it 'Two hosts has equal range of weight ' do
+    client.add({:host => 'example.com', :port => 80, :weight => 0})
+    client.add({:host => 'wp.pl', :port => 80, :weight => 5})
+    client.add({:host => 'onet.pl', :port => 80, :weight => 5})
+    client.configure({:mode => 'weighted'})
+    stat = client.node
+    expect(stat[:host]).to eq('wp.pl')
+    expect(stat[:port]).to eq(80)
+  end
+
+  it 'All hosts are disable' do
+    client.add({:host => 'example.com', :port => 80, :weight => 0, :state => 'disable'})
+    client.add({:host => 'wp.pl', :port => 80, :weight => 5, :state => 'disable'})
+    client.configure({:mode => 'weighted'})
+    expect(client.node).to eq(false)
+  end
+
+  it 'All hosts are offline' do
+    client.add({:host => 'example.com', :port => 1, :weight => 0})
+    client.add({:host => 'wp.pl', :port => 1, :weight => 5})
+    client.configure({:mode => 'weighted'})
+    expect(client.node).to eq(false)
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,2 @@
+$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
+require 'hawatel_tlb'

metadata

@@ -0,0 +1,127 @@
+--- !ruby/object:Gem::Specification
+name: hawatel_tlb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Daniel Iwaniuk
+- Przemyslaw Mantaj
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-04-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Hawatel_tlb is a ruby version load balancing which the purpose is to
+  dynamic return selected address IP/domainame based on specified algorithm
+email:
+- daniel.iwaniuk@hawatel.com
+- przemyslaw.mantaj@hawatel.com
+executables:
+- console
+- setup
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CONTRIBUTING.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- hawatel_tlb.gemspec
+- lib/hawatel_tlb.rb
+- lib/hawatel_tlb/client.rb
+- lib/hawatel_tlb/mode.rb
+- lib/hawatel_tlb/mode/dynamicratio.rb
+- lib/hawatel_tlb/mode/fastest.rb
+- lib/hawatel_tlb/mode/ratio.rb
+- lib/hawatel_tlb/mode/roundrobin.rb
+- lib/hawatel_tlb/mode/weighted.rb
+- lib/hawatel_tlb/version.rb
+- lib/hawatel_tlb/watchdog.rb
+- spec/client_spec.rb
+- spec/modes/dynamicratio_spec.rb
+- spec/modes/fastest_spec.rb
+- spec/modes/ratio_spec.rb
+- spec/modes/roundrobin_spec.rb
+- spec/modes/weighted_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/hawatel/hawatel_tlb
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '1.9'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.8
+signing_key: 
+specification_version: 4
+summary: Ruby gem for failover detection and balancing hosts group
+test_files:
+- spec/client_spec.rb
+- spec/modes/dynamicratio_spec.rb
+- spec/modes/fastest_spec.rb
+- spec/modes/ratio_spec.rb
+- spec/modes/roundrobin_spec.rb
+- spec/modes/weighted_spec.rb
+- spec/spec_helper.rb
+has_rdoc: