ebisu_connection 0.0.8 → 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3bea33e80ff8e417162f05398ebb64531751825b
+  data.tar.gz: a908cbe78a57d95209b8660de521a1371f0e5b05
+SHA512:
+  metadata.gz: c407ba2f24d9466e390eae96ff9931630b4f5ddd0361135d1e07c3d491e7d01e297c1f501ef0004130c5a49c6d0e9fd934877e2ff1ec8d244fd0ffd515851d2e
+  data.tar.gz: 871ee55618b50ffc9e833f9933d1706f1127a081d2d1c644d634cf26906d9c90e5f440fb3a537edb4b2735ab7f8cf069606003e74281b0b9bd8b1428aa143309

data/.gitignore

@@ -1,17 +1,8 @@
 *.gem
-*.rbc
 .bundle
-.config
-.yardoc
+.ruby-version
 Gemfile.lock
-InstalledFiles
-_yardoc
-coverage
-doc/
-lib/bundler/man
-pkg
-rdoc
-spec/reports
-test/tmp
-test/version_tmp
-tmp
+gemfiles/*.lock
+bundle
+log/*
+.*.sw[a-z]

data/Appraisals

@@ -0,0 +1,17 @@
+appraise "rails3" do
+  gem "activerecord", "~> 3.2.0"
+  gem "activesupport", "~> 3.2.0"
+  gem "railties", "~> 3.2.0"
+end
+
+appraise "rails40" do
+  gem "activerecord", "~> 4.0.0"
+  gem "activesupport", "~> 4.0.0"
+  gem "railties", "~> 4.0.0"
+end
+
+appraise "rails41" do
+  gem "activerecord", "~> 4.1.0"
+  gem "activesupport", "~> 4.1.0"
+  gem "railties", "~> 4.1.0"
+end

data/README.md

@@ -1,13 +1,11 @@
 # EbisuConnection
 
 EbisuConnection supports to connect with Mysql slave servers. It doesn't need Load Balancer.
-You can assign a performance weight to each slave server. And slave config is reflected dynamic. 
+You can assign a performance weight to each slave server. And slave config is reflected dynamic.
 EbisuConnection uses FreshConnection (https://github.com/tsukasaoishi/fresh_connection).
 
 ## Installation
 
-EbisuConnection has tested Rails3.2.16 and Rails4.0.2.
-
 Add this line to your application's Gemfile:
 
     gem 'ebisu_connection'
@@ -40,7 +38,7 @@ config/database.yml
         password: slave
         host: slave
 
-slave is base config to connect to slave servers.
+```slave``` is base config to connect to slave servers.
 Others will use the master setting. If you want to change, write in the slave.
 
 Config of each slave server fill out config/slave.yaml
@@ -58,13 +56,73 @@ config/slave.yaml is checked by end of action. If config changed, it's reflected
 
 String format is it. You can write config with hash.
 
-### Only master models
+### use multiple slave servers group
+If you may want to user multiple slave group, write multiple slave group to config/database.yml. 
 
-config/initializers/ebisu_connection.rb
+    production:
+      adapter: mysql2
+      encoding: utf8
+      reconnect: true
+      database: kaeru
+      pool: 5
+      username: master
+      password: master
+      host: localhost
+      socket: /var/run/mysqld/mysqld.sock
 
-    EbisuConnection::ConnectionManager.ignore_models = %w|Model1 Model2|
+      slave:
+        username: slave
+        password: slave
+        host: slave
+
+      admin_slave:
+        username: slave
+        password: slave
+        host: admin_slaves
+
+Config of each slave server fill out config/slave.yaml
+
+    production:
+      slave:
+        - "slave1, 10"
+        - "slave2, 20"
+        -
+          host: "slave3"
+          weight: 30
+      admin_slave:
+        - "slave3, 10"
+        - "slave4, 20"
 
-If models that ignore access to slave servers is exist, You can write model name at EbisuConnection::ConnectionManager.ignore models.
+
+And call establish_fresh_connection method in model that access to ```admin_slave``` slave group.
+
+    class AdminUser < ActiveRecord::Base
+      establish_fresh_connection :admin_slave
+    end
+
+The children is access to same slave group of parent.
+
+    class Parent < ActiveRecord::Base
+      establish_fresh_connection :admin_slave
+    end
+
+    class AdminUser < Parent
+    end
+
+    class Benefit < Parent
+    end
+
+AdminUser and Benefit access to ```admin_slave``` slave group.
+
+
+### Declare model that doesn't use slave db
+
+    class SomethingModel < ActiveRecord::Base
+      master_db_only!
+    end
+
+If model that always access to master servers is exist, You may want to write ```master_db_only!```  in model.
+The model that master_db_only model's child is always access to master db.
 
 ## Usage
 
@@ -91,3 +149,22 @@ In transaction, Always will be access to master server.
 3. Commit your changes (`git commit -am 'Add some feature'`)
 4. Push to the branch (`git push origin my-new-feature`)
 5. Create new Pull Request
+
+## Test
+
+I'm glad that you would do test!
+To run the test suite, you need mysql installed.
+How to setup your test environment.
+
+```bash
+bundle install --path bundle
+GEM_HOME=bundle/ruby/(your ruby version) gem install bundler --pre
+bundle exec appraisal install
+```
+
+This command run the spec suite for all rails versions supported.
+
+```base
+bundle exec appraisal rake spec
+```
+

data/Rakefile

@@ -1 +1,6 @@
 require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/ebisu_connection.gemspec

@@ -18,9 +18,11 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
-  spec.add_dependency 'activerecord', '>= 3.2.0'
-  spec.add_dependency 'fresh_connection', '>= 0.1.8'
+  spec.add_dependency 'fresh_connection', '~> 0.2.0'
 
   spec.add_development_dependency "bundler", "~> 1.3"
-  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rake", '>= 10.0.0'
+  spec.add_development_dependency "rspec", '>= 2.14.1'
+  spec.add_development_dependency 'mysql2', '>= 0.3.15'
+  spec.add_development_dependency 'appraisal', '>= 1.0.0'
 end

data/gemfiles/rails3.gemfile

@@ -0,0 +1,9 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "activerecord", "~> 3.2.0"
+gem "activesupport", "~> 3.2.0"
+gem "railties", "~> 3.2.0"
+
+gemspec :path => "../"

data/gemfiles/rails40.gemfile

@@ -0,0 +1,9 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "activerecord", "~> 4.0.0"
+gem "activesupport", "~> 4.0.0"
+gem "railties", "~> 4.0.0"
+
+gemspec :path => "../"

data/gemfiles/rails41.gemfile

@@ -0,0 +1,9 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "activerecord", "~> 4.1.0"
+gem "activesupport", "~> 4.1.0"
+gem "railties", "~> 4.1.0"
+
+gemspec :path => "../"

data/lib/ebisu_connection/conf_file.rb

@@ -2,7 +2,6 @@ module EbisuConnection
   class ConfFile
     class << self
       attr_writer :slaves_file, :check_interval
-      attr_accessor :slave_type
 
       def if_modify
         if time_to_check? && modify?
@@ -15,12 +14,18 @@ module EbisuConnection
         @spec = nil
       end
 
-      def slaves_conf
+      def slaves_conf(slave_group)
         @slaves_conf ||= get_slaves_conf
+        if @slaves_conf.is_a?(Hash)
+          @slaves_conf[slave_group] || @slaves_conf
+        else
+          @slaves_conf
+        end
       end
 
-      def spec
+      def spec(slave_group)
         @spec ||= get_spec
+        @spec.merge(@spec[slave_group] || {})
       end
 
       def slaves_file
@@ -50,13 +55,11 @@ module EbisuConnection
       def get_slaves_conf
         @file_mtime = File.mtime(slaves_file)
         conf = YAML.load_file(slaves_file)
-        conf = conf[Rails.env.to_s] if conf.is_a?(Hash)
-        slave_type ? conf[slave_type.to_s] : conf
+        conf[Rails.env.to_s] || {}
       end
 
       def get_spec
-        ret = ActiveRecord::Base.configurations[Rails.env]
-        ret.merge(ret["slave"] || {})
+        ActiveRecord::Base.configurations[Rails.env]
       end
     end
   end

data/lib/ebisu_connection/connection_manager.rb

@@ -3,17 +3,9 @@ require 'fresh_connection/abstract_connection_manager'
 
 module EbisuConnection
   class ConnectionManager < FreshConnection::AbstractConnectionManager
-    class << self
-      delegate :slaves_file, :slaves_file=, :check_interval, :check_interval=,
-        :slave_type, :slave_type=, :to => EbisuConnection::ConfFile
+    delegate :if_modify, :conf_clear!, :to => ConfFile
 
-      delegate :ignore_models=, :to => FreshConnection::SlaveConnection
-    end
-
-    delegate :if_modify, :conf_clear!, :slaves_conf, :spec,
-      :to => EbisuConnection::ConfFile
-
-    def initialize
+    def initialize(slave_group = "slave")
       super
       @slaves = {}
     end
@@ -87,7 +79,15 @@ module EbisuConnection
     end
 
     def get_slaves
-      EbisuConnection::Slaves.new(slaves_conf, spec)
+      SlaveGroup.new(slaves_conf, spec)
+    end
+
+    def slaves_conf
+      ConfFile.slaves_conf(@slave_group)
+    end
+
+    def spec
+      ConfFile.spec(@slave_group)
     end
   end
 end

data/lib/ebisu_connection/greatest_common_divisor.rb

@@ -0,0 +1,34 @@
+module EbisuConnection
+  class GreatestCommonDivisor
+    class << self
+      def calc(set)
+        self.new(set).calc
+      end
+    end
+
+    def initialize(set)
+      @set = set.sort.uniq
+    end
+
+    def calc
+      n = @set.shift
+      return n if n == 1 || @set.empty?
+
+      while !@set.empty?
+        m = @set.shift
+        n = gcd_euclid(m, n)
+      end
+      n
+    end
+
+    def gcd_euclid(m, n)
+      m, n = n, m if m < n
+      while n != 0
+        work = m % n
+        m = n
+        n = work
+      end
+      m
+    end
+  end
+end

data/lib/ebisu_connection/slave.rb

@@ -0,0 +1,41 @@
+require 'active_support/core_ext/hash/keys'
+require 'active_support/core_ext/object/blank'
+
+module EbisuConnection
+  class Slave
+    attr_reader :hostname, :weight
+
+    def initialize(conf, spec)
+      case conf
+      when String
+        host, weight = conf.split(/\s*,\s*/)
+        @hostname, port = host.split(/\s*:\s*/)
+      when Hash
+        conf.stringify_keys!
+        @hostname = conf["host"]
+        weight = conf["weight"]
+        port = conf["port"]
+      else
+        raise ArgumentError, "slaves config is invalid"
+      end
+
+      modify_spec = {"host" => @hostname}
+      modify_spec["port"] = port.to_i if port.present?
+
+      @spec = spec.merge(modify_spec)
+      @weight = (weight || 1).to_i
+    end
+
+    def connection
+      @connection ||= ActiveRecord::Base.send("mysql2_connection", @spec)
+    end
+
+    def disconnect!
+      if @connection
+        @connection.disconnect!
+        @connection = nil
+      end
+    rescue
+    end
+  end
+end

