Add function which shows memory usage.

function memctx_htab_sizes outputs allocated sizes and used sizes of aqo's memory contexts and hash tables a.lepikhov: I've changed implementation because the VIEW on memory contexts introduced in the next version of core PostgreSQL.

Author: Andrey Kazarinov

aqo--1.5--1.6.sql

@@ -98,3 +98,17 @@ AS 'MODULE_PATHNAME', 'aqo_queries'
 LANGUAGE C STRICT VOLATILE PARALLEL SAFE;
 
 CREATE VIEW aqo_queries AS SELECT * FROM aqo_queries();
+
+-- Show how much shared memory AQO are using at the moment
+CREATE FUNCTION aqo_memory_usage(
+OUT name text,
+OUT allocated_size int,
+OUT used_size int
+)
+RETURNS SETOF record
+AS $$
+SELECT name, allocated_size, size FROM pg_shmem_allocations
+WHERE name LIKE 'AQO%';
+$$ LANGUAGE SQL;
+COMMENT ON FUNCTION aqo_memory_usage() IS
+'Show how much shared memory AQO are using at the moment';

t/001_pgbench.pl

@@ -159,6 +159,9 @@
 WHERE v.exec_time > 0.");
 is($res, 3);
 
+$res = $node->safe_psql('postgres', "SELECT * FROM aqo_memory_usage() AS t1");
+note("MEMORY:\n$res\n");
+
 # ##############################################################################
 #
 # pgbench on a database with AQO in 'learn' mode.
@@ -183,6 +186,9 @@
 "$TRANSACTIONS", '-c', "$CLIENTS", '-j', "$THREADS" ],
 'pgbench in frozen mode');
 
+$res = $node->safe_psql('postgres', "SELECT * FROM aqo_memory_usage() AS t1");
+note("MEMORY:\n$res\n");
+
 # ##############################################################################
 #
 # Check procedure of ML-knowledge data cleaning.
@@ -298,6 +304,9 @@
 is($new_stat_count == $stat_count - $pgb_stat_count, 1,
 'Total number of samples in aqo_query_stat');
 
+$res = $node->safe_psql('postgres', "SELECT * FROM aqo_memory_usage() AS t1");
+note("MEMORY:\n$res\n");
+
 # ##############################################################################
 #
 # AQO works after moving to another schema