Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitd2a98e4

Browse files
danolivoAlena Rybakina
authored and
Alena Rybakina
committed
Some mistakes, revealed by regression tests, passed at Raspberry PI4:
1. Fix uint64 format in some output messages.2. Input parameters conversion mistake in aqo_queries_update.3. Unneeded routine 'get_aqo_schema'.4. Fix type of the first parameter (counter) in aqo_cardinality_error() and aqo_execution_time() routines.5. Fix aqo_data() routine.
1 parenteb87e74 commitd2a98e4

File tree

8 files changed

+69
-83
lines changed

8 files changed

+69
-83
lines changed

‎aqo--1.4--1.5.sql‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ CREATE FUNCTION aqo_data (
6262
OUT featuresdouble precision[][],
6363
OUT targetsdouble precision[],
6464
OUT reliabilitydouble precision[],
65-
OUT oidsinteger[]
65+
OUT oidsOid[]
6666
)
6767
RETURNS SETOF record
6868
AS'MODULE_PATHNAME','aqo_data'
@@ -106,7 +106,7 @@ LANGUAGE C VOLATILE;
106106
-- nexecs - number of executions of queries associated with this ID.
107107
--
108108
CREATE OR REPLACEFUNCTIONaqo_cardinality_error(controlledboolean)
109-
RETURNS TABLE(numbigint, idbigint, fshashbigint, errorfloat, nexecsbigint)
109+
RETURNS TABLE(numinteger, idbigint, fshashbigint, errordouble precision, nexecsbigint)
110110
AS'MODULE_PATHNAME','aqo_cardinality_error'
111111
LANGUAGE C STRICT VOLATILE;
112112
COMMENT ON FUNCTION aqo_cardinality_error(boolean) IS
@@ -119,7 +119,7 @@ COMMENT ON FUNCTION aqo_cardinality_error(boolean) IS
119119
-- Last case is possible in disabled mode with aqo.force_collect_stat = 'on'.
120120
--
121121
CREATE OR REPLACEFUNCTIONaqo_execution_time(controlledboolean)
122-
RETURNS TABLE(numbigint, idbigint, fshashbigint, exec_timefloat, nexecsbigint)
122+
RETURNS TABLE(numinteger, idbigint, fshashbigint, exec_timedouble precision, nexecsbigint)
123123
AS'MODULE_PATHNAME','aqo_execution_time'
124124
LANGUAGE C STRICT VOLATILE;
125125
COMMENT ON FUNCTION aqo_execution_time(boolean) IS

‎aqo.c‎

Lines changed: 0 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -359,52 +359,6 @@ _PG_init(void)
359359
RequestAddinShmemSpace(aqo_memsize());
360360
}
361361

362-
/*
363-
* Return AQO schema's Oid or InvalidOid if that's not possible.
364-
*/
365-
Oid
366-
get_aqo_schema(void)
367-
{
368-
Oidresult;
369-
Relationrel;
370-
SysScanDescscandesc;
371-
HeapTupletuple;
372-
ScanKeyDataentry[1];
373-
Oidext_oid;
374-
375-
/* It's impossible to fetch pg_aqo's schema now */
376-
if (!IsTransactionState())
377-
returnInvalidOid;
378-
379-
ext_oid=get_extension_oid("aqo", true);
380-
if (ext_oid==InvalidOid)
381-
returnInvalidOid;/* exit if pg_aqo does not exist */
382-
383-
ScanKeyInit(&entry[0],
384-
#ifPG_VERSION_NUM >=120000
385-
Anum_pg_extension_oid,
386-
#else
387-
ObjectIdAttributeNumber,
388-
#endif
389-
BTEqualStrategyNumber,F_OIDEQ,
390-
ObjectIdGetDatum(ext_oid));
391-
392-
rel=relation_open(ExtensionRelationId,AccessShareLock);
393-
scandesc=systable_beginscan(rel,ExtensionOidIndexId, true,
394-
NULL,1,entry);
395-
tuple=systable_getnext(scandesc);
396-
397-
/* We assume that there can be at most one matching tuple */
398-
if (HeapTupleIsValid(tuple))
399-
result= ((Form_pg_extension)GETSTRUCT(tuple))->extnamespace;
400-
else
401-
result=InvalidOid;
402-
403-
systable_endscan(scandesc);
404-
relation_close(rel,AccessShareLock);
405-
returnresult;
406-
}
407-
408362
/*
409363
* AQO is really needed for any activity?
410364
*/

‎aqo.h‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -297,7 +297,6 @@ extern double *selectivity_cache_find_global_relid(int clause_hash,
297297
intglobal_relid);
298298
externvoidselectivity_cache_clear(void);
299299

300-
externOidget_aqo_schema(void);
301300
externboolIsQueryDisabled(void);
302301

303302
externList*cur_classes;

‎expected/gucs.out‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -69,17 +69,17 @@ SELECT obj_description('aqo_reset'::regproc::oid);
6969
(1 row)
7070

7171
\df aqo_cardinality_error
72-
List of functions
73-
Schema | Name | Result data type | Argument data types | Type
74-
--------+-----------------------+------------------------------------------------------------------------------------+---------------------+------
75-
public | aqo_cardinality_error | TABLE(numbigint, id bigint, fshash bigint, error double precision, nexecs bigint) | controlled boolean | func
72+
List of functions
73+
Schema | Name | Result data type| Argument data types | Type
74+
--------+-----------------------+-------------------------------------------------------------------------------------+---------------------+------
75+
public | aqo_cardinality_error | TABLE(numinteger, id bigint, fshash bigint, error double precision, nexecs bigint) | controlled boolean | func
7676
(1 row)
7777

7878
\df aqo_execution_time
7979
List of functions
80-
Schema | Name | Result data type | Argument data types | Type
81-
--------+--------------------+----------------------------------------------------------------------------------------+---------------------+------
82-
public | aqo_execution_time | TABLE(numbigint, id bigint, fshash bigint, exec_time double precision, nexecs bigint) | controlled boolean | func
80+
Schema | Name | Result data type| Argument data types | Type
81+
--------+--------------------+-----------------------------------------------------------------------------------------+---------------------+------
82+
public | aqo_execution_time | TABLE(numinteger, id bigint, fshash bigint, exec_time double precision, nexecs bigint) | controlled boolean | func
8383
(1 row)
8484

8585
\df aqo_drop_class

‎learn_cache.c‎

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -112,11 +112,19 @@ lc_update_fss(uint64 fs, int fss, OkNNrdata *data, List *reloids)
112112
ptr= (char*)hdr+sizeof(dsm_block_hdr);/* start point of variable data */
113113

114114
/* copy the matrix into DSM storage */
115-
for (i=0;i<aqo_K;++i)
115+
116+
if (hdr->cols>0)
116117
{
117-
if (i<hdr->rows)
118+
for (i=0;i<aqo_K;++i)
119+
{
120+
if (i >=hdr->rows)
121+
break;
122+
123+
if (!ptr|| !data->matrix[i])
124+
elog(PANIC,"Something disruptive have happened! %d, %d (%d %d)",i,hdr->rows,found,hdr->cols);
118125
memcpy(ptr,data->matrix[i],sizeof(double)*hdr->cols);
119-
ptr+=sizeof(double)*data->cols;
126+
ptr+=sizeof(double)*data->cols;
127+
}
120128
}
121129

122130
/* copy targets into DSM storage */
@@ -177,7 +185,7 @@ lc_load_fss(uint64 fs, int fss, OkNNrdata *data, List **reloids)
177185
Assert(fss_htab&&aqo_learn_statement_timeout);
178186

179187
if (aqo_show_details)
180-
elog(NOTICE,"[AQO] Load ML data for fs%lu, fss %d from the cache",
188+
elog(NOTICE,"[AQO] Load ML data for fs"UINT64_FORMAT", fss %d from the cache",
181189
fs,fss);
182190

183191
LWLockAcquire(&aqo_state->lock,LW_SHARED);
@@ -213,6 +221,7 @@ init_with_dsm(OkNNrdata *data, dsm_block_hdr *hdr, List **reloids)
213221
Assert(LWLockHeldByMeInMode(&aqo_state->lock,LW_EXCLUSIVE)||
214222
LWLockHeldByMeInMode(&aqo_state->lock,LW_SHARED));
215223
Assert(hdr->magic==AQO_SHARED_MAGIC);
224+
Assert(hdr&&ptr);
216225

217226
data->rows=hdr->rows;
218227
data->cols=hdr->cols;
@@ -264,7 +273,7 @@ lc_flush_data(void)
264273
ptr=get_dsm_all(&size);
265274

266275
/* Iterate through records and store them into the aqo_data table */
267-
while(size>0)
276+
while(size>0)
268277
{
269278
dsm_block_hdr*hdr= (dsm_block_hdr*)ptr;
270279
OkNNrdatadata;

‎path_utils.c‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ create_aqo_plan_node()
5353
{
5454
AQOPlanNode*node= (AQOPlanNode*)newNode(sizeof(AQOPlanNode),
5555
T_ExtensibleNode);
56-
56+
Assert(node!=NULL);
5757
memcpy(node,&DefaultAQOPlanNode,sizeof(AQOPlanNode));
5858
node->rels=palloc(sizeof(RelSortOut));
5959
node->rels->hrels=NIL;
@@ -570,6 +570,7 @@ AQOnodeCopy(struct ExtensibleNode *enew, const struct ExtensibleNode *eold)
570570

571571
Assert(IsA(old,ExtensibleNode));
572572
Assert(strcmp(old->node.extnodename,AQO_PLAN_NODE)==0);
573+
Assert(new&&old);
573574

574575
/* Copy static fields in one command */
575576
memcpy(new,old,sizeof(AQOPlanNode));

‎postprocessing.c‎

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,7 @@ should_learn(PlanState *ps, AQOPlanNode *node, aqo_obj_stat *ctx,
320320
/* This node s*/
321321
if (aqo_show_details)
322322
elog(NOTICE,
323-
"[AQO] Learn on a plan node (%lu, %d), "
323+
"[AQO] Learn on a plan node ("UINT64_FORMAT", %d), "
324324
"predicted rows: %.0lf, updated prediction: %.0lf",
325325
query_context.query_hash,node->fss,predicted,*nrows);
326326

@@ -336,7 +336,7 @@ should_learn(PlanState *ps, AQOPlanNode *node, aqo_obj_stat *ctx,
336336
if (ctx->learn&&aqo_show_details&&
337337
fabs(*nrows-predicted) /predicted>0.2)
338338
elog(NOTICE,
339-
"[AQO] Learn on a finished plan node (%lu, %d), "
339+
"[AQO] Learn on a finished plan node ("UINT64_FORMAT", %d), "
340340
"predicted rows: %.0lf, updated prediction: %.0lf",
341341
query_context.query_hash,node->fss,predicted,*nrows);
342342

@@ -839,6 +839,7 @@ StoreToQueryEnv(QueryDesc *queryDesc)
839839
enr->md.reliddesc=InvalidOid;
840840
enr->md.tupdesc=NULL;
841841
enr->reldata=palloc0(qcsize);
842+
Assert(enr->reldata!=NULL);
842843
memcpy(enr->reldata,&query_context,qcsize);
843844

844845
if (newentry)
@@ -890,6 +891,7 @@ StorePlanInternals(QueryDesc *queryDesc)
890891
enr->md.reliddesc=InvalidOid;
891892
enr->md.tupdesc=NULL;
892893
enr->reldata=palloc0(sizeof(int));
894+
Assert(enr->reldata!=NULL);
893895
memcpy(enr->reldata,&njoins,sizeof(int));
894896

895897
if (newentry)
@@ -919,6 +921,7 @@ ExtractFromQueryEnv(QueryDesc *queryDesc)
919921
if (enr==NULL)
920922
return false;
921923

924+
Assert(enr->reldata!=NULL);
922925
memcpy(&query_context,enr->reldata,sizeof(QueryContextData));
923926

924927
return true;

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp