|
| 1 | +/* |
| 2 | + * pg_execplan.c |
| 3 | + * |
| 4 | + */ |
| 5 | + |
| 6 | +#include"postgres.h" |
| 7 | + |
| 8 | +#include"access/printtup.h" |
| 9 | +#include"commands/extension.h" |
| 10 | +#include"commands/prepare.h" |
| 11 | +#include"executor/executor.h" |
| 12 | +#include"nodes/plannodes.h" |
| 13 | +#include"tcop/pquery.h" |
| 14 | +#include"tcop/utility.h" |
| 15 | +#include"utils/builtins.h" |
| 16 | +#include"utils/memutils.h" |
| 17 | +#include"utils/plancache.h" |
| 18 | +#include"utils/snapmgr.h" |
| 19 | + |
| 20 | + |
| 21 | +#defineEXPLAN_DEBUG_LEVEL 0 |
| 22 | + |
| 23 | +PG_MODULE_MAGIC; |
| 24 | + |
| 25 | +PG_FUNCTION_INFO_V1(pg_store_query_plan); |
| 26 | +PG_FUNCTION_INFO_V1(pg_exec_query_plan); |
| 27 | + |
| 28 | +void_PG_init(void); |
| 29 | + |
| 30 | +/* |
| 31 | + * Module load/unload callback |
| 32 | + */ |
| 33 | +void |
| 34 | +_PG_init(void) |
| 35 | +{ |
| 36 | +return; |
| 37 | +} |
| 38 | + |
| 39 | +Datum |
| 40 | +pg_store_query_plan(PG_FUNCTION_ARGS) |
| 41 | +{ |
| 42 | +char*query_string=TextDatumGetCString(PG_GETARG_DATUM(1)), |
| 43 | +*filename=TextDatumGetCString(PG_GETARG_DATUM(0)), |
| 44 | +*plan_string; |
| 45 | +intnstmts; |
| 46 | +FILE*fout; |
| 47 | +MemoryContextoldcontext; |
| 48 | +List*parsetree_list; |
| 49 | +RawStmt*parsetree; |
| 50 | +List*querytree_list, |
| 51 | +*plantree_list; |
| 52 | +QueryDesc*queryDesc; |
| 53 | +size_tstring_len; |
| 54 | + |
| 55 | +if (EXPLAN_DEBUG_LEVEL>0) |
| 56 | +elog(LOG,"Store into %s plan of the query %s.",filename,query_string); |
| 57 | + |
| 58 | +oldcontext=MemoryContextSwitchTo(MessageContext); |
| 59 | + |
| 60 | +parsetree_list=pg_parse_query(query_string); |
| 61 | +nstmts=list_length(parsetree_list); |
| 62 | +if (nstmts!=1) |
| 63 | +elog(ERROR,"Query contains %d elements, but must contain only one.",nstmts); |
| 64 | + |
| 65 | +parsetree= (RawStmt*)linitial(parsetree_list); |
| 66 | +querytree_list=pg_analyze_and_rewrite(parsetree,query_string,NULL,0); |
| 67 | +plantree_list=pg_plan_queries(querytree_list,CURSOR_OPT_PARALLEL_OK,NULL); |
| 68 | + |
| 69 | +queryDesc=CreateQueryDesc((PlannedStmt*)linitial(plantree_list), |
| 70 | +query_string, |
| 71 | +InvalidSnapshot, |
| 72 | +InvalidSnapshot, |
| 73 | +None_Receiver, |
| 74 | +0, |
| 75 | +0); |
| 76 | + |
| 77 | +if (EXPLAN_DEBUG_LEVEL>0) |
| 78 | +elog(INFO,"BEFORE writing %s ...",filename); |
| 79 | + |
| 80 | +fout=fopen(filename,"wb"); |
| 81 | +Assert(fout!=NULL); |
| 82 | +string_len=strlen(query_string); |
| 83 | +fwrite(&string_len,sizeof(size_t),1,fout); |
| 84 | +fwrite(query_string,sizeof(char),string_len,fout); |
| 85 | + |
| 86 | +plan_string=nodeToString(queryDesc->plannedstmt); |
| 87 | +string_len=strlen(plan_string); |
| 88 | +fwrite(&string_len,sizeof(size_t),1,fout); |
| 89 | +fwrite(plan_string,sizeof(char),string_len,fout); |
| 90 | + |
| 91 | +fclose(fout); |
| 92 | +MemoryContextSwitchTo(oldcontext); |
| 93 | +PG_RETURN_VOID(); |
| 94 | +} |
| 95 | + |
| 96 | +staticvoid |
| 97 | +LoadPlanFromFile(constchar*filename,char**query_string,char**plan_string) |
| 98 | +{ |
| 99 | +FILE*fin; |
| 100 | +size_tstring_len; |
| 101 | +intnelems; |
| 102 | + |
| 103 | +fin=fopen(filename,"rb"); |
| 104 | +Assert(fin!=NULL); |
| 105 | + |
| 106 | +nelems=fread(&string_len,sizeof(size_t),1,fin); |
| 107 | +Assert(nelems==1); |
| 108 | +*query_string=palloc0(string_len+1); |
| 109 | +nelems=fread(*query_string,sizeof(char),string_len,fin); |
| 110 | +Assert(nelems==string_len); |
| 111 | + |
| 112 | +nelems=fread(&string_len,sizeof(size_t),1,fin); |
| 113 | +Assert(nelems==1); |
| 114 | +*plan_string=palloc0(string_len+1); |
| 115 | +nelems=fread(*plan_string,sizeof(char),string_len,fin); |
| 116 | +Assert(nelems==string_len); |
| 117 | + |
| 118 | +fclose(fin); |
| 119 | + |
| 120 | +} |
| 121 | + |
| 122 | +Datum |
| 123 | +pg_exec_query_plan(PG_FUNCTION_ARGS) |
| 124 | +{ |
| 125 | +char*filename=TextDatumGetCString(PG_GETARG_DATUM(0)), |
| 126 | +*query_string=NULL, |
| 127 | +*plan_string=NULL; |
| 128 | +PlannedStmt*pstmt; |
| 129 | +ParamListInfoparamLI=NULL; |
| 130 | +CachedPlanSource*psrc; |
| 131 | +CachedPlan*cplan; |
| 132 | +Portalportal; |
| 133 | +DestReceiver*receiver; |
| 134 | +int16format=0; |
| 135 | +inteflags=0; |
| 136 | + |
| 137 | +LoadPlanFromFile(filename,&query_string,&plan_string); |
| 138 | +pstmt= (PlannedStmt*)stringToNode(plan_string); |
| 139 | + |
| 140 | +psrc=CreateCachedPlan(NULL,query_string,query_string); |
| 141 | +CompleteCachedPlan(psrc,NIL,NULL,NULL,0,NULL,NULL, |
| 142 | +CURSOR_OPT_GENERIC_PLAN, false); |
| 143 | +StorePreparedStatement(query_string,psrc, false); |
| 144 | +SetRemoteSubplan(psrc,pstmt); |
| 145 | +cplan=GetCachedPlan(psrc,paramLI, false); |
| 146 | + |
| 147 | +if (EXPLAN_DEBUG_LEVEL>0) |
| 148 | +elog(INFO,"query: %s\n",query_string); |
| 149 | +if (EXPLAN_DEBUG_LEVEL>1) |
| 150 | +elog(INFO,"\nplan: %s\n",plan_string); |
| 151 | + |
| 152 | +receiver=CreateDestReceiver(DestDebug); |
| 153 | +portal=CreateNewPortal(); |
| 154 | +portal->visible= false; |
| 155 | +PortalDefineQuery(portal, |
| 156 | +NULL, |
| 157 | +query_string, |
| 158 | +query_string, |
| 159 | +cplan->stmt_list, |
| 160 | +cplan); |
| 161 | +PortalStart(portal,paramLI,eflags,InvalidSnapshot); |
| 162 | +PortalSetResultFormat(portal,0,&format); |
| 163 | +(void)PortalRun(portal, |
| 164 | +FETCH_ALL, |
| 165 | + true, |
| 166 | +receiver, |
| 167 | +receiver, |
| 168 | +query_string); |
| 169 | +receiver->rDestroy(receiver); |
| 170 | +PortalDrop(portal, false); |
| 171 | +DropPreparedStatement(query_string, false); |
| 172 | + |
| 173 | +if (EXPLAN_DEBUG_LEVEL>0) |
| 174 | +elog(INFO,"query execution finished.\n"); |
| 175 | + |
| 176 | +PG_RETURN_VOID(); |
| 177 | +} |