diff --git a/src/backend/optimizer/prep/prepkeyset.c b/src/backend/optimizer/prep/prepkeyset.c new file mode 100644 index 0000000000..a8301a8f94 --- /dev/null +++ b/src/backend/optimizer/prep/prepkeyset.c @@ -0,0 +1,213 @@ +/*------------------------------------------------------------------------- + * + * prepkeyset.c-- + * Special preperation for keyset queries. + * + * Copyright (c) 1994, Regents of the University of California + * + *------------------------------------------------------------------------- + */ +#include +#include + +#include "postgres.h" +#include "nodes/pg_list.h" +#include "nodes/parsenodes.h" +#include "utils/elog.h" + +#include "nodes/nodes.h" +#include "nodes/execnodes.h" +#include "nodes/plannodes.h" +#include "nodes/primnodes.h" +#include "nodes/relation.h" + +#include "catalog/pg_type.h" +#include "lib/stringinfo.h" +#include "optimizer/planmain.h" +/* + * Node_Copy-- + * a macro to simplify calling of copyObject on the specified field + */ +#define Node_Copy(from, newnode, field) newnode->field = copyObject(from->field) + +/***** DEBUG stuff +#define TABS {int i; printf("\n"); for (i = 0; igroupClause || + origNode->havingQual || + origNode->hasAggs || + origNode->utilityStmt || + origNode->unionClause || + origNode->unionall || + origNode->hasSubLinks || + origNode->commandType != CMD_SELECT) + return; + + /* Qualify single table query */ + + /* Qualify where clause */ + if ( ! inspectOrNode((Expr*)origNode->qual)) { + return; + } + + /* Copy essential elements into a union node */ + /* + elog(NOTICE, "OR_EXPR=%d, OP_EXPR=%d, AND_EXPR=%d", OR_EXPR, OP_EXPR, AND_EXPR); + elog(NOTICE, "T_List=%d, T_Expr=%d, T_Var=%d, T_Const=%d", T_List, T_Expr, T_Var, T_Const); + elog(NOTICE, "opType=%d", ((Expr*)origNode->qual)->opType); + */ + while (((Expr*)origNode->qual)->opType == OR_EXPR) { + Query *unionNode = makeNode(Query); + + /* Pull up Expr = */ + unionNode->qual = lsecond(((Expr*)origNode->qual)->args); + + /* Pull up balance of tree */ + origNode->qual = lfirst(((Expr*)origNode->qual)->args); + + /* + elog(NOTICE, "origNode: opType=%d, nodeTag=%d", ((Expr*)origNode->qual)->opType, nodeTag(origNode->qual)); + elog(NOTICE, "unionNode: opType=%d, nodeTag=%d", ((Expr*)unionNode->qual)->opType, nodeTag(unionNode->qual)); + */ + + unionNode->commandType = origNode->commandType; + unionNode->resultRelation = origNode->resultRelation; + unionNode->isPortal = origNode->isPortal; + unionNode->isBinary = origNode->isBinary; + + if (origNode->uniqueFlag) + unionNode->uniqueFlag = pstrdup(origNode->uniqueFlag); + + Node_Copy(origNode, unionNode, sortClause); + Node_Copy(origNode, unionNode, rtable); + Node_Copy(origNode, unionNode, targetList); + + origNode->unionClause = lappend(origNode->unionClause, unionNode); + } + return; +} + + + + +static int +inspectOrNode(Expr *expr) +{ + int fr = 0, sr = 0; + Expr *firstExpr, *secondExpr; + + if ( ! (expr && nodeTag(expr) == T_Expr && expr->opType == OR_EXPR)) + return 0; + + firstExpr = lfirst(expr->args); + secondExpr = lsecond(expr->args); + if (nodeTag(firstExpr) != T_Expr || nodeTag(secondExpr) != T_Expr) + return 0; + + if (firstExpr->opType == OR_EXPR) + fr = inspectOrNode(firstExpr); + else if (firstExpr->opType == OP_EXPR) /* Need to make sure it is last */ + fr = inspectOpNode(firstExpr); + else if (firstExpr->opType == AND_EXPR) /* Need to make sure it is last */ + fr = inspectAndNode(firstExpr); + + + if (secondExpr->opType == AND_EXPR) + sr = inspectAndNode(secondExpr); + else if (secondExpr->opType == OP_EXPR) + sr = inspectOpNode(secondExpr); + + return (fr && sr); +} + + +static int +inspectAndNode(Expr *expr) +{ + int fr = 0, sr = 0; + Expr *firstExpr, *secondExpr; + + if ( ! (expr && nodeTag(expr) == T_Expr && expr->opType == AND_EXPR)) + return 0; + + firstExpr = lfirst(expr->args); + secondExpr = lsecond(expr->args); + if (nodeTag(firstExpr) != T_Expr || nodeTag(secondExpr) != T_Expr) + return 0; + + if (firstExpr->opType == AND_EXPR) + fr = inspectAndNode(firstExpr); + else if (firstExpr->opType == OP_EXPR) + fr = inspectOpNode(firstExpr); + + if (secondExpr->opType == OP_EXPR) + sr = inspectOpNode(secondExpr); + + return (fr && sr); +} + + +static int +/****************************************************************** + * Return TRUE if T_Var = T_Const, else FALSE + * Actually it does not test for =. Need to do this! + ******************************************************************/ +inspectOpNode(Expr *expr) +{ + Expr *firstExpr, *secondExpr; + + if (nodeTag(expr) != T_Expr || expr->opType != OP_EXPR) + return 0; + + firstExpr = lfirst(expr->args); + secondExpr = lsecond(expr->args); + return (firstExpr && secondExpr && nodeTag(firstExpr) == T_Var && nodeTag(secondExpr) == T_Const); +}