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

Improved locking

Browse Source
This commit is contained in:
Andrew Kane
2024-01-16 13:34:55 -08:00
parent 719b4b7436
commit cad48d9203
4 changed files with 44 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/hnswbuild.c
View File

@@ -297,17 +297,17 @@ FlushPages(HnswBuildState * buildstate)
static bool
HnswAddDuplicateInMemory(HnswElement element, HnswElement dup)
{
SpinLockAcquire(&dup->lock);
LWLockAcquire(&dup->lock, LW_EXCLUSIVE);
if (dup->heaptidsLength == HNSW_HEAPTIDS)
{
SpinLockRelease(&dup->lock);
LWLockRelease(&dup->lock);
return false;
}
HnswAddHeapTid(dup, &element->heaptids[0]);
SpinLockRelease(&dup->lock);
LWLockRelease(&dup->lock);
return true;
}
@@ -371,9 +371,9 @@ HnswUpdateNeighborPagesInMemory(char *base, FmgrInfo *procinfo, Oid collation, H
Assert(neighborElement);
/* Use element for lock instead of hc since hc can be replaced */
SpinLockAcquire(&neighborElement->lock);
LWLockAcquire(&neighborElement->lock, LW_EXCLUSIVE);
HnswUpdateConnection(base, e, hc, lm, lc, NULL, NULL, procinfo, collation);
SpinLockRelease(&neighborElement->lock);
LWLockRelease(&neighborElement->lock);
}
}
}
@@ -482,7 +482,7 @@ InsertTuple(Relation index, Datum *values, bool *isnull, ItemPointer heaptid, Hn
HnswPtrStore(base, element->value, valuePtr);
/* Create element lock */
SpinLockInit(&element->lock);
LWLockInitialize(&element->lock, hnsw_lock_tranche_id);
/* Get entry point */
LWLockAcquire(&graph->entryLock, LW_EXCLUSIVE);
@@ -558,9 +558,9 @@ InitGraph(HnswGraph * graph, char *base, long memoryTotal)
graph->flushed = false;
graph->indtuples = 0;
SpinLockInit(&graph->lock);
LWLockInitialize(&graph->entryLock, entryLockTrancheId);
LWLockInitialize(&graph->allocatorLock, allocatorLockTrancheId);
LWLockInitialize(&graph->flushLock, flushLockTrancheId);
LWLockInitialize(&graph->entryLock, hnsw_lock_tranche_id);
LWLockInitialize(&graph->allocatorLock, hnsw_lock_tranche_id);
LWLockInitialize(&graph->flushLock, hnsw_lock_tranche_id);
}
/*