data/lib/ebisu_connection/slave_group.rb

@@ -0,0 +1,53 @@
+module EbisuConnection
+  class SlaveGroup
+    class AllSlavesHasGoneError < StandardError; end
+
+    def initialize(slaves_conf, spec)
+      @slaves = slaves_conf.map do |conf|
+        Slave.new(conf, spec)
+      end
+
+      recalc_roulette
+    end
+
+    def sample
+      raise AllSlavesHasGoneError if @slaves.blank?
+      @slaves[@roulette.sample]
+    end
+
+    def remove_connection(connection)
+      return unless s = @slaves.detect{|s| s.connection == connection}
+      s.disconnect! rescue nil
+      @slaves.delete(s)
+      raise AllSlavesHasGoneError if @slaves.blank?
+      recalc_roulette
+      nil
+    end
+
+    def all_disconnect!
+      @reserve_release = nil
+      @slaves.each {|s| s.disconnect!}
+    end
+
+    def reserve_release_connection!
+      @reserve_release = true
+    end
+
+    def reserved_release?
+      !!@reserve_release
+    end
+
+    private
+
+    def recalc_roulette
+      weight_list = @slaves.map {|s| s.weight }
+
+      @roulette = []
+      gcd = GreatestCommonDivisor.calc(weight_list)
+      weight_list.each_with_index do |w, index|
+        weight = w / gcd
+        @roulette.concat([index] * weight)
+      end
+    end
+  end
+end

data/lib/ebisu_connection/version.rb

@@ -1,3 +1,3 @@
 module EbisuConnection
-  VERSION = "0.0.8"
+  VERSION = "0.1.0"
 end

data/lib/ebisu_connection.rb

@@ -1,7 +1,17 @@
 require "fresh_connection"
-require "ebisu_connection/version"
-require "ebisu_connection/conf_file"
-require "ebisu_connection/slaves"
-require "ebisu_connection/connection_manager"
 
-FreshConnection::SlaveConnection.connection_manager = EbisuConnection::ConnectionManager
+module EbisuConnection
+  extend ActiveSupport::Autoload
+
+  autoload :ConfFile
+  autoload :ConnectionManager
+  autoload :SlaveGroup
+  autoload :Slave
+  autoload :GreatestCommonDivisor
+
+  class << self
+    delegate :slaves_file=, :check_interval=, :to => ConfFile
+  end
+end
+
+FreshConnection.connection_manager = EbisuConnection::ConnectionManager

data/spec/spec_helper.rb

@@ -0,0 +1,5 @@
+ENV["RAILS_ENV"]="test"
+$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
+require 'ebisu_connection'
+
+FreshConnection::Initializer.extend_active_record

data/spec/unit/greatest_common_divisor_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+describe EbisuConnection::GreatestCommonDivisor do
+  before(:all) do
+    @g = EbisuConnection::GreatestCommonDivisor
+  end
+
+  context ".calc" do
+    it "return nil if set is empty" do
+      expect(@g.calc([])).to be_nil
+    end
+
+    it "return first element if set has one element" do
+      expect(@g.calc([1])).to eq(1)
+    end
+
+    it "return first element if set has elements that is all same" do
+      set = [1] * 100
+      expect(@g.calc(set)).to eq(1)
+    end
+
+    it "return 1 if set includes 1 in elements" do
+      set = (1..100).to_a
+      expect(@g.calc(set)).to eq(1)
+    end
+
+    it "return gcd" do
+      expect(@g.calc([2,4])).to eq(2)
+      expect(@g.calc([2,4,6])).to eq(2)
+      expect(@g.calc([4,6])).to eq(2)
+      expect(@g.calc([3,4,6])).to eq(1)
+      expect(@g.calc([3,6])).to eq(3)
+      expect(@g.calc([10,10,2])).to eq(2)
+      expect(@g.calc([10,10,2,10,5])).to eq(1)
+      expect(@g.calc([10,10,10,5])).to eq(5)
+    end
+  end
+end

data/spec/unit/slave_group_spec.rb

@@ -0,0 +1,36 @@
+require 'spec_helper'
+
+describe EbisuConnection::SlaveGroup do
+  before(:all) do
+    @sg = EbisuConnection::SlaveGroup
+    @spec = {
+      "adapter" => "mysql2",
+      "database" => "ebisu_connection_test",
+      "username" => "root"
+    }
+  end
+
+  context "#sample" do
+    it "raise exception if slaves empty" do
+      inst = @sg.new([], {})
+      expect{
+        inst.sample
+      }.to raise_error(EbisuConnection::SlaveGroup::AllSlavesHasGoneError)
+    end
+
+    it "return slve instance object" do
+      inst = @sg.new(["h"], {})
+      expect(inst.sample).to be_a(EbisuConnection::Slave)
+    end
+  end
+
+  context "#remove_connection" do
+    it "raise exception AllSlavesHasGoneError when slaves size is one" do
+      inst = @sg.new(["localhost"], @spec)
+      c = inst.sample.connection
+      expect {
+        inst.remove_connection(c)
+      }.to raise_error(EbisuConnection::SlaveGroup::AllSlavesHasGoneError)
+    end
+  end
+end

data/spec/unit/slave_spec.rb

@@ -0,0 +1,58 @@
+require 'spec_helper'
+
+describe EbisuConnection::Slave do
+  before(:all) do
+    @spec = {
+      "adapter" => "mysql2",
+      "database" => "ebisu_connection_test",
+      "username" => "root"
+    }
+  end
+
+  context "initialize(conf is String)" do
+    it "hostname only" do
+      s = EbisuConnection::Slave.new("host_1", {})
+      expect(s.hostname).to eq("host_1")
+      expect(s.weight).to eq(1)
+    end
+
+    it "hostname and weight" do
+      s = EbisuConnection::Slave.new("host_1, 10", {})
+      expect(s.hostname).to eq("host_1")
+      expect(s.weight).to eq(10)
+    end
+
+    it "hostname, weight and port" do
+      s = EbisuConnection::Slave.new("host_1:1975, 10", {})
+      expect(s.hostname).to eq("host_1")
+      expect(s.weight).to eq(10)
+    end
+  end
+
+  context "initialize(conf is Hash)" do
+    it "hostname only" do
+      s = EbisuConnection::Slave.new({:host => "host_1"}, {})
+      expect(s.hostname).to eq("host_1")
+      expect(s.weight).to eq(1)
+    end
+
+    it "hostname and weight" do
+      s = EbisuConnection::Slave.new({:host => "host_1", :weight => 10}, {})
+      expect(s.hostname).to eq("host_1")
+      expect(s.weight).to eq(10)
+    end
+
+    it "hostname, weight and port" do
+      s = EbisuConnection::Slave.new({:host => "host_1", :weight => 10, :port => 1975}, {})
+      expect(s.hostname).to eq("host_1")
+      expect(s.weight).to eq(10)
+    end
+  end
+
+  context "#connection" do
+    it "return Mysql2Adapter object" do
+      s = EbisuConnection::Slave.new("localhost", @spec)
+      expect(s.connection).to be_a(ActiveRecord::ConnectionAdapters::Mysql2Adapter)
+    end
+  end
+end

