From 02b01e1ca900386d7cb9deb59fc70e640c59d8c5 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Sat, 12 Oct 2024 22:05:13 -0700 Subject: [PATCH] Show tuples with memory usage [skip ci] --- src/hnswscan.c | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/src/hnswscan.c b/src/hnswscan.c index 84e2332..ba3dea8 100644 --- a/src/hnswscan.c +++ b/src/hnswscan.c @@ -102,6 +102,17 @@ GetScanValue(IndexScanDesc scan) return value; } +#if defined(HNSW_MEMORY) +/* + * Show memory usage + */ +static void +ShowMemoryUsage(HnswScanOpaque so) +{ + elog(INFO, "memory: %zu KB, tuples: " INT64_FORMAT, MemoryContextMemAllocated(so->tmpCtx, false) / 1024, so->tuples); +} +#endif + /* * Prepare for an index scan */ @@ -209,7 +220,7 @@ hnswgettuple(IndexScanDesc scan, ScanDirection dir) so->first = false; #if defined(HNSW_MEMORY) - elog(INFO, "memory: %zu KB", MemoryContextMemAllocated(so->tmpCtx, false) / 1024); + ShowMemoryUsage(so); #endif } @@ -271,7 +282,7 @@ hnswgettuple(IndexScanDesc scan, ScanDirection dir) UnlockPage(scan->indexRelation, HNSW_SCAN_LOCK, ShareLock); #if defined(HNSW_MEMORY) - elog(INFO, "memory: %zu KB", MemoryContextMemAllocated(so->tmpCtx, false) / 1024); + ShowMemoryUsage(so); #endif }