mongodb/mongo-ruby-driver

# frozen_string_literal: true

require_relative '../base'

module Mongo
  module DriverBench
    module SingleDoc
      module InsertOne
        # Abstract base class for "insert one" benchmarks.
        #
        # @api private
        class Base < Mongo::DriverBench::SingleDoc::Base
          attr_reader :repetitions
          alias scale repetitions

          def before_task
            collection.drop
            collection.create
          end

          def do_task
            repetitions.times do
              collection.insert_one(dataset)
            end
          end
        end
      end
    end
  end
end