162 lines
6.0 KiB
JavaScript
162 lines
6.0 KiB
JavaScript
/**
|
|
* Verify that shard key appears in oplog records
|
|
*
|
|
* The only records that need to be checked are delete and update records, but updates happen on
|
|
* various paths that must all be checked.
|
|
*/
|
|
|
|
(function() {
|
|
"use strict";
|
|
|
|
function validateDocumentKeyInOplogForRemove(ns, _id, docKey) {
|
|
const deleteEntry = oplog.findOne({ns: ns, op: 'd', 'o._id': _id});
|
|
const o = docKey ? {_id: _id, x: docKey} : {_id: _id};
|
|
if (deleteEntry) {
|
|
assert.docEq(o, deleteEntry.o);
|
|
} else {
|
|
// Validate this is a batched delete, which includes the document key.
|
|
const elemMatch = docKey ? {'ns': ns, 'op': 'd', 'o._id': _id, 'o.x': docKey}
|
|
: {'ns': ns, 'op': 'd', 'o._id': _id};
|
|
const applyOpsEntry =
|
|
oplog.findOne({ns: 'admin.$cmd', op: 'c', 'o.applyOps': {$elemMatch: elemMatch}});
|
|
assert.docEq(o, applyOpsEntry.o.applyOps[0].o);
|
|
}
|
|
}
|
|
|
|
var st = new ShardingTest({name: 'test', shards: {rs0: {nodes: 1}}});
|
|
var db = st.s.getDB('test');
|
|
|
|
assert.commandWorked(st.s.adminCommand({enableSharding: 'test'}));
|
|
|
|
// 'test.un' is left unsharded.
|
|
assert.commandWorked(db.adminCommand({shardcollection: 'test.byId', key: {_id: 1}}));
|
|
assert.commandWorked(db.adminCommand({shardcollection: 'test.byX', key: {x: 1}}));
|
|
assert.commandWorked(db.adminCommand({shardcollection: 'test.byXId', key: {x: 1, _id: 1}}));
|
|
assert.commandWorked(db.adminCommand({shardcollection: 'test.byIdX', key: {_id: 1, x: 1}}));
|
|
|
|
assert.commandWorked(db.un.insert({_id: 10, x: 50, y: 60}));
|
|
assert.commandWorked(db.un.insert({_id: 30, x: 70, y: 80}));
|
|
|
|
assert.commandWorked(db.byId.insert({_id: 11, x: 51, y: 61}));
|
|
assert.commandWorked(db.byId.insert({_id: 31, x: 71, y: 81}));
|
|
|
|
assert.commandWorked(db.byX.insert({_id: 12, x: 52, y: 62}));
|
|
assert.commandWorked(db.byX.insert({_id: 32, x: 72, y: 82}));
|
|
|
|
assert.commandWorked(db.byXId.insert({_id: 13, x: 53, y: 63}));
|
|
assert.commandWorked(db.byXId.insert({_id: 33, x: 73, y: 83}));
|
|
|
|
assert.commandWorked(db.byIdX.insert({_id: 14, x: 54, y: 64}));
|
|
assert.commandWorked(db.byIdX.insert({_id: 34, x: 74, y: 84}));
|
|
|
|
var oplog = st.rs0.getPrimary().getDB('local').oplog.rs;
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test update command on 'un'");
|
|
|
|
assert.commandWorked(db.un.update({_id: 10, x: 50}, {$set: {y: 70}})); // in place
|
|
assert.commandWorked(db.un.update({_id: 30, x: 70}, {y: 75})); // replacement
|
|
|
|
// unsharded, only _id appears in o2:
|
|
|
|
var a = oplog.findOne({ns: 'test.un', op: 'u', 'o2._id': 10});
|
|
assert.eq(a.o2, {_id: 10});
|
|
|
|
var b = oplog.findOne({ns: 'test.un', op: 'u', 'o2._id': 30});
|
|
assert.eq(b.o2, {_id: 30});
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test update command on 'byId'");
|
|
|
|
assert.commandWorked(db.byId.update({_id: 11}, {$set: {y: 71}})); // in place
|
|
assert.commandWorked(db.byId.update({_id: 31}, {x: 71, y: 76})); // replacement
|
|
|
|
// sharded by {_id: 1}: only _id appears in o2:
|
|
|
|
a = oplog.findOne({ns: 'test.byId', op: 'u', 'o2._id': 11});
|
|
assert.eq(a.o2, {_id: 11});
|
|
|
|
b = oplog.findOne({ns: 'test.byId', op: 'u', 'o2._id': 31});
|
|
assert.eq(b.o2, {_id: 31});
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test update command on 'byX'");
|
|
|
|
assert.commandWorked(db.byX.update({x: 52}, {$set: {y: 72}})); // in place
|
|
assert.commandWorked(db.byX.update({x: 72}, {x: 72, y: 77})); // replacement
|
|
|
|
// sharded by {x: 1}: x appears in o2, followed by _id:
|
|
|
|
a = oplog.findOne({ns: 'test.byX', op: 'u', 'o2._id': 12});
|
|
assert.eq(a.o2, {x: 52, _id: 12});
|
|
|
|
b = oplog.findOne({ns: 'test.byX', op: 'u', 'o2._id': 32});
|
|
assert.eq(b.o2, {x: 72, _id: 32});
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test update command on 'byXId'");
|
|
|
|
assert.commandWorked(db.byXId.update({_id: 13, x: 53}, {$set: {y: 73}})); // in place
|
|
assert.commandWorked(db.byXId.update({_id: 33, x: 73}, {x: 73, y: 78})); // replacement
|
|
|
|
// sharded by {x: 1, _id: 1}: x appears in o2, followed by _id:
|
|
|
|
a = oplog.findOne({ns: 'test.byXId', op: 'u', 'o2._id': 13});
|
|
assert.eq(a.o2, {x: 53, _id: 13});
|
|
|
|
b = oplog.findOne({ns: 'test.byXId', op: 'u', 'o2._id': 33});
|
|
assert.eq(b.o2, {x: 73, _id: 33});
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test update command on 'byIdX'");
|
|
|
|
assert.commandWorked(db.byIdX.update({_id: 14, x: 54}, {$set: {y: 74}})); // in place
|
|
assert.commandWorked(db.byIdX.update({_id: 34, x: 74}, {x: 74, y: 79})); // replacement
|
|
|
|
// sharded by {_id: 1, x: 1}: _id appears in o2, followed by x:
|
|
|
|
a = oplog.findOne({ns: 'test.byIdX', op: 'u', 'o2._id': 14});
|
|
assert.eq(a.o2, {_id: 14, x: 54});
|
|
|
|
b = oplog.findOne({ns: 'test.byIdX', op: 'u', 'o2._id': 34});
|
|
assert.eq(b.o2, {_id: 34, x: 74});
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test remove command: 'un'");
|
|
|
|
assert.commandWorked(db.un.remove({_id: 10}));
|
|
assert.commandWorked(db.un.remove({_id: 30}));
|
|
|
|
validateDocumentKeyInOplogForRemove('test.un', 10);
|
|
validateDocumentKeyInOplogForRemove('test.un', 30);
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test remove command: 'byX'");
|
|
|
|
assert.commandWorked(db.byX.remove({_id: 12}));
|
|
assert.commandWorked(db.byX.remove({_id: 32}));
|
|
|
|
validateDocumentKeyInOplogForRemove('test.byX', 12, 52);
|
|
validateDocumentKeyInOplogForRemove('test.byX', 32, 72);
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test remove command: 'byXId'");
|
|
|
|
assert.commandWorked(db.byXId.remove({_id: 13}));
|
|
assert.commandWorked(db.byXId.remove({_id: 33}));
|
|
|
|
validateDocumentKeyInOplogForRemove('test.byXId', 13, 53);
|
|
validateDocumentKeyInOplogForRemove('test.byXId', 33, 73);
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
jsTest.log("Test remove command: 'byIdX'");
|
|
|
|
assert.commandWorked(db.byIdX.remove({_id: 14}));
|
|
assert.commandWorked(db.byIdX.remove({_id: 34}));
|
|
|
|
validateDocumentKeyInOplogForRemove('test.byIdX', 14, 54);
|
|
validateDocumentKeyInOplogForRemove('test.byIdX', 34, 74);
|
|
|
|
st.stop();
|
|
})();
|