CREATE OR REPLACEFUNCTIONaqo_cardinality_error(controlledboolean)

RETURNS TABLE(numbigint, idbigint, fshashbigint, error float, nexecsbigint)

AS $$

IF (controlled) THEN

RETURN QUERY

row_number() OVER (ORDER BY (cerror, query_id, fs_hash)DESC)AS nn,

query_id, fs_hash, cerror, execs

FROM (

aq.queryidAS query_id,

aq.fsAS fs_hash,

cardinality_error_with_aqo[array_length(cardinality_error_with_aqo,1)]AS cerror,

executions_with_aqoAS execs

FROM aqo_queries aqJOIN aqo_query_stat aqs

ONaq.queryid=aqs.queryid

WHERE TRUE= ANY (SELECT unnest(cardinality_error_with_aqo)IS NOT NULL)

)AS q1

ORDER BY nnASC;

ELSE

RETURN QUERY

row_number() OVER (ORDER BY (cerror, query_id, fs_hash)DESC)AS nn,

query_id, fs_hash, cerror, execs

FROM (

aq.queryidAS query_id,

aq.fsAS fs_hash,

(SELECTAVG(t)FROM unnest(cardinality_error_without_aqo) t)AS cerror,

executions_without_aqoAS execs

FROM aqo_queries aqJOIN aqo_query_stat aqs

ONaq.queryid=aqs.queryid

WHERE TRUE= ANY (SELECT unnest(cardinality_error_without_aqo)IS NOT NULL)

)AS q1

ORDER BY (nn)ASC;

END IF;

END;

$$ LANGUAGE plpgsql;

AS'MODULE_PATHNAME','aqo_cardinality_error'

LANGUAGE C STRICT VOLATILE;

'Get cardinality error of queries the last time they were executed. Order queries according to an error value.';

PG_FUNCTION_INFO_V1(aqo_reset);

PG_FUNCTION_INFO_V1(aqo_cleanup);

PG_FUNCTION_INFO_V1(aqo_drop_class);

PG_FUNCTION_INFO_V1(aqo_cardinality_error);

PG_RETURN_INT32(cnt);

}

typedefenum {

AQE_NN=0,AQE_QUERYID,AQE_FS,AQE_CERROR,AQE_NEXECS,AQE_TOTAL_NCOLS

}ce_output_order;

aqo_cardinality_error(PG_FUNCTION_ARGS)

{

boolcontrolled=PG_GETARG_BOOL(0);

ReturnSetInfo*rsinfo= (ReturnSetInfo*)fcinfo->resultinfo;

TupleDesctupDesc;

MemoryContextper_query_ctx;

MemoryContextoldcontext;

Tuplestorestate*tupstore;

Datumvalues[AQE_TOTAL_NCOLS];

boolnulls[AQE_TOTAL_NCOLS];

HASH_SEQ_STATUShash_seq;

QueriesEntry*qentry;

StatEntry*sentry;

intcounter=0;

if (rsinfo==NULL|| !IsA(rsinfo,ReturnSetInfo))

ereport(ERROR,

(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),

errmsg("set-valued function called in context that cannot accept a set")));

if (!(rsinfo->allowedModes&SFRM_Materialize))

ereport(ERROR,

(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),

errmsg("materialize mode required, but it is not allowed in this context")));

per_query_ctx=rsinfo->econtext->ecxt_per_query_memory;

oldcontext=MemoryContextSwitchTo(per_query_ctx);

if (get_call_result_type(fcinfo,NULL,&tupDesc)!=TYPEFUNC_COMPOSITE)

elog(ERROR,"return type must be a row type");

Assert(tupDesc->natts==AQE_TOTAL_NCOLS);

tupstore=tuplestore_begin_heap(true, false,work_mem);

rsinfo->returnMode=SFRM_Materialize;

rsinfo->setResult=tupstore;

rsinfo->setDesc=tupDesc;

MemoryContextSwitchTo(oldcontext);

LWLockAcquire(&aqo_state->queries_lock,LW_SHARED);

LWLockAcquire(&aqo_state->stat_lock,LW_SHARED);

hash_seq_init(&hash_seq,queries_htab);

while ((qentry=hash_seq_search(&hash_seq))!=NULL)

{

boolfound;

double*ce;

int64nexecs;

intnvals;

memset(nulls,0,AQE_TOTAL_NCOLS*sizeof(nulls[0]));

sentry= (StatEntry*)hash_search(stat_htab,&qentry->queryid,

HASH_FIND,&found);

if (!found)

continue;

nvals=controlled ?sentry->cur_stat_slot_aqo :sentry->cur_stat_slot;

if (nvals==0)

continue;

nexecs=controlled ?sentry->execs_with_aqo :sentry->execs_without_aqo;

ce=controlled ?sentry->est_error_aqo :sentry->est_error;

values[AQE_NN]=Int32GetDatum(counter++);

values[AQE_QUERYID]=Int64GetDatum(qentry->queryid);

values[AQE_FS]=Int64GetDatum(qentry->fs);

values[AQE_NEXECS]=Int64GetDatum(nexecs);

values[AQE_CERROR]=Float8GetDatum(ce[nvals-1]);

tuplestore_putvalues(tupstore,tupDesc,values,nulls);

}

LWLockRelease(&aqo_state->stat_lock);

LWLockRelease(&aqo_state->queries_lock);

tuplestore_donestoring(tupstore);

return (Datum)0;

}