lib/searchkick/tasks.rb in searchkick-0.2.2 vs lib/searchkick/tasks.rb in searchkick-0.2.3

@@ -1,9 +1,33 @@
 require "rake"
 
 namespace :searchkick do
-  desc "re-index elasticsearch"
+
+  desc "reindex model"
   task :reindex => :environment do
-    klass = ENV["CLASS"].constantize
-    klass.reindex
+    if ENV["CLASS"]
+      klass = ENV["CLASS"].constantize rescue nil
+      if klass
+        klass.reindex
+      else
+        abort "Could not find class: #{ENV["CLASS"]}"
+      end
+    else
+      abort "USAGE: rake searchkick:reindex CLASS=Product"
+    end
   end
+
+  if defined?(Rails)
+
+    namespace :reindex do
+      desc "reindex all models"
+      task :all => :environment do
+        Rails.application.eager_load!
+        (Searchkick::Reindex.instance_variable_get(:@descendents) || []).each do |model|
+          model.reindex
+        end
+      end
+    end
+
+  end
+
 end