active_record_upsert 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 8e82a70f09e1da45939fb0dba461762e5b7b2986
+  data.tar.gz: 736288186198273924c7f0caee5bb94e0a05ce52
+SHA512:
+  metadata.gz: edfd93a34947411bb9a390203bced749a2a95ca131e05ab1e0c31242f98197a4dfdcb92e9b9746538367377faf977dee1f452c6f7236ad88706aefc368e683ae
+  data.tar.gz: 452a0b7898226ba626657d11d426b6e1cec5a805e5978b67b5785cb17868f5509dc61c443699f7d25fd8b594804d896f08e577e88a13a76b880541f63bdb5753

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 2.3.0
+before_install: gem install bundler -v 1.11.2

data/Gemfile

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

data/README.md

@@ -0,0 +1,36 @@
+# ActiveRecordUpsert
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/active_record_upsert`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'active_record_upsert'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install active_record_upsert
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/active_record_upsert.
+

data/Rakefile

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

data/active_record_upsert.gemspec

@@ -0,0 +1,35 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'active_record_upsert/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "active_record_upsert"
+  spec.version       = ActiveRecordUpsert::VERSION
+  spec.authors       = ["Jesper Josefsson"]
+  spec.email         = ["jesper.josefsson@gmail.com"]
+
+  spec.summary       = %q{PG Upsert for AR}
+  
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "activerecord", "~> 5.0.0.beta2"
+  spec.add_runtime_dependency "arel", "~>7.0"
+
+  if defined?(JRUBY_VERSION)
+    spec.platform = 'java'
+    spec.add_runtime_dependency 'activerecord-jdbcpostgresql-adapter', '> 0'
+  else
+    spec.platform = Gem::Platform::RUBY
+    spec.add_runtime_dependency 'pg', '> 0'
+  end
+
+
+
+  spec.add_development_dependency "bundler", "~> 1.11"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+end

data/bin/console

@@ -0,0 +1,16 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require 'active_record'
+require 'active_record/connection_adapters/postgresql_adapter'
+require "active_record_upsert"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/active_record_upsert.rb

@@ -0,0 +1,16 @@
+require 'active_record_upsert/version'
+
+unless defined?(Arel)
+  raise 'ActiveRecordUpsert has to be required after ActiveRecord/Arel'
+end
+
+unless defined?(ActiveRecord)
+  raise 'ActiveRecordUpsert has to be required after ActiveRecord'
+end
+
+require 'active_record_upsert/arel'
+require 'active_record_upsert/active_record'
+
+module ActiveRecordUpsert
+  # Your code goes here...
+end

data/lib/active_record_upsert/active_record.rb

@@ -0,0 +1,5 @@
+require 'active_record/connection_adapters/postgresql_adapter'
+
+Dir.glob(File.join(__dir__, 'active_record/**/*.rb')) do |f|
+  require f
+end

data/lib/active_record_upsert/active_record/connection_adapters/abstract/database_statements.rb

@@ -0,0 +1,25 @@
+module ActiveRecordUpsert
+  module ActiveRecord
+    module ConnectionAdapters
+      module Abstract
+        module DatabaseStatementsExtensions
+          def exec_upsert(_sql, _name, _binds, _pk)
+            raise NotImplementedError
+          end
+
+          def upsert(arel, name = nil, pk = nil, id_value = nil, sequence_name = nil, binds = [])
+            sql, binds, pk, _sequence_name = sql_for_upsert(to_sql(arel, binds), pk, id_value, sequence_name, binds)
+            exec_upsert(sql, name, binds, pk)
+          end
+
+          def sql_for_upsert(sql, pk, id_value, sequence_name, binds)
+            [sql, binds, pk, sequence_name]
+          end
+        end
+
+        # ::ActiveRecord::ConnectionAdapters::DatabaseStatements.include(DatabaseStatementsExtensions)
+        ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.include(DatabaseStatementsExtensions)
+      end
+    end
+  end
+end

data/lib/active_record_upsert/active_record/connection_adapters/postgresql/database_statements.rb

@@ -0,0 +1,20 @@
+module ActiveRecordUpsert
+  module ActiveRecord
+    module ConnectionAdapters
+      module Postgresql
+        module DatabaseStatementsExtensions
+          def sql_for_upsert(sql, pk, id_value, sequence_name, binds)
+            sql = "#{sql} RETURNING *"
+            super
+          end
+
+          def exec_upsert(sql, name, binds, pk)
+            exec_query(sql, name, binds)
+          end
+        end
+
+        ::ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.include(DatabaseStatementsExtensions)
+      end
+    end
+  end
+end

data/lib/active_record_upsert/active_record/persistence.rb

@@ -0,0 +1,39 @@
+module ActiveRecordUpsert
+  module ActiveRecord
+    module PersistenceExtensions
+      def upsert(*args)
+        raise ReadOnlyRecord, "#{self.class} is marked as readonly" if readonly?
+        values = run_callbacks(:save) {
+          run_callbacks(:create) {
+            _upsert_record(*args)
+          }
+        }
+        assign_attributes(values.first.to_h)
+        self
+      rescue ::ActiveRecord::RecordInvalid
+        false
+      end
+
+      def _upsert_record(attribute_names = self.attribute_names)
+        attributes_values = arel_attributes_with_values_for_create(attribute_names)
+        values = self.class.unscoped.upsert attributes_values
+        @new_record = false
+        values
+      end
+
+      module ClassMethods
+        def upsert(attributes, &block)
+          if attributes.is_a?(Array)
+            attributes.collect { |attr| upsert(attr, &block) }
+          else
+            new(attributes, &block).upsert
+          end
+        end
+      end
+    end
+
+    puts 'JOJOJOJOJOJO'
+    ::ActiveRecord::Base.prepend(PersistenceExtensions)
+    ::ActiveRecord::Base.extend(PersistenceExtensions::ClassMethods)
+  end
+end

data/lib/active_record_upsert/active_record/relation.rb

@@ -0,0 +1,41 @@
+module ActiveRecordUpsert
+  module ActiveRecord
+    module RelationExtensions
+      def upsert(values) # :nodoc:
+        primary_key_value = nil
+
+        if primary_key && Hash === values
+          primary_key_value = values[values.keys.find { |k|
+            k.name == primary_key
+          }]
+        end
+
+        im = arel_table.create_insert
+        im.into arel_table
+
+        substitutes, binds = substitute_values values
+
+        cm = arel_table.create_on_conflict_do_update
+        cm.target = arel_table[primary_key]
+
+        filter = ->(o) { [primary_key, 'created_at'].include?(o.name) }
+        cm.set(substitutes.reject { |s| filter.call(s.first) })
+        on_conflict_binds = binds.reject(&filter)
+
+        im.on_conflict = cm.to_node
+
+        im.insert substitutes
+
+        @klass.connection.upsert(
+          im,
+          'SQL',
+          primary_key,
+          primary_key_value,
+          nil,
+          binds + on_conflict_binds)
+      end
+    end
+
+    ::ActiveRecord::Relation.include(RelationExtensions)
+  end
+end

data/lib/active_record_upsert/active_record/timestamp.rb

@@ -0,0 +1,22 @@
+module ActiveRecordUpsert
+  module ActiveRecord
+    module TimestampExtensions
+      def _upsert_record
+        if self.record_timestamps
+          current_time = current_time_from_proper_timezone
+
+          all_timestamp_attributes.each do |column|
+            column = column.to_s
+            if has_attribute?(column) && !attribute_present?(column)
+              write_attribute(column, current_time)
+            end
+          end
+        end
+
+        super
+      end
+    end
+
+    ::ActiveRecord::Base.prepend(TimestampExtensions)
+  end
+end

data/lib/active_record_upsert/arel.rb

@@ -0,0 +1,6 @@
+require 'active_record_upsert/arel/nodes/on_conflict_action'
+require 'active_record_upsert/arel/nodes/insert_statement'
+
+Dir.glob(File.join(__dir__, 'arel/**/*.rb')) do |f|
+  require f
+end

data/lib/active_record_upsert/arel/crud.rb

@@ -0,0 +1,18 @@
+# module ActiveRecordUpsert
+#   module Arel
+#     module CrudExtensions
+#       def create_on_conflict_do_update
+#         OnConflictDoUpdateManager.new
+#       end
+#     end
+#
+#     ::Arel::Crud.prepend(CrudExtensions)
+#   end
+# end
+module Arel
+  class Table
+    def create_on_conflict_do_update
+      OnConflictDoUpdateManager.new
+    end
+  end
+end

data/lib/active_record_upsert/arel/insert_manager.rb

@@ -0,0 +1,18 @@
+module ActiveRecordUpsert
+  module Arel
+    module InsertManagerExtensions
+      def on_conflict= node
+        @ast.on_conflict = node
+      end
+
+      def do_nothing_on_conflict(target)
+        @ast.on_conflict = Nodes::OnConflict.new.tap do |on_conflict|
+          on_conflict.target = target
+          on_conflict.action = Nodes::DoNothing.new
+        end
+      end
+    end
+
+    ::Arel::InsertManager.include(InsertManagerExtensions)
+  end
+end

data/lib/active_record_upsert/arel/nodes.rb

@@ -0,0 +1 @@
+

data/lib/active_record_upsert/arel/nodes/do_nothing.rb

@@ -0,0 +1,5 @@
+module Arel
+  module Nodes
+    DoNothing = Class.new(OnConflictAction)
+  end
+end

data/lib/active_record_upsert/arel/nodes/do_update_set.rb

@@ -0,0 +1,33 @@
+module Arel
+  module Nodes
+    class DoUpdateSet < OnConflictAction
+      attr_accessor :wheres, :values
+      attr_accessor :key
+
+      def initialize
+        @wheres   = []
+        @values   = []
+        @key      = nil
+      end
+
+      def initialize_copy other
+        super
+        @wheres = @wheres.clone
+        @values = @values.clone
+      end
+
+      def hash
+        [@relation, @wheres, @values, @key].hash
+      end
+
+      def eql? other
+        self.class == other.class &&
+          self.relation == other.relation &&
+          self.wheres == other.wheres &&
+          self.values == other.values &&
+          self.key == other.key
+      end
+      alias :== :eql?
+    end
+  end
+end

data/lib/active_record_upsert/arel/nodes/excluded_column.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    class ExcludedColumn < Arel::Nodes::Node
+      attr_reader :column
+      def initialize(column)
+        @column = column
+      end
+    end
+  end
+end

data/lib/active_record_upsert/arel/nodes/insert_statement.rb

@@ -0,0 +1,59 @@
+# module ActiveRecordUpsert
+#   module Arel
+#     module Nodes
+#       module InsertStatementExtensions
+#         attr_accessor :on_conflict
+#
+#         def initialize
+#           @on_conflict = nil
+#           super()
+#         end
+#
+#         def hash
+#           [@relation, @columns, @values, @select, @on_conflict].hash
+#         end
+#
+#         def eql? other
+#           self.class == other.class &&
+#             self.relation == other.relation &&
+#             self.columns == other.columns &&
+#             self.select == other.select &&
+#             self.values == other.values &&
+#             self.on_conflict == other.on_conflict
+#         end
+#       end
+#
+#       ::Arel::Nodes::InsertStatement.prepend(InsertStatementExtensions)
+#     end
+#   end
+# end
+
+module Arel
+  module Nodes
+    class InsertStatement
+      attr_accessor :on_conflict
+
+      def initialize
+        super()
+        @relation = nil
+        @columns  = []
+        @values   = nil
+        @select   = nil
+        @on_conflict = nil
+      end
+
+      def hash
+        [@relation, @columns, @values, @select, @on_conflict].hash
+      end
+
+      def eql? other
+        self.class == other.class &&
+          self.relation == other.relation &&
+          self.columns == other.columns &&
+          self.select == other.select &&
+          self.values == other.values &&
+          self.on_conflict == other.on_conflict
+      end
+    end
+  end
+end

data/lib/active_record_upsert/arel/nodes/on_conflict.rb

@@ -0,0 +1,23 @@
+module Arel
+  module Nodes
+    class OnConflict < Node
+      attr_accessor :target, :action
+
+      def initialize
+        super
+        @target = nil
+        @action = nil
+      end
+
+      def hash
+        [@target, @action].hash
+      end
+
+      def eql? other
+        self.class == other.class &&
+          self.target == other.target &&
+          self.update_statement == other.update_statement
+      end
+    end
+  end
+end

data/lib/active_record_upsert/arel/nodes/on_conflict_action.rb

@@ -0,0 +1,6 @@
+module Arel
+  module Nodes
+    class OnConflictAction < Node
+    end
+  end
+end

data/lib/active_record_upsert/arel/on_conflict_do_update_manager.rb

@@ -0,0 +1,47 @@
+module Arel
+  class OnConflictDoUpdateManager < Arel::TreeManager
+    def initialize
+      super
+      @ast = Nodes::OnConflict.new
+      @action = Nodes::DoUpdateSet.new
+      @ast.action = @action
+      @ctx = @ast
+    end
+
+    def target= column
+      @ast.target = column
+    end
+
+    def target(column)
+      @ast.target = column
+      self
+    end
+
+    def wheres= exprs
+      @action.wheres = exprs
+    end
+
+    def where expr
+      @action.wheres << expr
+      self
+    end
+
+    def to_node
+      @ast
+    end
+
+    def set values
+      if String === values
+        @action.values = [values]
+      else
+        @action.values = values.map { |column,value|
+          Nodes::Assignment.new(
+            Nodes::UnqualifiedColumn.new(column),
+            value
+          )
+        }
+      end
+      self
+    end
+  end
+end

data/lib/active_record_upsert/arel/visitors/to_sql.rb

@@ -0,0 +1,54 @@
+module ActiveRecordUpsert
+  module Arel
+    module Visitors
+      module ToSqlExtensions
+        def visit_Arel_Nodes_InsertStatement(o, collector)
+          collector = super
+          if o.on_conflict
+            maybe_visit o.on_conflict, collector
+          else
+            collector
+          end
+        end
+
+        def visit_Arel_Nodes_OnConflict o, collector
+          collector << "ON CONFLICT "
+          collector << " (#{quote_column_name o.target.name}) "
+          maybe_visit o.action, collector
+        end
+
+        def visit_Arel_Nodes_DoNothing _o, collector
+          collector << "DO NOTHING"
+        end
+
+        def visit_Arel_Nodes_DoUpdateSet o, collector
+          wheres = o.wheres
+
+          collector << "DO UPDATE "
+          unless o.values.empty?
+            collector << " SET "
+            collector = inject_join o.values, collector, ", "
+          end
+
+          unless wheres.empty?
+            collector << " WHERE "
+            collector = inject_join wheres, collector, " AND "
+          end
+
+          collector
+        end
+
+        def visit_Arel_Nodes_ExcludedColumn o, collector
+          collector << "EXCLUDED.#{quote_column_name o.column}"
+          collector
+        end
+
+        def table_exists?(name)
+          schema_cache.data_source_exists?(name)
+        end
+      end
+
+      ::Arel::Visitors::ToSql.prepend(ToSqlExtensions)
+    end
+  end
+end

data/lib/active_record_upsert/version.rb

@@ -0,0 +1,3 @@
+module ActiveRecordUpsert
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,156 @@
+--- !ruby/object:Gem::Specification
+name: active_record_upsert
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Jesper Josefsson
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2016-02-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.0.0.beta2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.0.0.beta2
+- !ruby/object:Gem::Dependency
+  name: arel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: 
+email:
+- jesper.josefsson@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- README.md
+- Rakefile
+- active_record_upsert.gemspec
+- bin/console
+- bin/setup
+- lib/active_record_upsert.rb
+- lib/active_record_upsert/active_record.rb
+- lib/active_record_upsert/active_record/connection_adapters/abstract/database_statements.rb
+- lib/active_record_upsert/active_record/connection_adapters/postgresql/database_statements.rb
+- lib/active_record_upsert/active_record/persistence.rb
+- lib/active_record_upsert/active_record/relation.rb
+- lib/active_record_upsert/active_record/timestamp.rb
+- lib/active_record_upsert/arel.rb
+- lib/active_record_upsert/arel/crud.rb
+- lib/active_record_upsert/arel/insert_manager.rb
+- lib/active_record_upsert/arel/nodes.rb
+- lib/active_record_upsert/arel/nodes/do_nothing.rb
+- lib/active_record_upsert/arel/nodes/do_update_set.rb
+- lib/active_record_upsert/arel/nodes/excluded_column.rb
+- lib/active_record_upsert/arel/nodes/insert_statement.rb
+- lib/active_record_upsert/arel/nodes/on_conflict.rb
+- lib/active_record_upsert/arel/nodes/on_conflict_action.rb
+- lib/active_record_upsert/arel/on_conflict_do_update_manager.rb
+- lib/active_record_upsert/arel/visitors/to_sql.rb
+- lib/active_record_upsert/version.rb
+homepage: 
+licenses: []
+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.5.1
+signing_key: 
+specification_version: 4
+summary: PG Upsert for AR
+test_files: []