wbtiger
/
foundationdb
mirror of https://github.com/apple/foundationdb.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed a memory bug in the watch map. A few other formatting tweaks, etc.

This commit is contained in:
A.J. Beamon 2022-01-05 11:26:43 -08:00
parent 431d3de43b
commit 138932c12f
4 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fdbclient/DatabaseContext.h
View File

@ -533,7 +533,7 @@ public:
EventCacheHolder connectToDatabaseEventCacheHolder;
private:
std::unordered_map<KeyRef, Reference<WatchMetadata>> watchMap;
std::unordered_map<Key, Reference<WatchMetadata>> watchMap;
};
#endif

19
fdbclient/NativeAPI.actor.cpp
View File

@ -1912,9 +1912,8 @@ Reference<WatchMetadata> DatabaseContext::getWatchMetadata(KeyRef key) const {
}
KeyRef DatabaseContext::setWatchMetadata(Reference<WatchMetadata> metadata) {
KeyRef keyRef = metadata->parameters->key;
watchMap[keyRef] = metadata;
return keyRef;
watchMap[metadata->parameters->key] = metadata;
return metadata->parameters->key;
}
void DatabaseContext::deleteWatchMetadata(KeyRef key) {
@ -2974,14 +2973,18 @@ ACTOR Future<Void> watchStorageServerResp(KeyRef key, Database cx) {
if (!metadata.isValid())
return Void();
if (watchVersion >= metadata->parameters->version) { // case 1: version_1 (SS) >= version_2 (map)
// case 1: version_1 (SS) >= version_2 (map)
if (watchVersion >= metadata->parameters->version) {
cx->deleteWatchMetadata(key);
if (metadata->watchPromise.canBeSet())
metadata->watchPromise.send(watchVersion);
} else { // ABA happens
}
// ABA happens
else {
TEST(true); // ABA issue where the version returned from the server is less than the version in the map
if (metadata->watchPromise.getFutureReferenceCount() ==
1) { // case 2: version_1 < version_2 and future_count == 1
// case 2: version_1 < version_2 and future_count == 1
if (metadata->watchPromise.getFutureReferenceCount() == 1) {
cx->deleteWatchMetadata(key);
}
}
@ -3067,7 +3070,7 @@ Future<Void> getWatchFuture(Database cx, Reference<WatchParameters> parameters)
// recreate in SS)
else if (parameters->version > metadata->parameters->version) {
TEST(true); // Setting a watch that has a different value than the one in the map but a higher version (newer)
cx->deleteWatchMetadata(parameters->key.contents());
cx->deleteWatchMetadata(parameters->key);
metadata->watchPromise.send(parameters->version);
metadata->watchFutureSS.cancel();

2
fdbclient/NativeAPI.actor.h
View File

@ -252,8 +252,6 @@ struct TransactionState : ReferenceCounted<TransactionState> {
std::shared_ptr<CoalescedKeyRangeMap<Value>> conflictingKeys;
// Only available so that Transaction can have a default constructor, for use in state variables
TransactionState() {}
TransactionState(TaskPriority taskID, SpanID spanID) : taskID(taskID), spanID(spanID) {}
TransactionState(Database cx, TaskPriority taskID, SpanID spanID, Reference<TransactionLogInfo> trLogInfo)

2
fdbserver/workloads/ReadHotDetection.actor.cpp
View File

@ -98,7 +98,7 @@ struct ReadHotDetectionWorkload : TestWorkload {
loop {
state Transaction tr(cx);
try {
StorageMetrics sm = wait(tr.getDatabase()->getStorageMetrics(self->wholeRange, 100));
StorageMetrics sm = wait(cx->getStorageMetrics(self->wholeRange, 100));
// TraceEvent("RHDCheckPhaseLog")
// .detail("KeyRangeSize", sm.bytes)
// .detail("KeyRangeReadBandwith", sm.bytesReadPerKSecond);