SERVER-47733 SymmetricEncryptorWindows shouldn't pad when update is called

1 files changed, 14 insertions, 1 deletions

diff --git a/src/mongo/db/storage/remove_saver.h b/src/mongo/db/storage/remove_saver.h
index 0704ddb4d49..463fe7f8410 100644
--- a/src/mongo/db/storage/remove_saver.h
+++ b/src/mongo/db/storage/remove_saver.h
@@ -51,7 +51,11 @@ class RemoveSaver {
     RemoveSaver& operator=(const RemoveSaver&) = delete;
 
 public:
-    RemoveSaver(const std::string& type, const std::string& ns, const std::string& why);
+    class Storage;
+    RemoveSaver(const std::string& type,
+                const std::string& ns,
+                const std::string& why,
+                std::unique_ptr<Storage> storage = std::make_unique<Storage>());
     ~RemoveSaver();
 
     /**
@@ -77,11 +81,20 @@ public:
     }
     void file() && = delete;
 
+    class Storage {
+    public:
+        virtual ~Storage() = default;
+        virtual std::unique_ptr<std::ostream> makeOstream(const boost::filesystem::path& file,
+                                                          const boost::filesystem::path& root);
+        virtual void dumpBuffer() {}
+    };
+
 private:
     boost::filesystem::path _root;
     boost::filesystem::path _file;
     std::unique_ptr<DataProtector> _protector;
     std::unique_ptr<std::ostream> _out;
+    std::unique_ptr<Storage> _storage;
 };
 
 }  // namespace mongo