Fix replay bug

src/storage/buffer/buffer_manager.cpp

@@ -126,7 +126,7 @@ BufferObj *BufferManager::GetBufferObject(UniquePtr<FileWorker> file_worker, boo
     if (auto iter1 = buffer_map_.find(file_path); iter1 != buffer_map_.end()) {
         BufferObj *buffer_obj = iter1->second.get();
         if (restart) {
-            buffer_obj->SetFileWorker(std::move(file_worker));
+            buffer_obj->UpdateFileWorkerInfo(std::move(file_worker));
         }
         return buffer_obj;
     }

src/storage/buffer/buffer_obj.cpp

@@ -13,7 +13,7 @@
 // limitations under the License.
 
 module;
-
+#include <cassert>
 import stl;
 import file_worker;
 import buffer_handle;
@@ -23,6 +23,7 @@ import logger;
 import third_party;
 import logger;
 import file_worker_type;
+import var_file_worker;
 
 module buffer_obj;
 
@@ -41,7 +42,20 @@ BufferObj::BufferObj(BufferManager *buffer_mgr, bool is_ephemeral, UniquePtr<Fil
 
 BufferObj::~BufferObj() = default;
 
-void BufferObj::SetFileWorker(UniquePtr<FileWorker> file_worker) { file_worker_ = std::move(file_worker); }
+void BufferObj::UpdateFileWorkerInfo(UniquePtr<FileWorker> file_worker) { 
+    switch (file_worker_->Type()) {
+        case FileWorkerType::kVarFile:{
+            assert(file_worker->Type() == FileWorkerType::kVarFile);
+            auto *var_file_worker = static_cast<VarFileWorker *>(file_worker_.get());
+            auto *new_var_file_worker = static_cast<VarFileWorker *>(var_file_worker);
+            var_file_worker->SetBufferSize(new_var_file_worker->GetBufferSize());
+            break;
+        }
+        default:{
+            break;
+        }
+    }
+}
 
 BufferHandle BufferObj::Load() {
     std::unique_lock<std::mutex> locker(w_locker_);

src/storage/buffer/buffer_obj.cppm

@@ -86,7 +86,7 @@ public:
     BufferObj(const BufferObj &) = delete;
     BufferObj &operator=(const BufferObj &) = delete;
 
-    void SetFileWorker(UniquePtr<FileWorker> file_worker);
+    void UpdateFileWorkerInfo(UniquePtr<FileWorker> file_worker);
 
 public:
     // called by ObjectHandle when load first time for that ObjectHandle

src/storage/buffer/file_worker/var_file_worker.cppm

@@ -44,6 +44,10 @@ public:
 
     SizeT GetMemoryCost() const override;
 
+    SizeT GetBufferSize() const { return buffer_size_; }
+
+    void SetBufferSize(SizeT buffer_size) { buffer_size_ = buffer_size; }
+
     FileWorkerType Type() const override { return FileWorkerType::kVarFile; }
 
 protected: