fix command info keyspace off

yourkeychen · Oct 24, 2021 · 4937e36 · 4937e36
1 parent a825c9d
commit 4937e36
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/pika_partition.cc b/src/pika_partition.cc
@@ -540,11 +540,11 @@ Status Partition::GetKeyNum(std::vector<blackwidow::KeyInfo>* key_info) {
   key_scan_info_.duration = -2;   // duration -2 mean the task in waiting status,
                                   // has not been scheduled for exec
   rocksdb::Status s = db_->GetKeyNum(key_info);
+  key_scan_info_.key_scaning_ = false;
   if (!s.ok()) {
     return Status::Corruption(s.ToString());
   }
   key_scan_info_.key_infos = *key_info;
   key_scan_info_.duration = time(NULL) - key_scan_info_.start_time;
-  key_scan_info_.key_scaning_ = false;
   return Status::OK();
 }
diff --git a/src/pika_table.cc b/src/pika_table.cc
@@ -190,6 +190,9 @@ void Table::RunKeyScan() {
 void Table::StopKeyScan() {
   slash::RWLock rwl(&partitions_rw_, false);
   slash::MutexLock ml(&key_scan_protector_);
+  if (!key_scan_info_.key_scaning_) {
+    return;
+  }
   for (const auto& item : partitions_) {
     item.second->db()->StopScanKeyNum();
   }