SERVER-26065 Write more generic optime comparator function

3 files changed, 192 insertions, 144 deletions

diff --git a/jstests/core/optime_cmp.js b/jstests/core/optime_cmp.js
new file mode 100644
index 00000000000..7ddd83bfc8e
--- /dev/null
+++ b/jstests/core/optime_cmp.js
@@ -0,0 +1,34 @@
+(function() {
+    'use strict';
+
+    // PV0
+    assert.eq(-1, rs.compareOpTimes(Timestamp(1, 0), Timestamp(2, 0)));
+    assert.eq(-1, rs.compareOpTimes(Timestamp(1, 0), Timestamp(1, 1)));
+
+    assert.eq(0, rs.compareOpTimes(Timestamp(2, 0), Timestamp(2, 0)));
+
+    assert.eq(1, rs.compareOpTimes(Timestamp(2, 0), Timestamp(1, 1)));
+    assert.eq(1, rs.compareOpTimes(Timestamp(1, 2), Timestamp(1, 1)));
+
+    // lhs PV0 rhs PV1
+    assert.throws(function() {
+        rs.compareOpTimes(Timestamp(1, 0), {ts: Timestamp(2, 0), t: 2});
+    });
+
+    // lhs PV1 rhs PV0
+    assert.throws(function() {
+        rs.compareOpTimes({ts: Timestamp(1, 0), t: 2}, Timestamp(2, 0));
+    });
+
+    // PV1
+    assert.eq(-1, rs.compareOpTimes({ts: Timestamp(2, 2), t: 2}, {ts: Timestamp(3, 1), t: 2}));
+    assert.eq(-1, rs.compareOpTimes({ts: Timestamp(2, 2), t: 2}, {ts: Timestamp(2, 4), t: 2}));
+    assert.eq(-1, rs.compareOpTimes({ts: Timestamp(3, 0), t: 2}, {ts: Timestamp(2, 0), t: 3}));
+
+    assert.eq(0, rs.compareOpTimes({ts: Timestamp(3, 0), t: 2}, {ts: Timestamp(3, 0), t: 2}));
+
+    assert.eq(1, rs.compareOpTimes({ts: Timestamp(3, 1), t: 2}, {ts: Timestamp(2, 2), t: 2}));
+    assert.eq(1, rs.compareOpTimes({ts: Timestamp(2, 4), t: 2}, {ts: Timestamp(2, 2), t: 2}));
+    assert.eq(1, rs.compareOpTimes({ts: Timestamp(2, 0), t: 3}, {ts: Timestamp(3, 0), t: 2}));
+
+})();
diff --git a/jstests/replsets/rslib.js b/jstests/replsets/rslib.js
index 6206a4a8168..d60f86462bf 100644
--- a/jstests/replsets/rslib.js
+++ b/jstests/replsets/rslib.js
@@ -10,6 +10,7 @@ var waitUntilAllNodesCaughtUp;
 var updateConfigIfNotDurable;
 var reInitiateWithoutThrowingOnAbortedMember;
 var awaitRSClientHosts;
+var getLastOpTime;
 
 (function() {
     "use strict";
@@ -356,4 +357,15 @@ var awaitRSClientHosts;
             return false;
         }, 'timed out waiting for replica set client to recognize hosts', timeout);
     };
+
+    /**
+     * Returns the last opTime of the connection based from replSetGetStatus. Can only
+     * be used on replica set nodes.
+     */
+    getLastOpTime = function(conn) {
+        var replSetStatus =
+            assert.commandWorked(conn.getDB("admin").runCommand({replSetGetStatus: 1}));
+        var connStatus = replSetStatus.members.filter(m => m.self)[0];
+        return connStatus.optime;
+    };
 }());
diff --git a/jstests/slow1/replsets_priority1.js b/jstests/slow1/replsets_priority1.js
index 4bbb856a4b6..dfd31173b6b 100644
--- a/jstests/slow1/replsets_priority1.js
+++ b/jstests/slow1/replsets_priority1.js
@@ -1,203 +1,205 @@
 // come up with random priorities and make sure that the right member gets
 // elected. then kill that member and make sure the next one gets elected.
 
-print("\n\n\nreplsets_priority1.js BEGIN\n");
+(function() {
 
-load("jstests/replsets/rslib.js");
+    "use strict";
 
-var rs = new ReplSetTest({name: 'testSet', nodes: 3, nodeOptions: {verbose: 2}});
-var nodes = rs.startSet();
-rs.initiate();
+    load("jstests/replsets/rslib.js");
 
-var master = rs.getPrimary();
+    var rs = new ReplSetTest({name: 'testSet', nodes: 3, nodeOptions: {verbose: 2}});
+    var nodes = rs.startSet();
+    rs.initiate();
 
-var everyoneOkSoon = function() {
-    var status;
-    assert.soon(function() {
-        var ok = true;
-        status = master.adminCommand({replSetGetStatus: 1});
+    var master = rs.getPrimary();
 
-        if (!status.members) {
-            return false;
-        }
+    var everyoneOkSoon = function() {
+        var status;
+        assert.soon(function() {
+            var ok = true;
+            status = master.adminCommand({replSetGetStatus: 1});
 
-        for (var i in status.members) {
-            if (status.members[i].health == 0) {
-                continue;
+            if (!status.members) {
+                return false;
             }
-            ok &= status.members[i].state == 1 || status.members[i].state == 2;
-        }
-        return ok;
-    }, tojson(status));
-};
 
-var checkPrimaryIs = function(node) {
+            for (var i in status.members) {
+                if (status.members[i].health == 0) {
+                    continue;
+                }
+                ok &= status.members[i].state == 1 || status.members[i].state == 2;
+            }
+            return ok;
+        }, tojson(status));
+    };
 
-    print("nreplsets_priority1.js checkPrimaryIs(" + node.host + ")");
+    var checkPrimaryIs = function(node) {
 
-    var status;
+        print("nreplsets_priority1.js checkPrimaryIs(" + node.host + ")");
 
-    assert.soon(function() {
-        var ok = true;
+        var status;
 
-        try {
-            status = master.adminCommand({replSetGetStatus: 1});
-        } catch (e) {
-            print(e);
-            print("nreplsets_priority1.js checkPrimaryIs reconnecting");
-            reconnect(master);
-            status = master.adminCommand({replSetGetStatus: 1});
-        }
+        assert.soon(function() {
+            var ok = true;
 
-        var str = "goal: " + node.host + "==1 states: ";
-        if (!status || !status.members) {
-            return false;
-        }
-        status.members.forEach(function(m) {
-            str += m.name + ": " + m.state + " ";
-
-            if (m.name == node.host) {
-                ok &= m.state == 1;
-            } else {
-                ok &= m.state != 1 || (m.state == 1 && m.health == 0);
+            try {
+                status = master.adminCommand({replSetGetStatus: 1});
+            } catch (e) {
+                print(e);
+                print("nreplsets_priority1.js checkPrimaryIs reconnecting");
+                reconnect(master);
+                status = master.adminCommand({replSetGetStatus: 1});
             }
-        });
-        print();
-        print(str);
-        print();
 
-        occasionally(function() {
-            print("\nstatus:");
-            printjson(status);
+            var str = "goal: " + node.host + "==1 states: ";
+            if (!status || !status.members) {
+                return false;
+            }
+            status.members.forEach(function(m) {
+                str += m.name + ": " + m.state + " ";
+
+                if (m.name == node.host) {
+                    ok &= m.state == 1;
+                } else {
+                    ok &= m.state != 1 || (m.state == 1 && m.health == 0);
+                }
+            });
+            print();
+            print(str);
             print();
-        }, 15);
 
-        return ok;
-    }, node.host + '==1', 240000, 1000);
+            occasionally(function() {
+                print("\nstatus:");
+                printjson(status);
+                print();
+            }, 15);
 
-    everyoneOkSoon();
-};
+            return ok;
+        }, node.host + '==1', 240000, 1000);
 
-everyoneOkSoon();
+        everyoneOkSoon();
+    };
 
-print("\n\nreplsets_priority1.js initial sync");
+    everyoneOkSoon();
 
-// intial sync
-master.getDB("foo").bar.insert({x: 1});
-rs.awaitReplication();
+    print("\n\nreplsets_priority1.js initial sync");
 
-print("\n\nreplsets_priority1.js starting loop");
+    // intial sync
+    master.getDB("foo").bar.insert({x: 1});
+    rs.awaitReplication();
 
-var n = 5;
-for (i = 0; i < n; i++) {
-    print("Round " + i + ": FIGHT!");
+    print("\n\nreplsets_priority1.js starting loop");
 
-    var max = null;
-    var second = null;
-    reconnect(master);
-    var config = master.getDB("local").system.replset.findOne();
+    var n = 5;
+    for (var i = 0; i < n; i++) {
+        print("Round " + i + ": FIGHT!");
 
-    var version = config.version;
-    config.version++;
+        var max = null;
+        var second = null;
+        reconnect(master);
+        var config = master.getDB("local").system.replset.findOne();
 
-    for (var j = 0; j < config.members.length; j++) {
-        var priority = Math.random() * 100;
-        print("random priority : " + priority);
-        config.members[j].priority = priority;
+        var version = config.version;
+        config.version++;
 
-        if (!max || priority > max.priority) {
-            max = config.members[j];
-        }
-    }
+        for (var j = 0; j < config.members.length; j++) {
+            var priority = Math.random() * 100;
+            print("random priority : " + priority);
+            config.members[j].priority = priority;
 
-    for (var j = 0; j < config.members.length; j++) {
-        if (config.members[j] == max) {
-            continue;
+            if (!max || priority > max.priority) {
+                max = config.members[j];
+            }
         }
-        if (!second || config.members[j].priority > second.priority) {
-            second = config.members[j];
+
+        for (var j = 0; j < config.members.length; j++) {
+            if (config.members[j] == max) {
+                continue;
+            }
+            if (!second || config.members[j].priority > second.priority) {
+                second = config.members[j];
+            }
         }
-    }
 
-    print("\n\nreplsets_priority1.js max is " + max.host + " with priority " + max.priority +
-          ", reconfiguring...");
+        print("\n\nreplsets_priority1.js max is " + max.host + " with priority " + max.priority +
+              ", reconfiguring...");
 
-    var count = 0;
-    while (config.version != version && count < 100) {
-        reconnect(master);
+        var count = 0;
+        while (config.version != version && count < 100) {
+            reconnect(master);
 
-        occasionally(function() {
-            print("version is " + version + ", trying to update to " + config.version);
-        });
+            occasionally(function() {
+                print("version is " + version + ", trying to update to " + config.version);
+            });
 
-        try {
-            master.adminCommand({replSetReconfig: config});
-            master = rs.getPrimary();
-            reconnect(master);
+            try {
+                master.adminCommand({replSetReconfig: config});
+                master = rs.getPrimary();
+                reconnect(master);
 
-            version = master.getDB("local").system.replset.findOne().version;
-        } catch (e) {
-            print("nreplsets_priority1.js Caught exception: " + e);
-        }
+                version = master.getDB("local").system.replset.findOne().version;
+            } catch (e) {
+                print("nreplsets_priority1.js Caught exception: " + e);
+            }
 
-        count++;
-    }
+            count++;
+        }
 
-    print("\nreplsets_priority1.js wait for 2 slaves");
+        print("\nreplsets_priority1.js wait for 2 slaves");
 
-    assert.soon(function() {
-        rs.getPrimary();
-        return rs.liveNodes.slaves.length == 2;
-    }, "2 slaves");
+        assert.soon(function() {
+            rs.getPrimary();
+            return rs.liveNodes.slaves.length == 2;
+        }, "2 slaves");
 
-    print("\nreplsets_priority1.js wait for new config version " + config.version);
+        print("\nreplsets_priority1.js wait for new config version " + config.version);
 
-    assert.soon(function() {
-        versions = [0, 0];
-        rs.liveNodes.slaves[0].setSlaveOk();
-        versions[0] = rs.liveNodes.slaves[0].getDB("local").system.replset.findOne().version;
-        rs.liveNodes.slaves[1].setSlaveOk();
-        versions[1] = rs.liveNodes.slaves[1].getDB("local").system.replset.findOne().version;
-        return versions[0] == config.version && versions[1] == config.version;
-    });
+        assert.soon(function() {
+            var versions = [0, 0];
+            rs.liveNodes.slaves[0].setSlaveOk();
+            versions[0] = rs.liveNodes.slaves[0].getDB("local").system.replset.findOne().version;
+            rs.liveNodes.slaves[1].setSlaveOk();
+            versions[1] = rs.liveNodes.slaves[1].getDB("local").system.replset.findOne().version;
+            return versions[0] == config.version && versions[1] == config.version;
+        });
 
-    print("replsets_priority1.js awaitReplication");
+        print("replsets_priority1.js awaitReplication");
 
-    // the reconfiguration needs to be replicated! the hb sends it out
-    // separately from the repl
-    rs.awaitReplication();
+        // the reconfiguration needs to be replicated! the hb sends it out
+        // separately from the repl
+        rs.awaitReplication();
 
-    print("reconfigured.  Checking statuses.");
+        print("reconfigured.  Checking statuses.");
 
-    checkPrimaryIs(max);
+        checkPrimaryIs(max);
 
-    // Wait for election oplog entry to be replicated, to avoid rollbacks later on.
-    rs.awaitReplication();
+        // Wait for election oplog entry to be replicated, to avoid rollbacks later on.
+        rs.awaitReplication();
 
-    print("rs.stop");
+        print("rs.stop");
 
-    rs.stop(max._id);
+        rs.stop(max._id);
 
-    var master = rs.getPrimary();
+        master = rs.getPrimary();
 
-    print("\nkilled max primary.  Checking statuses.");
+        print("\nkilled max primary.  Checking statuses.");
 
-    print("second is " + second.host + " with priority " + second.priority);
-    checkPrimaryIs(second);
+        print("second is " + second.host + " with priority " + second.priority);
+        checkPrimaryIs(second);
 
-    // Wait for election oplog entry to be replicated, to avoid rollbacks later on.
-    rs.awaitReplication();
+        // Wait for election oplog entry to be replicated, to avoid rollbacks later on.
+        rs.awaitReplication();
 
-    print("restart max " + max._id);
+        print("restart max " + max._id);
 
-    rs.restart(max._id);
-    master = rs.getPrimary();
+        rs.restart(max._id);
+        master = rs.getPrimary();
 
-    print("max restarted.  Checking statuses.");
-    checkPrimaryIs(max);
+        print("max restarted.  Checking statuses.");
+        checkPrimaryIs(max);
 
-    // Wait for election oplog entry to be replicated, to avoid rollbacks later on.
-    rs.awaitReplication();
-}
+        // Wait for election oplog entry to be replicated, to avoid rollbacks later on.
+        rs.awaitReplication();
+    }
 
-print("\n\n\n\n\nreplsets_priority1.js SUCCESS!\n\n");
+})();