From 8ebb3ffadae578c74a2fe8e576e5d1dd1a8ec2a2 Mon Sep 17 00:00:00 2001
From: Thomas Scherz <scherztc@ucmail.uc.edu>
Date: Wed, 11 Sep 2024 14:12:40 -0400
Subject: [PATCH] Adds MyReindex to assist with SOLR upgrade.

---
 lib/tasks/reindex.rake | 47 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 lib/tasks/reindex.rake

diff --git a/lib/tasks/reindex.rake b/lib/tasks/reindex.rake
new file mode 100644
index 00000000..5a4218cf
--- /dev/null
+++ b/lib/tasks/reindex.rake
@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+namespace :reindex do
+  desc "Reindex specified work, e.g., rake myreindex:reindex_by_id['c821gj76b']"
+  task :reindex_by_id, [:id] => :environment do |_, args|
+    object = find_or_warn(args[:id]) || next
+
+    if object.id.blank?
+      $stderr.puts "No such work exists for id: #{object.id}"
+      next
+    else
+      ActiveFedora::Base.find(object.id).update_index
+      puts "Reindexed work id #{object.id}"
+    end
+  end
+
+  desc "Reindex collections"
+  task reindex_collections: :environment do
+    count = 0
+    Collection.all.each do |object|
+      if object.id.blank?
+        $stderr.puts "No such work exists for id: #{object.id}"
+        next
+      else
+        ActiveFedora::Base.find(object.id).update_index
+        count += 1
+      end
+    end
+    puts "Reindexed works for #{count} objects"
+  end
+
+  desc "Reindex all objects"
+  task reindex_works: :environment do
+    count = 0
+    [Image, Document, ExternalObject, MedSym, Review].each do |model_class|
+      model_class.all.each do |object|
+        if object.id.blank?
+          $stderr.puts "No such work exists for id: #{object.id}"
+          next
+        else
+          ActiveFedora::Base.find(object.id).update_index
+          count += 1
+        end
+      end
+    end
+    puts "Reindexed works for #{count} objects"
+  end
+end