metadata

@@ -1,138 +1,154 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: ebisu_connection
-version: !ruby/object:Gem::Version 
-  hash: 15
-  prerelease: 
-  segments: 
-  - 0
-  - 0
-  - 8
-  version: 0.0.8
+version: !ruby/object:Gem::Version
+  version: 0.1.0
 platform: ruby
-authors: 
+authors:
 - tsukasaoishi
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2014-03-12 00:00:00 +09:00
-default_executable: 
-dependencies: 
-- !ruby/object:Gem::Dependency 
-  name: activerecord
+date: 2014-04-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: fresh_connection
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
   prerelease: false
-  requirement: &id001 !ruby/object:Gem::Requirement 
-    none: false
-    requirements: 
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
     - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 15
-        segments: 
-        - 3
-        - 2
-        - 0
-        version: 3.2.0
-  type: :runtime
-  version_requirements: *id001
-- !ruby/object:Gem::Dependency 
-  name: fresh_connection
+      - !ruby/object:Gem::Version
+        version: 10.0.0
+  type: :development
   prerelease: false
-  requirement: &id002 !ruby/object:Gem::Requirement 
-    none: false
-    requirements: 
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
     - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 11
-        segments: 
-        - 0
-        - 1
-        - 8
-        version: 0.1.8
-  type: :runtime
-  version_requirements: *id002
-- !ruby/object:Gem::Dependency 
-  name: bundler
+      - !ruby/object:Gem::Version
+        version: 10.0.0
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.14.1
+  type: :development
   prerelease: false
-  requirement: &id003 !ruby/object:Gem::Requirement 
-    none: false
-    requirements: 
-    - - ~>
-      - !ruby/object:Gem::Version 
-        hash: 9
-        segments: 
-        - 1
-        - 3
-        version: "1.3"
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.14.1
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.15
   type: :development
-  version_requirements: *id003
-- !ruby/object:Gem::Dependency 
-  name: rake
   prerelease: false
-  requirement: &id004 !ruby/object:Gem::Requirement 
-    none: false
-    requirements: 
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.3.15
+- !ruby/object:Gem::Dependency
+  name: appraisal
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
     - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
   type: :development
-  version_requirements: *id004
-description: EbisuConnection supports to connect with Mysql slave servers. It doesn't need Load Balancer.
-email: 
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+description: EbisuConnection supports to connect with Mysql slave servers. It doesn't
+  need Load Balancer.
+email:
 - tsukasa.oishi@gmail.com
 executables: []
-
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
-- .gitignore
+files:
+- ".gitignore"
+- Appraisals
 - Gemfile
 - LICENSE.txt
 - README.md
 - Rakefile
 - ebisu_connection.gemspec
