dynamoid-advanced-where 1.0.0

data/Rakefile

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

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "dynamoid/advanced/where"
+
+# 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/dynamoid_advanced_where.gemspec

@@ -0,0 +1,41 @@
+lib = File.expand_path('lib', __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'dynamoid_advanced_where/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'dynamoid-advanced-where'
+  spec.version       = DynamoidAdvancedWhere::VERSION
+  spec.authors       = ['Brian Malinconico']
+  spec.email         = ['bmalinconico@terminus.com']
+
+  spec.summary       = 'things'
+  spec.description   = 'things'
+  # spec.homepage      = "things"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  # if spec.respond_to?(:metadata)
+  #   spec.metadata["allowed_push_host"] = "TODO: Set to 'http://mygemserver.com'"
+  # else
+  #   raise "RubyGems 2.0 or newer is required to protect against " \
+  #     "public gem pushes."
+  # end
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |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 'dynamoid', '>= 3.2', '< 4'
+
+  spec.add_development_dependency 'bundle-audit'
+  spec.add_development_dependency 'bundler', '>= 1.16'
+  spec.add_development_dependency 'fasterer'
+  spec.add_development_dependency 'overcommit'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_development_dependency 'rubocop'
+  spec.add_development_dependency 'appraisal'
+end

data/gemfiles/.bundle/config

@@ -0,0 +1,2 @@
+---
+BUNDLE_RETRY: "1"

data/gemfiles/dynamoid_3.4.gemfile

@@ -0,0 +1,8 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "pry"
+gem "dynamoid", "~> 3.4.0"
+
+gemspec path: "../"

data/gemfiles/dynamoid_3.4.gemfile.lock

@@ -0,0 +1,118 @@
+PATH
+  remote: ..
+  specs:
+    dynamoid-advanced-where (0.1.0)
+      dynamoid (>= 3.2, < 4)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    activemodel (6.0.2.1)
+      activesupport (= 6.0.2.1)
+    activesupport (6.0.2.1)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 0.7, < 2)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+      zeitwerk (~> 2.2)
+    appraisal (2.2.0)
+      bundler
+      rake
+      thor (>= 0.14.0)
+    ast (2.4.0)
+    aws-eventstream (1.0.3)
+    aws-partitions (1.270.0)
+    aws-sdk-core (3.89.1)
+      aws-eventstream (~> 1.0, >= 1.0.2)
+      aws-partitions (~> 1, >= 1.239.0)
+      aws-sigv4 (~> 1.1)
+      jmespath (~> 1.0)
+    aws-sdk-dynamodb (1.41.0)
+      aws-sdk-core (~> 3, >= 3.71.0)
+      aws-sigv4 (~> 1.1)
+    aws-sigv4 (1.1.0)
+      aws-eventstream (~> 1.0, >= 1.0.2)
+    bundle-audit (0.1.0)
+      bundler-audit
+    bundler-audit (0.6.1)
+      bundler (>= 1.2.0, < 3)
+      thor (~> 0.18)
+    childprocess (3.0.0)
+    coderay (1.1.2)
+    colorize (0.8.1)
+    concurrent-ruby (1.1.5)
+    diff-lcs (1.3)
+    dynamoid (3.4.1)
+      activemodel (>= 4)
+      aws-sdk-dynamodb (~> 1)
+      concurrent-ruby (>= 1.0)
+      null-logger
+    fasterer (0.8.2)
+      colorize (~> 0.7)
+      ruby_parser (>= 3.14.1)
+    i18n (1.8.2)
+      concurrent-ruby (~> 1.0)
+    iniparse (1.4.4)
+    jaro_winkler (1.5.4)
+    jmespath (1.4.0)
+    method_source (0.9.2)
+    minitest (5.14.0)
+    null-logger (0.1.5)
+    overcommit (0.52.1)
+      childprocess (>= 0.6.3, < 4)
+      iniparse (~> 1.4)
+    parallel (1.19.1)
+    parser (2.7.0.2)
+      ast (~> 2.4.0)
+    pry (0.12.2)
+      coderay (~> 1.1.0)
+      method_source (~> 0.9.0)
+    rainbow (3.0.0)
+    rake (10.5.0)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.1)
+      rspec-support (~> 3.9.1)
+    rspec-expectations (3.9.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.2)
+    rubocop (0.79.0)
+      jaro_winkler (~> 1.5.1)
+      parallel (~> 1.10)
+      parser (>= 2.7.0.1)
+      rainbow (>= 2.2.2, < 4.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 1.4.0, < 1.7)
+    ruby-progressbar (1.10.1)
+    ruby_parser (3.14.1)
+      sexp_processor (~> 4.9)
+    sexp_processor (4.13.0)
+    thor (0.20.3)
+    thread_safe (0.3.6)
+    tzinfo (1.2.6)
+      thread_safe (~> 0.1)
+    unicode-display_width (1.6.1)
+    zeitwerk (2.2.2)
+
+PLATFORMS
+  ruby
+  x86_64-darwin-18
+
+DEPENDENCIES
+  appraisal
+  bundle-audit
+  bundler (>= 1.16)
+  dynamoid (~> 3.4.0)
+  dynamoid-advanced-where!
+  fasterer
+  overcommit
+  pry
+  rake (~> 10.0)
+  rspec (~> 3.0)
+  rubocop

