redis-search 1.0.1 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4553fdbea735f9b74b36c0e25bea9cc714f62d89
-  data.tar.gz: bfa28d488ee3db6d8a54b420b5bc4c42cd422468
+  metadata.gz: 804fc6a6a29426a456e61001196c3649286e0ec1
+  data.tar.gz: 580d8cea7d7bc1189bf3bf05ac701b4f1758de07
 SHA512:
-  metadata.gz: afde37ef18d54c870c2f3b52748c3d4c226e2a22e3dde0c911762a216f776426cc2fc2fe74b1b7cd765cf09f58ca361c2ea5642089af656836f9578f4a8cc9bd
-  data.tar.gz: c11756ffac26e7c85b02715d7c1b02b6a192651172778ce1169cc566deba5e513fd11c6827a606f32a2b3c82127929ec65427b8758da3a9e56fd03a43f0b1765
+  metadata.gz: aa20b21f334eb09449a3271a702a95f6c1f2f537eec92d7128046e887cfc6c995a35a27ae5a4c486045bda1b97f01650b656c2d8c02ce0faeeac6fcba5a0414e
+  data.tar.gz: 2ecdffdd2e841868404602a79d0445e98cf0e7e1ad6181417dd61c76b8a61bf0bce6c5e0ef568a5374d16e37b5328d6b1e1a622156572636f8ed035f01976c7c

data/lib/redis-search/base.rb

@@ -147,15 +147,17 @@ class Redis
         Redis::Search.complete(self.name, q, opts)
       end
 
-      def redis_search_index_batch_create(batch_size = 1000, progressbar = false)
+      def redis_search_index_batch_create(batch_size = 1000)
         count = 0
         if ancestors.collect(&:to_s).include?('ActiveRecord::Base'.freeze)
           find_in_batches(batch_size: batch_size) do |items|
             _redis_search_reindex_items(items)
+            count += item.count
           end
         elsif included_modules.collect(&:to_s).include?('Mongoid::Document'.freeze)
-          all.each_slice(batch_size) do |items|
+          self.all.each_slice(batch_size) do |items|
             _redis_search_reindex_items(items)
+            count += item.count
           end
         else
           puts 'skiped, not support this ORM in current.'
@@ -169,8 +171,7 @@ class Redis
       def _redis_search_reindex_items(items)
         items.each do |item|
           item.redis_search_index_create
-          count += 1
-          print DOT if progressbar
+          print DOT
           item = nil
         end
         items = nil

data/lib/redis-search/tasks.rb

@@ -32,10 +32,10 @@ namespace :redis_search do
       end
 
       klass = Object.const_get(ENV['CLASS'].to_s)
-      batch = ENV['BATCH'].to_i || 1000
+      batch = ENV['BATCH'].to_i > 0 ? ENV['BATCH'].to_i : 1000
       start = Time.now
       puts "Redis-Search index data to Redis from [#{klass}]"
-      count = klass.redis_search_index_batch_create(batch, true)
+      count = klass.redis_search_index_batch_create(batch)
       puts ''
       puts "Indexed #{count} rows  |  Time spend: #{(Time.now - start)}s"
       puts 'Rebuild Index done.'
@@ -63,7 +63,7 @@ namespace :redis_search do
       puts "Redis-Search index data to Redis from [#{dir}]"
       Redis::Search.indexed_models.each do |klass|
         puts "[#{klass}]"
-        count += klass.redis_search_index_batch_create(batch, true)
+        count += klass.redis_search_index_batch_create(batch)
         puts ''
       end
       puts "Indexed #{count} rows  |  Time spend: #{(Time.now - tm)}s"

data/lib/redis-search/version.rb

@@ -1,5 +1,5 @@
 class Redis
   module Search
-    VERSION = '1.0.1'
+    VERSION = '1.0.2'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: redis-search
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.2
 platform: ruby
 authors:
 - Jason Lee