green-em-pg 0.1.0

data/Gemfile

@@ -0,0 +1,10 @@
+source "http://rubygems.org"
+
+group :test do
+  gem "sequel"
+  gem "minitest"
+  gem "minitest-reporters"
+  gem "debugger"
+end
+
+gemspec

data/Gemfile.lock

@@ -0,0 +1,51 @@
+PATH
+  remote: .
+  specs:
+    green-em-pg (0.1.0)
+      em-pg (>= 0.1)
+      eventmachine (>= 0.12)
+      green (>= 0.1.1)
+      pg (>= 0.14)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    ansi (1.4.3)
+    builder (3.1.4)
+    columnize (0.3.6)
+    debugger (1.3.0)
+      columnize (>= 0.3.1)
+      debugger-linecache (~> 1.1.1)
+      debugger-ruby_core_source (~> 1.1.7)
+    debugger-linecache (1.1.2)
+      debugger-ruby_core_source (>= 1.1.1)
+    debugger-ruby_core_source (1.1.7)
+    em-pg (0.1.1)
+      eventmachine (>= 0.12)
+      pg (>= 0.14)
+    eventmachine (1.0.0)
+    green (0.1.1)
+      kgio (= 2.7.4)
+    hashie (1.2.0)
+    kgio (2.7.4)
+    minitest (4.5.0)
+    minitest-reporters (0.14.7)
+      ansi
+      builder
+      minitest (>= 2.12, < 5.0)
+      powerbar
+    pg (0.14.1)
+    powerbar (1.0.11)
+      ansi (~> 1.4.0)
+      hashie (>= 1.1.0)
+    sequel (3.43.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  debugger
+  green-em-pg!
+  minitest
+  minitest-reporters
+  sequel

data/README.md

@@ -0,0 +1,36 @@
+green-em-pg
+===========
+
+Simple wrapper around [em-pg](https://github.com/prepor/em-pg) for [Green](https://github.com/prepor/green). And [Sequel](http://sequel.rubyforge.org/) adapter.
+
+Usage
+-----
+
+```ruby
+gem "green-em-pg"
+```
+
+```ruby
+require "green-em/pg"
+
+db = Green::EM::PG.new host: "localhost",
+  port: 5432,
+  dbname: "test",
+  user: "postgres",
+  password: "postgres"
+  
+res = db.send_qeury "select * from test"
+
+puts res.inspect
+```
+
+Sequel
+------
+
+```ruby
+require "green-em/pg/sequal"
+url = "postgres://postgres:postgres@localhost:5432/test"
+db = Sequel.connect(url, pool_class: Green::EM::PG::Sequel::ConnectionPool)
+
+puts db[:test].all.inspect
+```

data/Rakefile

@@ -0,0 +1,121 @@
+require 'rubygems'
+require 'rake'
+require 'date'
+
+#############################################################################
+#
+# Helper functions
+#
+#############################################################################
+
+def name
+  @name ||= Dir['*.gemspec'].first.split('.').first
+end
+
+def version
+  line = File.read("lib/green-em/pg.rb")[/^\s*VERSION\s*=\s*.*/]
+  line.match(/.*VERSION\s*=\s*['"](.*)['"]/)[1]
+end
+
+def date
+  Date.today.to_s
+end
+
+def rubyforge_project
+  name
+end
+
+def gemspec_file
+  "#{name}.gemspec"
+end
+
+def gem_file
+  "#{name}-#{version}.gem"
+end
+
+def replace_header(head, header_name)
+  head.sub!(/(\.#{header_name}\s*= ').*'/) { "#{$1}#{send(header_name)}'"}
+end
+
+#############################################################################
+#
+# Standard tasks
+#
+#############################################################################
+
+task :default => :spec
+
+require 'rake/testtask'
+Rake::TestTask.new(:spec) do |test|
+  test.libs << 'lib' << 'spec'
+  test.pattern = 'spec/**/*_spec.rb'
+  test.verbose = true
+end
+
+desc "Open an irb session preloaded with this library"
+task :console do
+  sh "irb -rubygems -r ./lib/#{name}.rb"
+end
+
+#############################################################################
+#
+# Custom tasks (add your own tasks here)
+#
+#############################################################################
+
+
+
+#############################################################################
+#
+# Packaging tasks
+#
+#############################################################################
+
+desc "Create tag v#{version} and build and push #{gem_file} to Rubygems"
+task :release => :build do
+  unless `git branch` =~ /^\* master$/
+    puts "You must be on the master branch to release!"
+    exit!
+  end
+  sh "git commit --allow-empty -a -m 'Release #{version}'"
+  sh "git tag v#{version}"
+  sh "git push origin master"
+  sh "git push origin v#{version}"
+  sh "gem push pkg/#{name}-#{version}.gem"
+end
+
+desc "Build #{gem_file} into the pkg directory"
+task :build => :gemspec do
+  sh "mkdir -p pkg"
+  sh "gem build #{gemspec_file}"
+  sh "mv #{gem_file} pkg"
+end
+
+desc "Generate #{gemspec_file}"
+task :gemspec do
+  # read spec file and split out manifest section
+  spec = File.read(gemspec_file)
+  head, manifest, tail = spec.split("  # = MANIFEST =\n")
+
+  # replace name version and date
+  replace_header(head, :name)
+  replace_header(head, :version)
+  replace_header(head, :date)
+  #comment this out if your rubyforge_project has a different name
+  replace_header(head, :rubyforge_project)
+
+  # determine file list from git ls-files
+  files = `git ls-files`.
+    split("\n").
+    sort.
+    reject { |file| file =~ /^\./ }.
+    reject { |file| file =~ /^(rdoc|pkg)/ }.
+    map { |file| "    #{file}" }.
+    join("\n")
+
+  # piece file back together and write
+  manifest = "  s.files = %w[\n#{files}\n  ]\n"
+  spec = [head, manifest, tail].join("  # = MANIFEST =\n")
+  File.open(gemspec_file, 'w') { |io| io.write(spec) }
+  puts "Updated #{gemspec_file}"
+end

data/green-em-pg.gemspec

@@ -0,0 +1,30 @@
+spec = Gem::Specification.new do |s|
+  s.name = 'green-em-pg'
+  s.version = '0.1.0'
+  s.date = '2013-01-28'
+  s.summary = 'Async PostgreSQL client API for Ruby/EventMachine'
+  s.email = "ceo@prepor.ru"
+  s.homepage = "http://github.com/prepor/em-postgres"
+  s.description = 'Async PostgreSQL client API for Ruby/EventMachine'
+  s.has_rdoc = false
+  s.authors = ["Andrew Rudenko"]
+  s.add_dependency('eventmachine', '>= 0.12')
+  s.add_dependency('pg', '>= 0.14')
+  s.add_dependency('green', '>= 0.1.1')
+  s.add_dependency('em-pg', '>= 0.1')
+
+  # = MANIFEST =
+  s.files = %w[
+    Gemfile
+    Gemfile.lock
+    README.md
+    Rakefile
+    green-em-pg.gemspec
+    lib/green-em/pg.rb
+    lib/green-em/pg/sequel.rb
+    spec/green-em/pg/sequel_spec.rb
+    spec/green-em/pg_spec.rb
+    spec/spec_helper.rb
+  ]
+  # = MANIFEST =
+end

data/lib/green-em/pg.rb

@@ -0,0 +1,24 @@
+require 'green-em'
+require 'em/pg'
+class Green
+  module EM
+    class PG < ::EM::PG
+      VERSION = "0.1.0"
+      def self.new(*args)
+        Green.hub # ensure started green hub
+        db = allocate
+        db.instance_eval do
+          initialize(*args)
+        end
+        Green::EM.sync db
+        db
+      end
+
+      [:send_query, :send_prepare, :send_query_prepared, :send_describe_prepared, :send_describe_portal].each do |m|
+        define_method(m) do |*args|
+          Green::EM.sync super(*args)
+        end
+      end
+    end
+  end
+end

data/lib/green-em/pg/sequel.rb

@@ -0,0 +1,72 @@
+require 'pg'
+require 'forwardable'
+require 'green-em/pg'
+require 'sequel'
+require 'green/connection_pool'
+class Green
+  module EM
+    class PG
+      class Sequel
+        class ConnectionPool < ::Sequel::ConnectionPool
+          DEFAULT_SIZE = 4
+          attr_accessor :pool
+          def initialize(db, opts = {})
+            super
+            @pool = Green::ConnectionPool.new size: opts[:max_connection] || DEFAULT_SIZE,
+                                              disconnect_class: ::Sequel::DatabaseDisconnectError do
+              make_new(DEFAULT_SERVER)
+            end
+          end
+
+          def size
+            @pool.available.size
+          end
+
+          def hold(server = nil, &blk)
+            @pool.execute(&blk)
+          end
+
+          def disconnect(server = nil)
+            @pool.available.each{ |conn| db.disconnect_connection(conn) }
+            @pool.available.clear
+          end
+        end
+      end
+
+      class PGconn < Green::EM::PG
+        extend Forwardable
+
+        def_delegators :@pg, :status, :escape_string, :escape_bytea
+
+        class << self
+          alias :connect :new
+        end
+
+        { async_exec: :send_query,
+          prepare: :send_prepare,
+          exec_prepared: :send_query_prepared,
+          finish: :close }.each do |from, to|
+          define_method from do |*args|
+            send(to, *args)
+          end
+        end
+        # alias :async_exec :send_query
+        # alias :prepare :send_prepare
+        # alias :exec_prepared :send_query_prepared
+        # alias :finish :close
+
+        [:get_copy_data, :put_copy_data, :put_copy_end, :get_result, :wait_for_notify].each do |m|
+          define_method(m) do |*args|
+            raise "Unimplemented method #{m} in green postgres adapter"
+          end
+        end
+      end
+    end
+  end
+end
+
+PGconn = Green::EM::PG::PGconn
+
+require 'sequel/adapters/postgres'
+
+Sequel::Postgres::CONVERTED_EXCEPTIONS << ::EM::PG::Error

data/spec/green-em/pg/sequel_spec.rb

@@ -0,0 +1,64 @@
+require 'spec_helper'
+require 'green-em/pg/sequel'
+require 'green/group'
+
+describe Green::EM::PG do
+  DELAY = 1
+  QUERY = "select pg_sleep(#{DELAY})"
+
+
+  let(:url) { DB_URL }
+  let(:size) { 1 }
+  let(:db) { Sequel.connect(url, max_connection: size, pool_class: Green::EM::PG::Sequel::ConnectionPool, db_logger: Logger.new(nil)) }
+  let(:test) { db[:test] }
+
+  before do
+    db.create_table!(:test) do
+      text :name
+      integer :value, index: true
+    end
+  end
+
+  after do
+    db.drop_table?(:test)
+  end
+
+  it "should connect and execute query" do
+    test.insert name: "andrew", value: 42
+    test.where(name: "andrew").first[:value].must_equal 42
+  end
+
+
+  describe "pool size is exceeded" do
+    let(:size) { 1 }
+    it "should queue requests" do
+      start = Time.now.to_f
+
+      g = Green::Group.new
+      res = []
+      g.spawn { res << db[QUERY].all }
+      g.spawn { res << db[QUERY].all }
+      g.join
+      (Time.now.to_f - start.to_f).must_be_within_delta DELAY * 2, DELAY * 2 * 0.15
+      res.size.must_equal 2
+    end
+  end
+
+  describe "pool size is enough" do
+    let(:size) { 2 }
+    it "should parallel requests" do
+      start = Time.now.to_f
+
+      g = Green::Group.new
+      res = []
+      g.spawn { res << db[QUERY].all }
+      g.spawn { res << db[QUERY].all }
+      g.join
+
+      (Time.now.to_f - start.to_f).must_be_within_delta DELAY, DELAY * 0.30
+      res.size.must_equal 2
+    end
+  end
+
+
+end

data/spec/green-em/pg_spec.rb

@@ -0,0 +1,9 @@
+require 'spec_helper'
+
+describe Green::EM::PG do
+  let(:db) { Green::EM::PG.new DB_CONFIG }
+  it "should execute query" do
+    res = db.send_query "select 1;"
+    res.first["?column?"].must_equal "1"
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,26 @@
+ENV['BUNDLE_GEMFILE'] = File.expand_path('../../Gemfile', __FILE__)
+
+ENV['GREEN_HUB'] = 'EM'
+require 'bundler/setup'
+
+require 'green-em/pg'
+
+require 'minitest/spec'
+require 'minitest/autorun'
+require 'minitest/reporters'
+
+logger = Logger.new nil
+
+DB_CONFIG = {
+  host: "localhost",
+  port: 5432,
+  dbname: "test",
+  user: "postgres",
+  password: "postgres",
+}
+
+EM::PG.logger = logger
+
+DB_URL = "postgres://%s:%s@%s:%d/%s" % [DB_CONFIG[:user], DB_CONFIG[:password], DB_CONFIG[:host], DB_CONFIG[:port], DB_CONFIG[:dbname]]
+
+MiniTest::Reporters.use! MiniTest::Reporters::SpecReporter.new

metadata

@@ -0,0 +1,118 @@
+--- !ruby/object:Gem::Specification
+name: green-em-pg
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Andrew Rudenko
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.12'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.12'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.14'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.14'
+- !ruby/object:Gem::Dependency
+  name: green
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.1.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.1.1
+- !ruby/object:Gem::Dependency
+  name: em-pg
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.1'
+description: Async PostgreSQL client API for Ruby/EventMachine
+email: ceo@prepor.ru
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- green-em-pg.gemspec
+- lib/green-em/pg.rb
+- lib/green-em/pg/sequel.rb
+- spec/green-em/pg/sequel_spec.rb
+- spec/green-em/pg_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/prepor/em-postgres
+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.24
+signing_key: 
+specification_version: 3
+summary: Async PostgreSQL client API for Ruby/EventMachine
+test_files: []