fix invaild partition problem

Qihoo360 · Mar 2, 2017 · 4362c17 · 4362c17
1 parent 19fbefa
commit 4362c17
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/node/zp_data_table.cc b/src/node/zp_data_table.cc
@@ -60,15 +60,17 @@ Partition* Table::GetPartition(const std::string &key) {
   slash::RWLock l(&partition_rw_, false);
   if (partition_cnt_ > 0) {
     int partition_id = std::hash<std::string>()(key) % partition_cnt_;
-    return partitions_[partition_id]; 
+    auto it = partitions_.find(partition_id);
+    if (it != partitions_.end()) {
+      return it->second;
+    }
   }
   return NULL;
 }
 
 Partition* Table::GetPartitionById(const int partition_id) {
   slash::RWLock l(&partition_rw_, false);
   auto it = partitions_.find(partition_id);
-  //std::map<int, Partition*>::const_iterator it = partitions_.find(partition_id);
   if (it != partitions_.end()) {
     return it->second;
   }