SERVER-10159 Re-use Collection and index data structures rather than re-acquiring or copying

1 files changed, 10 insertions, 7 deletions

diff --git a/src/mongo/db/ops/update_lifecycle_impl.cpp b/src/mongo/db/ops/update_lifecycle_impl.cpp
index 9564d66b2e8..41ec120a31d 100644
--- a/src/mongo/db/ops/update_lifecycle_impl.cpp
+++ b/src/mongo/db/ops/update_lifecycle_impl.cpp
@@ -52,16 +52,19 @@ namespace mongo {
             ChunkVersion::IGNORED()) {
     }
 
-    const bool UpdateLifecycleImpl::canContinue() const {
+    void UpdateLifecycleImpl::setCollection(Collection* collection) {
+        _collection = collection;
+    }
+
+    bool UpdateLifecycleImpl::canContinue() const {
         // Collection needs to exist to continue
-        Collection* coll = cc().database()->getCollection(_nsString.ns());
-        return coll;
+        return _collection;
     }
 
-    const void UpdateLifecycleImpl::getIndexKeys(IndexPathSet* returnedIndexPathSet) const {
-        Collection* coll = cc().database()->getCollection(_nsString.ns());
-        if (coll)
-            *returnedIndexPathSet = coll->infoCache()->indexKeys();
+    const IndexPathSet* UpdateLifecycleImpl::getIndexKeys() const {
+        if (_collection)
+            return &_collection->infoCache()->indexKeys();
+        return NULL;
     }
 
     const std::vector<FieldRef*>* UpdateLifecycleImpl::getImmutableFields() const {