activerecord_pg_stuff 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 98e0fe7a1df9305832a41d308d6aba130aec4fb1
+  data.tar.gz: 8b5558b5366becdfb1fe1ec0ff77f312ab048e42
+SHA512:
+  metadata.gz: 8e20d0174e3ae5d40106b3453807871e7433c02bfdc92dcf25ccf8bfb80fd9fe5a2300053081eb1951d4013a2d91682e1823c25ff1e0597be37318b9957af228
+  data.tar.gz: bd22ac3135004e020aeec70bbc4ada73170473dbf790fe6a8d223688d5c36e6f7d9113961871f41ae6dc819eb4fc8c22b9f69e6d20a09747b905c184fb9cd909

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,3 @@
+--color
+-fd
+--order=rand

data/.travis.yml

@@ -0,0 +1,9 @@
+rvm:
+- 1.9.3
+- 2.0.0
+
+env:
+  global:
+    - DATABASE_URL=postgresql://postgres@localhost
+
+script: bundle exec rake

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in activerecord_pg_stuff.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Dmitry Galinsky
+
+MIT License
+
+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,72 @@
+# ActiveRecordPgStuff
+
+Adds support for working with temporary tables and pivot tables (PostgreSQL only).
+
+* [![Build Status](https://travis-ci.org/dima-exe/activerecord_pg_stuff.png?branch=master)](https://travis-ci.org/dima-exe/activerecord_pg_stuff)
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'activerecord_pg_stuff'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install activerecord_pg_stuff
+
+## Usage
+
+#### Temporary tables
+
+Example:
+
+    User.select(:id, :email).temporary_table do |rel|
+      rel.pluck(:id)
+    end
+
+#### Pivot tables
+
+Before using, you need to create the extension:
+
+    CREATE EXTENSION tablefunc
+
+Example:
+
+    CREATE TABLE payments (id integer, amount integer, seller_id integer, created_at timestamp)
+    INSERT INTO payments
+      VALUES (1, 1,  1, '2012-10-12 10:00 UTC'),
+             (2, 3,  1, '2012-11-12 10:00 UTC'),
+             (3, 5,  2, '2012-09-12 10:00 UTC'),
+             (4, 7,  2, '2012-10-12 10:00 UTC'),
+             (5, 11, 2, '2012-11-12 10:00 UTC'),
+             (6, 13, 2, '2012-11-12 10:00 UTC')
+
+    Payment
+      .select("SUM(amount) AS amount", :seller_id, "DATE_TRUNC('month', created_at) AS month")
+      .group("seller_id, DATE_TRUNC('month', created_at)")
+      .temporary_table do |rel|
+        # :month     - for row
+        # :seller_id - for column
+        # :amount    - for value
+        rel.pivot(:month, :seller_id, :amount)
+    end
+
+    expect(result.headers).to eq [nil, 1, 2]
+
+    expect(result.rows).to eq [
+      [ Time.utc(2012, 9,  1), nil, 5  ],
+      [ Time.utc(2012, 10, 1), 1,   7  ],
+      [ Time.utc(2012, 11, 1), 3,   24 ],
+    ]
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+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

data/Rakefile

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

data/activerecord_pg_stuff.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'activerecord_pg_stuff/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "activerecord_pg_stuff"
+  spec.version       = ActiveRecordPgStuff::VERSION
+  spec.authors       = ["Dmitry Galinsky"]
+  spec.email         = ["dima.exe@gmail.com"]
+  spec.description   = %q{ Adds support for working with temporary tables and pivot tables (PostgreSQL only) }
+  spec.summary       = %q{ Adds support for working with temporary tables and pivot tables (PostgreSQL only) }
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "activerecord", "~> 4.0"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "pg"
+end

data/lib/activerecord_pg_stuff.rb

@@ -0,0 +1,12 @@
+require 'active_record'
+require 'active_record/connection_adapters/postgresql_adapter'
+
+require File.expand_path("../activerecord_pg_stuff/version",                    __FILE__)
+require File.expand_path("../activerecord_pg_stuff/connection/temporary_table", __FILE__)
+require File.expand_path("../activerecord_pg_stuff/relation/temporary_table",   __FILE__)
+require File.expand_path("../activerecord_pg_stuff/relation/pivot",             __FILE__)
+
+ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.send :include, ActiveRecordPgStuff::Connection::TemporaryTable
+
+ActiveRecord::Relation.send :include, ActiveRecordPgStuff::Relation::TemporaryTable
+ActiveRecord::Relation.send :include, ActiveRecordPgStuff::Relation::Pivot

data/lib/activerecord_pg_stuff/connection/temporary_table.rb

@@ -0,0 +1,22 @@
+module ActiveRecordPgStuff
+  module Connection
+
+    module TemporaryTable
+
+      def with_temporary_table(name, sql, &block)
+        transaction do
+          begin
+            sql = sql.gsub(/\n/, ' ').gsub(/ +/, ' ').strip
+            sql = "CREATE TEMPORARY TABLE #{name} ON COMMIT DROP AS #{sql}"
+            conn.execute sql
+            yield name
+          ensure
+            execute("DROP TABLE IF EXISTS #{name}") rescue nil
+          end
+        end
+      end
+
+    end
+
+  end
+end

data/lib/activerecord_pg_stuff/relation/pivot.rb

@@ -0,0 +1,54 @@
+module ActiveRecordPgStuff
+  module Relation
+
+    class PivotResult
+      attr_reader :headers, :rows
+
+      def initialize(header_result, result)
+        @headers = [nil] + result_to_array(header_result).map(&:first)
+        @rows    =  result_to_array(result)
+      end
+
+      def each_row(&block)
+        rows.each(&block)
+      end
+
+      private
+        def result_to_array(result)
+          result.to_hash.map do |h|
+            result.columns.inject([]) do |a, col|
+              a << result.column_types[col].type_cast(h[col])
+            end
+          end
+        end
+    end
+
+    module Pivot
+
+      def pivot(row_id, col_id, val_id)
+
+        types_sql = %{ SELECT column_name, data_type FROM information_schema.columns WHERE table_name = #{connection.quote self.table_name} AND column_name IN (#{connection.quote row_id},#{connection.quote val_id}) }
+        types = connection.select_all types_sql
+        types = types.to_a.map(&:values).inject({}) do |a, v|
+          a[v[0]] = v[1]
+          a
+        end
+        row_type = types[row_id.to_s]
+        val_type = types[val_id.to_s]
+
+        cols = connection.select_all self.except(:select).select("DISTINCT #{col_id}").order(col_id).to_sql
+        cols_list = cols.rows.map(&:first).map do |c|
+          "#{col_id}_#{c} #{val_type}"
+        end.join(", ")
+
+        rel_1 = connection.quote self.select(row_id, col_id, val_id).order(row_id).to_sql
+        rel_2 = connection.quote self.except(:select).select("DISTINCT #{col_id}").order(col_id).to_sql
+        sql = %{ SELECT * FROM crosstab(#{rel_1}, #{rel_2}) AS (row_id #{row_type}, #{cols_list}) }
+        PivotResult.new cols, connection.select_all(sql)
+      end
+
+    end
+
+  end
+end
+

data/lib/activerecord_pg_stuff/relation/temporary_table.rb

@@ -0,0 +1,38 @@
+module ActiveRecordPgStuff
+  module Relation
+
+    module TemporaryTable
+
+      class Decorator
+        attr_reader :table_name, :arel_table, :quoted_table_name
+
+        def initialize(object, table_name)
+          @table_name        = table_name
+          @object            = object
+          @arel_table        = Arel::Table.new(table_name)
+          @quoted_table_name = @object.connection.quote_table_name(table_name)
+        end
+
+        def method_missing(name, *args, &block)
+          @object.send(name, *args, &block)
+        end
+
+        def respond_to?(name, *args)
+          @object.respond_to?(name, *args)
+        end
+      end
+
+      def temporary_table
+        tname = "temporary_#{self.table_name}_#{self.object_id}"
+        self.klass.connection.with_temporary_table tname, self.to_sql do |name|
+          dec     = Decorator.new self.klass, name
+          rel     = ActiveRecord::Relation.new dec, dec.arel_table
+          rel.readonly!
+          yield rel
+        end
+      end
+
+    end
+
+  end
+end

data/lib/activerecord_pg_stuff/version.rb

@@ -0,0 +1,3 @@
+module ActiveRecordPgStuff
+  VERSION = "0.0.1"
+end

data/spec/lib/connection_temporary_table_spec.rb

@@ -0,0 +1,35 @@
+require 'spec_helper'
+
+describe ActiveRecordPgStuff::Connection::TemporaryTable do
+
+  context "with_temporary_table" do
+    let(:sql) { "SELECT * FROM sellers WHERE id IN(1,2)" }
+
+    it "should create temporary table with 'name' and 'sql' and drop it after block executed" do
+      rs = conn.with_temporary_table 'sellers_tmp', sql do |name|
+        conn.select_all("SELECT * FROM #{name}").to_a.map(&:values)
+      end
+      expect(rs).to eq [%w{1 foo}, %w{2 bar}]
+      expect {
+        conn.execute 'SELECT * FROM sellers_tmp'
+      }.to raise_error(ActiveRecord::StatementInvalid, /PG::UndefinedTable/)
+    end
+
+    it "should create nested temporary tables" do
+      rs = conn.with_temporary_table 'sellers_tmp', sql do |name|
+        sql = "SELECT * FROM #{name} WHERE id = 1"
+        conn.with_temporary_table 'sellers_tmp_nested', sql do |name_nested|
+          conn.select_all("SELECT * FROM #{name_nested}").to_a.map(&:values)
+        end
+      end
+      expect(rs).to eq [%w{1 foo}]
+      expect {
+        conn.execute 'SELECT * FROM sellers_tmp'
+      }.to raise_error(ActiveRecord::StatementInvalid, /PG::UndefinedTable/)
+      expect {
+        conn.execute 'SELECT * FROM sellers_tmp_nested'
+      }.to raise_error(ActiveRecord::StatementInvalid, /PG::UndefinedTable/)
+    end
+
+  end
+end

data/spec/lib/relation_pivot_spec.rb

@@ -0,0 +1,25 @@
+require 'spec_helper'
+
+describe ActiveRecordPgStuff::Relation::TemporaryTable do
+
+  context "pivot" do
+
+    let(:rel) { Payment }
+
+    it "should create pivot table from relation" do
+      rs = rel.select(:seller_id, "SUM(amount) AS amount", "DATE_TRUNC('month', created_at) AS created_at")
+              .group("seller_id, DATE_TRUNC('month', created_at)").temporary_table do |tmp|
+                tmp.pivot :created_at, :seller_id, :amount
+              end
+
+      expect(rs.headers).to eq [nil,1,2]
+      expect(rs.rows).to eq [
+        [ Time.utc(2012, 9,  1), nil, 5  ],
+        [ Time.utc(2012, 10, 1), 1,   7  ],
+        [ Time.utc(2012, 11, 1), 3,   24 ],
+      ]
+    end
+
+  end
+
+end

data/spec/lib/relation_temporary_table_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+
+describe ActiveRecordPgStuff::Relation::TemporaryTable do
+
+  context "temporary_table" do
+
+    let(:rel) { Seller.where(id: [1,2]) }
+
+    it "should create temporary table from relation" do
+      rs = rel.temporary_table do |tmp|
+        tmp.where(id: [1,2]).load
+      end
+      expect(rs.map(&:class)).to eq [Seller, Seller]
+      expect(rs.map(&:class).map(&:table_name)).to eq %w{ sellers sellers }
+      expect(rs.map(&:id)).to eq [1,2]
+      expect(rs.map(&:readonly?)).to eq [true, true]
+    end
+
+    it "should create nested temporary tables from relation" do
+      rs = rel.select("id * 10 AS id").temporary_table do |tmp|
+        tmp.where(id: [10]).temporary_table do |nested_tmp|
+          nested_tmp.where(id: [10,20]).load
+        end
+      end
+      expect(rs.map(&:id)).to eq [10]
+    end
+
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,42 @@
+Bundler.require(:test)
+require 'rspec/autorun'
+require 'logger'
+require File.expand_path("../../lib/activerecord_pg_stuff", __FILE__)
+
+ActiveRecord::Base.logger = Logger.new(STDOUT)
+
+RSpec.configure do |c|
+  c.before(:suite) do
+    ActiveRecord::Base.establish_connection
+  end
+
+  c.before(:all) do
+    conn.execute 'CREATE EXTENSION IF NOT EXISTS tablefunc'
+    conn.execute "CREATE TABLE sellers (id integer, name varchar)"
+    conn.execute "INSERT INTO sellers VALUES(1, 'foo'), (2, 'bar'), (3, 'baz')"
+
+    conn.execute "CREATE TABLE payments (id integer, amount integer, seller_id integer, created_at timestamp)"
+    conn.execute "INSERT INTO payments VALUES(1, 1,  1, '2012-10-12 10:00 UTC')"
+    conn.execute "INSERT INTO payments VALUES(2, 3,  1, '2012-11-12 10:00 UTC')"
+    conn.execute "INSERT INTO payments VALUES(3, 5,  2, '2012-09-12 10:00 UTC')"
+    conn.execute "INSERT INTO payments VALUES(4, 7,  2, '2012-10-12 10:00 UTC')"
+    conn.execute "INSERT INTO payments VALUES(5, 11, 2, '2012-11-12 10:00 UTC')"
+    conn.execute "INSERT INTO payments VALUES(6, 13, 2, '2012-11-12 10:00 UTC')"
+  end
+
+  c.after(:all) do
+    conn.execute("DROP TABLE sellers")
+    conn.execute("DROP TABLE payments")
+  end
+end
+
+def conn
+  ActiveRecord::Base.connection
+end
+
+class Seller < ActiveRecord::Base
+end
+
+class Payment < ActiveRecord::Base
+end
+

metadata

@@ -0,0 +1,137 @@
+--- !ruby/object:Gem::Specification
+name: activerecord_pg_stuff
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Dmitry Galinsky
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-08-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '4.0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  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
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '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'
+- !ruby/object:Gem::Dependency
+  name: pg
+  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: ' Adds support for working with temporary tables and pivot tables (PostgreSQL
+  only) '
+email:
+- dima.exe@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- .travis.yml
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- activerecord_pg_stuff.gemspec
+- lib/activerecord_pg_stuff.rb
+- lib/activerecord_pg_stuff/connection/temporary_table.rb
+- lib/activerecord_pg_stuff/relation/pivot.rb
+- lib/activerecord_pg_stuff/relation/temporary_table.rb
+- lib/activerecord_pg_stuff/version.rb
+- spec/lib/connection_temporary_table_spec.rb
+- spec/lib/relation_pivot_spec.rb
+- spec/lib/relation_temporary_table_spec.rb
+- spec/spec_helper.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.2
+signing_key: 
+specification_version: 4
+summary: Adds support for working with temporary tables and pivot tables (PostgreSQL
+  only)
+test_files:
+- spec/lib/connection_temporary_table_spec.rb
+- spec/lib/relation_pivot_spec.rb
+- spec/lib/relation_temporary_table_spec.rb
+- spec/spec_helper.rb