2 * Copyright (c) 2009 Teodor Sigaev <teodor@sigaev.ru>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the author nor the names of any co-contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
17 * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
18 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33 #include <access/heapam.h>
34 #include <catalog/namespace.h>
35 #include <catalog/pg_class.h>
36 #include <nodes/pg_list.h>
37 #include <optimizer/plancat.h>
38 #include <storage/bufmgr.h>
39 #include <utils/builtins.h>
40 #include <utils/guc.h>
41 #include <utils/lsyscache.h>
42 #include <utils/rel.h>
46 static int nIndexesOut = 0;
47 static Oid *indexesOut = NULL;
48 get_relation_info_hook_type prevHook = NULL;
49 static bool fix_empty_table = false;
51 static char *indexesOutStr = "";
54 indexesOutAssign(const char * newval, bool doit, GucSource source)
63 rawname = pstrdup(newval);
65 if (!SplitIdentifierString(rawname, ',', &namelist))
70 nOids = list_length(namelist);
71 newOids = malloc(sizeof(Oid) * (nOids+1));
73 elog(ERROR,"could not allocate %d bytes", (int)(sizeof(Oid) * (nOids+1)));
78 char *curname = (char *) lfirst(l);
79 Oid indexOid = RangeVarGetRelid(makeRangeVarFromNameList(stringToQualifiedNameList(curname)), true);
81 if (indexOid == InvalidOid)
83 #if PG_VERSION_NUM >= 90100
86 elog(WARNING,"'%s' does not exist", curname);
89 else if ( get_rel_relkind(indexOid) != RELKIND_INDEX )
91 #if PG_VERSION_NUM >= 90100
94 elog(WARNING,"'%s' is not an index", curname);
99 newOids[i++] = indexOid;
106 indexesOut = newOids;
122 #if PG_VERSION_NUM >= 90100
125 indexesOutCheck(char **newval, void **extra, GucSource source)
129 val = (char*)indexesOutAssign(*newval, false, source);
140 indexesOutAssignNew(const char *newval, void *extra)
142 indexesOutAssign(newval, true, PGC_S_USER /* doesn't matter */);
148 indexFilter(PlannerInfo *root, Oid relationObjectId, bool inhparent, RelOptInfo *rel) {
151 for(i=0;i<nIndexesOut;i++)
155 foreach(l, rel->indexlist)
157 IndexOptInfo *info = (IndexOptInfo*)lfirst(l);
159 if (indexesOut[i] == info->indexoid)
161 rel->indexlist = list_delete_ptr(rel->indexlist, info);
167 if (fix_empty_table && rel)
176 execPlantuner(PlannerInfo *root, Oid relationObjectId, bool inhparent, RelOptInfo *rel) {
179 relation = heap_open(relationObjectId, NoLock);
180 if (relation->rd_rel->relkind == RELKIND_RELATION)
182 if (fix_empty_table && RelationGetNumberOfBlocks(relation) == 0)
185 * estimate_rel_size() could be too pessimistic for particular
192 indexFilter(root, relationObjectId, inhparent, rel);
194 heap_close(relation, NoLock);
197 * Call next hook if it exists
200 prevHook(root, relationObjectId, inhparent, rel);
204 IndexFilterShow(void)
210 len = 1 /* \0 */ + nIndexesOut * (2 * NAMEDATALEN + 2 /* ', ' */ + 1 /* . */);
211 ptr = val = palloc(len);
214 for(i=0; i<nIndexesOut; i++)
216 char *relname = get_rel_name(indexesOut[i]);
217 Oid nspOid = get_rel_namespace(indexesOut[i]);
218 char *nspname = get_namespace_name(nspOid);
220 if ( relname == NULL || nspOid == InvalidOid || nspname == NULL )
223 ptr += snprintf(ptr, len - (ptr - val), "%s%s.%s",
236 DefineCustomStringVariable(
237 "plantuner.forbid_index",
238 "List of forbidden indexes",
239 "Listed indexes will not be used in queries",
244 #if PG_VERSION_NUM >= 90100
253 DefineCustomBoolVariable(
254 "plantuner.fix_empty_table",
255 "Sets to zero estimations for empty tables",
256 "Sets to zero estimations for empty or newly created tables",
258 #if PG_VERSION_NUM >= 80400
262 #if PG_VERSION_NUM >= 80400
264 #if PG_VERSION_NUM >= 90100
272 if (get_relation_info_hook != execPlantuner )
274 prevHook = get_relation_info_hook;
275 get_relation_info_hook = execPlantuner;