mongodb/mongo-ruby-driver

# frozen_string_literal: true

require_relative '../base'

module Mongo
  module DriverBench
    module MultiDoc
      # Abstract base class for multi-doc benchmarks.
      #
      # @api private
      class Base < Mongo::DriverBench::Base
        private

        attr_reader :client, :collection

        def setup
          if file_name
            @dataset ||= load_file(file_name)
            @dataset_size ||= size_of_file(file_name) * scale
          end

          prepare_client
        end

        # The amount to scale the dataset size by (for scoring purposes).
        def scale
          10_000
        end

        def teardown
          cleanup_client
        end

        def prepare_client
          @client = new_client.use('perftest')
          @client.database.drop

          @collection = @client.database[:corpus].tap(&:create)
        end

        def cleanup_client
          @client.database.drop
        end
      end
    end
  end
end