Unverified Commit 53aca8ae authored by OSAMU SATO's avatar OSAMU SATO Committed by GitHub
Browse files

Add batch_size option to bin/tootctl search deploy (#17049)

parent 0909874c
......@@ -17,6 +17,7 @@ module Mastodon
option :concurrency, type: :numeric, default: 2, aliases: [:c], desc: 'Workload will be split between this number of threads'
option :batch_size, type: :numeric, default: 1_000, aliases: [:b], desc: 'Number of records in each batch'
option :only, type: :array, enum: %w(accounts tags statuses), desc: 'Only process these indices'
desc 'deploy', 'Create or upgrade ElasticSearch indices and populate them'
long_desc <<~LONG_DESC
......@@ -35,6 +36,11 @@ module Mastodon
if options[:batch_size] < 1
say('Cannot run with this batch_size setting, must be at least 1', :red)
indices = begin
if options[:only]
options[:only].map { |str| "#{str.camelize}Index".constantize }
......@@ -73,7 +79,7 @@ module Mastodon
# is uneconomical. So we only ever add.
indices.each do |index|
progress.title = "Importing #{index} "
batch_size = 1_000
batch_size = options[:batch_size]
slice_size = (batch_size / options[:concurrency]).ceil
index.adapter.default_scope.reorder(nil).find_in_batches(batch_size: batch_size) do |batch|
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment