Make CheckRange{Left,Right}Side into normal functions, WIP

This commit is contained in:
2024-09-04 12:04:58 -07:00
parent 394f09f9fb
commit 8a4b370e2a

View File

@@ -2532,58 +2532,29 @@ struct CheckRangeLeftSide {
bool ok; bool ok;
bool step() { bool step() {
if (remaining.size() == 0) { for (;; ++tls->range_read_iterations_accum) {
assert(searchPathLen >= prefixLen); if (remaining.size() == 0) {
ok = maxVersion(n) <= readVersion; assert(searchPathLen >= prefixLen);
return true;
}
if (searchPathLen >= prefixLen) {
if (!checkMaxBetweenExclusive(n, remaining[0], 256, readVersion, tls)) {
ok = false;
return true;
}
}
auto [child, maxV] = getChildAndMaxVersion(n, remaining[0]);
if (child == nullptr) {
auto c = getChildGeq(n, remaining[0]);
if (c != nullptr) {
if (searchPathLen < prefixLen) {
n = c;
return downLeftSpine();
}
n = c;
ok = maxVersion(n) <= readVersion; ok = maxVersion(n) <= readVersion;
return true; return true;
} else { }
n = nextSibling(n);
if (n == nullptr) { if (searchPathLen >= prefixLen) {
ok = true; if (!checkMaxBetweenExclusive(n, remaining[0], 256, readVersion, tls)) {
ok = false;
return true; return true;
} }
return downLeftSpine();
} }
}
n = child; auto [child, maxV] = getChildAndMaxVersion(n, remaining[0]);
remaining = remaining.subspan(1, remaining.size() - 1); if (child == nullptr) {
++searchPathLen; auto c = getChildGeq(n, remaining[0]);
if (c != nullptr) {
if (n->partialKeyLen > 0) {
int commonLen = std::min<int>(n->partialKeyLen, remaining.size());
int i = longestCommonPrefix(n->partialKey(), remaining.data(), commonLen);
searchPathLen += i;
if (i < commonLen) {
auto c = n->partialKey()[i] <=> remaining[i];
if (c > 0) {
if (searchPathLen < prefixLen) { if (searchPathLen < prefixLen) {
return downLeftSpine(); n = c;
} goto downLeftSpine;
if (n->entryPresent && n->entry.rangeVersion > readVersion) {
ok = false;
return true;
} }
n = c;
ok = maxVersion(n) <= readVersion; ok = maxVersion(n) <= readVersion;
return true; return true;
} else { } else {
@@ -2592,30 +2563,60 @@ struct CheckRangeLeftSide {
ok = true; ok = true;
return true; return true;
} }
return downLeftSpine(); goto downLeftSpine;
} }
} }
if (commonLen == n->partialKeyLen) {
// partial key matches n = child;
remaining = remaining.subspan(commonLen, remaining.size() - commonLen); remaining = remaining.subspan(1, remaining.size() - 1);
} else if (n->partialKeyLen > int(remaining.size())) { ++searchPathLen;
assert(searchPathLen >= prefixLen);
if (n->entryPresent && n->entry.rangeVersion > readVersion) { if (n->partialKeyLen > 0) {
ok = false; int commonLen = std::min<int>(n->partialKeyLen, remaining.size());
int i =
longestCommonPrefix(n->partialKey(), remaining.data(), commonLen);
searchPathLen += i;
if (i < commonLen) {
auto c = n->partialKey()[i] <=> remaining[i];
if (c > 0) {
if (searchPathLen < prefixLen) {
goto downLeftSpine;
}
if (n->entryPresent && n->entry.rangeVersion > readVersion) {
ok = false;
return true;
}
ok = maxVersion(n) <= readVersion;
return true;
} else {
n = nextSibling(n);
if (n == nullptr) {
ok = true;
return true;
}
goto downLeftSpine;
}
}
if (commonLen == n->partialKeyLen) {
// partial key matches
remaining =
remaining.subspan(commonLen, remaining.size() - commonLen);
} else if (n->partialKeyLen > int(remaining.size())) {
assert(searchPathLen >= prefixLen);
if (n->entryPresent && n->entry.rangeVersion > readVersion) {
ok = false;
return true;
}
ok = maxVersion(n) <= readVersion;
return true; return true;
} }
ok = maxVersion(n) <= readVersion; }
if (maxV <= readVersion) {
ok = true;
return true; return true;
} }
} }
if (maxV <= readVersion) { downLeftSpine:
ok = true;
return true;
}
return false;
}
bool downLeftSpine() {
for (; !n->entryPresent; n = getFirstChildExists(n)) { for (; !n->entryPresent; n = getFirstChildExists(n)) {
} }
ok = n->entry.rangeVersion <= readVersion; ok = n->entry.rangeVersion <= readVersion;
@@ -2817,7 +2818,7 @@ bool checkRangeRead(Node *n, std::span<const uint8_t> begin,
if (lcp == int(begin.size())) { if (lcp == int(begin.size())) {
CheckRangeRightSide checkRangeRightSide{n, end, lcp, readVersion, tls}; CheckRangeRightSide checkRangeRightSide{n, end, lcp, readVersion, tls};
while (!checkRangeRightSide.step()) for (; !checkRangeRightSide.step(); ++tls->range_read_iterations_accum)
; ;
return checkRangeRightSide.ok; return checkRangeRightSide.ok;
} }
@@ -2831,30 +2832,11 @@ bool checkRangeRead(Node *n, std::span<const uint8_t> begin,
// were false, then we would have returned above since lcp == begin.size(). // were false, then we would have returned above since lcp == begin.size().
assert(!(n->parent == nullptr && begin.size() == 0)); assert(!(n->parent == nullptr && begin.size() == 0));
CheckRangeLeftSide checkRangeLeftSide{n, begin, lcp + 1, readVersion, tls}; CheckRangeLeftSide checkRangeLeftSide{n, begin, lcp + 1, readVersion, tls};
for (; !checkRangeLeftSide.step(); ++tls->range_read_iterations_accum)
;
CheckRangeRightSide checkRangeRightSide{n, end, lcp + 1, readVersion, tls}; CheckRangeRightSide checkRangeRightSide{n, end, lcp + 1, readVersion, tls};
for (; !checkRangeRightSide.step(); ++tls->range_read_iterations_accum)
for (;;) { ;
bool leftDone = checkRangeLeftSide.step();
bool rightDone = checkRangeRightSide.step();
if (!leftDone && !rightDone) {
tls->range_read_iterations_accum += 2;
continue;
}
if (leftDone && rightDone) {
break;
} else if (leftDone) {
while (!checkRangeRightSide.step()) {
++tls->range_read_iterations_accum;
}
break;
} else {
assert(rightDone);
while (!checkRangeLeftSide.step()) {
++tls->range_read_iterations_accum;
}
}
break;
}
return checkRangeLeftSide.ok && checkRangeRightSide.ok; return checkRangeLeftSide.ok && checkRangeRightSide.ok;
} }