AO3-7283 Configurable number of shards for pseuds and tags index

app/models/search/pseud_indexer.rb

@@ -30,6 +30,15 @@ def self.mapping
     }
   end
 
+  def self.index_all(options = {})
+    unless options[:skip_delete]
+      delete_index
+      create_index(shards: ArchiveConfig.PSEUD_SHARDS)
+    end
+    options[:skip_delete] = true
+    super(options)
+  end
+
   def document(object)
     object.as_json(
       root: false,

app/models/search/tag_indexer.rb

@@ -29,6 +29,15 @@ def self.mapping
     }
   end
 
+  def self.index_all(options = {})
+    unless options[:skip_delete]
+      delete_index
+      create_index(shards: ArchiveConfig.TAG_SHARDS)
+    end
+    options[:skip_delete] = true
+    super(options)
+  end
+
   def self.settings
     {
       analysis: {

config/config.yml

@@ -787,5 +787,7 @@ PERCONA_ARGS: >
 # Production may have more than 5 for these
 BOOKMARKABLE_SHARDS: 5
 COLLECTION_SHARDS: 5
-WORKS_SHARDS: 5
+PSEUD_SHARDS: 5
+TAG_SHARDS: 5
 USER_SHARDS: 5
+WORKS_SHARDS: 5

spec/models/search/pseud_indexer_spec.rb

@@ -1,6 +1,15 @@
 require "spec_helper"
 
 describe PseudIndexer, pseud_search: true do
+  describe ".index_all" do
+    it "uses configured shard count when creating the index" do
+      allow(PseudIndexer).to receive(:index_from_db)
+      expect(PseudIndexer).to receive(:create_index).with(shards: ArchiveConfig.PSEUD_SHARDS)
+
+      PseudIndexer.index_all
+    end
+  end
+
   describe "#index_documents" do
     let(:pseud) { create(:pseud) }
     let(:indexer) { PseudIndexer.new([pseud.id]) }

spec/models/search/tag_indexer_spec.rb

@@ -0,0 +1,12 @@
+require "spec_helper"
+
+describe TagIndexer, tag_search: true do
+  describe ".index_all" do
+    it "uses configured shard count when creating the index" do
+      allow(TagIndexer).to receive(:index_from_db)
+      expect(TagIndexer).to receive(:create_index).with(shards: ArchiveConfig.TAG_SHARDS)
+
+      TagIndexer.index_all
+    end
+  end
+end