multithink 0.0.1

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,3 @@
+source "https://rubygems.org"
+
+gemspec(development_group: :runtime)

data/LICENSE

@@ -0,0 +1,41 @@
+Copyright (c) 2011 Mike Perham
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+Copyright (c) 2013 Joseph Glanville
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,4 @@
+MultiThink
+==========
+
+Simple RethinkDB connection pool heavily based on the work of the connection_pool gem.

data/Rakefile

@@ -0,0 +1,14 @@
+begin
+  require 'bundler'
+  Bundler::GemHelper.install_tasks
+rescue LoadError
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'test'
+  test.warning = true
+  test.pattern = 'test/**/test_*.rb'
+end
+
+task :default => :test

data/lib/multithink.rb

@@ -0,0 +1,54 @@
+require 'multithink/version'
+require 'multithink/timed_stack'
+
+class MultiThink
+  DEFAULTS = {size: 5, timeout: 5, servers: [{host: '127.0.0.1', port: 28015}]}
+
+  def initialize(options = {})
+    options = DEFAULTS.merge(options)
+
+    @size = options.fetch(:size)
+    @timeout = options.fetch(:timeout)
+    @servers = options.fetch(:servers)
+
+    @available = TimedStack.new(@size, @servers)
+    @key = :"current-#{@available.object_id}"
+  end
+
+  def with(options = {})
+    conn = checkout(options)
+    begin
+      yield conn
+    ensure
+      checkin
+    end
+  end
+
+  def checkout(options = {})
+    stack = ::Thread.current[@key] ||= []
+
+    if stack.empty?
+      timeout = options[:timeout] || @timeout
+      conn = @available.pop(timeout)
+    else
+      conn = stack.last
+    end
+
+    stack.push conn
+    conn
+  end
+
+  def checkin
+    stack = ::Thread.current[@key]
+    conn = stack.pop
+    if stack.empty?
+      @available << conn
+    end
+    nil
+  end
+
+  def shutdown
+    @available.shutdown
+  end
+
+end

data/lib/multithink/connection.rb

@@ -0,0 +1,25 @@
+require 'thread'
+require 'timeout'
+require 'rethinkdb'
+
+include RethinkDB::Shortcuts
+
+class MultiThink::Connection
+
+  def initialize(servers)
+    @servers = servers
+    connect
+  end
+
+  def connect
+    # TODO try all servers until we get a connection
+    options = @servers.first
+    @conn = r.connect(options)
+  end
+
+  def run(query)
+    # TODO handle connection failure
+    query.run(@conn)
+  end
+
+end

data/lib/multithink/timed_stack.rb

@@ -0,0 +1,63 @@
+require 'thread'
+require 'timeout'
+require_relative 'connection'
+
+class MultiThink::PoolShuttingDownError < RuntimeError; end
+
+class MultiThink::TimedStack
+
+  def initialize(size = 0, servers)
+    @que = Array.new(size) {|s| MultiThink::Connection.new(servers)}
+    @mutex = Mutex.new
+    @resource = ConditionVariable.new
+    @shutdown_block = nil
+  end
+
+  def push(obj)
+    @mutex.synchronize do
+      if @shutdown_block
+        @shutdown_block.call(obj)
+      else
+        @que.push obj
+      end
+
+      @resource.broadcast
+    end
+  end
+  alias_method :<<, :push
+
+  def pop(timeout=0.5)
+    deadline = Time.now + timeout
+    @mutex.synchronize do
+      loop do
+        raise MultiThink::PoolShuttingDownError if @shutdown_block
+        return @que.pop unless @que.empty?
+        to_wait = deadline - Time.now
+        raise Timeout::Error, "Waited #{timeout} sec" if to_wait <= 0
+        @resource.wait(@mutex, to_wait)
+      end
+    end
+  end
+
+  def shutdown(&block)
+    raise ArgumentError, "shutdown must receive a block" unless block_given?
+
+    @mutex.synchronize do
+      @shutdown_block = block
+      @resource.broadcast
+
+      @que.size.times do
+        conn = @que.pop
+        block.call(conn)
+      end
+    end
+  end
+
+  def empty?
+    @que.empty?
+  end
+
+  def length
+    @que.length
+  end
+end

data/lib/multithink/version.rb

@@ -0,0 +1,3 @@
+class MultiThink
+  VERSION = "0.0.1"
+end

data/multithink.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+require "./lib/multithink/version"
+
+Gem::Specification.new do |s|
+  s.name        = "multithink"
+  s.version     = MultiThink::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Joseph Glanville"]
+  s.email       = ["jpg@jpg.id.au"]
+  s.homepage    = "https://github.com/josephglanville/multithink"
+  s.description = s.summary = %q{Simple RethinkDB connection pool.}
+
+  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.require_paths = ["lib"]
+  s.license = "MIT"
+  s.add_development_dependency 'minitest', '>= 5.0.0'
+  s.add_development_dependency 'rake'
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification
+name: multithink
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Joseph Glanville
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-11-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 5.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 5.0.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'
+description: Simple RethinkDB connection pool.
+email:
+- jpg@jpg.id.au
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/multithink.rb
+- lib/multithink/connection.rb
+- lib/multithink/timed_stack.rb
+- lib/multithink/version.rb
+- multithink.gemspec
+homepage: https://github.com/josephglanville/multithink
+licenses:
+- MIT
+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.23
+signing_key: 
+specification_version: 3
+summary: Simple RethinkDB connection pool.
+test_files: []