CXX-180 Generate _id automatically for InsertWriteOperations

Author: TylerBrock

src/mongo/client/insert_write_operation.cpp

@@ -27,7 +27,7 @@ namespace mongo {
     } // namespace
 
     InsertWriteOperation::InsertWriteOperation(const BSONObj& doc)
-        : _doc(doc)
+        : _doc(_ensureId(doc))
     {}
 
     WriteOpType InsertWriteOperation::operationType() const {
@@ -63,4 +63,14 @@ namespace mongo {
         obj->appendElements(_doc);
     }
 
+    BSONObj InsertWriteOperation::_ensureId(const BSONObj& doc) {
+        if (doc.hasField("_id"))
+            return doc;
+
+        BSONObjBuilder bob;
+        bob.append("_id", OID::gen());
+        bob.appendElements(doc);
+        return bob.obj();
+    }
+
 } // namespace mongo

src/mongo/client/insert_write_operation.h

@@ -36,6 +36,8 @@ namespace mongo {
         virtual void appendSelfToBSONObj(BSONObjBuilder* obj) const;
 
     private:
+        static BSONObj _ensureId(const BSONObj& doc);
+
         const BSONObj _doc;
     };