database/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-10-23 10:48:54 +03:00
Code Activity

DRY code [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2026-10-18 03:04:52 -07:00
parent fdd3819331
commit c4b35f303f
2 changed files with 29 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/hnsw.h
View File

@@ -419,16 +419,20 @@ typedef union
typedef struct HnswSearchState
{
/* Info */
Relation index;
HnswUnvisited *unvisited;
int unvisitedLength;
int visited;
/* Prefetching */
#ifdef HNSW_USE_READ_STREAM
ReadStream *stream;
#elif defined(HNSW_USE_PREFETCH)
int maxPrefetch;
#endif
/* Batch state */
HnswUnvisited *unvisited;
int unvisitedLength;
int visited;
} HnswSearchState;
typedef struct HnswScanOpaqueData

33
src/hnswutils.c
View File

@@ -838,6 +838,22 @@ HnswLoadUnvisitedFromDisk(HnswElement element, HnswUnvisited * unvisited, int *u
#endif
}
/*
* Get next block number for search
*/
static BlockNumber
HnswNextBlock(HnswSearchState * searchState, OffsetNumber *offno)
{
HnswUnvisited *uv;
if (searchState->visited == searchState->unvisitedLength)
return InvalidBlockNumber;
uv = &searchState->unvisited[searchState->visited++];
*offno = ItemPointerGetOffsetNumber(&uv->indextid);
return ItemPointerGetBlockNumber(&uv->indextid);
}
#ifdef HNSW_USE_READ_STREAM
/*
* Get next block number for read stream
@@ -847,14 +863,12 @@ HnswReadStreamNextBlock(ReadStream *stream, void *callback_private_data, void *p
{
HnswSearchState *searchState = callback_private_data;
OffsetNumber *offno = per_buffer_data;
HnswUnvisited *uv;
BlockNumber blkno = HnswNextBlock(searchState, offno);
if (searchState->visited == searchState->unvisitedLength)
if (!BlockNumberIsValid(blkno))
return read_stream_pause(stream);
uv = &searchState->unvisited[searchState->visited++];
*offno = ItemPointerGetOffsetNumber(&uv->indextid);
return ItemPointerGetBlockNumber(&uv->indextid);
return blkno;
}
#endif
@@ -923,15 +937,12 @@ HnswNextBuffer(HnswSearchState * searchState, OffsetNumber *offno)
return buf;
#else
/* TODO DRY with HnswReadStreamNextBlock */
HnswUnvisited *uv;
BlockNumber blkno = HnswNextBlock(searchState, offno);
if (searchState->visited == searchState->unvisitedLength)
if (!BlockNumberIsValid(blkno))
return InvalidBuffer;
uv = &searchState->unvisited[searchState->visited++];
buf = ReadBuffer(searchState->index, ItemPointerGetBlockNumber(&uv->indextid));
*offno = ItemPointerGetOffsetNumber(&uv->indextid);
buf = ReadBuffer(searchState->index, blkno);
#ifdef HNSW_USE_PREFETCH
/* Prefetch another buffer if possible */