em-postgresql-sequel 0.1

data/lib/em-postgresql-sequel.rb

@@ -0,0 +1 @@
+["mutex", "fibered_connection_pool", "pgconn", "watcher"].each { |f| require "em-postgresql-sequel/#{f}" }

data/lib/em-postgresql-sequel/fibered_connection_pool.rb

@@ -0,0 +1,65 @@
+module EM
+  module Sequel
+    
+    class FiberedConnectionPool < ::Sequel::ConnectionPool
+      def initialize(opts={}, &block)
+        super
+        @available = []
+        @waiting = []
+        
+        Integer(opts[:max_connections] || 4).times do
+          @available << make_new(DEFAULT_SERVER)
+        end
+        
+        @mutex = Mutex.new
+      end
+      
+      def disconnect(opts={}, &block)
+        @mutex.synchronize do
+          block ||= @disconnection_proc
+          if block
+            m = Mutex.new
+            @available.each do |conn| 
+              m.synchronize do
+                block.call(conn)
+              end
+            end
+          end
+          @available.clear
+        end
+      end
+      
+      def size
+        @available.length
+      end
+      
+      def hold(server=nil, &blk)
+        if @available.empty?
+          @waiting << Fiber.current
+          Fiber.yield
+        end
+        
+        @waiting.delete Fiber.current
+        conn = @available.pop
+        
+        begin
+          blk.call(conn)
+        rescue ::Sequel::DatabaseDisconnectError
+          _conn = conn
+          conn = nil
+          @mutex.synchronize do
+            @disconnection_proc.call(_conn) if @disconnection_proc && _conn
+            @available << make_new(DEFAULT_SERVER)
+          end
+          raise
+        ensure
+          @available << conn if conn
+          if waiting = @waiting.shift
+            waiting.resume
+          end
+        end
+      end
+    end
+    
+  end
+end

data/lib/em-postgresql-sequel/mutex.rb

@@ -0,0 +1,32 @@
+module EM
+  module Sequel
+    class Mutex
+      def initialize
+        @waiting = []
+        @current = nil
+      end
+      
+      def synchronize
+        if @current
+          if @current == Fiber.current
+            raise "already in synchronize" 
+          else
+            @waiting << Fiber.current
+            Fiber.yield
+          end
+        end
+        
+        @current = Fiber.current
+        
+        begin
+          yield if block_given?
+        ensure
+          @current = nil
+          if waiting = @waiting.shift
+            waiting.resume
+          end
+        end
+      end
+    end
+  end
+end

data/lib/em-postgresql-sequel/pgconn.rb

@@ -0,0 +1,33 @@
+module Sequel
+  module Postgres
+    class ::PGconn
+      def async_exec(sql, args=nil)
+        send_query(sql, args)
+
+        deferrable = ::EM::DefaultDeferrable.new
+        ::EM.watch(self.socket, EM::Sequel::Postgres::Watcher, self, deferrable).notify_readable = true
+        
+        f = Fiber.current
+        
+        deferrable.callback do |res| 
+          # puts "!!! callback: #{res}"
+          
+          # check for alive?, otherwise we probably resume a dead fiber, because someone has killed our session e.g. "select pg_terminate_backend('procpid');"
+          f.resume(res) if f.alive?
+        end
+        
+        deferrable.errback  do |err| 
+          # puts "!!! errback: #{err}"
+
+          # check for alive?, otherwise we probably resume a dead fiber, because someone has killed our session e.g. "select pg_terminate_backend('procpid');"
+          f.resume(err) if f.alive?
+        end
+    
+        Fiber.yield.tap do |result|
+          raise result if result.is_a?(Exception)
+        end
+        
+      end
+    end
+  end
+end

data/lib/em-postgresql-sequel/watcher.rb

@@ -0,0 +1,22 @@
+module EM
+  module Sequel
+    module Postgres
+      module Watcher
+        def initialize(client, deferrable)
+          @client = client
+          @deferrable = deferrable
+        end
+
+        def notify_readable
+          detach
+          begin
+            @client.block
+            @deferrable.succeed(@client.get_last_result)
+          rescue Exception => e
+            @deferrable.fail(e)
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,49 @@
+--- !ruby/object:Gem::Specification
+name: em-postgresql-sequel
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Jan Zimmek
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-01-22 00:00:00.000000000Z
+dependencies: []
+description: glue together eventmachine, postgresql and sequel
+email: jan.zimmek@web.de
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/em-postgresql-sequel/fibered_connection_pool.rb
+- lib/em-postgresql-sequel/mutex.rb
+- lib/em-postgresql-sequel/pgconn.rb
+- lib/em-postgresql-sequel/watcher.rb
+- lib/em-postgresql-sequel.rb
+homepage: 
+licenses: []
+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.6
+signing_key: 
+specification_version: 3
+summary: glue together eventmachine, postgresql and sequel
+test_files: []