fix: crash during getting info about replication

dragonflydb · Dec 17, 2024 · 61a5a3b · 61a5a3b
1 parent 0fe5e86
commit 61a5a3b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/server/rdb_save.cc b/src/server/rdb_save.cc
@@ -1333,6 +1333,8 @@ size_t RdbSaver::Impl::GetTotalBuffersSize() const {
 
   auto cb = [this, &channel_bytes, &serializer_bytes](ShardId sid) {
     auto& snapshot = shard_snapshots_[sid];
+    if (!snapshot)
+      return;
     if (channel_.has_value())
       channel_bytes.fetch_add(channel_->GetSize(), memory_order_relaxed);
     serializer_bytes.store(snapshot->GetBufferCapacity() + snapshot->GetTempBuffersSize(),
@@ -1355,6 +1357,8 @@ RdbSaver::SnapshotStats RdbSaver::Impl::GetCurrentSnapshotProgress() const {
 
   auto cb = [this, &results](ShardId sid) {
     auto& snapshot = shard_snapshots_[sid];
+    if (!snapshot)
+      return;
     results[sid] = snapshot->GetCurrentSnapshotProgress();
   };