fix lock (#5)

Co-authored-by: wuxianrong <wuxianrong@360.cn>
OpenAtomFoundation · Mar 18, 2024 · ed727bb · ed727bb
1 parent a19a54a
commit ed727bb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/include/pika_db.h b/include/pika_db.h
@@ -101,6 +101,9 @@ class DB : public std::enable_shared_from_this<DB>, public pstd::noncopyable {
   std::shared_mutex& GetDBLock() {
     return dbs_rw_;
   }
+  std::shared_mutex& GetDBLocks() {
+    return db_rwlock_;
+  }
   void DBLock() {
     dbs_rw_.lock();
   }

diff --git a/src/pika_admin.cc b/src/pika_admin.cc
@@ -709,7 +709,7 @@ void FlushdbCmd::Execute() {
     if (db_->IsKeyScaning()) {
       res_.SetRes(CmdRes::kErrOther, "The keyscan operation is executing, Try again later");
     } else {
-      std::lock_guard l_prw(db_->GetDBLock());
+      std::lock_guard l_prw(db_->GetDBLocks());
       std::lock_guard s_prw(g_pika_rm->GetDBLock());
       FlushAllDBsWithoutLock();
       res_.SetRes(CmdRes::kOk);