ippeiukai

"use strict";

const Sequelize = require('sequelize');
const Promise = Sequelize.Promise;

const DEFAULT_BATCH_SIZE = 3000;


/**
 * Port of ActiveRecord::Base.find_each of Rails.

ippeiukai

module SingletonWithClassDelegator
  extend ActiveSupport::Concern

  included do
    include Singleton
    original_delegate_method = method(:delegate).unbind if self.respond_to?(:delegate)
    extend SingleForwardable
    if original_delegate_method.present?
      # restore the delegate of ActiveSupport that has been overshadowed by SingleForwardable
      define_singleton_method :delegate, original_delegate_method

ippeiukai

# module FooBar
#   extend DelegatedModuleFunction
#
#   define_delegated_module_functions do
#
#     def foo_bar
#       'Foo' + bar
#     end
#
#     private

ippeiukai

class IOStreamingResponseBody

  def self.render(options = {}, &block)
    instance = new(options)
    instance.define_singleton_method :render do |dummy_io|
      block.call(dummy_io)
    end
    instance
  end

ippeiukai

module EnumeratorWithPositionAndSize

  # @yields [original_params, position_and_size]
  # @yieldparam position_and_size [(Integer, Integer)] 2-element array of position (first being 1) and size; size may be nil (see Enumerator#size)
  # @example
  #  %w[a b c d e].each.with_position_and_size { |char, (position, size)| puts "#{char} (#{position}/#{size})" }
  def with_position_and_size
    return to_enum(__method__) unless block_given?
    my_size = self.size
    self.with_index(1) do |original_params, position|

ippeiukai

module ActiveRecordRelationExists

  # DANGER this switches ActiveRecord::Relation#exists from delegation to arel to our implementation that returns a String with a twist.
  #  This achieves the most common usage of #exists before 4.2. Proper fix will be: https://github.com/rails/rails/issues/16959
  def exists
    SqlStringWithNot.new("EXISTS (#{self.to_sql})")
  end

  class SqlStringWithNot < String

ippeiukai

# ActiveRecord's find_each and find_in_batches ported to Sequel.
# Sequel's paged_each is not practical when converting large data due to its use of transaction and offset.
#
# Usage:
#
#   SequelEachInBatches.find_each(dataset, keys) { |record| ... }
#
# It can also monkey patch Sequel::Dataset:
#
#   Sequel::Dataset.send(:include, SequelEachInBatches)

ippeiukai

module UseSingleForwardableWithActiveSupport
  extend ActiveSupport::Concern

  included do
    original_delegate_method = method(:delegate).unbind if self.respond_to?(:delegate)
    extend SingleForwardable
    if original_delegate_method.present?
      # restore the delegate of ActiveSupport that has been overshadowed by SingleForwardable
      define_singleton_method :delegate, original_delegate_method

ippeiukai

module MongoidEmbeddedObjectModification

  # Custom field behaves more like an immutable value.
  # Your changes to the object obtained with getter method will not take an effect unless you assign it back.
  # https://github.com/mongoid/mongoid/issues/3341
  def modify(mongoid_doc, field_name)
    obj = mongoid_doc.public_send(field_name)
    yield obj
    mongoid_doc.public_send("#{field_name}=", obj)
  end

ippeiukai

module MongoidEmbeddedObjectModification

  # Custom field behaves more like an immutable value.
  # Your changes to the object obtained with getter method will not take an effect unless you assign it back.
  # https://github.com/mongoid/mongoid/issues/3341
  def modify(mongoid_doc, field_name)
    obj = mongoid_doc.public_send(field_name)
    yield obj
    mongoid_doc.public_send("#{field_name}=", obj)
  end