data/gemfiles/dynamoid_latest.gemfile

@@ -0,0 +1,8 @@
+# This file was generated by Appraisal
+
+source "https://rubygems.org"
+
+gem "pry"
+gem "dynamoid", "~> 3.0"
+
+gemspec path: "../"

data/gemfiles/dynamoid_latest.gemfile.lock

@@ -0,0 +1,118 @@
+PATH
+  remote: ..
+  specs:
+    dynamoid-advanced-where (0.1.0)
+      dynamoid (>= 3.2, < 4)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    activemodel (6.0.2.1)
+      activesupport (= 6.0.2.1)
+    activesupport (6.0.2.1)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 0.7, < 2)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+      zeitwerk (~> 2.2)
+    appraisal (2.2.0)
+      bundler
+      rake
+      thor (>= 0.14.0)
+    ast (2.4.0)
+    aws-eventstream (1.0.3)
+    aws-partitions (1.270.0)
+    aws-sdk-core (3.89.1)
+      aws-eventstream (~> 1.0, >= 1.0.2)
+      aws-partitions (~> 1, >= 1.239.0)
+      aws-sigv4 (~> 1.1)
+      jmespath (~> 1.0)
+    aws-sdk-dynamodb (1.41.0)
+      aws-sdk-core (~> 3, >= 3.71.0)
+      aws-sigv4 (~> 1.1)
+    aws-sigv4 (1.1.0)
+      aws-eventstream (~> 1.0, >= 1.0.2)
+    bundle-audit (0.1.0)
+      bundler-audit
+    bundler-audit (0.6.1)
+      bundler (>= 1.2.0, < 3)
+      thor (~> 0.18)
+    childprocess (3.0.0)
+    coderay (1.1.2)
+    colorize (0.8.1)
+    concurrent-ruby (1.1.5)
+    diff-lcs (1.3)
+    dynamoid (3.4.1)
+      activemodel (>= 4)
+      aws-sdk-dynamodb (~> 1)
+      concurrent-ruby (>= 1.0)
+      null-logger
+    fasterer (0.8.2)
+      colorize (~> 0.7)
+      ruby_parser (>= 3.14.1)
+    i18n (1.8.2)
+      concurrent-ruby (~> 1.0)
+    iniparse (1.4.4)
+    jaro_winkler (1.5.4)
+    jmespath (1.4.0)
+    method_source (0.9.2)
+    minitest (5.14.0)
+    null-logger (0.1.5)
+    overcommit (0.52.1)
+      childprocess (>= 0.6.3, < 4)
+      iniparse (~> 1.4)
+    parallel (1.19.1)
+    parser (2.7.0.2)
+      ast (~> 2.4.0)
+    pry (0.12.2)
+      coderay (~> 1.1.0)
+      method_source (~> 0.9.0)
+    rainbow (3.0.0)
+    rake (10.5.0)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.1)
+      rspec-support (~> 3.9.1)
+    rspec-expectations (3.9.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.2)
+    rubocop (0.79.0)
+      jaro_winkler (~> 1.5.1)
+      parallel (~> 1.10)
+      parser (>= 2.7.0.1)
+      rainbow (>= 2.2.2, < 4.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 1.4.0, < 1.7)
+    ruby-progressbar (1.10.1)
+    ruby_parser (3.14.1)
+      sexp_processor (~> 4.9)
+    sexp_processor (4.13.0)
+    thor (0.20.3)
+    thread_safe (0.3.6)
+    tzinfo (1.2.6)
+      thread_safe (~> 0.1)
+    unicode-display_width (1.6.1)
+    zeitwerk (2.2.2)
+
+PLATFORMS
+  ruby
+  x86_64-darwin-18
+
+DEPENDENCIES
+  appraisal
+  bundle-audit
+  bundler (>= 1.16)
+  dynamoid (~> 3.0)
+  dynamoid-advanced-where!
+  fasterer
+  overcommit
+  pry
+  rake (~> 10.0)
+  rspec (~> 3.0)
+  rubocop

data/lib/dynamoid_advanced_where.rb

@@ -0,0 +1,8 @@
+require 'dynamoid'
+
+require "dynamoid_advanced_where/version"
+require "dynamoid_advanced_where/integrations/model"
+
+module DynamoidAdvancedWhere
+      # Your code goes here...
+end

data/lib/dynamoid_advanced_where/batched_updater.rb

@@ -0,0 +1,229 @@
+# frozen_string_literal: true
+
+module DynamoidAdvancedWhere
+  class BatchedUpdater
+    DEEP_MERGE_ATTRIBUTES = %i[expression_attribute_names expression_attribute_values].freeze
+
+    attr_accessor :query_builder, :_set_values, :_array_appends, :_set_appends, :_increments
+    delegate :klass, to: :query_builder
+
+    def initialize(query_builder:)
+      self.query_builder = query_builder
+      self._set_values = {}
+      self._set_appends = []
+      self._array_appends = []
+      self._increments = Hash.new(0)
+    end
+
+    def apply(hash_key, range_key = nil)
+      key_args = {
+        table_name: klass.table_name,
+        return_values: 'ALL_NEW',
+        key: {
+          klass.hash_key => hash_key,
+          klass.range_key => range_key
+        }.delete_if { |k, _v| k.nil? }
+      }
+      resp = client.update_item(update_item_arguments.merge(key_args))
+
+      klass.from_database(resp.attributes)
+    rescue Aws::DynamoDB::Errors::ConditionalCheckFailedException
+    end
+
+    def set_values(vals)
+      _set_values.merge!(vals)
+      self
+    end
+
+    def append_to(appends)
+      appends.each do |k, v|
+        case klass.attributes[k.to_sym][:type]
+        when :set
+          _set_appends << { k => v.to_set }
+        when :array
+          _array_appends << { k => v }
+        else
+          raise 'can only append to sets or arrays'
+        end
+      end
+
+      self
+    end
+
+    def increment(*fields, by: 1)
+      fields.each { |field| _increments[field] += by }
+      self
+    end
+
+    def decrement(*fields, by: 1)
+      increment(*fields, by: -1 * by)
+      self
+    end
+
+    private
+
+    def merge_multiple_sets(items_to_merge, result_base: {})
+      default = { collected_update_expression: [] }
+      result = result_base.merge(default)
+      items_to_merge.each do |update_data|
+        result[:collected_update_expression] << update_data.delete(:collected_update_expression)
+        result.merge!(update_data, &method(:hash_extendeer))
+      end
+
+      result[:collected_update_expression].flatten!
+      result[:collected_update_expression].reject!(&:blank?)
+
+      return default if result[:collected_update_expression].empty?
+
+      result
+    end
+
+    def field_update_arguments
+      merge_multiple_sets([set_values_update_args])
+    end
+
+    def update_item_arguments
+      filter = merge_multiple_sets(
+        [
+          field_update_arguments,
+          add_update_args,
+        ],
+        result_base: filter_builder.to_scan_filter,
+      )
+
+      filter[:update_expression] = filter.delete(:collected_update_expression).join(' ')
+      filter[:condition_expression] = filter.delete(:filter_expression)
+
+      filter
+    end
+
+    def args_to_update_command(update_args, command:)
+      return {} if update_args[:collected_update_expression].empty?
+
+      update_args.merge!(
+        collected_update_expression: [
+          "#{command} #{update_args[:collected_update_expression].join(', ')}"
+        ]
+      )
+    end
+
+    def set_values_update_args
+      args_to_update_command(
+        merge_multiple_sets(
+          [
+            explicit_set_args,
+            list_append_for_arrays,
+            increment_field_updates
+          ]
+        ),
+        command: 'SET'
+      )
+    end
+
+    def add_update_args
+      args_to_update_command(list_append_for_sets, command: 'ADD')
+    end
+
+    def explicit_set_args
+      builder_hash = { collected_update_expression: [] }
+
+      _set_values.each_with_object(builder_hash) do |(k, v), h|
+        prefix = merge_in_attr_placeholders(h, k, v)
+        h[:collected_update_expression] << "##{prefix} = :#{prefix}"
+      end
+    end
+
+
+    def increment_field_updates
+      return {} if _increments.empty?
+
+      zero_prefix = SecureRandom.hex
+
+      builder_hash = {
+        collected_update_expression: [],
+        expression_attribute_values: {
+          ":#{zero_prefix}": 0
+        }
+      }
+
+      _increments.each_with_object(builder_hash) do |(field, change), h|
+        prefix = merge_in_attr_placeholders(h, field, change)
+        builder_hash[:collected_update_expression] << "##{prefix} = if_not_exists(##{prefix}, :#{zero_prefix}) + :#{prefix}"
+      end
+    end
+
+    def list_append_for_sets
+      builder_hash = { collected_update_expression: [] }
+
+      _set_appends.each_with_object(builder_hash) do |to_append, h|
+        to_append.each do |k, v|
+          prefix = merge_in_attr_placeholders(h, k, v)
+          builder_hash[:collected_update_expression] << "##{prefix} :#{prefix}"
+        end
+      end
+    end
+
+    def list_append_for_arrays
+      empty_list_prefix = SecureRandom.hex
+
+      builder_hash = {
+        collected_update_expression: [],
+        expression_attribute_values: {
+          ":#{empty_list_prefix}": []
+        }
+      }
+
+      update_args = _array_appends.each_with_object(builder_hash) do |to_append, h|
+        to_append.each do |k, v|
+          prefix = merge_in_attr_placeholders(h, k, v)
+          builder_hash[:collected_update_expression] << "##{prefix}  = list_append(if_not_exists(##{prefix}, :#{empty_list_prefix}), :#{prefix})"
+        end
+      end
+
+      builder_hash[:collected_update_expression].empty? ? {} : update_args
+    end
+
+    def merge_in_attr_placeholders(hsh, field_name, value)
+      prefix, new_data = prefixerize(field_name, value)
+
+      hsh.merge!(new_data, &method(:hash_extendeer))
+
+      prefix
+    end
+
+    def prefixerize(field_name, value)
+      prefix = SecureRandom.hex
+
+      [
+        prefix,
+        {
+          expression_attribute_names: { "##{prefix}" => field_name },
+          expression_attribute_values: {
+            ":#{prefix}" => dump(value, klass.attributes[field_name])
+          }
+        }
+      ]
+    end
+
+    def hash_extendeer(key, old_value, new_value)
+      return new_value unless key.in?(DEEP_MERGE_ATTRIBUTES)
+
+      old_value.merge(new_value)
+    end
+
+    def client
+      Dynamoid.adapter.client
+    end
+
+    def filter_builder
+      @filter_builder ||= FilterBuilder.new(
+        root_node: query_builder.root_node,
+        klass: klass,
+      )
+    end
+
+    def dump(value, config)
+      Dynamoid::Dumping.dump_field(value, config)
+    end
+  end
+end