activerecord_upsert 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 8573e5d5eb0642aee3a01d2c5cb0dbfc4a953282e5600c567fb4f26bffc38470
+  data.tar.gz: a380f11680573b7dd735f838f85037577c85238fc1c66b02487659daae537be1
+SHA512:
+  metadata.gz: 9df3b5e9e725e8e0cda714511180b0858ce090e4b0e98a72e8f9d30f9df2de691750ec1961c17c079f4cc07c4dbb8021ef98e50886fd3e9caa93335b64ba5c79
+  data.tar.gz: 98cc6703921f84910452be1731db7328b9c70aaea49dce6cd9b3922015c59c06aa772f8bd91973b98488d30108fc03b748ee4e8be23a19358a03a87fc1002044

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+
+# rspec failure tracking
+.rspec_status

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.rubocop.yml

@@ -0,0 +1,201 @@
+require:
+  - rubocop-rspec
+  - rubocop-performance
+
+AllCops:
+  TargetRubyVersion: 2.6
+  Exclude:
+    - activerecord_upsert.gemspec
+    - bin/*
+
+Layout/AccessModifierIndentation:
+  EnforcedStyle: indent
+
+Layout/AlignArguments:
+  EnforcedStyle: with_fixed_indentation
+
+Layout/AlignHash:
+  EnforcedLastArgumentHashStyle: always_ignore
+
+Layout/AlignParameters:
+  EnforcedStyle: with_fixed_indentation
+  IndentationWidth: 2
+
+Layout/CaseIndentation:
+  EnforcedStyle: end
+  IndentOneStep: true
+
+Layout/DotPosition:
+  EnforcedStyle: leading
+
+Layout/EmptyLineAfterGuardClause:
+  Enabled: false
+
+Layout/EmptyLinesAroundExceptionHandlingKeywords:
+  Enabled: false
+
+Layout/EndAlignment:
+  EnforcedStyleAlignWith: variable
+
+Layout/IndentFirstHashElement:
+  EnforcedStyle: consistent
+
+Layout/MultilineMethodCallBraceLayout:
+  EnforcedStyle: new_line
+
+Layout/MultilineMethodCallIndentation:
+  EnforcedStyle: indented
+
+Layout/SpaceBeforeBlockBraces:
+  EnforcedStyle: space
+  EnforcedStyleForEmptyBraces: space
+
+Layout/SpaceInLambdaLiteral:
+  EnforcedStyle: require_space
+
+Layout/SpaceInsideArrayLiteralBrackets:
+  EnforcedStyle: no_space
+
+Layout/SpaceInsideBlockBraces:
+  EnforcedStyle: space
+  EnforcedStyleForEmptyBraces: no_space
+  SpaceBeforeBlockParameters: true
+
+Lint/AmbiguousBlockAssociation:
+  Exclude:
+    - spec/**/*
+
+Metrics/BlockLength:
+  Exclude:
+    - spec/**/*
+
+Metrics/LineLength:
+  Max: 150
+  IgnoreCopDirectives: true
+
+Performance/Casecmp:
+  Enabled: false
+
+RSpec/ContextWording:
+  Enabled: false
+
+RSpec/DescribeClass:
+  Exclude:
+    - spec/system/**/*
+
+RSpec/ExpectChange:
+  EnforcedStyle: block
+
+RSpec/LeadingSubject:
+  Enabled: false
+
+RSpec/LetSetup:
+  Enabled: false
+
+RSpec/MessageSpies:
+  EnforcedStyle: receive
+
+RSpec/MultipleExpectations:
+  Enabled: false
+
+RSpec/NamedSubject:
+  Enabled: false
+
+RSpec/NestedGroups:
+  Max: 6
+
+RSpec/NotToNot:
+  EnforcedStyle: to_not
+
+RSpec/SubjectStub:
+  Enabled: false
+
+RSpec/VerifiedDoubles:
+  Enabled: false
+
+Style/Alias:
+  EnforcedStyle: prefer_alias_method
+
+Style/CollectionMethods:
+  Enabled: true
+  PreferredMethods:
+    detect: detect
+    reduce: inject
+
+Style/CommentAnnotation:
+  Enabled: false
+
+Style/DateTime:
+  Enabled: false
+
+Style/Documentation:
+  Enabled: false
+
+Style/EmptyMethod:
+  EnforcedStyle: expanded
+
+Style/FormatStringToken:
+  EnforcedStyle: template
+
+Style/FrozenStringLiteralComment:
+  EnforcedStyle: never
+
+Style/HashSyntax:
+  Exclude:
+    - 'Rakefile'
+
+Style/MethodCallWithArgsParentheses:
+  Enabled: true
+  IgnoreMacros: true
+  IgnoredMethods:
+    - describe
+    - exit
+    - extend
+    - gem
+    - head
+    - helpers
+    - include
+    - load
+    - prepend
+    - raise
+    - require
+    - require_relative
+    - shared_examples
+    - sleep
+    - system
+    - throw
+    - to
+    - to_not
+  Exclude:
+    - /**/*.builder
+    - /**/*.jbuilder
+    - /**/*.rake
+    - Gemfile
+
+Style/NonNilCheck:
+  IncludeSemanticChanges: true
+
+Style/NumericLiterals:
+  Enabled: false
+
+Style/NumericPredicate:
+  EnforcedStyle: comparison
+
+Style/PercentLiteralDelimiters:
+  PreferredDelimiters:
+    default: "()"
+    "%i": "()"
+    "%I": "()"
+    "%w": "()"
+    "%W": "()"
+    "%r": "{}"
+
+Style/RegexpLiteral:
+  EnforcedStyle: mixed
+  AllowInnerSlashes: false
+
+Style/SafeNavigation:
+  Enabled: false
+
+Style/StabbyLambdaParentheses:
+  EnforcedStyle: require_parentheses

