set max size 1000

Andrey Kazarinov · Alena Rybakina · commit 4d3a687125f3 · 2022-08-17T09:57:31.000+03:00
diff --git a/aqo.c b/aqo.c
@@ -238,6 +238,19 @@ _PG_init(void)
 							NULL
 	);
 
+	DefineCustomIntVariable("aqo.max_size",
+							"Query max size in aqo_query_texts.",
+							NULL,
+							&max_size,
+							1000,
+							0, INT_MAX,
+							PGC_SUSET,
+							0,
+							NULL,
+							NULL,
+							NULL
+	);
+
 	prev_shmem_startup_hook						= shmem_startup_hook;
 	shmem_startup_hook							= aqo_init_shmem;
 	prev_planner_hook							= planner_hook;
diff --git a/aqo_shared.c b/aqo_shared.c
@@ -28,6 +28,7 @@ HTAB *fss_htab = NULL;
 static int aqo_htab_max_items = 1000;
 int fs_max_items = 1; /* Max number of different feature spaces in ML model */
 int fss_max_items = 1; /* Max number of different feature subspaces in ML model */
+int max_size = 1000;
 static uint32 temp_storage_size = 1024 * 1024 * 10; /* Storage size, in bytes */
 static dsm_segment *seg = NULL;
 
diff --git a/aqo_shared.h b/aqo_shared.h
@@ -52,6 +52,7 @@ extern HTAB *fss_htab;
 
 extern int fs_max_items; /* Max number of feature spaces that AQO can operate */
 extern int fss_max_items;
+extern int max_size;
 
 extern Size aqo_memsize(void);
 extern void reset_dsm_cache(void);
diff --git a/storage.c b/storage.c
@@ -969,6 +969,7 @@ aqo_qtext_store(uint64 queryid, const char *query_string)
 		}
 
 		entry->queryid = queryid;
+		size = size > max_size ? max_size : size;
 		entry->qtext_dp = dsa_allocate(qtext_dsa, size);
 		Assert(DsaPointerIsValid(entry->qtext_dp));
 		strptr = (char *) dsa_get_address(qtext_dsa, entry->qtext_dp);
