add the ability to reindex all models in one rake command

Mehul Kar
1 parent d3c5df9d
Showing 2 changed files with 26 additions and 4 deletions Show diff stats
README.md
lib/searchkick/tasks.rb
@@ -398,8 +398,15 @@ Product.search &quot;milk&quot;, load: false
 3. Deploy and reindex
+  You can reindex a particular class:
+
+  ```ruby
+  rake searchkick:reindex:class[Product] # or Product.reindex in the console
+  ```
+
+  Or you can reindex all models at the same time:
   ```ruby
-  rake searchkick:reindex CLASS=Product # or Product.reindex in the console
+  rake searchkick:reindex:all
   ```
 4. Once it finishes, replace search calls w/ searchkick calls
@@ -2,8 +2,23 @@ require &quot;rake&quot;
 namespace :searchkick do
   desc "re-index elasticsearch"
-  task :reindex => :environment do
-    klass = ENV["CLASS"].constantize
-    klass.reindex
+  namespace :reindex do
+    desc "reindex a Model by passing it as an argument"
+    task :class, [:klass] => [:environment] do |t, args|
+      begin
+        args[:klass].constantize.reindex
+      rescue
+        puts "#{args[:klass]} model not found"
+      end
+    end
+
+    desc "reindex all models"
+    task :all => [:environment] do
+      Dir[Rails.root + "app/models/**/*.rb"].each { |path| require path }
+      models = ActiveRecord::Base.descendants.map(&:name)
+      models.each do |model|
+        model.constantize.reindex if model.respond_to?(:search)
+      end
+    end
   end
 end
	@@ -2,8 +2,23 @@ require "rake"		@@ -2,8 +2,23 @@ require "rake"
2		2
3	namespace :searchkick do	3	namespace :searchkick do
4	desc "re-index elasticsearch"	4	desc "re-index elasticsearch"
5	- task :reindex => :environment do
6	- klass = ENV["CLASS"].constantize
7	- klass.reindex	5	+ namespace :reindex do
		6	+ desc "reindex a Model by passing it as an argument"
		7	+ task :class, [:klass] => [:environment] do \|t, args\|
		8	+ begin
		9	+ args[:klass].constantize.reindex
		10	+ rescue
		11	+ puts "#{args[:klass]} model not found"
		12	+ end
		13	+ end
		14	+
		15	+ desc "reindex all models"
		16	+ task :all => [:environment] do
		17	+ Dir[Rails.root + "app/models/*/.rb"].each { \|path\| require path }
		18	+ models = ActiveRecord::Base.descendants.map(&:name)
		19	+ models.each do \|model\|
		20	+ model.constantize.reindex if model.respond_to?(:search)
		21	+ end
		22	+ end
8	end	23	end
9	end	24	end