liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-15 09:06:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Look for duplicates last, since may need to fallback

Browse Source
This commit is contained in:
Andrew Kane
2023-08-10 00:00:27 -07:00
parent 7c0d94c99c
commit df68eb4570
1 changed files with 12 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/hnswutils.c
View File

@@ -937,27 +937,11 @@ HnswInsertElement(HnswElement element, HnswElement entryPoint, Relation index, F
if (removeEntryPoint) if (removeEntryPoint)
w = list_delete_ptr(w, entryCandidate); w = list_delete_ptr(w, entryCandidate);
/* Always call on inserts since duplicate update can fail */
neighbors[lc] = SelectNeighbors(w, lm, lc, procinfo, collation, NULL); neighbors[lc] = SelectNeighbors(w, lm, lc, procinfo, collation, NULL);
ep = w;
}
/* Look for duplicate */
if (level >= 0 && !vacuuming)
{
dup = HnswFindDuplicate(element, neighbors[0]);
if (dup != NULL)
return dup;
}
/* Update connections */
for (int lc = level; lc >= 0; lc--)
{
int lm = HnswGetLayerM(m, lc);
AddConnections(element, neighbors[lc], lm, lc); AddConnections(element, neighbors[lc], lm, lc);
/* Update connections */
if (!vacuuming && updateNeighbors == NULL) if (!vacuuming && updateNeighbors == NULL)
{ {
ListCell *lc2; ListCell *lc2;
@@ -965,10 +949,21 @@ HnswInsertElement(HnswElement element, HnswElement entryPoint, Relation index, F
foreach(lc2, neighbors[lc]) foreach(lc2, neighbors[lc])
HnswUpdateConnection(element, lfirst(lc2), lm, lc, NULL, index, procinfo, collation); HnswUpdateConnection(element, lfirst(lc2), lm, lc, NULL, index, procinfo, collation);
} }
ep = w;
} }
if (updateNeighbors != NULL) if (updateNeighbors != NULL)
*updateNeighbors = neighbors; *updateNeighbors = neighbors;
/* Look for duplicates */
/* This must come last, since duplicate updates can fail */
if (level >= 0 && !vacuuming)
{
dup = HnswFindDuplicate(element, neighbors[0]);
if (dup != NULL)
return dup;
}
return NULL; return NULL;
} }