SERVER-33366 OperationContext setRecoveryUnit() & releaseRecoveryUnit() should transfer using unique_ptr.

1 files changed, 4 insertions, 4 deletions

diff --git a/src/mongo/db/operation_context.cpp b/src/mongo/db/operation_context.cpp
index 091ffdd1ee4..94a5ea312d2 100644
--- a/src/mongo/db/operation_context.cpp
+++ b/src/mongo/db/operation_context.cpp
@@ -379,13 +379,13 @@ void OperationContext::setTxnNumber(TxnNumber txnNumber) {
     _txnNumber = txnNumber;
 }
 
-RecoveryUnit* OperationContext::releaseRecoveryUnit() {
-    return _recoveryUnit.release();
+std::unique_ptr<RecoveryUnit> OperationContext::releaseRecoveryUnit() {
+    return std::move(_recoveryUnit);
 }
 
 WriteUnitOfWork::RecoveryUnitState OperationContext::setRecoveryUnit(
-    RecoveryUnit* unit, WriteUnitOfWork::RecoveryUnitState state) {
-    _recoveryUnit.reset(unit);
+    std::unique_ptr<RecoveryUnit> unit, WriteUnitOfWork::RecoveryUnitState state) {
+    _recoveryUnit = std::move(unit);
     WriteUnitOfWork::RecoveryUnitState oldState = _ruState;
     _ruState = state;
     return oldState;