* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
+ * notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
* 3. Neither the name of the author nor the names of any co-contributors
- * may be used to endorse or promote products derived from this software
- * without specific prior written permission.
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
* OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
#include <postgres.h>
#include <fmgr.h>
+#include <miscadmin.h>
#include <access/heapam.h>
+#include <access/xact.h>
#include <catalog/namespace.h>
#include <catalog/pg_class.h>
#include <nodes/pg_list.h>
#include <utils/guc.h>
#include <utils/lsyscache.h>
#include <utils/rel.h>
+#if PG_VERSION_NUM >= 100000
+#include <utils/regproc.h>
+#include <utils/varlena.h>
+#endif
PG_MODULE_MAGIC;
-static int nDisabledIndexes = 0;
+#if PG_VERSION_NUM >= 130000
+#define heap_open(r, l) table_open(r, l)
+#define heap_close(r, l) table_close(r, l)
+#endif
+
+static int nDisabledIndexes = 0;
static Oid *disabledIndexes = NULL;
static char *disableIndexesOutStr = "";
-static int nEnabledIndexes = 0;
+static int nEnabledIndexes = 0;
static Oid *enabledIndexes = NULL;
static char *enableIndexesOutStr = "";
+static int nOnlyIndexes = 0;
+static Oid *onlyIndexes = NULL;
+static char *onlyIndexesOutStr = "";
+
get_relation_info_hook_type prevHook = NULL;
static bool fix_empty_table = false;
+static bool plantuner_enable_inited = false;
+static bool plantuner_only_inited = false;
+static bool plantuner_disable_inited = false;
+
+typedef enum IndexListKind {
+ EnabledKind,
+ DisabledKind,
+ OnlyKind
+} IndexListKind;
static const char *
-indexesAssign(const char * newval, bool doit, GucSource source, bool isDisable)
+indexesAssign(const char * newval, bool doit, GucSource source,
+ IndexListKind kind)
{
- char *rawname;
- List *namelist;
- ListCell *l;
+ char *rawname;
+ List *namelist;
+ ListCell *l;
Oid *newOids = NULL;
int nOids = 0,
i = 0;
if (!SplitIdentifierString(rawname, ',', &namelist))
goto cleanup;
- if (doit)
+ /*
+ * follow work could be done only in normal processing because of
+ * accsess to system catalog
+ */
+ if (MyBackendId == InvalidBackendId || !IsUnderPostmaster ||
+ !IsTransactionState())
+ {
+ /* reset init state */
+ switch(kind)
+ {
+ case EnabledKind:
+ plantuner_enable_inited = false;
+ break;
+ case DisabledKind:
+ plantuner_disable_inited = false;
+ break;
+ case OnlyKind:
+ plantuner_only_inited = false;
+ break;
+ default:
+ elog(ERROR, "wrong kind");
+ }
+
+ return newval;
+ }
+
+ if (doit)
{
nOids = list_length(namelist);
newOids = malloc(sizeof(Oid) * (nOids+1));
if (!newOids)
- elog(ERROR,"could not allocate %d bytes", (int)(sizeof(Oid) * (nOids+1)));
+ elog(ERROR,"could not allocate %d bytes",
+ (int)(sizeof(Oid) * (nOids+1)));
+ }
+
+ switch(kind)
+ {
+ case EnabledKind:
+ plantuner_enable_inited = true;
+ break;
+ case DisabledKind:
+ plantuner_disable_inited = true;
+ break;
+ case OnlyKind:
+ plantuner_only_inited = true;
+ break;
+ default:
+ elog(ERROR, "wrong kind");
}
foreach(l, namelist)
{
- char *curname = (char *) lfirst(l);
- Oid indexOid = RangeVarGetRelid(makeRangeVarFromNameList(stringToQualifiedNameList(curname)),
+ char *curname = (char *) lfirst(l);
#if PG_VERSION_NUM >= 90200
- NoLock,
+ Oid indexOid = RangeVarGetRelid(
+ makeRangeVarFromNameList(stringToQualifiedNameList(curname)),
+ NoLock, true);
+#else
+ Oid indexOid = RangeVarGetRelid(
+ makeRangeVarFromNameList(stringToQualifiedNameList(curname)),
+ true);
#endif
- true);
if (indexOid == InvalidOid)
{
}
}
- if (doit)
+ if (doit)
{
- if (isDisable)
- {
- nDisabledIndexes = nOids;
- disabledIndexes = newOids;
- }
- else
+ switch(kind)
{
- nEnabledIndexes = nOids;
- enabledIndexes = newOids;
+ case EnabledKind:
+ nEnabledIndexes = i;
+ if (enabledIndexes)
+ free(enabledIndexes);
+ enabledIndexes = newOids;
+ break;
+ case DisabledKind:
+ nDisabledIndexes = i;
+ if (disabledIndexes)
+ free(disabledIndexes);
+ disabledIndexes = newOids;
+ break;
+ case OnlyKind:
+ nOnlyIndexes = i;
+ if (onlyIndexes)
+ free(onlyIndexes);
+ onlyIndexes = newOids;
+ break;
+ default:
+ elog(ERROR, "wrong kind");
}
}
static const char *
assignDisabledIndexes(const char * newval, bool doit, GucSource source)
{
- return indexesAssign(newval, doit, source, true);
+ return indexesAssign(newval, doit, source, DisabledKind);
}
static const char *
assignEnabledIndexes(const char * newval, bool doit, GucSource source)
{
- return indexesAssign(newval, doit, source, false);
+ return indexesAssign(newval, doit, source, EnabledKind);
+}
+
+static const char *
+assignOnlyIndexes(const char * newval, bool doit, GucSource source)
+{
+ return indexesAssign(newval, doit, source, OnlyKind);
+}
+
+static void
+lateInit()
+{
+ if (!plantuner_only_inited)
+ indexesAssign(onlyIndexesOutStr, true, PGC_S_USER, OnlyKind);
+ if (!plantuner_enable_inited)
+ indexesAssign(enableIndexesOutStr, true, PGC_S_USER, EnabledKind);
+ if (!plantuner_disable_inited)
+ indexesAssign(disableIndexesOutStr, true, PGC_S_USER, DisabledKind);
}
#if PG_VERSION_NUM >= 90100
+static bool
+checkOnlyIndexes(char **newval, void **extra, GucSource source)
+{
+ char *val;
+
+ val = (char*)indexesAssign(*newval, false, source, OnlyKind);
+
+ if (val)
+ {
+ *newval = val;
+ return true;
+ }
+
+ return false;
+}
+
static bool
checkDisabledIndexes(char **newval, void **extra, GucSource source)
{
char *val;
- val = (char*)indexesAssign(*newval, false, source, true);
+ val = (char*)indexesAssign(*newval, false, source, DisabledKind);
if (val)
{
{
char *val;
- val = (char*)indexesAssign(*newval, false, source, false);
+ val = (char*)indexesAssign(*newval, false, source, EnabledKind);
if (val)
{
assignEnabledIndexes(newval, true, PGC_S_USER /* doesn't matter */);
}
+static void
+assignOnlyIndexesNew(const char *newval, void *extra)
+{
+ assignOnlyIndexes(newval, true, PGC_S_USER /* doesn't matter */);
+}
+
#endif
static void
-indexFilter(PlannerInfo *root, Oid relationObjectId, bool inhparent, RelOptInfo *rel) {
+indexFilter(PlannerInfo *root, Oid relationObjectId, bool inhparent,
+ RelOptInfo *rel)
+{
int i;
- for(i=0;i<nDisabledIndexes;i++)
+ lateInit();
+
+ if (nOnlyIndexes > 0)
+ {
+ ListCell *l;
+
+restart1:
+ foreach(l, rel->indexlist)
+ {
+ IndexOptInfo *info = (IndexOptInfo*)lfirst(l);
+ bool remove = true;
+
+ for(i=0; remove && i<nOnlyIndexes; i++)
+ if (onlyIndexes[i] == info->indexoid)
+ remove = false;
+
+ if (remove)
+ {
+ rel->indexlist = list_delete_ptr(rel->indexlist, info);
+ goto restart1;
+ }
+ }
+
+ return;
+ }
+
+ for(i=0; i<nDisabledIndexes; i++)
{
ListCell *l;
}
}
}
-
- if (fix_empty_table && rel)
- {
-
-
- }
-
}
static void
-execPlantuner(PlannerInfo *root, Oid relationObjectId, bool inhparent, RelOptInfo *rel) {
- Relation relation;
+execPlantuner(PlannerInfo *root, Oid relationObjectId, bool inhparent,
+ RelOptInfo *rel)
+{
+ Relation relation;
relation = heap_open(relationObjectId, NoLock);
if (relation->rd_rel->relkind == RELKIND_RELATION)
* estimate_rel_size() could be too pessimistic for particular
* workload
*/
- rel->pages = 0.0;
+ rel->pages = 1.0;
rel->tuples = 0.0;
}
heap_close(relation, NoLock);
/*
- * Call next hook if it exists
+ * Call next hook if it exists
*/
if (prevHook)
prevHook(root, relationObjectId, inhparent, rel);
}
static const char*
-IndexFilterShow(Oid* indexes, int nIndexes)
+IndexFilterShow(Oid* indexes, int nIndexes)
{
- char *val, *ptr;
- int i,
+ char *val, *ptr;
+ int i,
len;
+ lateInit();
+
len = 1 /* \0 */ + nIndexes * (2 * NAMEDATALEN + 2 /* ', ' */ + 1 /* . */);
ptr = val = palloc(len);
- *ptr ='\0';
+ *ptr =(char)'\0';
for(i=0; i<nIndexes; i++)
{
- char *relname = get_rel_name(indexes[i]);
- Oid nspOid = get_rel_namespace(indexes[i]);
- char *nspname = get_namespace_name(nspOid);
+ char *relname = get_rel_name(indexes[i]);
+ Oid nspOid = get_rel_namespace(indexes[i]);
+ char *nspname = get_namespace_name(nspOid);
if ( relname == NULL || nspOid == InvalidOid || nspname == NULL )
continue;
}
static const char*
-disabledIndexFilterShow()
+disabledIndexFilterShow(void)
{
return IndexFilterShow(disabledIndexes, nDisabledIndexes);
}
static const char*
-enabledIndexFilterShow()
+enabledIndexFilterShow(void)
{
return IndexFilterShow(enabledIndexes, nEnabledIndexes);
}
+static const char*
+onlyIndexFilterShow(void)
+{
+ return IndexFilterShow(onlyIndexes, nOnlyIndexes);
+}
+
void _PG_init(void);
void
-_PG_init(void)
+_PG_init(void)
{
- DefineCustomStringVariable(
+ DefineCustomStringVariable(
"plantuner.forbid_index",
"List of forbidden indexes (deprecated)",
"Listed indexes will not be used in queries (deprecated, use plantuner.disable_index)",
disabledIndexFilterShow
);
- DefineCustomStringVariable(
+ DefineCustomStringVariable(
"plantuner.disable_index",
"List of disabled indexes",
"Listed indexes will not be used in queries",
disabledIndexFilterShow
);
- DefineCustomStringVariable(
+ DefineCustomStringVariable(
"plantuner.enable_index",
"List of enabled indexes (overload plantuner.disable_index)",
"Listed indexes which could be used in queries even they are listed in plantuner.disable_index",
enabledIndexFilterShow
);
- DefineCustomBoolVariable(
+ DefineCustomStringVariable(
+ "plantuner.only_index",
+ "List of explicitly enabled indexes (overload plantuner.disable_index and plantuner.enable_index)",
+ "Only indexes in this list are allowed",
+ &onlyIndexesOutStr,
+ "",
+ PGC_USERSET,
+ 0,
+#if PG_VERSION_NUM >= 90100
+ checkOnlyIndexes,
+ assignOnlyIndexesNew,
+#else
+ assignOnlyIndexes,
+#endif
+ onlyIndexFilterShow
+ );
+
+ DefineCustomBoolVariable(
"plantuner.fix_empty_table",
"Sets to zero estimations for empty tables",
"Sets to zero estimations for empty or newly created tables",
projects / plantuner.git / blobdiff