class Mongo::BulkWrite
Constants
- SINGLE_STATEMENT_OPS
Attributes
@return [ Mongo::Collection
] collection The collection.
@return [ Hash, BSON::Document ] options The options.
@return [ Array<Hash, BSON::Document> ] requests The requests.
Public Class Methods
Create the new bulk write operation.
@api private
@example Create an ordered bulk write.
Mongo::BulkWrite.new(collection, [{ insert_one: { _id: 1 }}])
@example Create an unordered bulk write.
Mongo::BulkWrite.new(collection, [{ insert_one: { _id: 1 }}], ordered: false)
@example Create an ordered mixed bulk write.
Mongo::BulkWrite.new( collection, [ { insert_one: { _id: 1 }}, { update_one: { filter: { _id: 0 }, update: { '$set' => { name: 'test' }}}}, { delete_one: { filter: { _id: 2 }}} ] )
@param [ Mongo::Collection
] collection The collection. @param [ Array<Hash, BSON::Document> ] requests The requests. @param [ Hash, BSON::Document ] options The options.
@since 2.1.0
# File lib/mongo/bulk_write.rb, line 112 def initialize(collection, requests, options = {}) @collection = collection @requests = requests @options = options || {} end
Public Instance Methods
Execute the bulk write operation.
@example Execute the bulk write.
bulk_write.execute
@return [ Mongo::BulkWrite::Result
] The result.
@since 2.1.0
# File lib/mongo/bulk_write.rb, line 53 def execute operation_id = Monitoring.next_operation_id result_combiner = ResultCombiner.new operations = op_combiner.combine client.send(:with_session, @options) do |session| operations.each do |operation| if single_statement?(operation) write_with_retry(session, write_concern) do |server, txn_num| execute_operation( operation.keys.first, operation.values.first, server, operation_id, result_combiner, session, txn_num) end else legacy_write_with_retry do |server| execute_operation( operation.keys.first, operation.values.first, server, operation_id, result_combiner, session) end end end end result_combiner.result end
Is the bulk write ordered?
@api private
@example Is the bulk write ordered?
bulk_write.ordered?
@return [ true, false ] If the bulk write is ordered.
@since 2.1.0
# File lib/mongo/bulk_write.rb, line 128 def ordered? @ordered ||= options.fetch(:ordered, true) end
Get the write concern for the bulk write.
@api private
@example Get the write concern.
bulk_write.write_concern
@return [ WriteConcern
] The write concern.
@since 2.1.0
# File lib/mongo/bulk_write.rb, line 142 def write_concern @write_concern ||= options[:write_concern] ? WriteConcern.get(options[:write_concern]) : collection.write_concern end
Private Instance Methods
# File lib/mongo/bulk_write.rb, line 157 def base_spec(operation_id, session) { :db_name => database.name, :coll_name => collection.name, :write_concern => write_concern, :ordered => ordered?, :operation_id => operation_id, :bypass_document_validation => !!options[:bypass_document_validation], :options => options, :id_generator => client.options[:id_generator], :session => session } end
# File lib/mongo/bulk_write.rb, line 203 def delete_many(documents, server, operation_id, session, txn_num) Operation::Write::Bulk::Delete.new( base_spec(operation_id, session).merge(:deletes => documents) ).execute(server) end
# File lib/mongo/bulk_write.rb, line 197 def delete_one(documents, server, operation_id, session, txn_num) Operation::Write::Bulk::Delete.new( base_spec(operation_id, session).merge(:deletes => documents, :txn_num => txn_num) ).execute(server) end
# File lib/mongo/bulk_write.rb, line 171 def execute_operation(name, values, server, operation_id, combiner, session, txn_num = nil) raise Error::UnsupportedCollation.new if op_combiner.has_collation && !server.features.collation_enabled? raise Error::UnsupportedArrayFilters.new if op_combiner.has_array_filters && !server.features.array_filters_enabled? begin if values.size > server.max_write_batch_size split_execute(name, values, server, operation_id, combiner, session, txn_num) else combiner.combine!(send(name, values, server, operation_id, session, txn_num), values.size) end rescue Error::MaxBSONSize, Error::MaxMessageSize => e raise e if values.size <= 1 split_execute(name, values, server, operation_id, combiner, session, txn_num) end end
# File lib/mongo/bulk_write.rb, line 210 def insert_one(documents, server, operation_id, session, txn_num) Operation::Write::Bulk::Insert.new( base_spec(operation_id, session).merge(:documents => documents, :txn_num => txn_num) ).execute(server) end
# File lib/mongo/bulk_write.rb, line 186 def op_combiner @op_combiner ||= ordered? ? OrderedCombiner.new(requests) : UnorderedCombiner.new(requests) end
# File lib/mongo/bulk_write.rb, line 153 def single_statement?(operation) SINGLE_STATEMENT_OPS.include?(operation.keys.first) end
# File lib/mongo/bulk_write.rb, line 190 def split_execute(name, values, server, operation_id, combiner, session, txn_num) execute_operation(name, values.shift(values.size / 2), server, operation_id, combiner, session, txn_num) txn_num = session.next_txn_num if txn_num execute_operation(name, values, server, operation_id, combiner, session, txn_num) end
# File lib/mongo/bulk_write.rb, line 223 def update_many(documents, server, operation_id, session, txn_num) Operation::Write::Bulk::Update.new( base_spec(operation_id, session).merge(:updates => documents) ).execute(server) end
# File lib/mongo/bulk_write.rb, line 216 def update_one(documents, server, operation_id, session, txn_num) Operation::Write::Bulk::Update.new( base_spec(operation_id, session).merge(:updates => documents, :txn_num => txn_num) ).execute(server) end