check RI with ref_id before adding ifinite entry

dongchen-coder · Nov 3, 2019 · ceccef2 · ceccef2
1 parent 2f953ab
commit ceccef2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test_facility/utility/reference_lease.h b/test_facility/utility/reference_lease.h
@@ -73,7 +73,12 @@ void RIwithInfinite() {
 				cnt++;
 			}
 		}
-		(*RI[ref_it->first])[std::numeric_limits<uint64_t>::max()] = cnt;
+		if (RI.find(ref_it->first) != RI.end()) {
+			(*RI[ref_it->first])[std::numeric_limits<uint64_t>::max()] = cnt;
+		} else {
+			RI[ref_it->first] = new map<uint64_t, uint64_t>;
+			(*RI[ref_it->first])[std::numeric_limits<uint64_t>::max()] = cnt;
+		}
 	}
 	return;
 }