mongo/jstests/libs/trace_missing_docs.js

86 lines
2.5 KiB
JavaScript

//
// On error inserting documents, traces back and shows where the document was dropped
//
export function traceMissingDoc(coll, doc, mongos) {
if (mongos) coll = mongos.getCollection(coll + "");
else mongos = coll.getMongo();
let config = mongos.getDB("config");
let shards = config.shards.find().toArray();
for (var i = 0; i < shards.length; i++) {
shards[i].conn = new Mongo(shards[i].host);
}
let shardKeyPatt = config.collections.findOne({_id: coll + ""}).key;
// Project out the shard key
let shardKey = {};
for (let k in shardKeyPatt) {
if (doc[k] == undefined) {
jsTest.log(
"Shard key " +
tojson(shardKey) +
" not found in doc " +
tojson(doc) +
", falling back to _id search...",
);
shardKeyPatt = {_id: 1};
shardKey = {_id: doc["_id"]};
break;
}
shardKey[k] = doc[k];
}
if (doc["_id"] == undefined) {
jsTest.log("Id not found in doc " + tojson(doc) + " cannot trace oplog entries.");
return;
}
jsTest.log("Using shard key : " + tojson(shardKey));
let allOps = [];
for (var i = 0; i < shards.length; i++) {
let oplog = shards[i].conn.getCollection("local.oplog.rs");
if (!oplog.findOne()) {
jsTest.log("No oplog was found on shard " + shards[i]._id);
continue;
}
let addKeyQuery = function (query, prefix) {
for (let k in shardKey) {
query[prefix + "." + k] = shardKey[k];
}
return query;
};
let addToOps = function (cursor) {
cursor.forEach(function (doc) {
doc.shard = shards[i]._id;
doc.realTime = new Date(doc.ts.getTime() * 1000);
allOps.push(doc);
});
};
// Find ops
addToOps(oplog.find(addKeyQuery({op: "i"}, "o")));
let updateQuery = {$or: [addKeyQuery({op: "u"}, "o2"), {op: "u", "o2._id": doc["_id"]}]};
addToOps(oplog.find(updateQuery));
addToOps(oplog.find({op: "d", "o._id": doc["_id"]}));
}
let compareOps = function (opA, opB) {
return bsonWoCompare(opA.ts, opB.ts);
};
allOps.sort(compareOps);
jsTest.log.info("Ops found on each shard", {doc});
for (var i = 0; i < allOps.length; i++) {
jsTest.log.info({ops: allOps[i]});
}
return allOps;
}