Mirror/MarginaliaSearch
1
0
Fork 0
mirror of https://github.com/MarginaliaSearch/MarginaliaSearch.git synced 2025-02-23 13:09:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactoring BTreeReader and binary search code

Browse Source
This commit is contained in:
vlofgren 2022-06-20 12:35:58 +02:00
parent 8139ab0d1d
commit 1068694db6
1 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
marginalia_nu/src/main/java/nu/marginalia/util/btree/BTreeReader.java
View File

@ -5,6 +5,8 @@ import nu.marginalia.util.btree.model.BTreeHeader;
import nu.marginalia.util.multimap.MultimapFileLong;
import nu.marginalia.util.multimap.MultimapSearcher;
import javax.annotation.CheckReturnValue;
import static java.lang.Math.min;
public class BTreeReader {
@ -37,19 +39,22 @@ public class BTreeReader {
final long key = keyRaw & ctx.equalityMask();
final long dataAddress = header.dataOffsetLongs();
if (header.layers() == 0) { // For small data, we only have a data block
return dataSearcher.binarySearch(key, dataAddress, header.numEntries());
}
final long searchStart;
final long numEntries;
// Search index layers
long dataLayerOffset = searchIndex(header, key);
if (dataLayerOffset < 0) {
return dataLayerOffset;
if (header.layers() == 0) { // For small data, there is no index block, only a flat data block
searchStart = dataAddress;
numEntries = header.numEntries();
}
else {
long dataLayerOffset = searchIndex(header, key);
if (dataLayerOffset < 0) {
return dataLayerOffset;
}
// Search the corresponding data block
final long searchStart = dataAddress + dataLayerOffset * ctx.entrySize();
final long numEntries = min(header.numEntries() - dataLayerOffset, blockSize);
searchStart = dataAddress + dataLayerOffset * ctx.entrySize();
numEntries = min(header.numEntries() - dataLayerOffset, blockSize);
}
return dataSearcher.binarySearch(key, searchStart, numEntries);
}
@ -61,14 +66,15 @@ public class BTreeReader {
long layerOffset = 0;
for (int i = header.layers() - 1; i >= 0; --i) {
final long layerBlockOffset = header.relativeIndexLayerOffset(ctx, i) + layerOffset;
final long indexLayerBlockOffset = header.relativeIndexLayerOffset(ctx, i) + layerOffset;
final long nextLayerOffset = indexSearch(key, indexAddress + layerBlockOffset, blockSize);
final long nextLayerOffset = indexSearch(key, indexAddress + indexLayerBlockOffset, blockSize);
if (nextLayerOffset < 0)
return -1;
return nextLayerOffset;
layerOffset = blockSize *(nextLayerOffset + layerOffset);
layerOffset = blockSize * (nextLayerOffset + layerOffset);
}
return layerOffset;
}