fix Wal in manifest

Signed-off-by: Qi Xu <[email protected]>
tonyxuqqi · Mar 22, 2024 · 7f323a8 · 7f323a8
1 parent c98e2f6
commit 7f323a8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/db/wal_edit.cc b/db/wal_edit.cc
@@ -17,6 +17,8 @@ void WalAddition::EncodeTo(std::string* dst) const {
   if (metadata_.HasSyncedSize()) {
     PutVarint32(dst, static_cast<uint32_t>(WalAdditionTag::kSyncedSize));
     PutVarint64(dst, metadata_.GetSyncedSizeInBytes());
+    PutVarint32(dst, static_cast<uint32_t>(WalAdditionTag::kLastSyncSeq));
+    PutVarint64(dst, metadata_.GetLastSequence());
   }
 
   PutVarint32(dst, static_cast<uint32_t>(WalAdditionTag::kTerminate));
@@ -44,6 +46,13 @@ Status WalAddition::DecodeFrom(Slice* src) {
         metadata_.SetSyncedSizeInBytes(size);
         break;
       }
+      case WalAdditionTag::kLastSyncSeq: {
+        uint64_t lsn = 0;
+        if (!GetVarint64(src, &lsn)) {
+          return Status::Corruption(class_name, "Error decoding WAL file size");
+        }
+        metadata_.SetLastSequence(lsn); 
+      }
       // TODO: process future tags such as checksum.
       case WalAdditionTag::kTerminate:
         return Status::OK();

diff --git a/db/wal_edit.h b/db/wal_edit.h
@@ -45,6 +45,8 @@ class WalMetadata {
 
   uint64_t GetLastSequence() const { return last_sequence_number_; }
 
+  void SetLastSequence(uint64_t lsn) { last_sequence_number_ = lsn; }
+
  private:
   friend bool operator==(const WalMetadata& lhs, const WalMetadata& rhs);
   friend bool operator!=(const WalMetadata& lhs, const WalMetadata& rhs);
@@ -72,6 +74,8 @@ enum class WalAdditionTag : uint32_t {
   kTerminate = 1,
   // Synced Size in bytes.
   kSyncedSize = 2,
+
+  kLastSyncSeq = 3,
   // Add tags in the future, such as checksum?
 };