CS-6 fix for 'only' mode

Author: astaple

db/matcher.cpp

@@ -103,7 +103,6 @@ namespace mongo {
         if ( ! _needRecord ){
             return true;
         }
-
         return _recordMatcher.matches(recLoc.rec());
     }
 

db/query.cpp

@@ -481,12 +481,17 @@ namespace mongo {
                 findingStartCursor_ = new ClientCursor();
                 findingStartCursor_->c = qp().newReverseCursor();
                 findingStartCursor_->ns = qp().ns();
+                BSONElement tsElt = qp().query()[ "ts" ];
+                massert( "no ts field in query", !tsElt.eoo() );
+                BSONObjBuilder b;
+                b.append( tsElt );
+                BSONObj tsQuery = b.obj();
+                matcher_.reset(new KeyValJSMatcher(tsQuery, qp().indexKey()));
             } else {
                 c_ = qp().newCursor();
+                matcher_.reset(new KeyValJSMatcher(qp().query(), qp().indexKey()));
             }
 
-            matcher_.reset(new KeyValJSMatcher(qp().query(), qp().indexKey()));
-            
             if ( qp().scanAndOrderRequired() ) {
                 ordering_ = true;
                 so_.reset( new ScanAndOrder( ntoskip_, ntoreturn_, order_ ) );
@@ -498,9 +503,11 @@ namespace mongo {
                 if ( !findingStartCursor_ || !findingStartCursor_->c->ok() ) {
                     findingStart_ = false;
                     c_ = qp().newCursor();
+                    matcher_.reset(new KeyValJSMatcher(qp().query(), qp().indexKey()));
                 } else if ( !matcher_->matches( findingStartCursor_->c->currKey(), findingStartCursor_->c->currLoc() ) ) {
                     findingStart_ = false;
                     c_ = qp().newCursor( findingStartCursor_->c->currLoc() );
+                    matcher_.reset(new KeyValJSMatcher(qp().query(), qp().indexKey()));
                 } else {
                     findingStartCursor_->c->advance();
                     RARELY {

db/repl.cpp

@@ -981,7 +981,11 @@ namespace mongo {
 
     void ReplSource::syncToTailOfRemoteLog() {
         string _ns = ns();
-        BSONObj last = conn->findOne( _ns.c_str(), Query().sort( BSON( "$natural" << -1 ) ) );
+        BSONObjBuilder b;
+        if ( !only.empty() ) {
+            b.appendRegex("ns", string("^") + only);
+        }        
+        BSONObj last = conn->findOne( _ns.c_str(), Query( b.done() ).sort( BSON( "$natural" << -1 ) ) );
         if ( !last.isEmpty() ) {
             BSONElement ts = last.findElement( "ts" );
             massert( "non Date ts found", ts.type() == Date || ts.type() == Timestamp );

jstests/repl/repl4.js

@@ -28,6 +28,14 @@ doTest = function() {
     printjson( s.getDBNames() );
     assert.eq( -1, s.getDBNames().indexOf( "b" ) );
     assert.eq( 0, s.getDB( "b" ).b.find().count() );
+    
+    stopMongod( ports[ 1 ] );
+    
+    cm.save( { x:3 } );
+    bm.save( { x:4 } );
+    
+    s = startMongoProgram( "mongod", "--port", ports[ 1 ], "--dbpath", "/data/db/" + baseName + "-slave", "--slave", "--source", "127.0.0.1:" + ports[ 0 ], "--only", "c", "--nohttpinterface", "--noprealloc", "--bind_ip", "127.0.0.1" );
+    soonCount( "c", "c", 2 );
 }
 
 doTest();