SERVER-21404: Add rollback file coverage to ESE

1 files changed, 4 insertions, 0 deletions

diff --git a/src/mongo/db/dbhelpers.h b/src/mongo/db/dbhelpers.h
index 2c77bf8ca2d..5be0ba37185 100644
--- a/src/mongo/db/dbhelpers.h
+++ b/src/mongo/db/dbhelpers.h
@@ -33,11 +33,13 @@
 
 #include "mongo/db/db.h"
 #include "mongo/db/record_id.h"
+#include "mongo/db/storage/data_protector.h"
 
 namespace mongo {
 
 class Collection;
 class Cursor;
+class DataProtector;
 class OperationContext;
 struct KeyRange;
 struct WriteConcernOptions;
@@ -217,6 +219,7 @@ struct Helpers {
 
     public:
         RemoveSaver(const std::string& type, const std::string& ns, const std::string& why);
+        ~RemoveSaver();
 
         /**
          * Writes document to file. File is created lazily before writing the first document.
@@ -228,6 +231,7 @@ struct Helpers {
     private:
         boost::filesystem::path _root;
         boost::filesystem::path _file;
+        std::unique_ptr<DataProtector> _protector;
         std::unique_ptr<std::ostream> _out;
     };
 };