1cce7d176Sdrh /* 2b19a2bc6Sdrh ** 2001 September 15 3cce7d176Sdrh ** 4b19a2bc6Sdrh ** The author disclaims copyright to this source code. In place of 5b19a2bc6Sdrh ** a legal notice, here is a blessing: 6cce7d176Sdrh ** 7b19a2bc6Sdrh ** May you do good and not evil. 8b19a2bc6Sdrh ** May you find forgiveness for yourself and forgive others. 9b19a2bc6Sdrh ** May you share freely, never taking more than you give. 10cce7d176Sdrh ** 11cce7d176Sdrh ************************************************************************* 12cce7d176Sdrh ** This file contains C code routines that are called by the parser 13b19a2bc6Sdrh ** to handle SELECT statements in SQLite. 14cce7d176Sdrh ** 15*fbcd585fSdanielk1977 ** $Id: select.c,v 1.190 2004/06/15 02:44:19 danielk1977 Exp $ 16cce7d176Sdrh */ 17cce7d176Sdrh #include "sqliteInt.h" 18cce7d176Sdrh 19315555caSdrh 20cce7d176Sdrh /* 219bb61fe7Sdrh ** Allocate a new Select structure and return a pointer to that 229bb61fe7Sdrh ** structure. 23cce7d176Sdrh */ 244adee20fSdanielk1977 Select *sqlite3SelectNew( 25daffd0e5Sdrh ExprList *pEList, /* which columns to include in the result */ 26ad3cab52Sdrh SrcList *pSrc, /* the FROM clause -- which tables to scan */ 27daffd0e5Sdrh Expr *pWhere, /* the WHERE clause */ 28daffd0e5Sdrh ExprList *pGroupBy, /* the GROUP BY clause */ 29daffd0e5Sdrh Expr *pHaving, /* the HAVING clause */ 30daffd0e5Sdrh ExprList *pOrderBy, /* the ORDER BY clause */ 319bbca4c1Sdrh int isDistinct, /* true if the DISTINCT keyword is present */ 329bbca4c1Sdrh int nLimit, /* LIMIT value. -1 means not used */ 33ef0cae50Sdrh int nOffset /* OFFSET value. 0 means no offset */ 349bb61fe7Sdrh ){ 359bb61fe7Sdrh Select *pNew; 369bb61fe7Sdrh pNew = sqliteMalloc( sizeof(*pNew) ); 37daffd0e5Sdrh if( pNew==0 ){ 384adee20fSdanielk1977 sqlite3ExprListDelete(pEList); 394adee20fSdanielk1977 sqlite3SrcListDelete(pSrc); 404adee20fSdanielk1977 sqlite3ExprDelete(pWhere); 414adee20fSdanielk1977 sqlite3ExprListDelete(pGroupBy); 424adee20fSdanielk1977 sqlite3ExprDelete(pHaving); 434adee20fSdanielk1977 sqlite3ExprListDelete(pOrderBy); 44daffd0e5Sdrh }else{ 45b733d037Sdrh if( pEList==0 ){ 464adee20fSdanielk1977 pEList = sqlite3ExprListAppend(0, sqlite3Expr(TK_ALL,0,0,0), 0); 47b733d037Sdrh } 489bb61fe7Sdrh pNew->pEList = pEList; 499bb61fe7Sdrh pNew->pSrc = pSrc; 509bb61fe7Sdrh pNew->pWhere = pWhere; 519bb61fe7Sdrh pNew->pGroupBy = pGroupBy; 529bb61fe7Sdrh pNew->pHaving = pHaving; 539bb61fe7Sdrh pNew->pOrderBy = pOrderBy; 549bb61fe7Sdrh pNew->isDistinct = isDistinct; 5582c3d636Sdrh pNew->op = TK_SELECT; 569bbca4c1Sdrh pNew->nLimit = nLimit; 579bbca4c1Sdrh pNew->nOffset = nOffset; 587b58daeaSdrh pNew->iLimit = -1; 597b58daeaSdrh pNew->iOffset = -1; 60daffd0e5Sdrh } 619bb61fe7Sdrh return pNew; 629bb61fe7Sdrh } 639bb61fe7Sdrh 649bb61fe7Sdrh /* 6501f3f253Sdrh ** Given 1 to 3 identifiers preceeding the JOIN keyword, determine the 6601f3f253Sdrh ** type of join. Return an integer constant that expresses that type 6701f3f253Sdrh ** in terms of the following bit values: 6801f3f253Sdrh ** 6901f3f253Sdrh ** JT_INNER 7001f3f253Sdrh ** JT_OUTER 7101f3f253Sdrh ** JT_NATURAL 7201f3f253Sdrh ** JT_LEFT 7301f3f253Sdrh ** JT_RIGHT 7401f3f253Sdrh ** 7501f3f253Sdrh ** A full outer join is the combination of JT_LEFT and JT_RIGHT. 7601f3f253Sdrh ** 7701f3f253Sdrh ** If an illegal or unsupported join type is seen, then still return 7801f3f253Sdrh ** a join type, but put an error in the pParse structure. 7901f3f253Sdrh */ 804adee20fSdanielk1977 int sqlite3JoinType(Parse *pParse, Token *pA, Token *pB, Token *pC){ 8101f3f253Sdrh int jointype = 0; 8201f3f253Sdrh Token *apAll[3]; 8301f3f253Sdrh Token *p; 8401f3f253Sdrh static struct { 8501f3f253Sdrh const char *zKeyword; 8601f3f253Sdrh int nChar; 8701f3f253Sdrh int code; 8801f3f253Sdrh } keywords[] = { 8901f3f253Sdrh { "natural", 7, JT_NATURAL }, 90195e6967Sdrh { "left", 4, JT_LEFT|JT_OUTER }, 91195e6967Sdrh { "right", 5, JT_RIGHT|JT_OUTER }, 92195e6967Sdrh { "full", 4, JT_LEFT|JT_RIGHT|JT_OUTER }, 9301f3f253Sdrh { "outer", 5, JT_OUTER }, 9401f3f253Sdrh { "inner", 5, JT_INNER }, 9501f3f253Sdrh { "cross", 5, JT_INNER }, 9601f3f253Sdrh }; 9701f3f253Sdrh int i, j; 9801f3f253Sdrh apAll[0] = pA; 9901f3f253Sdrh apAll[1] = pB; 10001f3f253Sdrh apAll[2] = pC; 101195e6967Sdrh for(i=0; i<3 && apAll[i]; i++){ 10201f3f253Sdrh p = apAll[i]; 10301f3f253Sdrh for(j=0; j<sizeof(keywords)/sizeof(keywords[0]); j++){ 10401f3f253Sdrh if( p->n==keywords[j].nChar 1054adee20fSdanielk1977 && sqlite3StrNICmp(p->z, keywords[j].zKeyword, p->n)==0 ){ 10601f3f253Sdrh jointype |= keywords[j].code; 10701f3f253Sdrh break; 10801f3f253Sdrh } 10901f3f253Sdrh } 11001f3f253Sdrh if( j>=sizeof(keywords)/sizeof(keywords[0]) ){ 11101f3f253Sdrh jointype |= JT_ERROR; 11201f3f253Sdrh break; 11301f3f253Sdrh } 11401f3f253Sdrh } 115ad2d8307Sdrh if( 116ad2d8307Sdrh (jointype & (JT_INNER|JT_OUTER))==(JT_INNER|JT_OUTER) || 117195e6967Sdrh (jointype & JT_ERROR)!=0 118ad2d8307Sdrh ){ 11901f3f253Sdrh static Token dummy = { 0, 0 }; 12001f3f253Sdrh char *zSp1 = " ", *zSp2 = " "; 12101f3f253Sdrh if( pB==0 ){ pB = &dummy; zSp1 = 0; } 12201f3f253Sdrh if( pC==0 ){ pC = &dummy; zSp2 = 0; } 1234adee20fSdanielk1977 sqlite3SetNString(&pParse->zErrMsg, "unknown or unsupported join type: ", 0, 12401f3f253Sdrh pA->z, pA->n, zSp1, 1, pB->z, pB->n, zSp2, 1, pC->z, pC->n, 0); 12501f3f253Sdrh pParse->nErr++; 12601f3f253Sdrh jointype = JT_INNER; 127195e6967Sdrh }else if( jointype & JT_RIGHT ){ 1284adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 129da93d238Sdrh "RIGHT and FULL OUTER JOINs are not currently supported"); 130195e6967Sdrh jointype = JT_INNER; 13101f3f253Sdrh } 13201f3f253Sdrh return jointype; 13301f3f253Sdrh } 13401f3f253Sdrh 13501f3f253Sdrh /* 136ad2d8307Sdrh ** Return the index of a column in a table. Return -1 if the column 137ad2d8307Sdrh ** is not contained in the table. 138ad2d8307Sdrh */ 139ad2d8307Sdrh static int columnIndex(Table *pTab, const char *zCol){ 140ad2d8307Sdrh int i; 141ad2d8307Sdrh for(i=0; i<pTab->nCol; i++){ 1424adee20fSdanielk1977 if( sqlite3StrICmp(pTab->aCol[i].zName, zCol)==0 ) return i; 143ad2d8307Sdrh } 144ad2d8307Sdrh return -1; 145ad2d8307Sdrh } 146ad2d8307Sdrh 147ad2d8307Sdrh /* 148ad2d8307Sdrh ** Add a term to the WHERE expression in *ppExpr that requires the 149ad2d8307Sdrh ** zCol column to be equal in the two tables pTab1 and pTab2. 150ad2d8307Sdrh */ 151ad2d8307Sdrh static void addWhereTerm( 152ad2d8307Sdrh const char *zCol, /* Name of the column */ 153ad2d8307Sdrh const Table *pTab1, /* First table */ 154ad2d8307Sdrh const Table *pTab2, /* Second table */ 155ad2d8307Sdrh Expr **ppExpr /* Add the equality term to this expression */ 156ad2d8307Sdrh ){ 157ad2d8307Sdrh Token dummy; 158ad2d8307Sdrh Expr *pE1a, *pE1b, *pE1c; 159ad2d8307Sdrh Expr *pE2a, *pE2b, *pE2c; 160ad2d8307Sdrh Expr *pE; 161ad2d8307Sdrh 162ad2d8307Sdrh dummy.z = zCol; 163ad2d8307Sdrh dummy.n = strlen(zCol); 1644b59ab5eSdrh dummy.dyn = 0; 1654adee20fSdanielk1977 pE1a = sqlite3Expr(TK_ID, 0, 0, &dummy); 1664adee20fSdanielk1977 pE2a = sqlite3Expr(TK_ID, 0, 0, &dummy); 167ad2d8307Sdrh dummy.z = pTab1->zName; 168ad2d8307Sdrh dummy.n = strlen(dummy.z); 1694adee20fSdanielk1977 pE1b = sqlite3Expr(TK_ID, 0, 0, &dummy); 170ad2d8307Sdrh dummy.z = pTab2->zName; 171ad2d8307Sdrh dummy.n = strlen(dummy.z); 1724adee20fSdanielk1977 pE2b = sqlite3Expr(TK_ID, 0, 0, &dummy); 1734adee20fSdanielk1977 pE1c = sqlite3Expr(TK_DOT, pE1b, pE1a, 0); 1744adee20fSdanielk1977 pE2c = sqlite3Expr(TK_DOT, pE2b, pE2a, 0); 1754adee20fSdanielk1977 pE = sqlite3Expr(TK_EQ, pE1c, pE2c, 0); 1761f16230bSdrh ExprSetProperty(pE, EP_FromJoin); 177ad2d8307Sdrh if( *ppExpr ){ 1784adee20fSdanielk1977 *ppExpr = sqlite3Expr(TK_AND, *ppExpr, pE, 0); 179ad2d8307Sdrh }else{ 180ad2d8307Sdrh *ppExpr = pE; 181ad2d8307Sdrh } 182ad2d8307Sdrh } 183ad2d8307Sdrh 184ad2d8307Sdrh /* 1851f16230bSdrh ** Set the EP_FromJoin property on all terms of the given expression. 1861cc093c2Sdrh ** 187e78e8284Sdrh ** The EP_FromJoin property is used on terms of an expression to tell 1881cc093c2Sdrh ** the LEFT OUTER JOIN processing logic that this term is part of the 1891f16230bSdrh ** join restriction specified in the ON or USING clause and not a part 1901f16230bSdrh ** of the more general WHERE clause. These terms are moved over to the 1911f16230bSdrh ** WHERE clause during join processing but we need to remember that they 1921f16230bSdrh ** originated in the ON or USING clause. 1931cc093c2Sdrh */ 1941cc093c2Sdrh static void setJoinExpr(Expr *p){ 1951cc093c2Sdrh while( p ){ 1961f16230bSdrh ExprSetProperty(p, EP_FromJoin); 1971cc093c2Sdrh setJoinExpr(p->pLeft); 1981cc093c2Sdrh p = p->pRight; 1991cc093c2Sdrh } 2001cc093c2Sdrh } 2011cc093c2Sdrh 2021cc093c2Sdrh /* 203ad2d8307Sdrh ** This routine processes the join information for a SELECT statement. 204ad2d8307Sdrh ** ON and USING clauses are converted into extra terms of the WHERE clause. 205ad2d8307Sdrh ** NATURAL joins also create extra WHERE clause terms. 206ad2d8307Sdrh ** 207ad2d8307Sdrh ** This routine returns the number of errors encountered. 208ad2d8307Sdrh */ 209ad2d8307Sdrh static int sqliteProcessJoin(Parse *pParse, Select *p){ 210ad2d8307Sdrh SrcList *pSrc; 211ad2d8307Sdrh int i, j; 212ad2d8307Sdrh pSrc = p->pSrc; 213ad2d8307Sdrh for(i=0; i<pSrc->nSrc-1; i++){ 214ad2d8307Sdrh struct SrcList_item *pTerm = &pSrc->a[i]; 215ad2d8307Sdrh struct SrcList_item *pOther = &pSrc->a[i+1]; 216ad2d8307Sdrh 217ad2d8307Sdrh if( pTerm->pTab==0 || pOther->pTab==0 ) continue; 218ad2d8307Sdrh 219ad2d8307Sdrh /* When the NATURAL keyword is present, add WHERE clause terms for 220ad2d8307Sdrh ** every column that the two tables have in common. 221ad2d8307Sdrh */ 222ad2d8307Sdrh if( pTerm->jointype & JT_NATURAL ){ 223ad2d8307Sdrh Table *pTab; 224ad2d8307Sdrh if( pTerm->pOn || pTerm->pUsing ){ 2254adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "a NATURAL join may not have " 226ad2d8307Sdrh "an ON or USING clause", 0); 227ad2d8307Sdrh return 1; 228ad2d8307Sdrh } 229ad2d8307Sdrh pTab = pTerm->pTab; 230ad2d8307Sdrh for(j=0; j<pTab->nCol; j++){ 231ad2d8307Sdrh if( columnIndex(pOther->pTab, pTab->aCol[j].zName)>=0 ){ 232ad2d8307Sdrh addWhereTerm(pTab->aCol[j].zName, pTab, pOther->pTab, &p->pWhere); 233ad2d8307Sdrh } 234ad2d8307Sdrh } 235ad2d8307Sdrh } 236ad2d8307Sdrh 237ad2d8307Sdrh /* Disallow both ON and USING clauses in the same join 238ad2d8307Sdrh */ 239ad2d8307Sdrh if( pTerm->pOn && pTerm->pUsing ){ 2404adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot have both ON and USING " 241da93d238Sdrh "clauses in the same join"); 242ad2d8307Sdrh return 1; 243ad2d8307Sdrh } 244ad2d8307Sdrh 245ad2d8307Sdrh /* Add the ON clause to the end of the WHERE clause, connected by 246ad2d8307Sdrh ** and AND operator. 247ad2d8307Sdrh */ 248ad2d8307Sdrh if( pTerm->pOn ){ 2491cc093c2Sdrh setJoinExpr(pTerm->pOn); 250ad2d8307Sdrh if( p->pWhere==0 ){ 251ad2d8307Sdrh p->pWhere = pTerm->pOn; 252ad2d8307Sdrh }else{ 2534adee20fSdanielk1977 p->pWhere = sqlite3Expr(TK_AND, p->pWhere, pTerm->pOn, 0); 254ad2d8307Sdrh } 255ad2d8307Sdrh pTerm->pOn = 0; 256ad2d8307Sdrh } 257ad2d8307Sdrh 258ad2d8307Sdrh /* Create extra terms on the WHERE clause for each column named 259ad2d8307Sdrh ** in the USING clause. Example: If the two tables to be joined are 260ad2d8307Sdrh ** A and B and the USING clause names X, Y, and Z, then add this 261ad2d8307Sdrh ** to the WHERE clause: A.X=B.X AND A.Y=B.Y AND A.Z=B.Z 262ad2d8307Sdrh ** Report an error if any column mentioned in the USING clause is 263ad2d8307Sdrh ** not contained in both tables to be joined. 264ad2d8307Sdrh */ 265ad2d8307Sdrh if( pTerm->pUsing ){ 266ad2d8307Sdrh IdList *pList; 267ad2d8307Sdrh int j; 268ad2d8307Sdrh assert( i<pSrc->nSrc-1 ); 269ad2d8307Sdrh pList = pTerm->pUsing; 270ad2d8307Sdrh for(j=0; j<pList->nId; j++){ 271bf5cd97eSdrh if( columnIndex(pTerm->pTab, pList->a[j].zName)<0 || 272bf5cd97eSdrh columnIndex(pOther->pTab, pList->a[j].zName)<0 ){ 2734adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot join using column %s - column " 274da93d238Sdrh "not present in both tables", pList->a[j].zName); 275ad2d8307Sdrh return 1; 276ad2d8307Sdrh } 277bf5cd97eSdrh addWhereTerm(pList->a[j].zName, pTerm->pTab, pOther->pTab, &p->pWhere); 278ad2d8307Sdrh } 279ad2d8307Sdrh } 280ad2d8307Sdrh } 281ad2d8307Sdrh return 0; 282ad2d8307Sdrh } 283ad2d8307Sdrh 284ad2d8307Sdrh /* 2859bb61fe7Sdrh ** Delete the given Select structure and all of its substructures. 2869bb61fe7Sdrh */ 2874adee20fSdanielk1977 void sqlite3SelectDelete(Select *p){ 28882c3d636Sdrh if( p==0 ) return; 2894adee20fSdanielk1977 sqlite3ExprListDelete(p->pEList); 2904adee20fSdanielk1977 sqlite3SrcListDelete(p->pSrc); 2914adee20fSdanielk1977 sqlite3ExprDelete(p->pWhere); 2924adee20fSdanielk1977 sqlite3ExprListDelete(p->pGroupBy); 2934adee20fSdanielk1977 sqlite3ExprDelete(p->pHaving); 2944adee20fSdanielk1977 sqlite3ExprListDelete(p->pOrderBy); 2954adee20fSdanielk1977 sqlite3SelectDelete(p->pPrior); 296a76b5dfcSdrh sqliteFree(p->zSelect); 2979bb61fe7Sdrh sqliteFree(p); 2989bb61fe7Sdrh } 2999bb61fe7Sdrh 3009bb61fe7Sdrh /* 3012282792aSdrh ** Delete the aggregate information from the parse structure. 3022282792aSdrh */ 3031d83f052Sdrh static void sqliteAggregateInfoReset(Parse *pParse){ 3042282792aSdrh sqliteFree(pParse->aAgg); 3052282792aSdrh pParse->aAgg = 0; 3062282792aSdrh pParse->nAgg = 0; 3072282792aSdrh pParse->useAgg = 0; 3082282792aSdrh } 3092282792aSdrh 3102282792aSdrh /* 311c926afbcSdrh ** Insert code into "v" that will push the record on the top of the 312c926afbcSdrh ** stack into the sorter. 313c926afbcSdrh */ 314c926afbcSdrh static void pushOntoSorter(Parse *pParse, Vdbe *v, ExprList *pOrderBy){ 315c926afbcSdrh int i; 316c926afbcSdrh for(i=0; i<pOrderBy->nExpr; i++){ 3174adee20fSdanielk1977 sqlite3ExprCode(pParse, pOrderBy->a[i].pExpr); 318c926afbcSdrh } 319ffbc3088Sdrh sqlite3VdbeAddOp(v, OP_MakeKey, pOrderBy->nExpr, 0); 3204adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_SortPut, 0, 0); 321c926afbcSdrh } 322c926afbcSdrh 323c926afbcSdrh /* 3242282792aSdrh ** This routine generates the code for the inside of the inner loop 3252282792aSdrh ** of a SELECT. 32682c3d636Sdrh ** 32738640e15Sdrh ** If srcTab and nColumn are both zero, then the pEList expressions 32838640e15Sdrh ** are evaluated in order to get the data for this row. If nColumn>0 32938640e15Sdrh ** then data is pulled from srcTab and pEList is used only to get the 33038640e15Sdrh ** datatypes for each column. 3312282792aSdrh */ 3322282792aSdrh static int selectInnerLoop( 3332282792aSdrh Parse *pParse, /* The parser context */ 334df199a25Sdrh Select *p, /* The complete select statement being coded */ 3352282792aSdrh ExprList *pEList, /* List of values being extracted */ 33682c3d636Sdrh int srcTab, /* Pull data from this table */ 337967e8b73Sdrh int nColumn, /* Number of columns in the source table */ 3382282792aSdrh ExprList *pOrderBy, /* If not NULL, sort results using this key */ 3392282792aSdrh int distinct, /* If >=0, make sure results are distinct */ 3402282792aSdrh int eDest, /* How to dispose of the results */ 3412282792aSdrh int iParm, /* An argument to the disposal method */ 3422282792aSdrh int iContinue, /* Jump here to continue with next row */ 34384ac9d02Sdanielk1977 int iBreak, /* Jump here to break out of the inner loop */ 34484ac9d02Sdanielk1977 char *aff /* affinity string if eDest is SRT_Union */ 3452282792aSdrh ){ 3462282792aSdrh Vdbe *v = pParse->pVdbe; 3472282792aSdrh int i; 34838640e15Sdrh 349daffd0e5Sdrh if( v==0 ) return 0; 35038640e15Sdrh assert( pEList!=0 ); 3512282792aSdrh 352df199a25Sdrh /* If there was a LIMIT clause on the SELECT statement, then do the check 353df199a25Sdrh ** to see if this row should be output. 354df199a25Sdrh */ 355df199a25Sdrh if( pOrderBy==0 ){ 3567b58daeaSdrh if( p->iOffset>=0 ){ 3574adee20fSdanielk1977 int addr = sqlite3VdbeCurrentAddr(v); 3584adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemIncr, p->iOffset, addr+2); 3594adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, iContinue); 360df199a25Sdrh } 3617b58daeaSdrh if( p->iLimit>=0 ){ 3624adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemIncr, p->iLimit, iBreak); 363df199a25Sdrh } 364df199a25Sdrh } 365df199a25Sdrh 366967e8b73Sdrh /* Pull the requested columns. 3672282792aSdrh */ 36838640e15Sdrh if( nColumn>0 ){ 369967e8b73Sdrh for(i=0; i<nColumn; i++){ 3704adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Column, srcTab, i); 37182c3d636Sdrh } 37238640e15Sdrh }else{ 37338640e15Sdrh nColumn = pEList->nExpr; 37438640e15Sdrh for(i=0; i<pEList->nExpr; i++){ 3754adee20fSdanielk1977 sqlite3ExprCode(pParse, pEList->a[i].pExpr); 37638640e15Sdrh } 37782c3d636Sdrh } 3782282792aSdrh 379daffd0e5Sdrh /* If the DISTINCT keyword was present on the SELECT statement 380daffd0e5Sdrh ** and this row has been seen before, then do not make this row 381daffd0e5Sdrh ** part of the result. 3822282792aSdrh */ 383f5905aa7Sdrh if( distinct>=0 && pEList && pEList->nExpr>0 ){ 3840bd1f4eaSdrh #if NULL_ALWAYS_DISTINCT 3854adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_IsNull, -pEList->nExpr, sqlite3VdbeCurrentAddr(v)+7); 3860bd1f4eaSdrh #endif 387d3d39e93Sdrh /* Deliberately leave the affinity string off of the following OP_MakeKey */ 3884adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeKey, pEList->nExpr, 1); 3894adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Distinct, distinct, sqlite3VdbeCurrentAddr(v)+3); 3904adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, pEList->nExpr+1, 0); 3914adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, iContinue); 3920f69c1e3Sdanielk1977 sqlite3VdbeAddOp(v, OP_String8, 0, 0); 3934adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutStrKey, distinct, 0); 3942282792aSdrh } 39582c3d636Sdrh 396c926afbcSdrh switch( eDest ){ 39782c3d636Sdrh /* In this mode, write each query result to the key of the temporary 39882c3d636Sdrh ** table iParm. 3992282792aSdrh */ 400c926afbcSdrh case SRT_Union: { 4014adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeRecord, nColumn, NULL_ALWAYS_DISTINCT); 40284ac9d02Sdanielk1977 sqlite3VdbeChangeP3(v, -1, aff, P3_STATIC); 4030f69c1e3Sdanielk1977 sqlite3VdbeAddOp(v, OP_String8, 0, 0); 4044adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutStrKey, iParm, 0); 405c926afbcSdrh break; 406c926afbcSdrh } 40782c3d636Sdrh 4085974a30fSdrh /* Store the result as data using a unique key. 4095974a30fSdrh */ 410c926afbcSdrh case SRT_Table: 411c926afbcSdrh case SRT_TempTable: { 4124adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeRecord, nColumn, 0); 413c926afbcSdrh if( pOrderBy ){ 414c926afbcSdrh pushOntoSorter(pParse, v, pOrderBy); 415c926afbcSdrh }else{ 4164adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NewRecno, iParm, 0); 4174adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pull, 1, 0); 4184adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutIntKey, iParm, 0); 419c926afbcSdrh } 420c926afbcSdrh break; 421c926afbcSdrh } 4225974a30fSdrh 42382c3d636Sdrh /* Construct a record from the query result, but instead of 42482c3d636Sdrh ** saving that record, use it as a key to delete elements from 42582c3d636Sdrh ** the temporary table iParm. 42682c3d636Sdrh */ 427c926afbcSdrh case SRT_Except: { 4280bd1f4eaSdrh int addr; 4294adee20fSdanielk1977 addr = sqlite3VdbeAddOp(v, OP_MakeRecord, nColumn, NULL_ALWAYS_DISTINCT); 43084ac9d02Sdanielk1977 sqlite3VdbeChangeP3(v, -1, aff, P3_STATIC); 4314adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NotFound, iParm, addr+3); 4324adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Delete, iParm, 0); 433c926afbcSdrh break; 434c926afbcSdrh } 4352282792aSdrh 4362282792aSdrh /* If we are creating a set for an "expr IN (SELECT ...)" construct, 4372282792aSdrh ** then there should be a single item on the stack. Write this 4382282792aSdrh ** item into the set table with bogus data. 4392282792aSdrh */ 440c926afbcSdrh case SRT_Set: { 4414adee20fSdanielk1977 int addr1 = sqlite3VdbeCurrentAddr(v); 44252b36cabSdrh int addr2; 443e014a838Sdanielk1977 444967e8b73Sdrh assert( nColumn==1 ); 4454adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NotNull, -1, addr1+3); 4464adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 1, 0); 4474adee20fSdanielk1977 addr2 = sqlite3VdbeAddOp(v, OP_Goto, 0, 0); 448c926afbcSdrh if( pOrderBy ){ 449c926afbcSdrh pushOntoSorter(pParse, v, pOrderBy); 450c926afbcSdrh }else{ 451e014a838Sdanielk1977 char const *affStr; 452e014a838Sdanielk1977 char aff = (iParm>>16)&0xFF; 453e014a838Sdanielk1977 aff = sqlite3CompareAffinity(pEList->a[0].pExpr, aff); 454e014a838Sdanielk1977 affStr = sqlite3AffinityString(aff); 45584ac9d02Sdanielk1977 sqlite3VdbeOp3(v, OP_MakeKey, 1, 0, affStr, P3_STATIC); 4560f69c1e3Sdanielk1977 sqlite3VdbeAddOp(v, OP_String8, 0, 0); 457e014a838Sdanielk1977 sqlite3VdbeAddOp(v, OP_PutStrKey, (iParm&0x0000FFFF), 0); 458c926afbcSdrh } 4594adee20fSdanielk1977 sqlite3VdbeChangeP2(v, addr2, sqlite3VdbeCurrentAddr(v)); 460c926afbcSdrh break; 461c926afbcSdrh } 46282c3d636Sdrh 4632282792aSdrh /* If this is a scalar select that is part of an expression, then 4642282792aSdrh ** store the results in the appropriate memory cell and break out 4652282792aSdrh ** of the scan loop. 4662282792aSdrh */ 467c926afbcSdrh case SRT_Mem: { 468967e8b73Sdrh assert( nColumn==1 ); 469c926afbcSdrh if( pOrderBy ){ 470c926afbcSdrh pushOntoSorter(pParse, v, pOrderBy); 471c926afbcSdrh }else{ 4724adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemStore, iParm, 1); 4734adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, iBreak); 474c926afbcSdrh } 475c926afbcSdrh break; 476c926afbcSdrh } 4772282792aSdrh 478f46f905aSdrh /* Send the data to the callback function. 479f46f905aSdrh */ 480f46f905aSdrh case SRT_Callback: 481f46f905aSdrh case SRT_Sorter: { 482f46f905aSdrh if( pOrderBy ){ 483ce665cf6Sdrh sqlite3VdbeAddOp(v, OP_MakeRecord, nColumn, 0); 484f46f905aSdrh pushOntoSorter(pParse, v, pOrderBy); 485f46f905aSdrh }else{ 486f46f905aSdrh assert( eDest==SRT_Callback ); 4874adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Callback, nColumn, 0); 488f46f905aSdrh } 489f46f905aSdrh break; 490f46f905aSdrh } 491f46f905aSdrh 492142e30dfSdrh /* Invoke a subroutine to handle the results. The subroutine itself 493142e30dfSdrh ** is responsible for popping the results off of the stack. 494142e30dfSdrh */ 495142e30dfSdrh case SRT_Subroutine: { 496ac82fcf5Sdrh if( pOrderBy ){ 4974adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeRecord, nColumn, 0); 498ac82fcf5Sdrh pushOntoSorter(pParse, v, pOrderBy); 499ac82fcf5Sdrh }else{ 5004adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Gosub, 0, iParm); 501ac82fcf5Sdrh } 502142e30dfSdrh break; 503142e30dfSdrh } 504142e30dfSdrh 505d7489c39Sdrh /* Discard the results. This is used for SELECT statements inside 506d7489c39Sdrh ** the body of a TRIGGER. The purpose of such selects is to call 507d7489c39Sdrh ** user-defined functions that have side effects. We do not care 508d7489c39Sdrh ** about the actual results of the select. 509d7489c39Sdrh */ 510c926afbcSdrh default: { 511f46f905aSdrh assert( eDest==SRT_Discard ); 5124adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, nColumn, 0); 513c926afbcSdrh break; 514c926afbcSdrh } 515c926afbcSdrh } 51682c3d636Sdrh return 0; 51782c3d636Sdrh } 51882c3d636Sdrh 51982c3d636Sdrh /* 520d8bc7086Sdrh ** If the inner loop was generated using a non-null pOrderBy argument, 521d8bc7086Sdrh ** then the results were placed in a sorter. After the loop is terminated 522d8bc7086Sdrh ** we need to run the sorter and output the results. The following 523d8bc7086Sdrh ** routine generates the code needed to do that. 524d8bc7086Sdrh */ 525c926afbcSdrh static void generateSortTail( 526ffbc3088Sdrh Parse *pParse, /* The parsing context */ 527c926afbcSdrh Select *p, /* The SELECT statement */ 528c926afbcSdrh Vdbe *v, /* Generate code into this VDBE */ 529c926afbcSdrh int nColumn, /* Number of columns of data */ 530c926afbcSdrh int eDest, /* Write the sorted results here */ 531c926afbcSdrh int iParm /* Optional parameter associated with eDest */ 532c926afbcSdrh ){ 5334adee20fSdanielk1977 int end1 = sqlite3VdbeMakeLabel(v); 5344adee20fSdanielk1977 int end2 = sqlite3VdbeMakeLabel(v); 535d8bc7086Sdrh int addr; 536ffbc3088Sdrh KeyInfo *pInfo; 537ffbc3088Sdrh ExprList *pOrderBy; 538ffbc3088Sdrh int nCol, i; 539ffbc3088Sdrh sqlite *db = pParse->db; 540ffbc3088Sdrh 541f46f905aSdrh if( eDest==SRT_Sorter ) return; 542ffbc3088Sdrh pOrderBy = p->pOrderBy; 543ffbc3088Sdrh nCol = pOrderBy->nExpr; 544ffbc3088Sdrh pInfo = sqliteMalloc( sizeof(*pInfo) + nCol*(sizeof(CollSeq*)+1) ); 545ffbc3088Sdrh if( pInfo==0 ) return; 546ffbc3088Sdrh pInfo->aSortOrder = (char*)&pInfo->aColl[nCol]; 547ffbc3088Sdrh pInfo->nField = nCol; 548ffbc3088Sdrh for(i=0; i<nCol; i++){ 5490202b29eSdanielk1977 /* If a collation sequence was specified explicity, then it 5500202b29eSdanielk1977 ** is stored in pOrderBy->a[i].zName. Otherwise, use the default 5510202b29eSdanielk1977 ** collation type for the expression. 5520202b29eSdanielk1977 */ 5537cedc8d4Sdanielk1977 pInfo->aColl[i] = sqlite3ExprCollSeq(pParse, pOrderBy->a[i].pExpr); 5540202b29eSdanielk1977 if( !pInfo->aColl[i] ){ 555ffbc3088Sdrh pInfo->aColl[i] = db->pDfltColl; 5560202b29eSdanielk1977 } 557ffbc3088Sdrh pInfo->aSortOrder[i] = pOrderBy->a[i].sortOrder; 558ffbc3088Sdrh } 559ffbc3088Sdrh sqlite3VdbeOp3(v, OP_Sort, 0, 0, (char*)pInfo, P3_KEYINFO_HANDOFF); 5604adee20fSdanielk1977 addr = sqlite3VdbeAddOp(v, OP_SortNext, 0, end1); 5617b58daeaSdrh if( p->iOffset>=0 ){ 5624adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemIncr, p->iOffset, addr+4); 5634adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 1, 0); 5644adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, addr); 565df199a25Sdrh } 5667b58daeaSdrh if( p->iLimit>=0 ){ 5674adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemIncr, p->iLimit, end2); 568df199a25Sdrh } 569c926afbcSdrh switch( eDest ){ 570c926afbcSdrh case SRT_Table: 571c926afbcSdrh case SRT_TempTable: { 5724adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NewRecno, iParm, 0); 5734adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pull, 1, 0); 5744adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutIntKey, iParm, 0); 575c926afbcSdrh break; 576c926afbcSdrh } 577c926afbcSdrh case SRT_Set: { 578c926afbcSdrh assert( nColumn==1 ); 5794adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NotNull, -1, sqlite3VdbeCurrentAddr(v)+3); 5804adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 1, 0); 5814adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, sqlite3VdbeCurrentAddr(v)+3); 582d6861792Sdrh sqlite3VdbeOp3(v, OP_MakeKey, 1, 0, "n", P3_STATIC); 5830f69c1e3Sdanielk1977 sqlite3VdbeAddOp(v, OP_String8, 0, 0); 584e014a838Sdanielk1977 sqlite3VdbeAddOp(v, OP_PutStrKey, (iParm&0x0000FFFF), 0); 585c926afbcSdrh break; 586c926afbcSdrh } 587c926afbcSdrh case SRT_Mem: { 588c926afbcSdrh assert( nColumn==1 ); 5894adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemStore, iParm, 1); 5904adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, end1); 591c926afbcSdrh break; 592c926afbcSdrh } 593ce665cf6Sdrh case SRT_Callback: 594ac82fcf5Sdrh case SRT_Subroutine: { 595ac82fcf5Sdrh int i; 59684ac9d02Sdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, p->pEList->nExpr, 0); 59784ac9d02Sdanielk1977 sqlite3VdbeAddOp(v, OP_Pull, 1, 0); 598ac82fcf5Sdrh for(i=0; i<nColumn; i++){ 5994adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Column, -1-i, i); 600ac82fcf5Sdrh } 601ce665cf6Sdrh if( eDest==SRT_Callback ){ 602ce665cf6Sdrh sqlite3VdbeAddOp(v, OP_Callback, nColumn, 0); 603ce665cf6Sdrh }else{ 6044adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Gosub, 0, iParm); 605ce665cf6Sdrh } 60684ac9d02Sdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 2, 0); 607ac82fcf5Sdrh break; 608ac82fcf5Sdrh } 609c926afbcSdrh default: { 610f46f905aSdrh /* Do nothing */ 611c926afbcSdrh break; 612c926afbcSdrh } 613c926afbcSdrh } 6144adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, addr); 6154adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, end2); 6164adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 1, 0); 6174adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, end1); 6184adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_SortReset, 0, 0); 619d8bc7086Sdrh } 620d8bc7086Sdrh 621d8bc7086Sdrh /* 622517eb646Sdanielk1977 ** Return a pointer to a string containing the 'declaration type' of the 623517eb646Sdanielk1977 ** expression pExpr. The string may be treated as static by the caller. 624e78e8284Sdrh ** 625517eb646Sdanielk1977 ** If the declaration type is the exact datatype definition extracted from 626517eb646Sdanielk1977 ** the original CREATE TABLE statement if the expression is a column. 627e78e8284Sdrh ** 628517eb646Sdanielk1977 ** The declaration type for an expression is either TEXT, NUMERIC or ANY. 629517eb646Sdanielk1977 ** The declaration type for a ROWID field is INTEGER. 630fcb78a49Sdrh */ 631517eb646Sdanielk1977 static const char *columnType(Parse *pParse, SrcList *pTabList, Expr *pExpr){ 632517eb646Sdanielk1977 char const *zType = 0; 633517eb646Sdanielk1977 int j; 634517eb646Sdanielk1977 if( pExpr==0 ) return 0; 635517eb646Sdanielk1977 if( pExpr->op==TK_COLUMN && pTabList ){ 6366a3ea0e6Sdrh Table *pTab; 637517eb646Sdanielk1977 int iCol = pExpr->iColumn; 638517eb646Sdanielk1977 for(j=0; j<pTabList->nSrc && pTabList->a[j].iCursor!=pExpr->iTable; j++){} 6396a3ea0e6Sdrh assert( j<pTabList->nSrc ); 6406a3ea0e6Sdrh pTab = pTabList->a[j].pTab; 641fcb78a49Sdrh if( iCol<0 ) iCol = pTab->iPKey; 642fcb78a49Sdrh assert( iCol==-1 || (iCol>=0 && iCol<pTab->nCol) ); 643fcb78a49Sdrh if( iCol<0 ){ 644fcb78a49Sdrh zType = "INTEGER"; 645fcb78a49Sdrh }else{ 646fcb78a49Sdrh zType = pTab->aCol[iCol].zType; 647*fbcd585fSdanielk1977 if( !zType ) zType = ""; 648fcb78a49Sdrh } 649fcb78a49Sdrh }else{ 650517eb646Sdanielk1977 switch( sqlite3ExprType(pExpr) ){ 651736c22b8Sdrh case SQLITE_AFF_TEXT: zType = "TEXT"; break; 652736c22b8Sdrh case SQLITE_AFF_NUMERIC: zType = "NUMERIC"; break; 653736c22b8Sdrh default: zType = "ANY"; break; 654736c22b8Sdrh } 655fcb78a49Sdrh } 656517eb646Sdanielk1977 return zType; 657517eb646Sdanielk1977 } 658517eb646Sdanielk1977 659517eb646Sdanielk1977 /* 660517eb646Sdanielk1977 ** Generate code that will tell the VDBE the declaration types of columns 661517eb646Sdanielk1977 ** in the result set. 662517eb646Sdanielk1977 */ 663517eb646Sdanielk1977 static void generateColumnTypes( 664517eb646Sdanielk1977 Parse *pParse, /* Parser context */ 665517eb646Sdanielk1977 SrcList *pTabList, /* List of tables */ 666517eb646Sdanielk1977 ExprList *pEList /* Expressions defining the result set */ 667517eb646Sdanielk1977 ){ 668517eb646Sdanielk1977 Vdbe *v = pParse->pVdbe; 669517eb646Sdanielk1977 int i; 670517eb646Sdanielk1977 for(i=0; i<pEList->nExpr; i++){ 671517eb646Sdanielk1977 Expr *p = pEList->a[i].pExpr; 672517eb646Sdanielk1977 const char *zType = columnType(pParse, pTabList, p); 673517eb646Sdanielk1977 if( p==0 ) continue; 674*fbcd585fSdanielk1977 /* The vdbe must make it's own copy of the column-type, in case the 675*fbcd585fSdanielk1977 ** schema is reset before this virtual machine is deleted. 676*fbcd585fSdanielk1977 */ 677*fbcd585fSdanielk1977 sqlite3VdbeSetColName(v, i+pEList->nExpr, zType, strlen(zType)); 678fcb78a49Sdrh } 679fcb78a49Sdrh } 680fcb78a49Sdrh 681fcb78a49Sdrh /* 682fcb78a49Sdrh ** Generate code that will tell the VDBE the names of columns 683fcb78a49Sdrh ** in the result set. This information is used to provide the 684fcabd464Sdrh ** azCol[] values in the callback. 68582c3d636Sdrh */ 686832508b7Sdrh static void generateColumnNames( 687832508b7Sdrh Parse *pParse, /* Parser context */ 688ad3cab52Sdrh SrcList *pTabList, /* List of tables */ 689832508b7Sdrh ExprList *pEList /* Expressions defining the result set */ 690832508b7Sdrh ){ 691d8bc7086Sdrh Vdbe *v = pParse->pVdbe; 6926a3ea0e6Sdrh int i, j; 693fcabd464Sdrh sqlite *db = pParse->db; 694fcabd464Sdrh int fullNames, shortNames; 695fcabd464Sdrh 6963cf86063Sdanielk1977 /* If this is an EXPLAIN, skip this step */ 6973cf86063Sdanielk1977 if( pParse->explain ){ 69861de0d1bSdanielk1977 return; 6993cf86063Sdanielk1977 } 7003cf86063Sdanielk1977 701d6502758Sdrh assert( v!=0 ); 7026f8a503dSdanielk1977 if( pParse->colNamesSet || v==0 || sqlite3_malloc_failed ) return; 703d8bc7086Sdrh pParse->colNamesSet = 1; 704fcabd464Sdrh fullNames = (db->flags & SQLITE_FullColNames)!=0; 705fcabd464Sdrh shortNames = (db->flags & SQLITE_ShortColNames)!=0; 70622322fd4Sdanielk1977 sqlite3VdbeSetNumCols(v, pEList->nExpr); 70782c3d636Sdrh for(i=0; i<pEList->nExpr; i++){ 70882c3d636Sdrh Expr *p; 7095a38705eSdrh p = pEList->a[i].pExpr; 7105a38705eSdrh if( p==0 ) continue; 71182c3d636Sdrh if( pEList->a[i].zName ){ 71282c3d636Sdrh char *zName = pEList->a[i].zName; 713d8123366Sdanielk1977 sqlite3VdbeSetColName(v, i, zName, strlen(zName)); 71482c3d636Sdrh continue; 71582c3d636Sdrh } 716fa173a76Sdrh if( p->op==TK_COLUMN && pTabList ){ 7176a3ea0e6Sdrh Table *pTab; 71897665873Sdrh char *zCol; 7198aff1015Sdrh int iCol = p->iColumn; 7206a3ea0e6Sdrh for(j=0; j<pTabList->nSrc && pTabList->a[j].iCursor!=p->iTable; j++){} 7216a3ea0e6Sdrh assert( j<pTabList->nSrc ); 7226a3ea0e6Sdrh pTab = pTabList->a[j].pTab; 7238aff1015Sdrh if( iCol<0 ) iCol = pTab->iPKey; 72497665873Sdrh assert( iCol==-1 || (iCol>=0 && iCol<pTab->nCol) ); 725b1363206Sdrh if( iCol<0 ){ 726b1363206Sdrh zCol = "_ROWID_"; 727b1363206Sdrh }else{ 728b1363206Sdrh zCol = pTab->aCol[iCol].zName; 729b1363206Sdrh } 730fcabd464Sdrh if( !shortNames && !fullNames && p->span.z && p->span.z[0] ){ 7313cf86063Sdanielk1977 sqlite3VdbeSetColName(v, i, p->span.z, p->span.n); 732fcabd464Sdrh }else if( fullNames || (!shortNames && pTabList->nSrc>1) ){ 73382c3d636Sdrh char *zName = 0; 73482c3d636Sdrh char *zTab; 73582c3d636Sdrh 7366a3ea0e6Sdrh zTab = pTabList->a[j].zAlias; 737fcabd464Sdrh if( fullNames || zTab==0 ) zTab = pTab->zName; 7384adee20fSdanielk1977 sqlite3SetString(&zName, zTab, ".", zCol, 0); 7393cf86063Sdanielk1977 sqlite3VdbeSetColName(v, i, zName, P3_DYNAMIC); 74082c3d636Sdrh }else{ 7413cf86063Sdanielk1977 sqlite3VdbeSetColName(v, i, zCol, 0); 74282c3d636Sdrh } 7436977fea8Sdrh }else if( p->span.z && p->span.z[0] ){ 7443cf86063Sdanielk1977 sqlite3VdbeSetColName(v, i, p->span.z, p->span.n); 7453cf86063Sdanielk1977 /* sqlite3VdbeCompressSpace(v, addr); */ 7461bee3d7bSdrh }else{ 7471bee3d7bSdrh char zName[30]; 7481bee3d7bSdrh assert( p->op!=TK_COLUMN || pTabList==0 ); 7491bee3d7bSdrh sprintf(zName, "column%d", i+1); 7503cf86063Sdanielk1977 sqlite3VdbeSetColName(v, i, zName, 0); 75182c3d636Sdrh } 75282c3d636Sdrh } 75376d505baSdanielk1977 generateColumnTypes(pParse, pTabList, pEList); 7545080aaa7Sdrh } 75582c3d636Sdrh 75682c3d636Sdrh /* 757d8bc7086Sdrh ** Name of the connection operator, used for error messages. 758d8bc7086Sdrh */ 759d8bc7086Sdrh static const char *selectOpName(int id){ 760d8bc7086Sdrh char *z; 761d8bc7086Sdrh switch( id ){ 762d8bc7086Sdrh case TK_ALL: z = "UNION ALL"; break; 763d8bc7086Sdrh case TK_INTERSECT: z = "INTERSECT"; break; 764d8bc7086Sdrh case TK_EXCEPT: z = "EXCEPT"; break; 765d8bc7086Sdrh default: z = "UNION"; break; 766d8bc7086Sdrh } 767d8bc7086Sdrh return z; 768d8bc7086Sdrh } 769d8bc7086Sdrh 770d8bc7086Sdrh /* 771315555caSdrh ** Forward declaration 772315555caSdrh */ 773315555caSdrh static int fillInColumnList(Parse*, Select*); 774315555caSdrh 775315555caSdrh /* 77622f70c32Sdrh ** Given a SELECT statement, generate a Table structure that describes 77722f70c32Sdrh ** the result set of that SELECT. 77822f70c32Sdrh */ 7794adee20fSdanielk1977 Table *sqlite3ResultSetOfSelect(Parse *pParse, char *zTabName, Select *pSelect){ 78022f70c32Sdrh Table *pTab; 781b733d037Sdrh int i, j; 78222f70c32Sdrh ExprList *pEList; 783b733d037Sdrh Column *aCol; 78422f70c32Sdrh 78522f70c32Sdrh if( fillInColumnList(pParse, pSelect) ){ 78622f70c32Sdrh return 0; 78722f70c32Sdrh } 78822f70c32Sdrh pTab = sqliteMalloc( sizeof(Table) ); 78922f70c32Sdrh if( pTab==0 ){ 79022f70c32Sdrh return 0; 79122f70c32Sdrh } 79222f70c32Sdrh pTab->zName = zTabName ? sqliteStrDup(zTabName) : 0; 79322f70c32Sdrh pEList = pSelect->pEList; 79422f70c32Sdrh pTab->nCol = pEList->nExpr; 795417be79cSdrh assert( pTab->nCol>0 ); 796b733d037Sdrh pTab->aCol = aCol = sqliteMalloc( sizeof(pTab->aCol[0])*pTab->nCol ); 79722f70c32Sdrh for(i=0; i<pTab->nCol; i++){ 798517eb646Sdanielk1977 Expr *pR; 799517eb646Sdanielk1977 char *zType; 800517eb646Sdanielk1977 Expr *p = pEList->a[i].pExpr; 80122f70c32Sdrh if( pEList->a[i].zName ){ 802b733d037Sdrh aCol[i].zName = sqliteStrDup(pEList->a[i].zName); 803517eb646Sdanielk1977 }else if( p->op==TK_DOT 804b733d037Sdrh && (pR=p->pRight)!=0 && pR->token.z && pR->token.z[0] ){ 805b733d037Sdrh int cnt; 8064adee20fSdanielk1977 sqlite3SetNString(&aCol[i].zName, pR->token.z, pR->token.n, 0); 807b733d037Sdrh for(j=cnt=0; j<i; j++){ 8084adee20fSdanielk1977 if( sqlite3StrICmp(aCol[j].zName, aCol[i].zName)==0 ){ 809b733d037Sdrh int n; 810b733d037Sdrh char zBuf[30]; 811b733d037Sdrh sprintf(zBuf,"_%d",++cnt); 812b733d037Sdrh n = strlen(zBuf); 8134adee20fSdanielk1977 sqlite3SetNString(&aCol[i].zName, pR->token.z, pR->token.n, zBuf,n,0); 814b733d037Sdrh j = -1; 815b733d037Sdrh } 816b733d037Sdrh } 817b733d037Sdrh }else if( p->span.z && p->span.z[0] ){ 8184adee20fSdanielk1977 sqlite3SetNString(&pTab->aCol[i].zName, p->span.z, p->span.n, 0); 81922f70c32Sdrh }else{ 82022f70c32Sdrh char zBuf[30]; 82122f70c32Sdrh sprintf(zBuf, "column%d", i+1); 82222f70c32Sdrh pTab->aCol[i].zName = sqliteStrDup(zBuf); 82322f70c32Sdrh } 824e014a838Sdanielk1977 825517eb646Sdanielk1977 zType = sqliteStrDup(columnType(pParse, pSelect->pSrc ,p)); 826517eb646Sdanielk1977 pTab->aCol[i].zType = zType; 827517eb646Sdanielk1977 pTab->aCol[i].affinity = SQLITE_AFF_NUMERIC; 828517eb646Sdanielk1977 if( zType ){ 829517eb646Sdanielk1977 pTab->aCol[i].affinity = sqlite3AffinityType(zType, strlen(zType)); 830517eb646Sdanielk1977 } 8317cedc8d4Sdanielk1977 pTab->aCol[i].pColl = sqlite3ExprCollSeq(pParse, p); 8320202b29eSdanielk1977 if( !pTab->aCol[i].pColl ){ 8330202b29eSdanielk1977 pTab->aCol[i].pColl = pParse->db->pDfltColl; 8340202b29eSdanielk1977 } 83522f70c32Sdrh } 83622f70c32Sdrh pTab->iPKey = -1; 83722f70c32Sdrh return pTab; 83822f70c32Sdrh } 83922f70c32Sdrh 84022f70c32Sdrh /* 841ad2d8307Sdrh ** For the given SELECT statement, do three things. 842d8bc7086Sdrh ** 843ad3cab52Sdrh ** (1) Fill in the pTabList->a[].pTab fields in the SrcList that 84463eb5f29Sdrh ** defines the set of tables that should be scanned. For views, 84563eb5f29Sdrh ** fill pTabList->a[].pSelect with a copy of the SELECT statement 84663eb5f29Sdrh ** that implements the view. A copy is made of the view's SELECT 84763eb5f29Sdrh ** statement so that we can freely modify or delete that statement 84863eb5f29Sdrh ** without worrying about messing up the presistent representation 84963eb5f29Sdrh ** of the view. 850d8bc7086Sdrh ** 851ad2d8307Sdrh ** (2) Add terms to the WHERE clause to accomodate the NATURAL keyword 852ad2d8307Sdrh ** on joins and the ON and USING clause of joins. 853ad2d8307Sdrh ** 854ad2d8307Sdrh ** (3) Scan the list of columns in the result set (pEList) looking 85554473229Sdrh ** for instances of the "*" operator or the TABLE.* operator. 85654473229Sdrh ** If found, expand each "*" to be every column in every table 85754473229Sdrh ** and TABLE.* to be every column in TABLE. 858d8bc7086Sdrh ** 859d8bc7086Sdrh ** Return 0 on success. If there are problems, leave an error message 860d8bc7086Sdrh ** in pParse and return non-zero. 861d8bc7086Sdrh */ 862d8bc7086Sdrh static int fillInColumnList(Parse *pParse, Select *p){ 86354473229Sdrh int i, j, k, rc; 864ad3cab52Sdrh SrcList *pTabList; 865daffd0e5Sdrh ExprList *pEList; 866a76b5dfcSdrh Table *pTab; 867daffd0e5Sdrh 868daffd0e5Sdrh if( p==0 || p->pSrc==0 ) return 1; 869daffd0e5Sdrh pTabList = p->pSrc; 870daffd0e5Sdrh pEList = p->pEList; 871d8bc7086Sdrh 872d8bc7086Sdrh /* Look up every table in the table list. 873d8bc7086Sdrh */ 874ad3cab52Sdrh for(i=0; i<pTabList->nSrc; i++){ 875d8bc7086Sdrh if( pTabList->a[i].pTab ){ 876d8bc7086Sdrh /* This routine has run before! No need to continue */ 877d8bc7086Sdrh return 0; 878d8bc7086Sdrh } 879daffd0e5Sdrh if( pTabList->a[i].zName==0 ){ 88022f70c32Sdrh /* A sub-query in the FROM clause of a SELECT */ 88122f70c32Sdrh assert( pTabList->a[i].pSelect!=0 ); 882ad2d8307Sdrh if( pTabList->a[i].zAlias==0 ){ 883ad2d8307Sdrh char zFakeName[60]; 884ad2d8307Sdrh sprintf(zFakeName, "sqlite_subquery_%p_", 885ad2d8307Sdrh (void*)pTabList->a[i].pSelect); 8864adee20fSdanielk1977 sqlite3SetString(&pTabList->a[i].zAlias, zFakeName, 0); 887ad2d8307Sdrh } 88822f70c32Sdrh pTabList->a[i].pTab = pTab = 8894adee20fSdanielk1977 sqlite3ResultSetOfSelect(pParse, pTabList->a[i].zAlias, 89022f70c32Sdrh pTabList->a[i].pSelect); 89122f70c32Sdrh if( pTab==0 ){ 892daffd0e5Sdrh return 1; 893daffd0e5Sdrh } 8945cf590c1Sdrh /* The isTransient flag indicates that the Table structure has been 8955cf590c1Sdrh ** dynamically allocated and may be freed at any time. In other words, 8965cf590c1Sdrh ** pTab is not pointing to a persistent table structure that defines 8975cf590c1Sdrh ** part of the schema. */ 89822f70c32Sdrh pTab->isTransient = 1; 89922f70c32Sdrh }else{ 900a76b5dfcSdrh /* An ordinary table or view name in the FROM clause */ 901a76b5dfcSdrh pTabList->a[i].pTab = pTab = 9024adee20fSdanielk1977 sqlite3LocateTable(pParse,pTabList->a[i].zName,pTabList->a[i].zDatabase); 903a76b5dfcSdrh if( pTab==0 ){ 904d8bc7086Sdrh return 1; 905d8bc7086Sdrh } 906a76b5dfcSdrh if( pTab->pSelect ){ 90763eb5f29Sdrh /* We reach here if the named table is a really a view */ 9084adee20fSdanielk1977 if( sqlite3ViewGetColumnNames(pParse, pTab) ){ 909417be79cSdrh return 1; 910417be79cSdrh } 91163eb5f29Sdrh /* If pTabList->a[i].pSelect!=0 it means we are dealing with a 91263eb5f29Sdrh ** view within a view. The SELECT structure has already been 91363eb5f29Sdrh ** copied by the outer view so we can skip the copy step here 91463eb5f29Sdrh ** in the inner view. 91563eb5f29Sdrh */ 91663eb5f29Sdrh if( pTabList->a[i].pSelect==0 ){ 9174adee20fSdanielk1977 pTabList->a[i].pSelect = sqlite3SelectDup(pTab->pSelect); 918a76b5dfcSdrh } 919d8bc7086Sdrh } 92022f70c32Sdrh } 92163eb5f29Sdrh } 922d8bc7086Sdrh 923ad2d8307Sdrh /* Process NATURAL keywords, and ON and USING clauses of joins. 924ad2d8307Sdrh */ 925ad2d8307Sdrh if( sqliteProcessJoin(pParse, p) ) return 1; 926ad2d8307Sdrh 9277c917d19Sdrh /* For every "*" that occurs in the column list, insert the names of 92854473229Sdrh ** all columns in all tables. And for every TABLE.* insert the names 92954473229Sdrh ** of all columns in TABLE. The parser inserted a special expression 9307c917d19Sdrh ** with the TK_ALL operator for each "*" that it found in the column list. 9317c917d19Sdrh ** The following code just has to locate the TK_ALL expressions and expand 9327c917d19Sdrh ** each one to the list of all columns in all tables. 93354473229Sdrh ** 93454473229Sdrh ** The first loop just checks to see if there are any "*" operators 93554473229Sdrh ** that need expanding. 936d8bc7086Sdrh */ 9377c917d19Sdrh for(k=0; k<pEList->nExpr; k++){ 93854473229Sdrh Expr *pE = pEList->a[k].pExpr; 93954473229Sdrh if( pE->op==TK_ALL ) break; 94054473229Sdrh if( pE->op==TK_DOT && pE->pRight && pE->pRight->op==TK_ALL 94154473229Sdrh && pE->pLeft && pE->pLeft->op==TK_ID ) break; 9427c917d19Sdrh } 94354473229Sdrh rc = 0; 9447c917d19Sdrh if( k<pEList->nExpr ){ 94554473229Sdrh /* 94654473229Sdrh ** If we get here it means the result set contains one or more "*" 94754473229Sdrh ** operators that need to be expanded. Loop through each expression 94854473229Sdrh ** in the result set and expand them one by one. 94954473229Sdrh */ 9507c917d19Sdrh struct ExprList_item *a = pEList->a; 9517c917d19Sdrh ExprList *pNew = 0; 9527c917d19Sdrh for(k=0; k<pEList->nExpr; k++){ 95354473229Sdrh Expr *pE = a[k].pExpr; 95454473229Sdrh if( pE->op!=TK_ALL && 95554473229Sdrh (pE->op!=TK_DOT || pE->pRight==0 || pE->pRight->op!=TK_ALL) ){ 95654473229Sdrh /* This particular expression does not need to be expanded. 95754473229Sdrh */ 9584adee20fSdanielk1977 pNew = sqlite3ExprListAppend(pNew, a[k].pExpr, 0); 9597c917d19Sdrh pNew->a[pNew->nExpr-1].zName = a[k].zName; 9607c917d19Sdrh a[k].pExpr = 0; 9617c917d19Sdrh a[k].zName = 0; 9627c917d19Sdrh }else{ 96354473229Sdrh /* This expression is a "*" or a "TABLE.*" and needs to be 96454473229Sdrh ** expanded. */ 96554473229Sdrh int tableSeen = 0; /* Set to 1 when TABLE matches */ 96654473229Sdrh Token *pName; /* text of name of TABLE */ 96754473229Sdrh if( pE->op==TK_DOT && pE->pLeft ){ 96854473229Sdrh pName = &pE->pLeft->token; 96954473229Sdrh }else{ 97054473229Sdrh pName = 0; 97154473229Sdrh } 972ad3cab52Sdrh for(i=0; i<pTabList->nSrc; i++){ 973d8bc7086Sdrh Table *pTab = pTabList->a[i].pTab; 97454473229Sdrh char *zTabName = pTabList->a[i].zAlias; 97554473229Sdrh if( zTabName==0 || zTabName[0]==0 ){ 97654473229Sdrh zTabName = pTab->zName; 97754473229Sdrh } 97854473229Sdrh if( pName && (zTabName==0 || zTabName[0]==0 || 9794adee20fSdanielk1977 sqlite3StrNICmp(pName->z, zTabName, pName->n)!=0 || 980c754fa54Sdrh zTabName[pName->n]!=0) ){ 98154473229Sdrh continue; 98254473229Sdrh } 98354473229Sdrh tableSeen = 1; 984d8bc7086Sdrh for(j=0; j<pTab->nCol; j++){ 98522f70c32Sdrh Expr *pExpr, *pLeft, *pRight; 986ad2d8307Sdrh char *zName = pTab->aCol[j].zName; 987ad2d8307Sdrh 988ad2d8307Sdrh if( i>0 && (pTabList->a[i-1].jointype & JT_NATURAL)!=0 && 989ad2d8307Sdrh columnIndex(pTabList->a[i-1].pTab, zName)>=0 ){ 990ad2d8307Sdrh /* In a NATURAL join, omit the join columns from the 991ad2d8307Sdrh ** table on the right */ 992ad2d8307Sdrh continue; 993ad2d8307Sdrh } 9944adee20fSdanielk1977 if( i>0 && sqlite3IdListIndex(pTabList->a[i-1].pUsing, zName)>=0 ){ 995ad2d8307Sdrh /* In a join with a USING clause, omit columns in the 996ad2d8307Sdrh ** using clause from the table on the right. */ 997ad2d8307Sdrh continue; 998ad2d8307Sdrh } 9994adee20fSdanielk1977 pRight = sqlite3Expr(TK_ID, 0, 0, 0); 100022f70c32Sdrh if( pRight==0 ) break; 1001ad2d8307Sdrh pRight->token.z = zName; 1002ad2d8307Sdrh pRight->token.n = strlen(zName); 10034b59ab5eSdrh pRight->token.dyn = 0; 10044b59ab5eSdrh if( zTabName && pTabList->nSrc>1 ){ 10054adee20fSdanielk1977 pLeft = sqlite3Expr(TK_ID, 0, 0, 0); 10064adee20fSdanielk1977 pExpr = sqlite3Expr(TK_DOT, pLeft, pRight, 0); 100722f70c32Sdrh if( pExpr==0 ) break; 10084b59ab5eSdrh pLeft->token.z = zTabName; 10094b59ab5eSdrh pLeft->token.n = strlen(zTabName); 10104b59ab5eSdrh pLeft->token.dyn = 0; 10114adee20fSdanielk1977 sqlite3SetString((char**)&pExpr->span.z, zTabName, ".", zName, 0); 10126977fea8Sdrh pExpr->span.n = strlen(pExpr->span.z); 10136977fea8Sdrh pExpr->span.dyn = 1; 10146977fea8Sdrh pExpr->token.z = 0; 10156977fea8Sdrh pExpr->token.n = 0; 10166977fea8Sdrh pExpr->token.dyn = 0; 101722f70c32Sdrh }else{ 101822f70c32Sdrh pExpr = pRight; 10196977fea8Sdrh pExpr->span = pExpr->token; 102022f70c32Sdrh } 10214adee20fSdanielk1977 pNew = sqlite3ExprListAppend(pNew, pExpr, 0); 1022d8bc7086Sdrh } 1023d8bc7086Sdrh } 102454473229Sdrh if( !tableSeen ){ 1025f5db2d3eSdrh if( pName ){ 10264adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "no such table: %T", pName); 1027f5db2d3eSdrh }else{ 10284adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "no tables specified"); 1029f5db2d3eSdrh } 103054473229Sdrh rc = 1; 103154473229Sdrh } 10327c917d19Sdrh } 10337c917d19Sdrh } 10344adee20fSdanielk1977 sqlite3ExprListDelete(pEList); 10357c917d19Sdrh p->pEList = pNew; 1036d8bc7086Sdrh } 103754473229Sdrh return rc; 1038d8bc7086Sdrh } 1039d8bc7086Sdrh 1040d8bc7086Sdrh /* 1041ff78bd2fSdrh ** This routine recursively unlinks the Select.pSrc.a[].pTab pointers 1042ff78bd2fSdrh ** in a select structure. It just sets the pointers to NULL. This 1043ff78bd2fSdrh ** routine is recursive in the sense that if the Select.pSrc.a[].pSelect 1044ff78bd2fSdrh ** pointer is not NULL, this routine is called recursively on that pointer. 1045ff78bd2fSdrh ** 1046ff78bd2fSdrh ** This routine is called on the Select structure that defines a 1047ff78bd2fSdrh ** VIEW in order to undo any bindings to tables. This is necessary 1048ff78bd2fSdrh ** because those tables might be DROPed by a subsequent SQL command. 10495cf590c1Sdrh ** If the bindings are not removed, then the Select.pSrc->a[].pTab field 10505cf590c1Sdrh ** will be left pointing to a deallocated Table structure after the 10515cf590c1Sdrh ** DROP and a coredump will occur the next time the VIEW is used. 1052ff78bd2fSdrh */ 10534adee20fSdanielk1977 void sqlite3SelectUnbind(Select *p){ 1054ff78bd2fSdrh int i; 1055ad3cab52Sdrh SrcList *pSrc = p->pSrc; 1056ff78bd2fSdrh Table *pTab; 1057ff78bd2fSdrh if( p==0 ) return; 1058ad3cab52Sdrh for(i=0; i<pSrc->nSrc; i++){ 1059ff78bd2fSdrh if( (pTab = pSrc->a[i].pTab)!=0 ){ 1060ff78bd2fSdrh if( pTab->isTransient ){ 10614adee20fSdanielk1977 sqlite3DeleteTable(0, pTab); 1062ff78bd2fSdrh } 1063ff78bd2fSdrh pSrc->a[i].pTab = 0; 1064ff78bd2fSdrh if( pSrc->a[i].pSelect ){ 10654adee20fSdanielk1977 sqlite3SelectUnbind(pSrc->a[i].pSelect); 1066ff78bd2fSdrh } 1067ff78bd2fSdrh } 1068ff78bd2fSdrh } 1069ff78bd2fSdrh } 1070ff78bd2fSdrh 1071ff78bd2fSdrh /* 1072d8bc7086Sdrh ** This routine associates entries in an ORDER BY expression list with 1073d8bc7086Sdrh ** columns in a result. For each ORDER BY expression, the opcode of 1074967e8b73Sdrh ** the top-level node is changed to TK_COLUMN and the iColumn value of 1075d8bc7086Sdrh ** the top-level node is filled in with column number and the iTable 1076d8bc7086Sdrh ** value of the top-level node is filled with iTable parameter. 1077d8bc7086Sdrh ** 1078d8bc7086Sdrh ** If there are prior SELECT clauses, they are processed first. A match 1079d8bc7086Sdrh ** in an earlier SELECT takes precedence over a later SELECT. 1080d8bc7086Sdrh ** 1081d8bc7086Sdrh ** Any entry that does not match is flagged as an error. The number 1082d8bc7086Sdrh ** of errors is returned. 1083d8bc7086Sdrh */ 1084d8bc7086Sdrh static int matchOrderbyToColumn( 1085d8bc7086Sdrh Parse *pParse, /* A place to leave error messages */ 1086d8bc7086Sdrh Select *pSelect, /* Match to result columns of this SELECT */ 1087d8bc7086Sdrh ExprList *pOrderBy, /* The ORDER BY values to match against columns */ 1088e4de1febSdrh int iTable, /* Insert this value in iTable */ 1089d8bc7086Sdrh int mustComplete /* If TRUE all ORDER BYs must match */ 1090d8bc7086Sdrh ){ 1091d8bc7086Sdrh int nErr = 0; 1092d8bc7086Sdrh int i, j; 1093d8bc7086Sdrh ExprList *pEList; 1094d8bc7086Sdrh 1095daffd0e5Sdrh if( pSelect==0 || pOrderBy==0 ) return 1; 1096d8bc7086Sdrh if( mustComplete ){ 1097d8bc7086Sdrh for(i=0; i<pOrderBy->nExpr; i++){ pOrderBy->a[i].done = 0; } 1098d8bc7086Sdrh } 1099d8bc7086Sdrh if( fillInColumnList(pParse, pSelect) ){ 1100d8bc7086Sdrh return 1; 1101d8bc7086Sdrh } 1102d8bc7086Sdrh if( pSelect->pPrior ){ 110392cd52f5Sdrh if( matchOrderbyToColumn(pParse, pSelect->pPrior, pOrderBy, iTable, 0) ){ 110492cd52f5Sdrh return 1; 110592cd52f5Sdrh } 1106d8bc7086Sdrh } 1107d8bc7086Sdrh pEList = pSelect->pEList; 1108d8bc7086Sdrh for(i=0; i<pOrderBy->nExpr; i++){ 1109d8bc7086Sdrh Expr *pE = pOrderBy->a[i].pExpr; 1110e4de1febSdrh int iCol = -1; 1111d8bc7086Sdrh if( pOrderBy->a[i].done ) continue; 11124adee20fSdanielk1977 if( sqlite3ExprIsInteger(pE, &iCol) ){ 1113e4de1febSdrh if( iCol<=0 || iCol>pEList->nExpr ){ 11144adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 1115da93d238Sdrh "ORDER BY position %d should be between 1 and %d", 1116e4de1febSdrh iCol, pEList->nExpr); 1117e4de1febSdrh nErr++; 1118e4de1febSdrh break; 1119e4de1febSdrh } 1120fcb78a49Sdrh if( !mustComplete ) continue; 1121e4de1febSdrh iCol--; 1122e4de1febSdrh } 1123e4de1febSdrh for(j=0; iCol<0 && j<pEList->nExpr; j++){ 11244cfa7934Sdrh if( pEList->a[j].zName && (pE->op==TK_ID || pE->op==TK_STRING) ){ 1125a76b5dfcSdrh char *zName, *zLabel; 1126a76b5dfcSdrh zName = pEList->a[j].zName; 1127a76b5dfcSdrh assert( pE->token.z ); 1128a76b5dfcSdrh zLabel = sqliteStrNDup(pE->token.z, pE->token.n); 11294adee20fSdanielk1977 sqlite3Dequote(zLabel); 11304adee20fSdanielk1977 if( sqlite3StrICmp(zName, zLabel)==0 ){ 1131e4de1febSdrh iCol = j; 1132d8bc7086Sdrh } 11336e142f54Sdrh sqliteFree(zLabel); 1134d8bc7086Sdrh } 11354adee20fSdanielk1977 if( iCol<0 && sqlite3ExprCompare(pE, pEList->a[j].pExpr) ){ 1136e4de1febSdrh iCol = j; 1137d8bc7086Sdrh } 1138e4de1febSdrh } 1139e4de1febSdrh if( iCol>=0 ){ 1140967e8b73Sdrh pE->op = TK_COLUMN; 1141e4de1febSdrh pE->iColumn = iCol; 1142d8bc7086Sdrh pE->iTable = iTable; 1143d8bc7086Sdrh pOrderBy->a[i].done = 1; 1144d8bc7086Sdrh } 1145e4de1febSdrh if( iCol<0 && mustComplete ){ 11464adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 1147da93d238Sdrh "ORDER BY term number %d does not match any result column", i+1); 1148d8bc7086Sdrh nErr++; 1149d8bc7086Sdrh break; 1150d8bc7086Sdrh } 1151d8bc7086Sdrh } 1152d8bc7086Sdrh return nErr; 1153d8bc7086Sdrh } 1154d8bc7086Sdrh 1155d8bc7086Sdrh /* 1156d8bc7086Sdrh ** Get a VDBE for the given parser context. Create a new one if necessary. 1157d8bc7086Sdrh ** If an error occurs, return NULL and leave a message in pParse. 1158d8bc7086Sdrh */ 11594adee20fSdanielk1977 Vdbe *sqlite3GetVdbe(Parse *pParse){ 1160d8bc7086Sdrh Vdbe *v = pParse->pVdbe; 1161d8bc7086Sdrh if( v==0 ){ 11624adee20fSdanielk1977 v = pParse->pVdbe = sqlite3VdbeCreate(pParse->db); 1163d8bc7086Sdrh } 1164d8bc7086Sdrh return v; 1165d8bc7086Sdrh } 1166d8bc7086Sdrh 1167d3d39e93Sdrh #if 0 /***** This routine needs deleting *****/ 116884ac9d02Sdanielk1977 static void multiSelectAffinity(Select *p, char *zAff){ 116984ac9d02Sdanielk1977 int i; 117084ac9d02Sdanielk1977 117184ac9d02Sdanielk1977 if( !p ) return; 117284ac9d02Sdanielk1977 multiSelectAffinity(p->pPrior, zAff); 117384ac9d02Sdanielk1977 117484ac9d02Sdanielk1977 for(i=0; i<p->pEList->nExpr; i++){ 117584ac9d02Sdanielk1977 if( zAff[i]=='\0' ){ 117684ac9d02Sdanielk1977 zAff[i] = sqlite3ExprAffinity(p->pEList->a[i].pExpr); 117784ac9d02Sdanielk1977 } 117884ac9d02Sdanielk1977 } 117984ac9d02Sdanielk1977 } 1180d3d39e93Sdrh #endif 118184ac9d02Sdanielk1977 1182d8bc7086Sdrh /* 11837b58daeaSdrh ** Compute the iLimit and iOffset fields of the SELECT based on the 11847b58daeaSdrh ** nLimit and nOffset fields. nLimit and nOffset hold the integers 11857b58daeaSdrh ** that appear in the original SQL statement after the LIMIT and OFFSET 11867b58daeaSdrh ** keywords. Or that hold -1 and 0 if those keywords are omitted. 11877b58daeaSdrh ** iLimit and iOffset are the integer memory register numbers for 11887b58daeaSdrh ** counters used to compute the limit and offset. If there is no 11897b58daeaSdrh ** limit and/or offset, then iLimit and iOffset are negative. 11907b58daeaSdrh ** 11917b58daeaSdrh ** This routine changes the values if iLimit and iOffset only if 11927b58daeaSdrh ** a limit or offset is defined by nLimit and nOffset. iLimit and 11937b58daeaSdrh ** iOffset should have been preset to appropriate default values 11947b58daeaSdrh ** (usually but not always -1) prior to calling this routine. 11957b58daeaSdrh ** Only if nLimit>=0 or nOffset>0 do the limit registers get 11967b58daeaSdrh ** redefined. The UNION ALL operator uses this property to force 11977b58daeaSdrh ** the reuse of the same limit and offset registers across multiple 11987b58daeaSdrh ** SELECT statements. 11997b58daeaSdrh */ 12007b58daeaSdrh static void computeLimitRegisters(Parse *pParse, Select *p){ 12017b58daeaSdrh /* 12027b58daeaSdrh ** If the comparison is p->nLimit>0 then "LIMIT 0" shows 12037b58daeaSdrh ** all rows. It is the same as no limit. If the comparision is 12047b58daeaSdrh ** p->nLimit>=0 then "LIMIT 0" show no rows at all. 12057b58daeaSdrh ** "LIMIT -1" always shows all rows. There is some 12067b58daeaSdrh ** contraversy about what the correct behavior should be. 12077b58daeaSdrh ** The current implementation interprets "LIMIT 0" to mean 12087b58daeaSdrh ** no rows. 12097b58daeaSdrh */ 12107b58daeaSdrh if( p->nLimit>=0 ){ 12117b58daeaSdrh int iMem = pParse->nMem++; 12124adee20fSdanielk1977 Vdbe *v = sqlite3GetVdbe(pParse); 12137b58daeaSdrh if( v==0 ) return; 12144adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, -p->nLimit, 0); 12154adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemStore, iMem, 1); 12167b58daeaSdrh p->iLimit = iMem; 12177b58daeaSdrh } 12187b58daeaSdrh if( p->nOffset>0 ){ 12197b58daeaSdrh int iMem = pParse->nMem++; 12204adee20fSdanielk1977 Vdbe *v = sqlite3GetVdbe(pParse); 12217b58daeaSdrh if( v==0 ) return; 12224adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, -p->nOffset, 0); 12234adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemStore, iMem, 1); 12247b58daeaSdrh p->iOffset = iMem; 12257b58daeaSdrh } 12267b58daeaSdrh } 12277b58daeaSdrh 12287b58daeaSdrh /* 1229d3d39e93Sdrh ** Generate VDBE instructions that will open a transient table that 1230d3d39e93Sdrh ** will be used for an index or to store keyed results for a compound 1231d3d39e93Sdrh ** select. In other words, open a transient table that needs a 1232d3d39e93Sdrh ** KeyInfo structure. The number of columns in the KeyInfo is determined 1233d3d39e93Sdrh ** by the result set of the SELECT statement in the second argument. 1234d3d39e93Sdrh ** 1235dc1bdc4fSdanielk1977 ** Specifically, this routine is called to open an index table for 1236dc1bdc4fSdanielk1977 ** DISTINCT, UNION, INTERSECT and EXCEPT select statements (but not 1237dc1bdc4fSdanielk1977 ** UNION ALL). 1238dc1bdc4fSdanielk1977 ** 1239d3d39e93Sdrh ** Make the new table a KeyAsData table if keyAsData is true. 1240dc1bdc4fSdanielk1977 ** 1241dc1bdc4fSdanielk1977 ** The value returned is the address of the OP_OpenTemp instruction. 1242d3d39e93Sdrh */ 1243dc1bdc4fSdanielk1977 static int openTempIndex(Parse *pParse, Select *p, int iTab, int keyAsData){ 1244d3d39e93Sdrh KeyInfo *pKeyInfo; 1245736c22b8Sdrh int nColumn; 1246d3d39e93Sdrh sqlite *db = pParse->db; 1247d3d39e93Sdrh int i; 1248d3d39e93Sdrh Vdbe *v = pParse->pVdbe; 1249dc1bdc4fSdanielk1977 int addr; 1250d3d39e93Sdrh 1251736c22b8Sdrh if( fillInColumnList(pParse, p) ){ 1252dc1bdc4fSdanielk1977 return 0; 1253736c22b8Sdrh } 1254736c22b8Sdrh nColumn = p->pEList->nExpr; 1255d3d39e93Sdrh pKeyInfo = sqliteMalloc( sizeof(*pKeyInfo)+nColumn*sizeof(CollSeq*) ); 1256dc1bdc4fSdanielk1977 if( pKeyInfo==0 ) return 0; 1257dc1bdc4fSdanielk1977 pKeyInfo->enc = pParse->db->enc; 1258d3d39e93Sdrh pKeyInfo->nField = nColumn; 1259d3d39e93Sdrh for(i=0; i<nColumn; i++){ 1260dc1bdc4fSdanielk1977 pKeyInfo->aColl[i] = sqlite3ExprCollSeq(pParse, p->pEList->a[i].pExpr); 1261dc1bdc4fSdanielk1977 if( !pKeyInfo->aColl[i] ){ 1262d3d39e93Sdrh pKeyInfo->aColl[i] = db->pDfltColl; 1263d3d39e93Sdrh } 1264dc1bdc4fSdanielk1977 } 1265dc1bdc4fSdanielk1977 addr = sqlite3VdbeOp3(v, OP_OpenTemp, iTab, 0, 1266dc1bdc4fSdanielk1977 (char*)pKeyInfo, P3_KEYINFO_HANDOFF); 1267d3d39e93Sdrh if( keyAsData ){ 1268d3d39e93Sdrh sqlite3VdbeAddOp(v, OP_KeyAsData, iTab, 1); 1269d3d39e93Sdrh } 1270dc1bdc4fSdanielk1977 return addr; 1271dc1bdc4fSdanielk1977 } 1272dc1bdc4fSdanielk1977 1273dc1bdc4fSdanielk1977 static int multiSelectOpenTempAddr(Select *p, int addr, IdList **ppOpenTemp){ 1274dc1bdc4fSdanielk1977 if( !p->ppOpenTemp ){ 1275dc1bdc4fSdanielk1977 *ppOpenTemp = sqlite3IdListAppend(0, 0); 1276dc1bdc4fSdanielk1977 p->ppOpenTemp = ppOpenTemp; 1277dc1bdc4fSdanielk1977 }else{ 1278dc1bdc4fSdanielk1977 *p->ppOpenTemp = sqlite3IdListAppend(*p->ppOpenTemp, 0); 1279dc1bdc4fSdanielk1977 } 1280dc1bdc4fSdanielk1977 if( !(*p->ppOpenTemp) ){ 1281dc1bdc4fSdanielk1977 return SQLITE_NOMEM; 1282dc1bdc4fSdanielk1977 } 1283dc1bdc4fSdanielk1977 (*p->ppOpenTemp)->a[(*p->ppOpenTemp)->nId-1].idx = addr; 1284dc1bdc4fSdanielk1977 return SQLITE_OK; 1285dc1bdc4fSdanielk1977 } 1286dc1bdc4fSdanielk1977 1287dc1bdc4fSdanielk1977 static CollSeq *multiSelectCollSeq(Parse *pParse, Select *p, int iCol){ 1288dc1bdc4fSdanielk1977 CollSeq *pRet = 0; 1289dc1bdc4fSdanielk1977 if( p->pPrior ){ 1290dc1bdc4fSdanielk1977 pRet = multiSelectCollSeq(pParse, p->pPrior, iCol); 1291dc1bdc4fSdanielk1977 } 1292dc1bdc4fSdanielk1977 if( !pRet ){ 1293dc1bdc4fSdanielk1977 pRet = sqlite3ExprCollSeq(pParse, p->pEList->a[iCol].pExpr); 1294dc1bdc4fSdanielk1977 } 1295dc1bdc4fSdanielk1977 return pRet; 1296d3d39e93Sdrh } 1297d3d39e93Sdrh 1298d3d39e93Sdrh /* 129982c3d636Sdrh ** This routine is called to process a query that is really the union 130082c3d636Sdrh ** or intersection of two or more separate queries. 1301c926afbcSdrh ** 1302e78e8284Sdrh ** "p" points to the right-most of the two queries. the query on the 1303e78e8284Sdrh ** left is p->pPrior. The left query could also be a compound query 1304e78e8284Sdrh ** in which case this routine will be called recursively. 1305e78e8284Sdrh ** 1306e78e8284Sdrh ** The results of the total query are to be written into a destination 1307e78e8284Sdrh ** of type eDest with parameter iParm. 1308e78e8284Sdrh ** 1309e78e8284Sdrh ** Example 1: Consider a three-way compound SQL statement. 1310e78e8284Sdrh ** 1311e78e8284Sdrh ** SELECT a FROM t1 UNION SELECT b FROM t2 UNION SELECT c FROM t3 1312e78e8284Sdrh ** 1313e78e8284Sdrh ** This statement is parsed up as follows: 1314e78e8284Sdrh ** 1315e78e8284Sdrh ** SELECT c FROM t3 1316e78e8284Sdrh ** | 1317e78e8284Sdrh ** `-----> SELECT b FROM t2 1318e78e8284Sdrh ** | 13194b11c6d3Sjplyon ** `------> SELECT a FROM t1 1320e78e8284Sdrh ** 1321e78e8284Sdrh ** The arrows in the diagram above represent the Select.pPrior pointer. 1322e78e8284Sdrh ** So if this routine is called with p equal to the t3 query, then 1323e78e8284Sdrh ** pPrior will be the t2 query. p->op will be TK_UNION in this case. 1324e78e8284Sdrh ** 1325e78e8284Sdrh ** Notice that because of the way SQLite parses compound SELECTs, the 1326e78e8284Sdrh ** individual selects always group from left to right. 132782c3d636Sdrh */ 132884ac9d02Sdanielk1977 static int multiSelect( 132984ac9d02Sdanielk1977 Parse *pParse, 133084ac9d02Sdanielk1977 Select *p, 133184ac9d02Sdanielk1977 int eDest, 133284ac9d02Sdanielk1977 int iParm, 133384ac9d02Sdanielk1977 char *aff /* If eDest is SRT_Union, the affinity string */ 133484ac9d02Sdanielk1977 ){ 133584ac9d02Sdanielk1977 int rc = SQLITE_OK; /* Success code from a subroutine */ 133610e5e3cfSdrh Select *pPrior; /* Another SELECT immediately to our left */ 133710e5e3cfSdrh Vdbe *v; /* Generate code to this VDBE */ 1338dc1bdc4fSdanielk1977 IdList *pOpenTemp = 0; 133982c3d636Sdrh 13407b58daeaSdrh /* Make sure there is no ORDER BY or LIMIT clause on prior SELECTs. Only 13417b58daeaSdrh ** the last SELECT in the series may have an ORDER BY or LIMIT. 134282c3d636Sdrh */ 134384ac9d02Sdanielk1977 if( p==0 || p->pPrior==0 ){ 134484ac9d02Sdanielk1977 rc = 1; 134584ac9d02Sdanielk1977 goto multi_select_end; 134684ac9d02Sdanielk1977 } 1347d8bc7086Sdrh pPrior = p->pPrior; 1348d8bc7086Sdrh if( pPrior->pOrderBy ){ 13494adee20fSdanielk1977 sqlite3ErrorMsg(pParse,"ORDER BY clause should come after %s not before", 1350da93d238Sdrh selectOpName(p->op)); 135184ac9d02Sdanielk1977 rc = 1; 135284ac9d02Sdanielk1977 goto multi_select_end; 135382c3d636Sdrh } 13547b58daeaSdrh if( pPrior->nLimit>=0 || pPrior->nOffset>0 ){ 13554adee20fSdanielk1977 sqlite3ErrorMsg(pParse,"LIMIT clause should come after %s not before", 13567b58daeaSdrh selectOpName(p->op)); 135784ac9d02Sdanielk1977 rc = 1; 135884ac9d02Sdanielk1977 goto multi_select_end; 13597b58daeaSdrh } 136082c3d636Sdrh 1361d8bc7086Sdrh /* Make sure we have a valid query engine. If not, create a new one. 1362d8bc7086Sdrh */ 13634adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 136484ac9d02Sdanielk1977 if( v==0 ){ 136584ac9d02Sdanielk1977 rc = 1; 136684ac9d02Sdanielk1977 goto multi_select_end; 136784ac9d02Sdanielk1977 } 1368d8bc7086Sdrh 13691cc3d75fSdrh /* Create the destination temporary table if necessary 13701cc3d75fSdrh */ 13711cc3d75fSdrh if( eDest==SRT_TempTable ){ 1372b4964b72Sdanielk1977 assert( p->pEList ); 13734adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_OpenTemp, iParm, 0); 1374b4964b72Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, iParm, p->pEList->nExpr); 13751cc3d75fSdrh eDest = SRT_Table; 13761cc3d75fSdrh } 13771cc3d75fSdrh 1378f46f905aSdrh /* Generate code for the left and right SELECT statements. 1379d8bc7086Sdrh */ 138082c3d636Sdrh switch( p->op ){ 1381f46f905aSdrh case TK_ALL: { 1382f46f905aSdrh if( p->pOrderBy==0 ){ 13837b58daeaSdrh pPrior->nLimit = p->nLimit; 13847b58daeaSdrh pPrior->nOffset = p->nOffset; 1385dc1bdc4fSdanielk1977 pPrior->ppOpenTemp = p->ppOpenTemp; 138684ac9d02Sdanielk1977 rc = sqlite3Select(pParse, pPrior, eDest, iParm, 0, 0, 0, aff); 138784ac9d02Sdanielk1977 if( rc ){ 138884ac9d02Sdanielk1977 goto multi_select_end; 138984ac9d02Sdanielk1977 } 1390f46f905aSdrh p->pPrior = 0; 13917b58daeaSdrh p->iLimit = pPrior->iLimit; 13927b58daeaSdrh p->iOffset = pPrior->iOffset; 13937b58daeaSdrh p->nLimit = -1; 13947b58daeaSdrh p->nOffset = 0; 139584ac9d02Sdanielk1977 rc = sqlite3Select(pParse, p, eDest, iParm, 0, 0, 0, aff); 1396f46f905aSdrh p->pPrior = pPrior; 139784ac9d02Sdanielk1977 if( rc ){ 139884ac9d02Sdanielk1977 goto multi_select_end; 139984ac9d02Sdanielk1977 } 1400f46f905aSdrh break; 1401f46f905aSdrh } 1402f46f905aSdrh /* For UNION ALL ... ORDER BY fall through to the next case */ 1403f46f905aSdrh } 140482c3d636Sdrh case TK_EXCEPT: 140582c3d636Sdrh case TK_UNION: { 1406d8bc7086Sdrh int unionTab; /* Cursor number of the temporary table holding result */ 1407d8bc7086Sdrh int op; /* One of the SRT_ operations to apply to self */ 1408d8bc7086Sdrh int priorOp; /* The SRT_ operation to apply to prior selects */ 14097b58daeaSdrh int nLimit, nOffset; /* Saved values of p->nLimit and p->nOffset */ 1410c926afbcSdrh ExprList *pOrderBy; /* The ORDER BY clause for the right SELECT */ 1411dc1bdc4fSdanielk1977 int addr; 141282c3d636Sdrh 1413d8bc7086Sdrh priorOp = p->op==TK_ALL ? SRT_Table : SRT_Union; 14147b58daeaSdrh if( eDest==priorOp && p->pOrderBy==0 && p->nLimit<0 && p->nOffset==0 ){ 1415d8bc7086Sdrh /* We can reuse a temporary table generated by a SELECT to our 1416c926afbcSdrh ** right. 1417d8bc7086Sdrh */ 141882c3d636Sdrh unionTab = iParm; 141982c3d636Sdrh }else{ 1420d8bc7086Sdrh /* We will need to create our own temporary table to hold the 1421d8bc7086Sdrh ** intermediate results. 1422d8bc7086Sdrh */ 142382c3d636Sdrh unionTab = pParse->nTab++; 1424d8bc7086Sdrh if( p->pOrderBy 1425d8bc7086Sdrh && matchOrderbyToColumn(pParse, p, p->pOrderBy, unionTab, 1) ){ 142684ac9d02Sdanielk1977 rc = 1; 142784ac9d02Sdanielk1977 goto multi_select_end; 1428d8bc7086Sdrh } 1429dc1bdc4fSdanielk1977 addr = sqlite3VdbeAddOp(v, OP_OpenTemp, unionTab, 0); 1430d8bc7086Sdrh if( p->op!=TK_ALL ){ 1431dc1bdc4fSdanielk1977 rc = multiSelectOpenTempAddr(p, addr, &pOpenTemp); 1432dc1bdc4fSdanielk1977 if( rc!=SQLITE_OK ){ 1433dc1bdc4fSdanielk1977 goto multi_select_end; 1434dc1bdc4fSdanielk1977 } 1435dc1bdc4fSdanielk1977 sqlite3VdbeAddOp(v, OP_KeyAsData, unionTab, 1); 143682c3d636Sdrh } 143784ac9d02Sdanielk1977 assert( p->pEList ); 1438d8bc7086Sdrh } 1439d8bc7086Sdrh 1440d8bc7086Sdrh /* Code the SELECT statements to our left 1441d8bc7086Sdrh */ 1442dc1bdc4fSdanielk1977 pPrior->ppOpenTemp = p->ppOpenTemp; 144384ac9d02Sdanielk1977 rc = sqlite3Select(pParse, pPrior, priorOp, unionTab, 0, 0, 0, aff); 144484ac9d02Sdanielk1977 if( rc ){ 144584ac9d02Sdanielk1977 goto multi_select_end; 144684ac9d02Sdanielk1977 } 144784ac9d02Sdanielk1977 if( p->op==TK_ALL ){ 144884ac9d02Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, unionTab, pPrior->pEList->nExpr); 144984ac9d02Sdanielk1977 } 1450d8bc7086Sdrh 1451d8bc7086Sdrh /* Code the current SELECT statement 1452d8bc7086Sdrh */ 1453d8bc7086Sdrh switch( p->op ){ 1454d8bc7086Sdrh case TK_EXCEPT: op = SRT_Except; break; 1455d8bc7086Sdrh case TK_UNION: op = SRT_Union; break; 1456d8bc7086Sdrh case TK_ALL: op = SRT_Table; break; 1457d8bc7086Sdrh } 145882c3d636Sdrh p->pPrior = 0; 1459c926afbcSdrh pOrderBy = p->pOrderBy; 1460c926afbcSdrh p->pOrderBy = 0; 14617b58daeaSdrh nLimit = p->nLimit; 14627b58daeaSdrh p->nLimit = -1; 14637b58daeaSdrh nOffset = p->nOffset; 14647b58daeaSdrh p->nOffset = 0; 146584ac9d02Sdanielk1977 rc = sqlite3Select(pParse, p, op, unionTab, 0, 0, 0, aff); 146682c3d636Sdrh p->pPrior = pPrior; 1467c926afbcSdrh p->pOrderBy = pOrderBy; 14687b58daeaSdrh p->nLimit = nLimit; 14697b58daeaSdrh p->nOffset = nOffset; 147084ac9d02Sdanielk1977 if( rc ){ 147184ac9d02Sdanielk1977 goto multi_select_end; 147284ac9d02Sdanielk1977 } 147384ac9d02Sdanielk1977 1474d8bc7086Sdrh 1475d8bc7086Sdrh /* Convert the data in the temporary table into whatever form 1476d8bc7086Sdrh ** it is that we currently need. 1477d8bc7086Sdrh */ 1478c926afbcSdrh if( eDest!=priorOp || unionTab!=iParm ){ 14796b56344dSdrh int iCont, iBreak, iStart; 148082c3d636Sdrh assert( p->pEList ); 148141202ccaSdrh if( eDest==SRT_Callback ){ 14826a3ea0e6Sdrh generateColumnNames(pParse, 0, p->pEList); 148341202ccaSdrh } 14844adee20fSdanielk1977 iBreak = sqlite3VdbeMakeLabel(v); 14854adee20fSdanielk1977 iCont = sqlite3VdbeMakeLabel(v); 14864adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Rewind, unionTab, iBreak); 14877b58daeaSdrh computeLimitRegisters(pParse, p); 14884adee20fSdanielk1977 iStart = sqlite3VdbeCurrentAddr(v); 148938640e15Sdrh rc = selectInnerLoop(pParse, p, p->pEList, unionTab, p->pEList->nExpr, 1490d8bc7086Sdrh p->pOrderBy, -1, eDest, iParm, 149184ac9d02Sdanielk1977 iCont, iBreak, 0); 149284ac9d02Sdanielk1977 if( rc ){ 149384ac9d02Sdanielk1977 rc = 1; 149484ac9d02Sdanielk1977 goto multi_select_end; 149584ac9d02Sdanielk1977 } 14964adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iCont); 14974adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Next, unionTab, iStart); 14984adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iBreak); 14994adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, unionTab, 0); 150082c3d636Sdrh } 150182c3d636Sdrh break; 150282c3d636Sdrh } 150382c3d636Sdrh case TK_INTERSECT: { 150482c3d636Sdrh int tab1, tab2; 15056b56344dSdrh int iCont, iBreak, iStart; 15067b58daeaSdrh int nLimit, nOffset; 1507dc1bdc4fSdanielk1977 int addr; 150882c3d636Sdrh 1509d8bc7086Sdrh /* INTERSECT is different from the others since it requires 15106206d50aSdrh ** two temporary tables. Hence it has its own case. Begin 1511d8bc7086Sdrh ** by allocating the tables we will need. 1512d8bc7086Sdrh */ 151382c3d636Sdrh tab1 = pParse->nTab++; 151482c3d636Sdrh tab2 = pParse->nTab++; 1515d8bc7086Sdrh if( p->pOrderBy && matchOrderbyToColumn(pParse,p,p->pOrderBy,tab1,1) ){ 151684ac9d02Sdanielk1977 rc = 1; 151784ac9d02Sdanielk1977 goto multi_select_end; 1518d8bc7086Sdrh } 1519dc1bdc4fSdanielk1977 1520dc1bdc4fSdanielk1977 addr = sqlite3VdbeAddOp(v, OP_OpenTemp, tab1, 0); 1521dc1bdc4fSdanielk1977 rc = multiSelectOpenTempAddr(p, addr, &pOpenTemp); 1522dc1bdc4fSdanielk1977 if( rc!=SQLITE_OK ){ 1523dc1bdc4fSdanielk1977 goto multi_select_end; 1524dc1bdc4fSdanielk1977 } 1525dc1bdc4fSdanielk1977 sqlite3VdbeAddOp(v, OP_KeyAsData, tab1, 1); 152684ac9d02Sdanielk1977 assert( p->pEList ); 1527d8bc7086Sdrh 1528d8bc7086Sdrh /* Code the SELECTs to our left into temporary table "tab1". 1529d8bc7086Sdrh */ 1530dc1bdc4fSdanielk1977 pPrior->ppOpenTemp = p->ppOpenTemp; 153184ac9d02Sdanielk1977 rc = sqlite3Select(pParse, pPrior, SRT_Union, tab1, 0, 0, 0, aff); 153284ac9d02Sdanielk1977 if( rc ){ 153384ac9d02Sdanielk1977 goto multi_select_end; 153484ac9d02Sdanielk1977 } 1535d8bc7086Sdrh 1536d8bc7086Sdrh /* Code the current SELECT into temporary table "tab2" 1537d8bc7086Sdrh */ 1538dc1bdc4fSdanielk1977 addr = sqlite3VdbeAddOp(v, OP_OpenTemp, tab2, 0); 1539dc1bdc4fSdanielk1977 rc = multiSelectOpenTempAddr(p, addr, &pOpenTemp); 1540dc1bdc4fSdanielk1977 if( rc!=SQLITE_OK ){ 1541dc1bdc4fSdanielk1977 goto multi_select_end; 1542dc1bdc4fSdanielk1977 } 1543dc1bdc4fSdanielk1977 sqlite3VdbeAddOp(v, OP_KeyAsData, tab2, 1); 154482c3d636Sdrh p->pPrior = 0; 15457b58daeaSdrh nLimit = p->nLimit; 15467b58daeaSdrh p->nLimit = -1; 15477b58daeaSdrh nOffset = p->nOffset; 15487b58daeaSdrh p->nOffset = 0; 154984ac9d02Sdanielk1977 rc = sqlite3Select(pParse, p, SRT_Union, tab2, 0, 0, 0, aff); 155082c3d636Sdrh p->pPrior = pPrior; 15517b58daeaSdrh p->nLimit = nLimit; 15527b58daeaSdrh p->nOffset = nOffset; 155384ac9d02Sdanielk1977 if( rc ){ 155484ac9d02Sdanielk1977 goto multi_select_end; 155584ac9d02Sdanielk1977 } 1556d8bc7086Sdrh 1557d8bc7086Sdrh /* Generate code to take the intersection of the two temporary 1558d8bc7086Sdrh ** tables. 1559d8bc7086Sdrh */ 156082c3d636Sdrh assert( p->pEList ); 156141202ccaSdrh if( eDest==SRT_Callback ){ 15626a3ea0e6Sdrh generateColumnNames(pParse, 0, p->pEList); 156341202ccaSdrh } 15644adee20fSdanielk1977 iBreak = sqlite3VdbeMakeLabel(v); 15654adee20fSdanielk1977 iCont = sqlite3VdbeMakeLabel(v); 15664adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Rewind, tab1, iBreak); 15677b58daeaSdrh computeLimitRegisters(pParse, p); 15684adee20fSdanielk1977 iStart = sqlite3VdbeAddOp(v, OP_FullKey, tab1, 0); 15694adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NotFound, tab2, iCont); 157038640e15Sdrh rc = selectInnerLoop(pParse, p, p->pEList, tab1, p->pEList->nExpr, 1571d8bc7086Sdrh p->pOrderBy, -1, eDest, iParm, 157284ac9d02Sdanielk1977 iCont, iBreak, 0); 157384ac9d02Sdanielk1977 if( rc ){ 157484ac9d02Sdanielk1977 rc = 1; 157584ac9d02Sdanielk1977 goto multi_select_end; 157684ac9d02Sdanielk1977 } 15774adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iCont); 15784adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Next, tab1, iStart); 15794adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iBreak); 15804adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, tab2, 0); 15814adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, tab1, 0); 158282c3d636Sdrh break; 158382c3d636Sdrh } 158482c3d636Sdrh } 158582c3d636Sdrh assert( p->pEList && pPrior->pEList ); 158682c3d636Sdrh if( p->pEList->nExpr!=pPrior->pEList->nExpr ){ 15874adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "SELECTs to the left and right of %s" 1588da93d238Sdrh " do not have the same number of result columns", selectOpName(p->op)); 158984ac9d02Sdanielk1977 rc = 1; 159084ac9d02Sdanielk1977 goto multi_select_end; 15912282792aSdrh } 159284ac9d02Sdanielk1977 1593dc1bdc4fSdanielk1977 if( p->pOrderBy || (pOpenTemp && pOpenTemp->nId>0) ){ 1594dc1bdc4fSdanielk1977 int nCol = p->pEList->nExpr; 1595dc1bdc4fSdanielk1977 int i; 1596dc1bdc4fSdanielk1977 KeyInfo *pKeyInfo = sqliteMalloc(sizeof(*pKeyInfo)+nCol*sizeof(CollSeq*)); 1597dc1bdc4fSdanielk1977 if( !pKeyInfo ){ 1598dc1bdc4fSdanielk1977 rc = SQLITE_NOMEM; 1599dc1bdc4fSdanielk1977 goto multi_select_end; 1600dc1bdc4fSdanielk1977 } 1601dc1bdc4fSdanielk1977 1602dc1bdc4fSdanielk1977 pKeyInfo->enc = pParse->db->enc; 1603dc1bdc4fSdanielk1977 pKeyInfo->nField = nCol; 1604dc1bdc4fSdanielk1977 1605dc1bdc4fSdanielk1977 for(i=0; i<nCol; i++){ 1606dc1bdc4fSdanielk1977 pKeyInfo->aColl[i] = multiSelectCollSeq(pParse, p, i); 1607dc1bdc4fSdanielk1977 if( !pKeyInfo->aColl[i] ){ 1608dc1bdc4fSdanielk1977 pKeyInfo->aColl[i] = pParse->db->pDfltColl; 1609dc1bdc4fSdanielk1977 } 1610dc1bdc4fSdanielk1977 } 1611dc1bdc4fSdanielk1977 1612dc1bdc4fSdanielk1977 for(i=0; pOpenTemp && i<pOpenTemp->nId; i++){ 1613dc1bdc4fSdanielk1977 int p3type = (i==0?P3_KEYINFO_HANDOFF:P3_KEYINFO); 1614dc1bdc4fSdanielk1977 int addr = pOpenTemp->a[i].idx; 1615dc1bdc4fSdanielk1977 sqlite3VdbeChangeP3(v, addr, (char *)pKeyInfo, p3type); 1616dc1bdc4fSdanielk1977 } 1617dc1bdc4fSdanielk1977 1618dc1bdc4fSdanielk1977 if( p->pOrderBy ){ 1619dc1bdc4fSdanielk1977 for(i=0; i<p->pOrderBy->nExpr; i++){ 1620dc1bdc4fSdanielk1977 Expr *pExpr = p->pOrderBy->a[i].pExpr; 1621dc1bdc4fSdanielk1977 char *zName = p->pOrderBy->a[i].zName; 1622dc1bdc4fSdanielk1977 assert( pExpr->op==TK_COLUMN && pExpr->iColumn<nCol ); 1623dc1bdc4fSdanielk1977 assert( !pExpr->pColl ); 1624dc1bdc4fSdanielk1977 if( zName ){ 1625dc1bdc4fSdanielk1977 pExpr->pColl = sqlite3LocateCollSeq(pParse, zName, -1); 162684ac9d02Sdanielk1977 }else{ 1627dc1bdc4fSdanielk1977 pExpr->pColl = pKeyInfo->aColl[pExpr->iColumn]; 162884ac9d02Sdanielk1977 } 162984ac9d02Sdanielk1977 } 1630dc1bdc4fSdanielk1977 generateSortTail(pParse, p, v, p->pEList->nExpr, eDest, iParm); 1631dc1bdc4fSdanielk1977 } 1632dc1bdc4fSdanielk1977 1633dc1bdc4fSdanielk1977 if( !pOpenTemp ){ 1634dc1bdc4fSdanielk1977 /* This happens for UNION ALL ... ORDER BY */ 1635dc1bdc4fSdanielk1977 sqliteFree(pKeyInfo); 1636dc1bdc4fSdanielk1977 } 1637dc1bdc4fSdanielk1977 } 1638dc1bdc4fSdanielk1977 1639dc1bdc4fSdanielk1977 multi_select_end: 1640dc1bdc4fSdanielk1977 if( pOpenTemp ){ 1641dc1bdc4fSdanielk1977 sqlite3IdListDelete(pOpenTemp); 1642dc1bdc4fSdanielk1977 } 1643dc1bdc4fSdanielk1977 p->ppOpenTemp = 0; 164484ac9d02Sdanielk1977 return rc; 16452282792aSdrh } 16462282792aSdrh 16472282792aSdrh /* 1648832508b7Sdrh ** Scan through the expression pExpr. Replace every reference to 16496a3ea0e6Sdrh ** a column in table number iTable with a copy of the iColumn-th 165084e59207Sdrh ** entry in pEList. (But leave references to the ROWID column 16516a3ea0e6Sdrh ** unchanged.) 1652832508b7Sdrh ** 1653832508b7Sdrh ** This routine is part of the flattening procedure. A subquery 1654832508b7Sdrh ** whose result set is defined by pEList appears as entry in the 1655832508b7Sdrh ** FROM clause of a SELECT such that the VDBE cursor assigned to that 1656832508b7Sdrh ** FORM clause entry is iTable. This routine make the necessary 1657832508b7Sdrh ** changes to pExpr so that it refers directly to the source table 1658832508b7Sdrh ** of the subquery rather the result set of the subquery. 1659832508b7Sdrh */ 16606a3ea0e6Sdrh static void substExprList(ExprList*,int,ExprList*); /* Forward Decl */ 16616a3ea0e6Sdrh static void substExpr(Expr *pExpr, int iTable, ExprList *pEList){ 1662832508b7Sdrh if( pExpr==0 ) return; 166350350a15Sdrh if( pExpr->op==TK_COLUMN && pExpr->iTable==iTable ){ 166450350a15Sdrh if( pExpr->iColumn<0 ){ 166550350a15Sdrh pExpr->op = TK_NULL; 166650350a15Sdrh }else{ 1667832508b7Sdrh Expr *pNew; 166884e59207Sdrh assert( pEList!=0 && pExpr->iColumn<pEList->nExpr ); 1669832508b7Sdrh assert( pExpr->pLeft==0 && pExpr->pRight==0 && pExpr->pList==0 ); 1670832508b7Sdrh pNew = pEList->a[pExpr->iColumn].pExpr; 1671832508b7Sdrh assert( pNew!=0 ); 1672832508b7Sdrh pExpr->op = pNew->op; 1673d94a6698Sdrh assert( pExpr->pLeft==0 ); 16744adee20fSdanielk1977 pExpr->pLeft = sqlite3ExprDup(pNew->pLeft); 1675d94a6698Sdrh assert( pExpr->pRight==0 ); 16764adee20fSdanielk1977 pExpr->pRight = sqlite3ExprDup(pNew->pRight); 1677d94a6698Sdrh assert( pExpr->pList==0 ); 16784adee20fSdanielk1977 pExpr->pList = sqlite3ExprListDup(pNew->pList); 1679832508b7Sdrh pExpr->iTable = pNew->iTable; 1680832508b7Sdrh pExpr->iColumn = pNew->iColumn; 1681832508b7Sdrh pExpr->iAgg = pNew->iAgg; 16824adee20fSdanielk1977 sqlite3TokenCopy(&pExpr->token, &pNew->token); 16834adee20fSdanielk1977 sqlite3TokenCopy(&pExpr->span, &pNew->span); 168450350a15Sdrh } 1685832508b7Sdrh }else{ 16866a3ea0e6Sdrh substExpr(pExpr->pLeft, iTable, pEList); 16876a3ea0e6Sdrh substExpr(pExpr->pRight, iTable, pEList); 16886a3ea0e6Sdrh substExprList(pExpr->pList, iTable, pEList); 1689832508b7Sdrh } 1690832508b7Sdrh } 1691832508b7Sdrh static void 16926a3ea0e6Sdrh substExprList(ExprList *pList, int iTable, ExprList *pEList){ 1693832508b7Sdrh int i; 1694832508b7Sdrh if( pList==0 ) return; 1695832508b7Sdrh for(i=0; i<pList->nExpr; i++){ 16966a3ea0e6Sdrh substExpr(pList->a[i].pExpr, iTable, pEList); 1697832508b7Sdrh } 1698832508b7Sdrh } 1699832508b7Sdrh 1700832508b7Sdrh /* 17011350b030Sdrh ** This routine attempts to flatten subqueries in order to speed 17021350b030Sdrh ** execution. It returns 1 if it makes changes and 0 if no flattening 17031350b030Sdrh ** occurs. 17041350b030Sdrh ** 17051350b030Sdrh ** To understand the concept of flattening, consider the following 17061350b030Sdrh ** query: 17071350b030Sdrh ** 17081350b030Sdrh ** SELECT a FROM (SELECT x+y AS a FROM t1 WHERE z<100) WHERE a>5 17091350b030Sdrh ** 17101350b030Sdrh ** The default way of implementing this query is to execute the 17111350b030Sdrh ** subquery first and store the results in a temporary table, then 17121350b030Sdrh ** run the outer query on that temporary table. This requires two 17131350b030Sdrh ** passes over the data. Furthermore, because the temporary table 17141350b030Sdrh ** has no indices, the WHERE clause on the outer query cannot be 1715832508b7Sdrh ** optimized. 17161350b030Sdrh ** 1717832508b7Sdrh ** This routine attempts to rewrite queries such as the above into 17181350b030Sdrh ** a single flat select, like this: 17191350b030Sdrh ** 17201350b030Sdrh ** SELECT x+y AS a FROM t1 WHERE z<100 AND a>5 17211350b030Sdrh ** 17221350b030Sdrh ** The code generated for this simpification gives the same result 1723832508b7Sdrh ** but only has to scan the data once. And because indices might 1724832508b7Sdrh ** exist on the table t1, a complete scan of the data might be 1725832508b7Sdrh ** avoided. 17261350b030Sdrh ** 1727832508b7Sdrh ** Flattening is only attempted if all of the following are true: 17281350b030Sdrh ** 1729832508b7Sdrh ** (1) The subquery and the outer query do not both use aggregates. 17301350b030Sdrh ** 1731832508b7Sdrh ** (2) The subquery is not an aggregate or the outer query is not a join. 1732832508b7Sdrh ** 17338af4d3acSdrh ** (3) The subquery is not the right operand of a left outer join, or 17348af4d3acSdrh ** the subquery is not itself a join. (Ticket #306) 1735832508b7Sdrh ** 1736832508b7Sdrh ** (4) The subquery is not DISTINCT or the outer query is not a join. 1737832508b7Sdrh ** 1738832508b7Sdrh ** (5) The subquery is not DISTINCT or the outer query does not use 1739832508b7Sdrh ** aggregates. 1740832508b7Sdrh ** 1741832508b7Sdrh ** (6) The subquery does not use aggregates or the outer query is not 1742832508b7Sdrh ** DISTINCT. 1743832508b7Sdrh ** 174408192d5fSdrh ** (7) The subquery has a FROM clause. 174508192d5fSdrh ** 1746df199a25Sdrh ** (8) The subquery does not use LIMIT or the outer query is not a join. 1747df199a25Sdrh ** 1748df199a25Sdrh ** (9) The subquery does not use LIMIT or the outer query does not use 1749df199a25Sdrh ** aggregates. 1750df199a25Sdrh ** 1751df199a25Sdrh ** (10) The subquery does not use aggregates or the outer query does not 1752df199a25Sdrh ** use LIMIT. 1753df199a25Sdrh ** 1754174b6195Sdrh ** (11) The subquery and the outer query do not both have ORDER BY clauses. 1755174b6195Sdrh ** 17563fc673e6Sdrh ** (12) The subquery is not the right term of a LEFT OUTER JOIN or the 17573fc673e6Sdrh ** subquery has no WHERE clause. (added by ticket #350) 17583fc673e6Sdrh ** 1759832508b7Sdrh ** In this routine, the "p" parameter is a pointer to the outer query. 1760832508b7Sdrh ** The subquery is p->pSrc->a[iFrom]. isAgg is true if the outer query 1761832508b7Sdrh ** uses aggregates and subqueryIsAgg is true if the subquery uses aggregates. 1762832508b7Sdrh ** 1763665de47aSdrh ** If flattening is not attempted, this routine is a no-op and returns 0. 1764832508b7Sdrh ** If flattening is attempted this routine returns 1. 1765832508b7Sdrh ** 1766832508b7Sdrh ** All of the expression analysis must occur on both the outer query and 1767832508b7Sdrh ** the subquery before this routine runs. 17681350b030Sdrh */ 17698c74a8caSdrh static int flattenSubquery( 17708c74a8caSdrh Parse *pParse, /* The parsing context */ 17718c74a8caSdrh Select *p, /* The parent or outer SELECT statement */ 17728c74a8caSdrh int iFrom, /* Index in p->pSrc->a[] of the inner subquery */ 17738c74a8caSdrh int isAgg, /* True if outer SELECT uses aggregate functions */ 17748c74a8caSdrh int subqueryIsAgg /* True if the subquery uses aggregate functions */ 17758c74a8caSdrh ){ 17760bb28106Sdrh Select *pSub; /* The inner query or "subquery" */ 1777ad3cab52Sdrh SrcList *pSrc; /* The FROM clause of the outer query */ 1778ad3cab52Sdrh SrcList *pSubSrc; /* The FROM clause of the subquery */ 17790bb28106Sdrh ExprList *pList; /* The result set of the outer query */ 17806a3ea0e6Sdrh int iParent; /* VDBE cursor number of the pSub result set temp table */ 1781832508b7Sdrh int i; 1782832508b7Sdrh Expr *pWhere; 17831350b030Sdrh 1784832508b7Sdrh /* Check to see if flattening is permitted. Return 0 if not. 1785832508b7Sdrh */ 1786832508b7Sdrh if( p==0 ) return 0; 1787832508b7Sdrh pSrc = p->pSrc; 1788ad3cab52Sdrh assert( pSrc && iFrom>=0 && iFrom<pSrc->nSrc ); 1789832508b7Sdrh pSub = pSrc->a[iFrom].pSelect; 1790832508b7Sdrh assert( pSub!=0 ); 1791832508b7Sdrh if( isAgg && subqueryIsAgg ) return 0; 1792ad3cab52Sdrh if( subqueryIsAgg && pSrc->nSrc>1 ) return 0; 1793832508b7Sdrh pSubSrc = pSub->pSrc; 1794832508b7Sdrh assert( pSubSrc ); 1795c31c2eb8Sdrh if( pSubSrc->nSrc==0 ) return 0; 1796df199a25Sdrh if( (pSub->isDistinct || pSub->nLimit>=0) && (pSrc->nSrc>1 || isAgg) ){ 1797df199a25Sdrh return 0; 1798df199a25Sdrh } 1799d11d382cSdrh if( (p->isDistinct || p->nLimit>=0) && subqueryIsAgg ) return 0; 1800174b6195Sdrh if( p->pOrderBy && pSub->pOrderBy ) return 0; 1801832508b7Sdrh 18028af4d3acSdrh /* Restriction 3: If the subquery is a join, make sure the subquery is 18038af4d3acSdrh ** not used as the right operand of an outer join. Examples of why this 18048af4d3acSdrh ** is not allowed: 18058af4d3acSdrh ** 18068af4d3acSdrh ** t1 LEFT OUTER JOIN (t2 JOIN t3) 18078af4d3acSdrh ** 18088af4d3acSdrh ** If we flatten the above, we would get 18098af4d3acSdrh ** 18108af4d3acSdrh ** (t1 LEFT OUTER JOIN t2) JOIN t3 18118af4d3acSdrh ** 18128af4d3acSdrh ** which is not at all the same thing. 18138af4d3acSdrh */ 18148af4d3acSdrh if( pSubSrc->nSrc>1 && iFrom>0 && (pSrc->a[iFrom-1].jointype & JT_OUTER)!=0 ){ 18158af4d3acSdrh return 0; 18168af4d3acSdrh } 18178af4d3acSdrh 18183fc673e6Sdrh /* Restriction 12: If the subquery is the right operand of a left outer 18193fc673e6Sdrh ** join, make sure the subquery has no WHERE clause. 18203fc673e6Sdrh ** An examples of why this is not allowed: 18213fc673e6Sdrh ** 18223fc673e6Sdrh ** t1 LEFT OUTER JOIN (SELECT * FROM t2 WHERE t2.x>0) 18233fc673e6Sdrh ** 18243fc673e6Sdrh ** If we flatten the above, we would get 18253fc673e6Sdrh ** 18263fc673e6Sdrh ** (t1 LEFT OUTER JOIN t2) WHERE t2.x>0 18273fc673e6Sdrh ** 18283fc673e6Sdrh ** But the t2.x>0 test will always fail on a NULL row of t2, which 18293fc673e6Sdrh ** effectively converts the OUTER JOIN into an INNER JOIN. 18303fc673e6Sdrh */ 18313fc673e6Sdrh if( iFrom>0 && (pSrc->a[iFrom-1].jointype & JT_OUTER)!=0 18323fc673e6Sdrh && pSub->pWhere!=0 ){ 18333fc673e6Sdrh return 0; 18343fc673e6Sdrh } 18353fc673e6Sdrh 18360bb28106Sdrh /* If we reach this point, it means flattening is permitted for the 183763eb5f29Sdrh ** iFrom-th entry of the FROM clause in the outer query. 1838832508b7Sdrh */ 1839c31c2eb8Sdrh 1840c31c2eb8Sdrh /* Move all of the FROM elements of the subquery into the 1841c31c2eb8Sdrh ** the FROM clause of the outer query. Before doing this, remember 1842c31c2eb8Sdrh ** the cursor number for the original outer query FROM element in 1843c31c2eb8Sdrh ** iParent. The iParent cursor will never be used. Subsequent code 1844c31c2eb8Sdrh ** will scan expressions looking for iParent references and replace 1845c31c2eb8Sdrh ** those references with expressions that resolve to the subquery FROM 1846c31c2eb8Sdrh ** elements we are now copying in. 1847c31c2eb8Sdrh */ 18486a3ea0e6Sdrh iParent = pSrc->a[iFrom].iCursor; 1849c31c2eb8Sdrh { 1850c31c2eb8Sdrh int nSubSrc = pSubSrc->nSrc; 18518af4d3acSdrh int jointype = pSrc->a[iFrom].jointype; 1852c31c2eb8Sdrh 1853c31c2eb8Sdrh if( pSrc->a[iFrom].pTab && pSrc->a[iFrom].pTab->isTransient ){ 18544adee20fSdanielk1977 sqlite3DeleteTable(0, pSrc->a[iFrom].pTab); 1855c31c2eb8Sdrh } 1856f26e09c8Sdrh sqliteFree(pSrc->a[iFrom].zDatabase); 1857c31c2eb8Sdrh sqliteFree(pSrc->a[iFrom].zName); 1858c31c2eb8Sdrh sqliteFree(pSrc->a[iFrom].zAlias); 1859c31c2eb8Sdrh if( nSubSrc>1 ){ 1860c31c2eb8Sdrh int extra = nSubSrc - 1; 1861c31c2eb8Sdrh for(i=1; i<nSubSrc; i++){ 18624adee20fSdanielk1977 pSrc = sqlite3SrcListAppend(pSrc, 0, 0); 1863c31c2eb8Sdrh } 1864c31c2eb8Sdrh p->pSrc = pSrc; 1865c31c2eb8Sdrh for(i=pSrc->nSrc-1; i-extra>=iFrom; i--){ 1866c31c2eb8Sdrh pSrc->a[i] = pSrc->a[i-extra]; 1867c31c2eb8Sdrh } 1868c31c2eb8Sdrh } 1869c31c2eb8Sdrh for(i=0; i<nSubSrc; i++){ 1870c31c2eb8Sdrh pSrc->a[i+iFrom] = pSubSrc->a[i]; 1871c31c2eb8Sdrh memset(&pSubSrc->a[i], 0, sizeof(pSubSrc->a[i])); 1872c31c2eb8Sdrh } 18738af4d3acSdrh pSrc->a[iFrom+nSubSrc-1].jointype = jointype; 1874c31c2eb8Sdrh } 1875c31c2eb8Sdrh 1876c31c2eb8Sdrh /* Now begin substituting subquery result set expressions for 1877c31c2eb8Sdrh ** references to the iParent in the outer query. 1878c31c2eb8Sdrh ** 1879c31c2eb8Sdrh ** Example: 1880c31c2eb8Sdrh ** 1881c31c2eb8Sdrh ** SELECT a+5, b*10 FROM (SELECT x*3 AS a, y+10 AS b FROM t1) WHERE a>b; 1882c31c2eb8Sdrh ** \ \_____________ subquery __________/ / 1883c31c2eb8Sdrh ** \_____________________ outer query ______________________________/ 1884c31c2eb8Sdrh ** 1885c31c2eb8Sdrh ** We look at every expression in the outer query and every place we see 1886c31c2eb8Sdrh ** "a" we substitute "x*3" and every place we see "b" we substitute "y+10". 1887c31c2eb8Sdrh */ 18886a3ea0e6Sdrh substExprList(p->pEList, iParent, pSub->pEList); 1889832508b7Sdrh pList = p->pEList; 1890832508b7Sdrh for(i=0; i<pList->nExpr; i++){ 18916977fea8Sdrh Expr *pExpr; 18926977fea8Sdrh if( pList->a[i].zName==0 && (pExpr = pList->a[i].pExpr)->span.z!=0 ){ 18936977fea8Sdrh pList->a[i].zName = sqliteStrNDup(pExpr->span.z, pExpr->span.n); 1894832508b7Sdrh } 1895832508b7Sdrh } 18961b2e0329Sdrh if( isAgg ){ 18976a3ea0e6Sdrh substExprList(p->pGroupBy, iParent, pSub->pEList); 18986a3ea0e6Sdrh substExpr(p->pHaving, iParent, pSub->pEList); 18991b2e0329Sdrh } 1900174b6195Sdrh if( pSub->pOrderBy ){ 1901174b6195Sdrh assert( p->pOrderBy==0 ); 1902174b6195Sdrh p->pOrderBy = pSub->pOrderBy; 1903174b6195Sdrh pSub->pOrderBy = 0; 1904174b6195Sdrh }else if( p->pOrderBy ){ 19056a3ea0e6Sdrh substExprList(p->pOrderBy, iParent, pSub->pEList); 1906174b6195Sdrh } 1907832508b7Sdrh if( pSub->pWhere ){ 19084adee20fSdanielk1977 pWhere = sqlite3ExprDup(pSub->pWhere); 1909832508b7Sdrh }else{ 1910832508b7Sdrh pWhere = 0; 1911832508b7Sdrh } 1912832508b7Sdrh if( subqueryIsAgg ){ 1913832508b7Sdrh assert( p->pHaving==0 ); 19141b2e0329Sdrh p->pHaving = p->pWhere; 19151b2e0329Sdrh p->pWhere = pWhere; 19166a3ea0e6Sdrh substExpr(p->pHaving, iParent, pSub->pEList); 19171b2e0329Sdrh if( pSub->pHaving ){ 19184adee20fSdanielk1977 Expr *pHaving = sqlite3ExprDup(pSub->pHaving); 19191b2e0329Sdrh if( p->pHaving ){ 19204adee20fSdanielk1977 p->pHaving = sqlite3Expr(TK_AND, p->pHaving, pHaving, 0); 19211b2e0329Sdrh }else{ 19221b2e0329Sdrh p->pHaving = pHaving; 19231b2e0329Sdrh } 19241b2e0329Sdrh } 19251b2e0329Sdrh assert( p->pGroupBy==0 ); 19264adee20fSdanielk1977 p->pGroupBy = sqlite3ExprListDup(pSub->pGroupBy); 1927832508b7Sdrh }else if( p->pWhere==0 ){ 1928832508b7Sdrh p->pWhere = pWhere; 1929832508b7Sdrh }else{ 19306a3ea0e6Sdrh substExpr(p->pWhere, iParent, pSub->pEList); 1931832508b7Sdrh if( pWhere ){ 19324adee20fSdanielk1977 p->pWhere = sqlite3Expr(TK_AND, p->pWhere, pWhere, 0); 1933832508b7Sdrh } 1934832508b7Sdrh } 1935c31c2eb8Sdrh 1936c31c2eb8Sdrh /* The flattened query is distinct if either the inner or the 1937c31c2eb8Sdrh ** outer query is distinct. 1938c31c2eb8Sdrh */ 1939832508b7Sdrh p->isDistinct = p->isDistinct || pSub->isDistinct; 19408c74a8caSdrh 1941c31c2eb8Sdrh /* Transfer the limit expression from the subquery to the outer 1942c31c2eb8Sdrh ** query. 1943c31c2eb8Sdrh */ 1944df199a25Sdrh if( pSub->nLimit>=0 ){ 1945df199a25Sdrh if( p->nLimit<0 ){ 1946df199a25Sdrh p->nLimit = pSub->nLimit; 1947df199a25Sdrh }else if( p->nLimit+p->nOffset > pSub->nLimit+pSub->nOffset ){ 1948df199a25Sdrh p->nLimit = pSub->nLimit + pSub->nOffset - p->nOffset; 1949df199a25Sdrh } 1950df199a25Sdrh } 1951df199a25Sdrh p->nOffset += pSub->nOffset; 19528c74a8caSdrh 1953c31c2eb8Sdrh /* Finially, delete what is left of the subquery and return 1954c31c2eb8Sdrh ** success. 1955c31c2eb8Sdrh */ 19564adee20fSdanielk1977 sqlite3SelectDelete(pSub); 1957832508b7Sdrh return 1; 19581350b030Sdrh } 19591350b030Sdrh 19601350b030Sdrh /* 19619562b551Sdrh ** Analyze the SELECT statement passed in as an argument to see if it 19629562b551Sdrh ** is a simple min() or max() query. If it is and this query can be 19639562b551Sdrh ** satisfied using a single seek to the beginning or end of an index, 1964e78e8284Sdrh ** then generate the code for this SELECT and return 1. If this is not a 19659562b551Sdrh ** simple min() or max() query, then return 0; 19669562b551Sdrh ** 19679562b551Sdrh ** A simply min() or max() query looks like this: 19689562b551Sdrh ** 19699562b551Sdrh ** SELECT min(a) FROM table; 19709562b551Sdrh ** SELECT max(a) FROM table; 19719562b551Sdrh ** 19729562b551Sdrh ** The query may have only a single table in its FROM argument. There 19739562b551Sdrh ** can be no GROUP BY or HAVING or WHERE clauses. The result set must 19749562b551Sdrh ** be the min() or max() of a single column of the table. The column 19759562b551Sdrh ** in the min() or max() function must be indexed. 19769562b551Sdrh ** 19774adee20fSdanielk1977 ** The parameters to this routine are the same as for sqlite3Select(). 19789562b551Sdrh ** See the header comment on that routine for additional information. 19799562b551Sdrh */ 19809562b551Sdrh static int simpleMinMaxQuery(Parse *pParse, Select *p, int eDest, int iParm){ 19819562b551Sdrh Expr *pExpr; 19829562b551Sdrh int iCol; 19839562b551Sdrh Table *pTab; 19849562b551Sdrh Index *pIdx; 19859562b551Sdrh int base; 19869562b551Sdrh Vdbe *v; 19879562b551Sdrh int seekOp; 19889562b551Sdrh int cont; 19896e17529eSdrh ExprList *pEList, *pList, eList; 19909562b551Sdrh struct ExprList_item eListItem; 19916e17529eSdrh SrcList *pSrc; 19926e17529eSdrh 19939562b551Sdrh 19949562b551Sdrh /* Check to see if this query is a simple min() or max() query. Return 19959562b551Sdrh ** zero if it is not. 19969562b551Sdrh */ 19979562b551Sdrh if( p->pGroupBy || p->pHaving || p->pWhere ) return 0; 19986e17529eSdrh pSrc = p->pSrc; 19996e17529eSdrh if( pSrc->nSrc!=1 ) return 0; 20006e17529eSdrh pEList = p->pEList; 20016e17529eSdrh if( pEList->nExpr!=1 ) return 0; 20026e17529eSdrh pExpr = pEList->a[0].pExpr; 20039562b551Sdrh if( pExpr->op!=TK_AGG_FUNCTION ) return 0; 20046e17529eSdrh pList = pExpr->pList; 20056e17529eSdrh if( pList==0 || pList->nExpr!=1 ) return 0; 20066977fea8Sdrh if( pExpr->token.n!=3 ) return 0; 20074adee20fSdanielk1977 if( sqlite3StrNICmp(pExpr->token.z,"min",3)==0 ){ 20080bce8354Sdrh seekOp = OP_Rewind; 20094adee20fSdanielk1977 }else if( sqlite3StrNICmp(pExpr->token.z,"max",3)==0 ){ 20100bce8354Sdrh seekOp = OP_Last; 20110bce8354Sdrh }else{ 20120bce8354Sdrh return 0; 20130bce8354Sdrh } 20146e17529eSdrh pExpr = pList->a[0].pExpr; 20159562b551Sdrh if( pExpr->op!=TK_COLUMN ) return 0; 20169562b551Sdrh iCol = pExpr->iColumn; 20176e17529eSdrh pTab = pSrc->a[0].pTab; 20189562b551Sdrh 20199562b551Sdrh /* If we get to here, it means the query is of the correct form. 202017f71934Sdrh ** Check to make sure we have an index and make pIdx point to the 202117f71934Sdrh ** appropriate index. If the min() or max() is on an INTEGER PRIMARY 202217f71934Sdrh ** key column, no index is necessary so set pIdx to NULL. If no 202317f71934Sdrh ** usable index is found, return 0. 20249562b551Sdrh */ 20259562b551Sdrh if( iCol<0 ){ 20269562b551Sdrh pIdx = 0; 20279562b551Sdrh }else{ 2028dc1bdc4fSdanielk1977 CollSeq *pColl = sqlite3ExprCollSeq(pParse, pExpr); 20299562b551Sdrh for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){ 20309562b551Sdrh assert( pIdx->nColumn>=1 ); 2031dc1bdc4fSdanielk1977 if( pIdx->aiColumn[0]==iCol && pIdx->keyInfo.aColl[0]==pColl ) break; 20329562b551Sdrh } 20339562b551Sdrh if( pIdx==0 ) return 0; 20349562b551Sdrh } 20359562b551Sdrh 2036e5f50722Sdrh /* Identify column types if we will be using the callback. This 20379562b551Sdrh ** step is skipped if the output is going to a table or a memory cell. 2038e5f50722Sdrh ** The column names have already been generated in the calling function. 20399562b551Sdrh */ 20404adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 20419562b551Sdrh if( v==0 ) return 0; 20429562b551Sdrh 20430c37e630Sdrh /* If the output is destined for a temporary table, open that table. 20440c37e630Sdrh */ 20450c37e630Sdrh if( eDest==SRT_TempTable ){ 20464adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_OpenTemp, iParm, 0); 2047b4964b72Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, iParm, 1); 20480c37e630Sdrh } 20490c37e630Sdrh 205017f71934Sdrh /* Generating code to find the min or the max. Basically all we have 205117f71934Sdrh ** to do is find the first or the last entry in the chosen index. If 205217f71934Sdrh ** the min() or max() is on the INTEGER PRIMARY KEY, then find the first 205317f71934Sdrh ** or last entry in the main table. 20549562b551Sdrh */ 20554adee20fSdanielk1977 sqlite3CodeVerifySchema(pParse, pTab->iDb); 20566e17529eSdrh base = pSrc->a[0].iCursor; 20577b58daeaSdrh computeLimitRegisters(pParse, p); 20586e17529eSdrh if( pSrc->a[0].pSelect==0 ){ 20594adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, pTab->iDb, 0); 2060d3d39e93Sdrh sqlite3VdbeAddOp(v, OP_OpenRead, base, pTab->tnum); 2061b4964b72Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, base, pTab->nCol); 20626e17529eSdrh } 20634adee20fSdanielk1977 cont = sqlite3VdbeMakeLabel(v); 20649562b551Sdrh if( pIdx==0 ){ 20654adee20fSdanielk1977 sqlite3VdbeAddOp(v, seekOp, base, 0); 20669562b551Sdrh }else{ 20674adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, pIdx->iDb, 0); 2068d3d39e93Sdrh sqlite3VdbeOp3(v, OP_OpenRead, base+1, pIdx->tnum, 2069d3d39e93Sdrh (char*)&pIdx->keyInfo, P3_KEYINFO); 20704adee20fSdanielk1977 sqlite3VdbeAddOp(v, seekOp, base+1, 0); 20714adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_IdxRecno, base+1, 0); 20724adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, base+1, 0); 20737cf6e4deSdrh sqlite3VdbeAddOp(v, OP_MoveGe, base, 0); 20749562b551Sdrh } 20755cf8e8c7Sdrh eList.nExpr = 1; 20765cf8e8c7Sdrh memset(&eListItem, 0, sizeof(eListItem)); 20775cf8e8c7Sdrh eList.a = &eListItem; 20785cf8e8c7Sdrh eList.a[0].pExpr = pExpr; 207984ac9d02Sdanielk1977 selectInnerLoop(pParse, p, &eList, 0, 0, 0, -1, eDest, iParm, cont, cont, 0); 20804adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, cont); 20814adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, base, 0); 20826e17529eSdrh 20839562b551Sdrh return 1; 20849562b551Sdrh } 20859562b551Sdrh 20869562b551Sdrh /* 20879bb61fe7Sdrh ** Generate code for the given SELECT statement. 20889bb61fe7Sdrh ** 2089fef5208cSdrh ** The results are distributed in various ways depending on the 2090fef5208cSdrh ** value of eDest and iParm. 2091fef5208cSdrh ** 2092fef5208cSdrh ** eDest Value Result 2093fef5208cSdrh ** ------------ ------------------------------------------- 2094fef5208cSdrh ** SRT_Callback Invoke the callback for each row of the result. 2095fef5208cSdrh ** 2096fef5208cSdrh ** SRT_Mem Store first result in memory cell iParm 2097fef5208cSdrh ** 2098e014a838Sdanielk1977 ** SRT_Set Store results as keys of table iParm. 2099fef5208cSdrh ** 210082c3d636Sdrh ** SRT_Union Store results as a key in a temporary table iParm 210182c3d636Sdrh ** 21024b11c6d3Sjplyon ** SRT_Except Remove results from the temporary table iParm. 2103c4a3c779Sdrh ** 2104c4a3c779Sdrh ** SRT_Table Store results in temporary table iParm 21059bb61fe7Sdrh ** 2106e78e8284Sdrh ** The table above is incomplete. Additional eDist value have be added 2107e78e8284Sdrh ** since this comment was written. See the selectInnerLoop() function for 2108e78e8284Sdrh ** a complete listing of the allowed values of eDest and their meanings. 2109e78e8284Sdrh ** 21109bb61fe7Sdrh ** This routine returns the number of errors. If any errors are 21119bb61fe7Sdrh ** encountered, then an appropriate error message is left in 21129bb61fe7Sdrh ** pParse->zErrMsg. 21139bb61fe7Sdrh ** 21149bb61fe7Sdrh ** This routine does NOT free the Select structure passed in. The 21159bb61fe7Sdrh ** calling function needs to do that. 21161b2e0329Sdrh ** 21171b2e0329Sdrh ** The pParent, parentTab, and *pParentAgg fields are filled in if this 21181b2e0329Sdrh ** SELECT is a subquery. This routine may try to combine this SELECT 21191b2e0329Sdrh ** with its parent to form a single flat query. In so doing, it might 21201b2e0329Sdrh ** change the parent query from a non-aggregate to an aggregate query. 21211b2e0329Sdrh ** For that reason, the pParentAgg flag is passed as a pointer, so it 21221b2e0329Sdrh ** can be changed. 2123e78e8284Sdrh ** 2124e78e8284Sdrh ** Example 1: The meaning of the pParent parameter. 2125e78e8284Sdrh ** 2126e78e8284Sdrh ** SELECT * FROM t1 JOIN (SELECT x, count(*) FROM t2) JOIN t3; 2127e78e8284Sdrh ** \ \_______ subquery _______/ / 2128e78e8284Sdrh ** \ / 2129e78e8284Sdrh ** \____________________ outer query ___________________/ 2130e78e8284Sdrh ** 2131e78e8284Sdrh ** This routine is called for the outer query first. For that call, 2132e78e8284Sdrh ** pParent will be NULL. During the processing of the outer query, this 2133e78e8284Sdrh ** routine is called recursively to handle the subquery. For the recursive 2134e78e8284Sdrh ** call, pParent will point to the outer query. Because the subquery is 2135e78e8284Sdrh ** the second element in a three-way join, the parentTab parameter will 2136e78e8284Sdrh ** be 1 (the 2nd value of a 0-indexed array.) 21379bb61fe7Sdrh */ 21384adee20fSdanielk1977 int sqlite3Select( 2139cce7d176Sdrh Parse *pParse, /* The parser context */ 21409bb61fe7Sdrh Select *p, /* The SELECT statement being coded. */ 2141e78e8284Sdrh int eDest, /* How to dispose of the results */ 2142e78e8284Sdrh int iParm, /* A parameter used by the eDest disposal method */ 2143832508b7Sdrh Select *pParent, /* Another SELECT for which this is a sub-query */ 2144832508b7Sdrh int parentTab, /* Index in pParent->pSrc of this query */ 214584ac9d02Sdanielk1977 int *pParentAgg, /* True if pParent uses aggregate functions */ 214684ac9d02Sdanielk1977 char *aff /* If eDest is SRT_Union, the affinity string */ 2147cce7d176Sdrh ){ 2148d8bc7086Sdrh int i; 2149cce7d176Sdrh WhereInfo *pWInfo; 2150cce7d176Sdrh Vdbe *v; 2151cce7d176Sdrh int isAgg = 0; /* True for select lists like "count(*)" */ 2152a2e00042Sdrh ExprList *pEList; /* List of columns to extract. */ 2153ad3cab52Sdrh SrcList *pTabList; /* List of tables to select from */ 21549bb61fe7Sdrh Expr *pWhere; /* The WHERE clause. May be NULL */ 21559bb61fe7Sdrh ExprList *pOrderBy; /* The ORDER BY clause. May be NULL */ 21562282792aSdrh ExprList *pGroupBy; /* The GROUP BY clause. May be NULL */ 21572282792aSdrh Expr *pHaving; /* The HAVING clause. May be NULL */ 215819a775c2Sdrh int isDistinct; /* True if the DISTINCT keyword is present */ 215919a775c2Sdrh int distinct; /* Table to use for the distinct set */ 21601d83f052Sdrh int rc = 1; /* Value to return from this function */ 21619bb61fe7Sdrh 21626f8a503dSdanielk1977 if( sqlite3_malloc_failed || pParse->nErr || p==0 ) return 1; 21634adee20fSdanielk1977 if( sqlite3AuthCheck(pParse, SQLITE_SELECT, 0, 0, 0) ) return 1; 2164daffd0e5Sdrh 216582c3d636Sdrh /* If there is are a sequence of queries, do the earlier ones first. 216682c3d636Sdrh */ 216782c3d636Sdrh if( p->pPrior ){ 216884ac9d02Sdanielk1977 return multiSelect(pParse, p, eDest, iParm, aff); 216982c3d636Sdrh } 217082c3d636Sdrh 217182c3d636Sdrh /* Make local copies of the parameters for this query. 217282c3d636Sdrh */ 21739bb61fe7Sdrh pTabList = p->pSrc; 21749bb61fe7Sdrh pWhere = p->pWhere; 21759bb61fe7Sdrh pOrderBy = p->pOrderBy; 21762282792aSdrh pGroupBy = p->pGroupBy; 21772282792aSdrh pHaving = p->pHaving; 217819a775c2Sdrh isDistinct = p->isDistinct; 21799bb61fe7Sdrh 21806a3ea0e6Sdrh /* Allocate VDBE cursors for each table in the FROM clause 218110e5e3cfSdrh */ 21824adee20fSdanielk1977 sqlite3SrcListAssignCursors(pParse, pTabList); 218310e5e3cfSdrh 21849bb61fe7Sdrh /* 21859bb61fe7Sdrh ** Do not even attempt to generate any code if we have already seen 21869bb61fe7Sdrh ** errors before this routine starts. 21879bb61fe7Sdrh */ 21881d83f052Sdrh if( pParse->nErr>0 ) goto select_end; 2189cce7d176Sdrh 2190e78e8284Sdrh /* Expand any "*" terms in the result set. (For example the "*" in 2191e78e8284Sdrh ** "SELECT * FROM t1") The fillInColumnlist() routine also does some 2192e78e8284Sdrh ** other housekeeping - see the header comment for details. 2193cce7d176Sdrh */ 2194d8bc7086Sdrh if( fillInColumnList(pParse, p) ){ 21951d83f052Sdrh goto select_end; 2196cce7d176Sdrh } 2197ad2d8307Sdrh pWhere = p->pWhere; 2198d8bc7086Sdrh pEList = p->pEList; 21991d83f052Sdrh if( pEList==0 ) goto select_end; 2200cce7d176Sdrh 22012282792aSdrh /* If writing to memory or generating a set 22022282792aSdrh ** only a single column may be output. 220319a775c2Sdrh */ 2204fef5208cSdrh if( (eDest==SRT_Mem || eDest==SRT_Set) && pEList->nExpr>1 ){ 22054adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "only a single result allowed for " 2206da93d238Sdrh "a SELECT that is part of an expression"); 22071d83f052Sdrh goto select_end; 220819a775c2Sdrh } 220919a775c2Sdrh 2210c926afbcSdrh /* ORDER BY is ignored for some destinations. 22112282792aSdrh */ 2212c926afbcSdrh switch( eDest ){ 2213c926afbcSdrh case SRT_Union: 2214c926afbcSdrh case SRT_Except: 2215c926afbcSdrh case SRT_Discard: 2216f93bbbeaSdanielk1977 case SRT_Set: 2217acd4c695Sdrh pOrderBy = 0; 2218c926afbcSdrh break; 2219c926afbcSdrh default: 2220c926afbcSdrh break; 22212282792aSdrh } 22222282792aSdrh 222310e5e3cfSdrh /* At this point, we should have allocated all the cursors that we 2224832508b7Sdrh ** need to handle subquerys and temporary tables. 222510e5e3cfSdrh ** 2226967e8b73Sdrh ** Resolve the column names and do a semantics check on all the expressions. 22272282792aSdrh */ 22284794b980Sdrh for(i=0; i<pEList->nExpr; i++){ 22294adee20fSdanielk1977 if( sqlite3ExprResolveIds(pParse, pTabList, 0, pEList->a[i].pExpr) ){ 22301d83f052Sdrh goto select_end; 2231cce7d176Sdrh } 22324adee20fSdanielk1977 if( sqlite3ExprCheck(pParse, pEList->a[i].pExpr, 1, &isAgg) ){ 22331d83f052Sdrh goto select_end; 2234cce7d176Sdrh } 2235cce7d176Sdrh } 2236cce7d176Sdrh if( pWhere ){ 22374adee20fSdanielk1977 if( sqlite3ExprResolveIds(pParse, pTabList, pEList, pWhere) ){ 22381d83f052Sdrh goto select_end; 2239cce7d176Sdrh } 22404adee20fSdanielk1977 if( sqlite3ExprCheck(pParse, pWhere, 0, 0) ){ 22411d83f052Sdrh goto select_end; 2242cce7d176Sdrh } 2243cce7d176Sdrh } 2244c66c5a26Sdrh if( pHaving ){ 2245c66c5a26Sdrh if( pGroupBy==0 ){ 22464adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "a GROUP BY clause is required before HAVING"); 2247c66c5a26Sdrh goto select_end; 2248c66c5a26Sdrh } 22494adee20fSdanielk1977 if( sqlite3ExprResolveIds(pParse, pTabList, pEList, pHaving) ){ 2250c66c5a26Sdrh goto select_end; 2251c66c5a26Sdrh } 22524adee20fSdanielk1977 if( sqlite3ExprCheck(pParse, pHaving, 1, &isAgg) ){ 2253c66c5a26Sdrh goto select_end; 2254c66c5a26Sdrh } 2255c66c5a26Sdrh } 2256cce7d176Sdrh if( pOrderBy ){ 2257cce7d176Sdrh for(i=0; i<pOrderBy->nExpr; i++){ 2258e4de1febSdrh int iCol; 225988eee38aSdrh Expr *pE = pOrderBy->a[i].pExpr; 22604adee20fSdanielk1977 if( sqlite3ExprIsInteger(pE, &iCol) && iCol>0 && iCol<=pEList->nExpr ){ 22614adee20fSdanielk1977 sqlite3ExprDelete(pE); 22624adee20fSdanielk1977 pE = pOrderBy->a[i].pExpr = sqlite3ExprDup(pEList->a[iCol-1].pExpr); 226388eee38aSdrh } 22644adee20fSdanielk1977 if( sqlite3ExprResolveIds(pParse, pTabList, pEList, pE) ){ 226588eee38aSdrh goto select_end; 226688eee38aSdrh } 22674adee20fSdanielk1977 if( sqlite3ExprCheck(pParse, pE, isAgg, 0) ){ 226888eee38aSdrh goto select_end; 226988eee38aSdrh } 22704adee20fSdanielk1977 if( sqlite3ExprIsConstant(pE) ){ 22714adee20fSdanielk1977 if( sqlite3ExprIsInteger(pE, &iCol)==0 ){ 22724adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 2273da93d238Sdrh "ORDER BY terms must not be non-integer constants"); 22741d83f052Sdrh goto select_end; 2275e4de1febSdrh }else if( iCol<=0 || iCol>pEList->nExpr ){ 22764adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 2277da93d238Sdrh "ORDER BY column number %d out of range - should be " 2278e4de1febSdrh "between 1 and %d", iCol, pEList->nExpr); 2279e4de1febSdrh goto select_end; 2280e4de1febSdrh } 2281cce7d176Sdrh } 2282cce7d176Sdrh } 2283cce7d176Sdrh } 22842282792aSdrh if( pGroupBy ){ 22852282792aSdrh for(i=0; i<pGroupBy->nExpr; i++){ 228688eee38aSdrh int iCol; 22872282792aSdrh Expr *pE = pGroupBy->a[i].pExpr; 22884adee20fSdanielk1977 if( sqlite3ExprIsInteger(pE, &iCol) && iCol>0 && iCol<=pEList->nExpr ){ 22894adee20fSdanielk1977 sqlite3ExprDelete(pE); 22904adee20fSdanielk1977 pE = pGroupBy->a[i].pExpr = sqlite3ExprDup(pEList->a[iCol-1].pExpr); 22919208643dSdrh } 22924adee20fSdanielk1977 if( sqlite3ExprResolveIds(pParse, pTabList, pEList, pE) ){ 22931d83f052Sdrh goto select_end; 22942282792aSdrh } 22954adee20fSdanielk1977 if( sqlite3ExprCheck(pParse, pE, isAgg, 0) ){ 22961d83f052Sdrh goto select_end; 22972282792aSdrh } 22984adee20fSdanielk1977 if( sqlite3ExprIsConstant(pE) ){ 22994adee20fSdanielk1977 if( sqlite3ExprIsInteger(pE, &iCol)==0 ){ 23004adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 2301da93d238Sdrh "GROUP BY terms must not be non-integer constants"); 230288eee38aSdrh goto select_end; 230388eee38aSdrh }else if( iCol<=0 || iCol>pEList->nExpr ){ 23044adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 2305da93d238Sdrh "GROUP BY column number %d out of range - should be " 230688eee38aSdrh "between 1 and %d", iCol, pEList->nExpr); 230788eee38aSdrh goto select_end; 230888eee38aSdrh } 230988eee38aSdrh } 23102282792aSdrh } 23112282792aSdrh } 2312cce7d176Sdrh 2313d820cb1bSdrh /* Begin generating code. 2314d820cb1bSdrh */ 23154adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 2316d820cb1bSdrh if( v==0 ) goto select_end; 2317d820cb1bSdrh 2318e78e8284Sdrh /* Identify column names if we will be using them in a callback. This 2319e78e8284Sdrh ** step is skipped if the output is going to some other destination. 23200bb28106Sdrh */ 23210bb28106Sdrh if( eDest==SRT_Callback ){ 23226a3ea0e6Sdrh generateColumnNames(pParse, pTabList, pEList); 23230bb28106Sdrh } 23240bb28106Sdrh 2325d3d39e93Sdrh #if 1 /* I do not think we need the following code any more.... */ 232684ac9d02Sdanielk1977 /* If the destination is SRT_Union, then set the number of columns in 232784ac9d02Sdanielk1977 ** the records that will be inserted into the temporary table. The caller 232884ac9d02Sdanielk1977 ** couldn't do this, in case the select statement is of the form 232984ac9d02Sdanielk1977 ** "SELECT * FROM ....". 233084ac9d02Sdanielk1977 ** 233184ac9d02Sdanielk1977 ** We need to do this before we start inserting records into the 233284ac9d02Sdanielk1977 ** temporary table (which has had OP_KeyAsData executed on it), because 233384ac9d02Sdanielk1977 ** it is required by the key comparison function. So do it now, even 233484ac9d02Sdanielk1977 ** though this means that OP_SetNumColumns may be executed on the same 233584ac9d02Sdanielk1977 ** cursor more than once. 233684ac9d02Sdanielk1977 */ 233784ac9d02Sdanielk1977 if( eDest==SRT_Union ){ 233884ac9d02Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, iParm, pEList->nExpr); 233984ac9d02Sdanielk1977 } 2340d3d39e93Sdrh #endif 234184ac9d02Sdanielk1977 2342d820cb1bSdrh /* Generate code for all sub-queries in the FROM clause 2343d820cb1bSdrh */ 2344ad3cab52Sdrh for(i=0; i<pTabList->nSrc; i++){ 23455cf590c1Sdrh const char *zSavedAuthContext; 2346c31c2eb8Sdrh int needRestoreContext; 2347c31c2eb8Sdrh 2348a76b5dfcSdrh if( pTabList->a[i].pSelect==0 ) continue; 23495cf590c1Sdrh if( pTabList->a[i].zName!=0 ){ 23505cf590c1Sdrh zSavedAuthContext = pParse->zAuthContext; 23515cf590c1Sdrh pParse->zAuthContext = pTabList->a[i].zName; 2352c31c2eb8Sdrh needRestoreContext = 1; 2353c31c2eb8Sdrh }else{ 2354c31c2eb8Sdrh needRestoreContext = 0; 23555cf590c1Sdrh } 23564adee20fSdanielk1977 sqlite3Select(pParse, pTabList->a[i].pSelect, SRT_TempTable, 235784ac9d02Sdanielk1977 pTabList->a[i].iCursor, p, i, &isAgg, 0); 2358c31c2eb8Sdrh if( needRestoreContext ){ 23595cf590c1Sdrh pParse->zAuthContext = zSavedAuthContext; 23605cf590c1Sdrh } 2361832508b7Sdrh pTabList = p->pSrc; 2362832508b7Sdrh pWhere = p->pWhere; 2363c31c2eb8Sdrh if( eDest!=SRT_Union && eDest!=SRT_Except && eDest!=SRT_Discard ){ 2364832508b7Sdrh pOrderBy = p->pOrderBy; 2365acd4c695Sdrh } 2366832508b7Sdrh pGroupBy = p->pGroupBy; 2367832508b7Sdrh pHaving = p->pHaving; 2368832508b7Sdrh isDistinct = p->isDistinct; 23691b2e0329Sdrh } 23701b2e0329Sdrh 23716e17529eSdrh /* Check for the special case of a min() or max() function by itself 23726e17529eSdrh ** in the result set. 23736e17529eSdrh */ 23746e17529eSdrh if( simpleMinMaxQuery(pParse, p, eDest, iParm) ){ 23756e17529eSdrh rc = 0; 23766e17529eSdrh goto select_end; 23776e17529eSdrh } 23786e17529eSdrh 23791b2e0329Sdrh /* Check to see if this is a subquery that can be "flattened" into its parent. 23801b2e0329Sdrh ** If flattening is a possiblity, do so and return immediately. 23811b2e0329Sdrh */ 23821b2e0329Sdrh if( pParent && pParentAgg && 23838c74a8caSdrh flattenSubquery(pParse, pParent, parentTab, *pParentAgg, isAgg) ){ 23841b2e0329Sdrh if( isAgg ) *pParentAgg = 1; 23851b2e0329Sdrh return rc; 23861b2e0329Sdrh } 2387832508b7Sdrh 23887cedc8d4Sdanielk1977 /* If there is an ORDER BY clause, resolve any collation sequences 23897cedc8d4Sdanielk1977 ** names that have been explicitly specified. 23907cedc8d4Sdanielk1977 */ 23917cedc8d4Sdanielk1977 if( pOrderBy ){ 23927cedc8d4Sdanielk1977 for(i=0; i<pOrderBy->nExpr; i++){ 23937cedc8d4Sdanielk1977 if( pOrderBy->a[i].zName ){ 23947cedc8d4Sdanielk1977 pOrderBy->a[i].pExpr->pColl = 23957cedc8d4Sdanielk1977 sqlite3LocateCollSeq(pParse, pOrderBy->a[i].zName, -1); 23967cedc8d4Sdanielk1977 } 23977cedc8d4Sdanielk1977 } 23987cedc8d4Sdanielk1977 if( pParse->nErr ){ 23997cedc8d4Sdanielk1977 goto select_end; 24007cedc8d4Sdanielk1977 } 24017cedc8d4Sdanielk1977 } 24027cedc8d4Sdanielk1977 24037b58daeaSdrh /* Set the limiter. 24047b58daeaSdrh */ 24057b58daeaSdrh computeLimitRegisters(pParse, p); 24067b58daeaSdrh 24072d0794e3Sdrh /* If the output is destined for a temporary table, open that table. 24082d0794e3Sdrh */ 24092d0794e3Sdrh if( eDest==SRT_TempTable ){ 24104adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_OpenTemp, iParm, 0); 2411b4964b72Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, iParm, pEList->nExpr); 24122d0794e3Sdrh } 24132d0794e3Sdrh 24142282792aSdrh /* Do an analysis of aggregate expressions. 2415efb7251dSdrh */ 2416d820cb1bSdrh sqliteAggregateInfoReset(pParse); 2417bb999ef6Sdrh if( isAgg || pGroupBy ){ 24180bce8354Sdrh assert( pParse->nAgg==0 ); 2419bb999ef6Sdrh isAgg = 1; 24202282792aSdrh for(i=0; i<pEList->nExpr; i++){ 24214adee20fSdanielk1977 if( sqlite3ExprAnalyzeAggregates(pParse, pEList->a[i].pExpr) ){ 24221d83f052Sdrh goto select_end; 24232282792aSdrh } 24242282792aSdrh } 24252282792aSdrh if( pGroupBy ){ 24262282792aSdrh for(i=0; i<pGroupBy->nExpr; i++){ 24274adee20fSdanielk1977 if( sqlite3ExprAnalyzeAggregates(pParse, pGroupBy->a[i].pExpr) ){ 24281d83f052Sdrh goto select_end; 24292282792aSdrh } 24302282792aSdrh } 24312282792aSdrh } 24324adee20fSdanielk1977 if( pHaving && sqlite3ExprAnalyzeAggregates(pParse, pHaving) ){ 24331d83f052Sdrh goto select_end; 24342282792aSdrh } 2435191b690eSdrh if( pOrderBy ){ 2436191b690eSdrh for(i=0; i<pOrderBy->nExpr; i++){ 24374adee20fSdanielk1977 if( sqlite3ExprAnalyzeAggregates(pParse, pOrderBy->a[i].pExpr) ){ 24381d83f052Sdrh goto select_end; 2439191b690eSdrh } 2440191b690eSdrh } 2441191b690eSdrh } 2442efb7251dSdrh } 2443efb7251dSdrh 24442282792aSdrh /* Reset the aggregator 2445cce7d176Sdrh */ 2446cce7d176Sdrh if( isAgg ){ 2447ce2663ccSdanielk1977 int addr = sqlite3VdbeAddOp(v, OP_AggReset, 0, pParse->nAgg); 2448e5095355Sdrh for(i=0; i<pParse->nAgg; i++){ 24490bce8354Sdrh FuncDef *pFunc; 24500bce8354Sdrh if( (pFunc = pParse->aAgg[i].pFunc)!=0 && pFunc->xFinalize!=0 ){ 2451f9b596ebSdrh sqlite3VdbeOp3(v, OP_AggInit, 0, i, (char*)pFunc, P3_FUNCDEF); 2452e5095355Sdrh } 2453e5095355Sdrh } 24541bee3d7bSdrh if( pGroupBy==0 ){ 24550f69c1e3Sdanielk1977 sqlite3VdbeAddOp(v, OP_String8, 0, 0); 24564adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_AggFocus, 0, 0); 2457ce2663ccSdanielk1977 }else{ 2458ce2663ccSdanielk1977 int sz = sizeof(KeyInfo) + pGroupBy->nExpr*sizeof(CollSeq*); 2459ce2663ccSdanielk1977 KeyInfo *pKey = (KeyInfo *)sqliteMalloc(sz); 2460ce2663ccSdanielk1977 if( 0==pKey ){ 2461ce2663ccSdanielk1977 goto select_end; 2462ce2663ccSdanielk1977 } 2463ce2663ccSdanielk1977 pKey->enc = pParse->db->enc; 2464ce2663ccSdanielk1977 pKey->nField = pGroupBy->nExpr; 2465ce2663ccSdanielk1977 for(i=0; i<pGroupBy->nExpr; i++){ 2466ce2663ccSdanielk1977 pKey->aColl[i] = sqlite3ExprCollSeq(pParse, pGroupBy->a[i].pExpr); 2467ce2663ccSdanielk1977 if( !pKey->aColl[i] ){ 2468ce2663ccSdanielk1977 pKey->aColl[i] = pParse->db->pDfltColl; 2469ce2663ccSdanielk1977 } 2470ce2663ccSdanielk1977 } 2471ce2663ccSdanielk1977 sqlite3VdbeChangeP3(v, addr, (char *)pKey, P3_KEYINFO_HANDOFF); 24721bee3d7bSdrh } 2473cce7d176Sdrh } 2474cce7d176Sdrh 247519a775c2Sdrh /* Initialize the memory cell to NULL 247619a775c2Sdrh */ 2477fef5208cSdrh if( eDest==SRT_Mem ){ 24780f69c1e3Sdanielk1977 sqlite3VdbeAddOp(v, OP_String8, 0, 0); 24794adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemStore, iParm, 1); 248019a775c2Sdrh } 248119a775c2Sdrh 2482832508b7Sdrh /* Open a temporary table to use for the distinct set. 2483cce7d176Sdrh */ 248419a775c2Sdrh if( isDistinct ){ 2485832508b7Sdrh distinct = pParse->nTab++; 2486d3d39e93Sdrh openTempIndex(pParse, p, distinct, 0); 2487832508b7Sdrh }else{ 2488832508b7Sdrh distinct = -1; 2489efb7251dSdrh } 2490832508b7Sdrh 2491832508b7Sdrh /* Begin the database scan 2492832508b7Sdrh */ 24934adee20fSdanielk1977 pWInfo = sqlite3WhereBegin(pParse, pTabList, pWhere, 0, 249468d2e591Sdrh pGroupBy ? 0 : &pOrderBy); 24951d83f052Sdrh if( pWInfo==0 ) goto select_end; 2496cce7d176Sdrh 24972282792aSdrh /* Use the standard inner loop if we are not dealing with 24982282792aSdrh ** aggregates 2499cce7d176Sdrh */ 2500da9d6c45Sdrh if( !isAgg ){ 2501df199a25Sdrh if( selectInnerLoop(pParse, p, pEList, 0, 0, pOrderBy, distinct, eDest, 250284ac9d02Sdanielk1977 iParm, pWInfo->iContinue, pWInfo->iBreak, aff) ){ 25031d83f052Sdrh goto select_end; 2504cce7d176Sdrh } 2505da9d6c45Sdrh } 2506cce7d176Sdrh 2507e3184744Sdrh /* If we are dealing with aggregates, then do the special aggregate 25082282792aSdrh ** processing. 2509efb7251dSdrh */ 25102282792aSdrh else{ 2511268380caSdrh AggExpr *pAgg; 25122282792aSdrh if( pGroupBy ){ 25131bee3d7bSdrh int lbl1; 25142282792aSdrh for(i=0; i<pGroupBy->nExpr; i++){ 25154adee20fSdanielk1977 sqlite3ExprCode(pParse, pGroupBy->a[i].pExpr); 2516efb7251dSdrh } 2517d3d39e93Sdrh /* No affinity string is attached to the following OP_MakeKey 2518d3d39e93Sdrh ** because we do not need to do any coercion of datatypes. */ 25194adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeKey, pGroupBy->nExpr, 0); 25204adee20fSdanielk1977 lbl1 = sqlite3VdbeMakeLabel(v); 25214adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_AggFocus, 0, lbl1); 2522268380caSdrh for(i=0, pAgg=pParse->aAgg; i<pParse->nAgg; i++, pAgg++){ 2523268380caSdrh if( pAgg->isAgg ) continue; 25244adee20fSdanielk1977 sqlite3ExprCode(pParse, pAgg->pExpr); 25254adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_AggSet, 0, i); 25262282792aSdrh } 25274adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, lbl1); 25282282792aSdrh } 2529268380caSdrh for(i=0, pAgg=pParse->aAgg; i<pParse->nAgg; i++, pAgg++){ 25302282792aSdrh Expr *pE; 2531268380caSdrh int nExpr; 2532268380caSdrh FuncDef *pDef; 2533268380caSdrh if( !pAgg->isAgg ) continue; 2534268380caSdrh assert( pAgg->pFunc!=0 ); 2535268380caSdrh assert( pAgg->pFunc->xStep!=0 ); 2536268380caSdrh pDef = pAgg->pFunc; 2537268380caSdrh pE = pAgg->pExpr; 2538268380caSdrh assert( pE!=0 ); 25392282792aSdrh assert( pE->op==TK_AGG_FUNCTION ); 2540f9b596ebSdrh nExpr = sqlite3ExprCodeExprList(pParse, pE->pList); 25414adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, i, 0); 2542dc1bdc4fSdanielk1977 if( pDef->needCollSeq ){ 2543dc1bdc4fSdanielk1977 CollSeq *pColl = 0; 2544dc1bdc4fSdanielk1977 int j; 2545dc1bdc4fSdanielk1977 for(j=0; !pColl && j<nExpr; j++){ 2546dc1bdc4fSdanielk1977 pColl = sqlite3ExprCollSeq(pParse, pE->pList->a[j].pExpr); 2547dc1bdc4fSdanielk1977 } 2548dc1bdc4fSdanielk1977 if( !pColl ) pColl = pParse->db->pDfltColl; 2549dc1bdc4fSdanielk1977 sqlite3VdbeOp3(v, OP_CollSeq, 0, 0, (char *)pColl, P3_COLLSEQ); 2550dc1bdc4fSdanielk1977 } 25514adee20fSdanielk1977 sqlite3VdbeOp3(v, OP_AggFunc, 0, nExpr, (char*)pDef, P3_POINTER); 25522282792aSdrh } 25532282792aSdrh } 25542282792aSdrh 2555cce7d176Sdrh /* End the database scan loop. 2556cce7d176Sdrh */ 25574adee20fSdanielk1977 sqlite3WhereEnd(pWInfo); 2558cce7d176Sdrh 25592282792aSdrh /* If we are processing aggregates, we need to set up a second loop 25602282792aSdrh ** over all of the aggregate values and process them. 25612282792aSdrh */ 25622282792aSdrh if( isAgg ){ 25634adee20fSdanielk1977 int endagg = sqlite3VdbeMakeLabel(v); 25642282792aSdrh int startagg; 25654adee20fSdanielk1977 startagg = sqlite3VdbeAddOp(v, OP_AggNext, 0, endagg); 25662282792aSdrh pParse->useAgg = 1; 25672282792aSdrh if( pHaving ){ 25684adee20fSdanielk1977 sqlite3ExprIfFalse(pParse, pHaving, startagg, 1); 25692282792aSdrh } 2570df199a25Sdrh if( selectInnerLoop(pParse, p, pEList, 0, 0, pOrderBy, distinct, eDest, 257184ac9d02Sdanielk1977 iParm, startagg, endagg, aff) ){ 25721d83f052Sdrh goto select_end; 25732282792aSdrh } 25744adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, startagg); 25754adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, endagg); 25764adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Noop, 0, 0); 25772282792aSdrh pParse->useAgg = 0; 25782282792aSdrh } 25792282792aSdrh 2580cce7d176Sdrh /* If there is an ORDER BY clause, then we need to sort the results 2581cce7d176Sdrh ** and send them to the callback one by one. 2582cce7d176Sdrh */ 2583cce7d176Sdrh if( pOrderBy ){ 2584ffbc3088Sdrh generateSortTail(pParse, p, v, pEList->nExpr, eDest, iParm); 2585cce7d176Sdrh } 25866a535340Sdrh 2587f620b4e2Sdrh /* If this was a subquery, we have now converted the subquery into a 2588f620b4e2Sdrh ** temporary table. So delete the subquery structure from the parent 2589f620b4e2Sdrh ** to prevent this subquery from being evaluated again and to force the 2590f620b4e2Sdrh ** the use of the temporary table. 2591f620b4e2Sdrh */ 2592f620b4e2Sdrh if( pParent ){ 2593f620b4e2Sdrh assert( pParent->pSrc->nSrc>parentTab ); 2594f620b4e2Sdrh assert( pParent->pSrc->a[parentTab].pSelect==p ); 25954adee20fSdanielk1977 sqlite3SelectDelete(p); 2596f620b4e2Sdrh pParent->pSrc->a[parentTab].pSelect = 0; 2597f620b4e2Sdrh } 2598f620b4e2Sdrh 25991d83f052Sdrh /* The SELECT was successfully coded. Set the return code to 0 26001d83f052Sdrh ** to indicate no errors. 26011d83f052Sdrh */ 26021d83f052Sdrh rc = 0; 26031d83f052Sdrh 26041d83f052Sdrh /* Control jumps to here if an error is encountered above, or upon 26051d83f052Sdrh ** successful coding of the SELECT. 26061d83f052Sdrh */ 26071d83f052Sdrh select_end: 26081d83f052Sdrh sqliteAggregateInfoReset(pParse); 26091d83f052Sdrh return rc; 2610cce7d176Sdrh } 2611