+- gemfiles/rails3.gemfile
+- gemfiles/rails40.gemfile
+- gemfiles/rails41.gemfile
 - lib/ebisu_connection.rb
 - lib/ebisu_connection/conf_file.rb
 - lib/ebisu_connection/connection_manager.rb
-- lib/ebisu_connection/slaves.rb
+- lib/ebisu_connection/greatest_common_divisor.rb
+- lib/ebisu_connection/slave.rb
+- lib/ebisu_connection/slave_group.rb
 - lib/ebisu_connection/version.rb
-has_rdoc: true
-homepage: ""
-licenses: 
+- spec/spec_helper.rb
+- spec/unit/greatest_common_divisor_spec.rb
+- spec/unit/slave_group_spec.rb
+- spec/unit/slave_spec.rb
+homepage: ''
+licenses:
 - MIT
+metadata: {}
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
-  none: false
-  requirements: 
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
   - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
-  none: false
-  requirements: 
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
   - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
+    - !ruby/object:Gem::Version
+      version: '0'
 requirements: []
-
 rubyforge_project: 
-rubygems_version: 1.6.2
+rubygems_version: 2.2.0
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: EbisuConnection supports to connect with Mysql slave servers.
-test_files: []
-
+test_files:
+- spec/spec_helper.rb
+- spec/unit/greatest_common_divisor_spec.rb
+- spec/unit/slave_group_spec.rb
+- spec/unit/slave_spec.rb

data/lib/ebisu_connection/slaves.rb

@@ -1,119 +0,0 @@
-module EbisuConnection
-  class Slaves
-    class AllSlavesHasGoneError < StandardError; end
-
-    class Slave
-      attr_reader :hostname, :weight
-
-      def initialize(conf, spec)
-        case conf
-        when String
-          @hostname, weight = conf.split(/\s*,\s*/)
-          edit_spec = {"host" => @hostname}
-        when Hash
-          conf = stringify_keys(conf)
-          weight = conf.delete("weight")
-          edit_spec = conf
-          @hostname = conf["host"]
-        else
-          raise ArgumentError, "slaves config is invalid"
-        end
-
-        @spec = spec.merge(edit_spec)
-        @weight = (weight || 1).to_i
-      end
-
-      def connection
-        @connection ||= ActiveRecord::Base.send("#{@spec["adapter"]}_connection", @spec)
-      end
-
-      def disconnect!
-        if @connection
-          @connection.disconnect!
-          @connection = nil
-        end
-      rescue
-      end
-
-      private
-
-      def stringify_keys(hash)
-        stringify_hash = {}
-        hash.each do |k,v|
-          stringify_hash[k.to_s] = v
-        end
-        stringify_hash
-      end
-    end
-
-    def initialize(slaves_conf, spec)
-      @slaves = slaves_conf.map do |conf|
-        Slave.new(conf, spec)
-      end
-
-      recalc_roulette
-    end
-
-    def sample
-      raise AllSlavesHasGoneError if @slaves.blank?
-      @slaves[@roulette.sample]
-    end
-
-    def remove_connection(connection)
-      return unless s = @slaves.detect{|s| s.connection == connection}
-      s.disconnect! rescue nil
-      @slaves.delete(s)
-      raise AllSlavesHasGoneError if @slaves.blank?
-      recalc_roulette
-      nil
-    end
-
-    def all_disconnect!
-      @reserve_release = nil
-      @slaves.each {|s| s.disconnect!}
-    end
-
-    def reserve_release_connection!
-      @reserve_release = true
-    end
-
-    def reserved_release?
-      !!@reserve_release
-    end
-
-    private
-
-    def recalc_roulette
-      weight_list = @slaves.map {|s| s.weight }
-
-      @roulette = []
-      gcd = get_gcd(weight_list)
-      weight_list.each_with_index do |w, index|
-        weight = w / gcd
-        @roulette.concat([index] * weight)
-      end
-    end
-
-    def get_gcd(list)
-      list = list.sort.uniq
-      n = list.shift
-      return n if n == 1 || list.empty?
-
-      while !list.empty?
-        m = list.shift
-        n = gcd_euclid(m, n)
-      end
-      n
-    end
-
-    def gcd_euclid(m, n)
-      m, n = n, m if m < n
-      while n != 0
-        work = m % n
-        m = n
-        n = work
-      end
-      m
-    end
-  end
-end