data/.travis.yml

@@ -0,0 +1,14 @@
+sudo: false
+language: ruby
+cache:
+  - bundler
+rvm:
+  - 2.2.9
+  - 2.3.8
+  - 2.4.6
+  - 2.5.5
+  - 2.6.3
+  - 2.7.0.preview2
+matrix:
+  allow_failures:
+    - rvm: 2.7.0.preview1

data/CHANGELOG.md

@@ -0,0 +1,4 @@
+# CHANGELOG
+
+## v0.1.0
+- Initial version

data/Gemfile

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

data/README.md

@@ -0,0 +1,37 @@
+# ActiveRecord Upsert
+
+Adds the ability to use MySQL's `ON DUPLICATE KEY UPDATE` to upsert records in ActiveRecord.
+
+**Note**: This will conflict with Rails 6, which adds its own upsert, so requires Rails 5.2 currently. 
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'activerecord_upsert'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install activerecord_upsert
+
+## Usage
+
+```ruby
+Model.where(foo: :bar).upsert(count: 'count + 1')
+```
+
+## 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]/activerecord_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/activerecord_upsert.gemspec

@@ -0,0 +1,42 @@
+lib = File.expand_path('lib', __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'active_record_upsert/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'activerecord_upsert'
+  spec.version       = ActiveRecordUpsert::VERSION
+  spec.authors       = ['Daniel Vandersluis']
+  spec.email         = ['daniel.vandersluis@gmail.com']
+
+  spec.summary       = %|Support MySQL's INSERT ... ON DUPLICATE KEY UPDATE in ActiveRecord|
+  spec.homepage      = 'https://github.com/dvandersluis/activerecord_upsert'
+
+  if spec.respond_to?(:metadata)
+    spec.metadata['homepage_uri'] = spec.homepage
+    spec.metadata['changelog_uri'] = "https://github.com/dvandersluis/activerecord_upsert/CHANGELOG.md"
+  else
+    raise 'RubyGems 2.0 or newer is required to protect against ' \
+      'public gem pushes.'
+  end
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = 'exe'
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'mysql2'
+  spec.add_dependency 'activerecord', '>= 5.2', '< 6'
+
+  spec.add_development_dependency('bundler', '~> 2.0')
+  spec.add_development_dependency('database_cleaner', '~> 1.7.0')
+  spec.add_development_dependency('pry', '~> 0.12.0')
+  spec.add_development_dependency('rake', '~> 10.0')
+  spec.add_development_dependency('rspec', '~> 3.0')
+  spec.add_development_dependency('rubocop', '~> 0.75.0')
+  spec.add_development_dependency('rubocop-rspec', '~> 1.36.0')
+  spec.add_development_dependency('rubocop-performance', '~> 1.5.0')
+end

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "activerecord_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(__FILE__)

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,21 @@
+require 'active_record_upsert/version'
+
+require 'arel'
+require 'active_record'
+
+require 'active_record_upsert/arel/insert_manager'
+require 'active_record_upsert/arel/insert_statement'
+require 'active_record_upsert/arel/visitors/mysql'
+
+require 'active_record_upsert/active_record/persistence'
+require 'active_record_upsert/active_record/relation'
+
+module ActiveRecordUpsert
+end
+
+Arel::Nodes::InsertStatement.include ActiveRecordUpsert::Arel::InsertStatement
+Arel::InsertManager.include ActiveRecordUpsert::Arel::InsertManager
+Arel::Visitors::MySQL.prepend ActiveRecordUpsert::Arel::Visitors::MySQL
+
+ActiveRecord::Relation.include ActiveRecordUpsert::ActiveRecord::Relation
+ActiveRecord::Persistence::ClassMethods.include ActiveRecordUpsert::ActiveRecord::Persistence

data/lib/active_record_upsert/active_record/persistence.rb

@@ -0,0 +1,46 @@
+module ActiveRecordUpsert
+  module ActiveRecord
+    module Persistence
+      def _upsert_record(values, on_duplicate)
+        set_timestamps(values)
+        set_primary_key_value(values)
+
+        im = compile_insert(values, on_duplicate)
+        connection.insert(im, "#{self} Upsert", primary_key || false, values[primary_key])
+      end
+
+      private
+
+      def set_timestamps(values) # rubocop:disable Naming/AccessorMethodName
+        return unless record_timestamps
+
+        current_time = current_time_from_proper_timezone
+
+        all_timestamp_attributes_in_model.each do |column|
+          values[column] = current_time unless values.include?(column)
+        end
+      end
+
+      def set_primary_key_value(values) # rubocop:disable Naming/AccessorMethodName
+        primary_key = self.primary_key
+        return unless primary_key && values.is_a?(Hash)
+
+        primary_key_value = values[primary_key]
+        primary_key_value ||= next_sequence_value if prefetch_primary_key?
+        values[primary_key] = primary_key_value
+      end
+
+      def compile_insert(values, on_duplicate)
+        if values.empty?
+          im = arel_table.compile_insert(connection.empty_insert_statement_value(primary_key))
+          im.into(arel_table)
+        else
+          im = arel_table.compile_insert(_substitute_values(values))
+        end
+
+        im.on_duplicate_update(on_duplicate)
+        im
+      end
+    end
+  end
+end

data/lib/active_record_upsert/active_record/relation.rb

@@ -0,0 +1,9 @@
+module ActiveRecordUpsert
+  module ActiveRecord
+    module Relation
+      def upsert(on_duplicate = nil)
+        scoping { klass._upsert_record(values_for_create, on_duplicate) }
+      end
+    end
+  end
+end

data/lib/active_record_upsert/arel/insert_manager.rb

@@ -0,0 +1,50 @@
+module ActiveRecordUpsert
+  module Arel
+    module InsertManager
+      def on_duplicate_update(expr = nil)
+        @ast.on_duplicate = OnDuplicate.new(expr, @ast.columns)
+      end
+
+      class OnDuplicate < ::Arel::Nodes::Node
+        attr_accessor :columns
+
+        def initialize(expr, columns)
+          @expr = expr
+          @columns = columns
+        end
+
+        def expr # rubocop:disable Metrics/MethodLength
+          case @expr
+            when nil
+              eq(first_col, first_col)
+
+            when String
+              sql(@expr)
+
+            when Hash
+              @expr.map do |key, val|
+                eq(sql(key.to_s), sql(val))
+              end
+
+            else
+              @expr
+          end
+        end
+
+        private
+
+        def first_col
+          sql(columns.first.name.to_s)
+        end
+
+        def sql(stmt)
+          ::Arel.sql(stmt)
+        end
+
+        def eq(left, right)
+          ::Arel::Nodes::Equality.new(left, right)
+        end
+      end
+    end
+  end
+end

data/lib/active_record_upsert/arel/insert_statement.rb

@@ -0,0 +1,12 @@
+module ActiveRecordUpsert
+  module Arel
+    module InsertStatement
+      attr_accessor :on_duplicate
+
+      def initialize
+        super()
+        @on_duplicate = nil
+      end
+    end
+  end
+end

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

@@ -0,0 +1,15 @@
+module ActiveRecordUpsert
+  module Arel
+    module Visitors
+      module MySQL
+        def visit_Arel_Nodes_InsertStatement(obj, collector) # rubocop:disable Naming/MethodName
+          collector = super(obj, collector)
+          return collector unless obj.on_duplicate
+
+          collector << ' ON DUPLICATE KEY UPDATE'
+          maybe_visit(obj.on_duplicate.expr, collector)
+        end
+      end
+    end
+  end
+end

data/lib/active_record_upsert/version.rb

@@ -0,0 +1,3 @@
+module ActiveRecordUpsert
+  VERSION = '0.1.0'.freeze
+end

data/lib/activerecord_upsert.rb

@@ -0,0 +1 @@
+require 'active_record_upsert'

metadata

@@ -0,0 +1,209 @@
+--- !ruby/object:Gem::Specification
+name: activerecord_upsert
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Daniel Vandersluis
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2019-10-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  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: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5.2'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5.2'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '6'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: database_cleaner
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.7.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.7.0
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.12.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.12.0
+- !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'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.75.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.75.0
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.36.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.36.0
+- !ruby/object:Gem::Dependency
+  name: rubocop-performance
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.5.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.5.0
+description: 
+email:
+- daniel.vandersluis@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- ".travis.yml"
+- CHANGELOG.md
+- Gemfile
+- README.md
+- Rakefile
+- activerecord_upsert.gemspec
+- bin/console
+- bin/setup
+- lib/active_record_upsert.rb
+- lib/active_record_upsert/active_record/persistence.rb
+- lib/active_record_upsert/active_record/relation.rb
+- lib/active_record_upsert/arel/insert_manager.rb
+- lib/active_record_upsert/arel/insert_statement.rb
+- lib/active_record_upsert/arel/visitors/mysql.rb
+- lib/active_record_upsert/version.rb
+- lib/activerecord_upsert.rb
+homepage: https://github.com/dvandersluis/activerecord_upsert
+licenses: []
+metadata:
+  homepage_uri: https://github.com/dvandersluis/activerecord_upsert
+  changelog_uri: https://github.com/dvandersluis/activerecord_upsert/CHANGELOG.md
+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: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Support MySQL's INSERT ... ON DUPLICATE KEY UPDATE in ActiveRecord
+test_files: []