lib/rocket_job/batch/performance.rb in rocketjob-6.0.0.rc1 vs lib/rocket_job/batch/performance.rb in rocketjob-6.0.0.rc2
- old
+ new
@@ -20,11 +20,11 @@
raise "Please start workers before starting the performance test" if servers.zero?
count_running_workers
puts "Loading job with #{count} records/lines"
- job = RocketJob::Jobs::PerformanceJob.new(log_level: :warn)
+ job = RocketJob::Jobs::PerformanceJob.new(log_level: :warn)
job.input_category.slice_size = slice_size
if encrypt
job.input_category.serializer = :encrypt
job.output_category.serializer = :encrypt
elsif !compress
@@ -62,10 +62,10 @@
end
end
# Parse command line options
def parse(argv)
- parser = OptionParser.new do |o|
+ parser = OptionParser.new do |o|
o.on("-c", "--count COUNT", "Count of records to enqueue") do |arg|
self.count = arg.to_i
end
o.on("-m", "--mongo MONGO_CONFIG_FILE_NAME", "Location of mongoid.yml config file") do |arg|
self.mongo_config = arg