bugfix: MemTableList::RemoveOldMemTables invalid iterator after remov… (#6013)

Summary:
Fix issue https://github.com/facebook/rocksdb/issues/6012.

I found that it may be caused by the following codes in function _RemoveOldMemTables()_ in **db/memtable_list.cc**  :
```
  for (auto it = memlist.rbegin(); it != memlist.rend(); ++it) {
    MemTable* mem = *it;
    if (mem->GetNextLogNumber() > log_number) {
      break;
    }
    current_->Remove(mem, to_delete);
```

The iterator **it** turns invalid after `current_->Remove(mem, to_delete);`
Pull Request resolved: https://github.com/facebook/rocksdb/pull/6013

Test Plan:
```
make check
```

Differential Revision: D18401107

Pulled By: riversand963

fbshipit-source-id: bf0da3b868ed70f7aff24cf7b3e2049c0c5c7a4e
main
蔡渠棠 5 years ago committed by Facebook Github Bot
parent c17384fea4
commit f29e6b3be2
  1. 7
      db/memtable_list.cc

@ -736,17 +736,24 @@ void MemTableList::RemoveOldMemTables(uint64_t log_number,
assert(to_delete != nullptr); assert(to_delete != nullptr);
InstallNewVersion(); InstallNewVersion();
auto& memlist = current_->memlist_; auto& memlist = current_->memlist_;
autovector<MemTable*> old_memtables;
for (auto it = memlist.rbegin(); it != memlist.rend(); ++it) { for (auto it = memlist.rbegin(); it != memlist.rend(); ++it) {
MemTable* mem = *it; MemTable* mem = *it;
if (mem->GetNextLogNumber() > log_number) { if (mem->GetNextLogNumber() > log_number) {
break; break;
} }
old_memtables.push_back(mem);
}
for (auto it = old_memtables.begin(); it != old_memtables.end(); ++it) {
MemTable* mem = *it;
current_->Remove(mem, to_delete); current_->Remove(mem, to_delete);
--num_flush_not_started_; --num_flush_not_started_;
if (0 == num_flush_not_started_) { if (0 == num_flush_not_started_) {
imm_flush_needed.store(false, std::memory_order_release); imm_flush_needed.store(false, std::memory_order_release);
} }
} }
UpdateMemoryUsageExcludingLast(); UpdateMemoryUsageExcludingLast();
ResetTrimHistoryNeeded(); ResetTrimHistoryNeeded();
} }

Loading…
Cancel
Save