quick_random_records 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6610b33086b881bea06147c1096cfdce6c4ef795
-  data.tar.gz: a620b9ec21e99e90c206618157e3e6e2ec0066e0
+  metadata.gz: 8460dba37c8d16afde91b5f8ed2e85edb122c5ef
+  data.tar.gz: 3ff4a0cdc2df474bfb3e11718e25669582b75797
 SHA512:
-  metadata.gz: 37750f06ce6f2f07fb840664485ca2a5f41fdcb8a6ee9fde89a48ec2598015b6d084482b85014b064813235ac27fc2c8d9759fdc7dbfd7a0e6cdae9f8815b8a0
-  data.tar.gz: b7a01f2b84b59723a1e95a0d19ad0bfbcd83c86d6c8da5a5163b74107773bab3f635b2ddb1f7cb9183e46603849efdb9d8b6693b57ce2ed77a0acd259fc5f1e4
+  metadata.gz: 33fe51141b1fb07b2a0fdd77d7f188c69da47bcee9854de1513446a205d59b5e2fdad410764840211a8a38d573ad82f8bf13b9c83d656735cf0f2d0d2f8c24db
+  data.tar.gz: 2019ea12883b3bc1c6e848709b5c9ba2c44e95a933242a8231e0886a9c91842113d20801b2764b94a55be4a1fef49973b65ef9f4f166c3d013e25596a7e9f360

data/Gemfile

@@ -3,7 +3,6 @@ source 'https://rubygems.org'
 # Specify your gem's dependencies in quick_random_records.gemspec
 
 gem 'activerecord'
-gem 'rails_or'
 
 group :test do
   gem 'simplecov'

data/lib/quick_random_records/version.rb

@@ -1,3 +1,3 @@
 module QuickRandomRecords
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/quick_random_records.rb

@@ -1,27 +1,61 @@
 require 'quick_random_records/version'
 require 'active_record'
-require 'rails_or'
 
 class ActiveRecord::Base
-  def self.random_records(quantity)
-    id_range = 1..self.last.id
-    sample_ids = [*id_range].sample(quantity)
-    samples = self.where(id: sample_ids)
+  def self.random_records(quantity, strategy: 1, multiple: 1.25, loop_limit: 3)
+    case strategy
+    when 1
+      self.sample_complement_records(quantity, multiple, loop_limit)
+    when 2
+      self.order_rand_limit_records(quantity)
+    when 3
+      self.pluck_sample_records(quantity)
+    else
+      "this gem doesn't support strategy other than 1, 2, 3"
+    end
+  end
+
+  private
+
+  def self.sample_complement_records(quantity, multiple, loop_limit)
+    min_max = self.pluck('MIN(id), MAX(id)').first
+    id_range = (min_max[0]..min_max[1])
 
-    while samples.size < quantity
-      complement = []
+    samples = [*id_range].sample(quantity * multiple)
+    exist_samples = self.where(id: samples).pluck(:id)
+    exist_samples_size = exist_samples.size
+    deficit = quantity - exist_samples_size
+    exist_samples_size = 1 if exist_samples_size.zero?
+    deficit_weight = quantity * multiple / exist_samples_size
+    n = 1
 
-      while complement.empty?
-        complement_id = rand(id_range)
-        next if sample_ids.include?(complement_id)
+    while deficit > 0 && n <= loop_limit
+      complements = ([*id_range] - samples).sample(deficit * multiple * deficit_weight * n)
+      exist_complements = self.where(id: complements).pluck(:id)
 
-        sample_ids << complement_id
-        complement = self.where(id: complement_id)
-      end
+      deficit -= exist_complements.size
+      samples += complements
+      exist_samples += exist_complements
+      n += 1
+    end
+
+    self.where(id: exist_samples[0...quantity])
+  end
 
-      samples = samples.or(complement)
+  def self.order_rand_limit_records(quantity)
+   adapter_type = connection.adapter_name.downcase.to_sym
+    case adapter_type
+    when :mysql, :mysql2
+      self.order("RAND()").limit(quantity)
+    when :sqlite, :postgresql
+      self.order("RANDOM()").limit(quantity)
+    else
+      raise NotImplementedError, "Unknown adapter type '#{adapter_type}'"
     end
+  end
 
-    samples
+  def self.pluck_sample_records(quantity)
+    ids = self.pluck(:id).sample(quantity)
+    self.where(id: ids)
   end
 end

data/quick_random_records.gemspec

@@ -25,5 +25,4 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "minitest", "~> 5.0"
 
   spec.add_dependency "activerecord", ">= 3"
-  spec.add_dependency "rails_or"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: quick_random_records
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - derekfan
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-05-23 00:00:00.000000000 Z
+date: 2018-05-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -80,20 +80,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '3'
-- !ruby/object:Gem::Dependency
-  name: rails_or
-  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'
 description: Returns random records for Ruby Models fast and quick
 email:
 - haoping.fan@gmail.com