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

Keep track of tries [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2026-10-07 00:09:18 -07:00
parent 6220c5b791
commit 11cb15219e
1 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/hnswinsert.c
View File

@@ -17,6 +17,8 @@
#include "varatt.h"
#endif
#define HNSW_MAX_FSM_TRIES 3
/*
* Get the insert page
*/
@@ -169,6 +171,7 @@ AddElementOnDisk(Relation index, HnswElement e, int m, BlockNumber *updatedInser
bool tryFsm;
int fsmTries = 0;
bool fsmPage;
BlockNumber fsmPages[HNSW_MAX_FSM_TRIES];
/* Calculate sizes */
etupSize = HNSW_ELEMENT_TUPLE_SIZE(VARSIZE_ANY(HnswPtrAccess(base, e->value)));
@@ -196,10 +199,28 @@ AddElementOnDisk(Relation index, HnswElement e, int m, BlockNumber *updatedInser
/* Try free space map */
if (tryFsm)
{
/* TODO do not retry same page */
currentPage = GetPageWithFreeSpace(index, combinedSize);
fsmPages[fsmTries] = currentPage;
/*
* Even if a page is returned, it may not be possible to add the
* tuples (see HnswFreeOffset for details), so avoid checking the
* same page again
*/
if (BlockNumberIsValid(currentPage))
{
for (int i = 0; i < fsmTries; i++)
{
if (fsmPages[i] == currentPage)
{
currentPage = InvalidBlockNumber;
break;
}
}
}
fsmPage = BlockNumberIsValid(currentPage);
tryFsm = ++fsmTries < 3 && BlockNumberIsValid(currentPage);
tryFsm = ++fsmTries < HNSW_MAX_FSM_TRIES && BlockNumberIsValid(currentPage);
}
/* Start at insert page if free space map yields nothing */