A couple of bugfixes here: don't enable statement timeout in parallel worker and minor DSM cache fix.

Author: danolivo

learn_cache.c

@@ -227,7 +227,7 @@ init_with_dsm(OkNNrdata *data, dsm_block_hdr *hdr, List **reloids)
 Assert(LWLockHeldByMeInMode(&aqo_state->lock, LW_EXCLUSIVE) ||
 LWLockHeldByMeInMode(&aqo_state->lock, LW_SHARED));
 Assert(hdr->magic == AQO_SHARED_MAGIC);
-Assert(hdr && ptr);
+Assert(hdr && ptr && hdr->rows > 0);
 
 data->rows = hdr->rows;
 data->cols = hdr->cols;
@@ -245,6 +245,12 @@ init_with_dsm(OkNNrdata *data, dsm_block_hdr *hdr, List **reloids)
 }
 }
 
+/*
+* Kludge code. But we should rewrite this code because now all knowledge
+* base lives in non-transactional shared memory.
+*/
+ptr = (char *) hdr + sizeof(dsm_block_hdr) + (sizeof(double) * data->cols * aqo_K);
+
 memcpy(data->targets, ptr, sizeof(double) * hdr->rows);
 ptr += sizeof(double) * aqo_K;
 memcpy(data->rfactors, ptr, sizeof(double) * hdr->rows);
@@ -261,7 +267,7 @@ init_with_dsm(OkNNrdata *data, dsm_block_hdr *hdr, List **reloids)
 return calculate_size(hdr->cols, *reloids);
 }
 
-/* It is just read operation. No any interest in size calculation. */
+/* It is just a read operation. No any interest in size calculation. */
 return 0;
 }
 

postprocessing.c

@@ -638,6 +638,13 @@ set_timeout_if_need(QueryDesc *queryDesc)
 {
 TimestampTz	fin_time;
 
+if (IsParallelWorker())
+/*
+* AQO timeout should stop only main worker. Other workers would be
+* terminated by a regular ERROR machinery.
+*/
+return false;
+
 if (!get_timeout_active(STATEMENT_TIMEOUT) || !aqo_learn_statement_timeout)
 return false;