atomically 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 053a70728eaf97d497e01c5eb97d28b8606a13f6
-  data.tar.gz: d19a6e6f011e0fa73f8ed40a6f48fc514d426c16
+SHA256:
+  metadata.gz: d38cf83ab4f1e792cfefbb2c01ac897377fb7af2a81da6f09f932df4d5fa76f2
+  data.tar.gz: 88640837b4c69ef162d036b2a3e6ad61087e94a6baa79cde026fb748f09b58f1
 SHA512:
-  metadata.gz: 642a937243e5e133294cfc5d97be7201cad09041717ff13ec512f8f23d4fc128d18dfd78d25d4130d0c0526655ccd232b25ada2decf284b1e55ece0c990b2dea
-  data.tar.gz: 247cbb367d404627ab799af68f35de8dd2294e0a7b1d585c061e8b5d4a25ae5691af646577a51dd8a9b1e696304b1cd784ddc45773c7ec55acb74529d1e7fd77
+  metadata.gz: cd968dcc060705bc3d0174daf532fa4c80b8dfadc5845a7406bcf68d930744c6ce78e2f04b97091eb4cea27818ed44df1e4b3c47eefac5e0643ab492b9bed21b
+  data.tar.gz: 7577dd5ef6f32a4d3753c71637d218358d48aeac8b7c675c9b780d906af56f08bd9cf5cb14d7b98fd12a9275d28f6d45ec8a5baaccb3b177257e300a04c0c890

data/README.md

@@ -31,20 +31,28 @@ Or install it yourself as:
 ### create_or_plus
 
 Import an array of records. When key is duplicate, plus the old value with new value.
-It is useful to add `items` to user when `user_items` may not exist.
+It is useful to add `items` to `user` when `user_items` may not exist.
 
 First two args (columns, values) are the same with the [import](https://github.com/zdennis/activerecord-import#columns-and-arrays) method.
 
 Example:
-
 ```rb
-columns = [:user_id, :item_id, :count]
+columns = [:user_id, :item_id, :quantity]
 values = [[user.id, item1.id, 3], [user.id, item2.id, 2]]
-on_duplicate_update_columns = [:count]
+on_duplicate_update_columns = [:quantity]
 
 UserItem.atomically.create_or_plus(columns, values, on_duplicate_update_columns)
 ```
 
+### pay_all
+Reduce the quantity of items and return how many rows and updated if all of them is enough.
+Do nothing and return zero if any of them is not enough.
+
+Example:
+```rb
+user.user_items.atomically.pay_all({ item1.id => 4, item2.id => 3 }, [:quantity], primary_key: :item_id)
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test DB=mysql` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/atomically.gemspec

@@ -34,6 +34,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "mysql2", ">= 0.3"
   spec.add_development_dependency "pluck_all", ">= 2.0.3"
 
-  spec.add_dependency "activerecord-import", ">= 0.27.0"
   spec.add_dependency "activerecord", ">= 3"
+  spec.add_dependency "activerecord-import", ">= 0.27.0"
+  spec.add_dependency "rails_or", ">= 1.1.8"
 end

data/lib/atomically/active_record/extension.rb

@@ -3,7 +3,7 @@ require 'atomically/query_service'
 
 class ActiveRecord::Relation
   def atomically
-    Atomically::QueryService.new(klass)
+    Atomically::QueryService.new(klass, relation: self)
   end
 end
 

data/lib/atomically/patches/from.rb

@@ -0,0 +1,8 @@
+require 'active_record'
+
+class << ActiveRecord::Base
+  def from(value) # For Rails 3
+    value = "(#{value.to_sql}) subquery" if value.is_a?( ActiveRecord::Relation )
+    return super
+  end
+end

data/lib/atomically/patches/none.rb

@@ -0,0 +1,7 @@
+require 'active_record'
+
+class << ActiveRecord::Base
+  def none # For Rails 3
+    where('1=0')
+  end
+end

data/lib/atomically/query_service.rb

@@ -1,21 +1,52 @@
+# frozen_string_literal: true
+
 require 'activerecord-import'
+require 'rails_or'
+require 'atomically/patches/none' if not ActiveRecord::Base.respond_to?(:none)
+require 'atomically/patches/from' if Gem::Version.new(ActiveRecord::VERSION::STRING) < Gem::Version.new('4.0.0')
 
 class Atomically::QueryService
-  def initialize(klass)
+  def initialize(klass, relation: nil)
     @klass = klass
+    @relation = relation || @klass
   end
 
   def create_or_plus(columns, data, update_columns)
     @klass.import(columns, data, on_duplicate_key_update: on_duplicate_key_plus_sql(update_columns))
   end
 
+  def pay_all(hash, update_columns, primary_key: :id) # { id => pay_count }
+    return 0 if hash.blank?
+
+    update_columns = update_columns.map(&method(:quote_column))
+
+    query = hash.inject(@klass.none) do |relation, (id, pay_count)|
+      condition = @relation.where(primary_key => id)
+      update_columns.each{|s| condition = condition.where("#{s} >= ?", pay_count) }
+      next relation.or(condition)
+    end
+
+    raw_when_sql = hash.map{|id, pay_count| "WHEN #{sanitize(id)} THEN #{sanitize(-pay_count)}" }.join("\n")
+    update_sqls = update_columns.map.with_index do |column, idx|
+      value = idx == 0 ? "(@change := \nCASE #{quote_column(primary_key)}\n#{raw_when_sql}\nEND)" : '@change'
+      next "#{column} = #{column} + #{value}"
+    end
+
+    return query.where("(#{@klass.from(query).select('COUNT(*)').to_sql}) = ?", hash.size)
+                .update_all(update_sqls.join(', '))
+  end
+
   private
 
   def on_duplicate_key_plus_sql( columns)
-    columns.lazy
-           .map(&@klass.connection.method(:quote_column_name))
-           .map{|s| "#{s} = #{s} + VALUES(#{s})" }
-           .force
-           .join(', ')
+    columns.lazy.map(&method(:quote_column)).map{|s| "#{s} = #{s} + VALUES(#{s})" }.force.join(', ')
+  end
+
+  def quote_column(column)
+    @klass.connection.quote_column_name(column)
+  end
+
+  def sanitize(value)
+    @klass.connection.quote(value)
   end
 end

data/lib/atomically/version.rb

@@ -1,3 +1,3 @@
 module Atomically
-  VERSION = '1.0.0'
+  VERSION = '1.0.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: atomically
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - khiav reoy
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-11-21 00:00:00.000000000 Z
+date: 2018-11-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -94,6 +94,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 2.0.3
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3'
 - !ruby/object:Gem::Dependency
   name: activerecord-import
   requirement: !ruby/object:Gem::Requirement
@@ -109,19 +123,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: 0.27.0
 - !ruby/object:Gem::Dependency
-  name: activerecord
+  name: rails_or
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '3'
+        version: 1.1.8
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '3'
+        version: 1.1.8
 description: ''
 email:
 - mrtmrt15xn@yahoo.com.tw
@@ -149,6 +163,8 @@ files:
 - gemfiles/5.2.gemfile
 - lib/atomically.rb
 - lib/atomically/active_record/extension.rb
+- lib/atomically/patches/from.rb
+- lib/atomically/patches/none.rb
 - lib/atomically/query_service.rb
 - lib/atomically/version.rb
 homepage: https://github.com/khiav223577/atomically
@@ -171,7 +187,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.13
+rubygems_version: 2.7.6
 signing_key: 
 specification_version: 4
 summary: ''