xref: /sqlite-3.40.0/src/select.c (revision fa173a76)
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*fa173a76Sdrh ** $Id: select.c,v 1.104 2002/07/10 21:26:01 drh Exp $
16cce7d176Sdrh */
17cce7d176Sdrh #include "sqliteInt.h"
18cce7d176Sdrh 
19cce7d176Sdrh /*
209bb61fe7Sdrh ** Allocate a new Select structure and return a pointer to that
219bb61fe7Sdrh ** structure.
22cce7d176Sdrh */
239bb61fe7Sdrh Select *sqliteSelectNew(
24daffd0e5Sdrh   ExprList *pEList,     /* which columns to include in the result */
25ad3cab52Sdrh   SrcList *pSrc,        /* the FROM clause -- which tables to scan */
26daffd0e5Sdrh   Expr *pWhere,         /* the WHERE clause */
27daffd0e5Sdrh   ExprList *pGroupBy,   /* the GROUP BY clause */
28daffd0e5Sdrh   Expr *pHaving,        /* the HAVING clause */
29daffd0e5Sdrh   ExprList *pOrderBy,   /* the ORDER BY clause */
309bbca4c1Sdrh   int isDistinct,       /* true if the DISTINCT keyword is present */
319bbca4c1Sdrh   int nLimit,           /* LIMIT value.  -1 means not used */
329bbca4c1Sdrh   int nOffset           /* OFFSET value.  -1 means not used */
339bb61fe7Sdrh ){
349bb61fe7Sdrh   Select *pNew;
359bb61fe7Sdrh   pNew = sqliteMalloc( sizeof(*pNew) );
36daffd0e5Sdrh   if( pNew==0 ){
37daffd0e5Sdrh     sqliteExprListDelete(pEList);
38ad3cab52Sdrh     sqliteSrcListDelete(pSrc);
39daffd0e5Sdrh     sqliteExprDelete(pWhere);
40daffd0e5Sdrh     sqliteExprListDelete(pGroupBy);
41daffd0e5Sdrh     sqliteExprDelete(pHaving);
42daffd0e5Sdrh     sqliteExprListDelete(pOrderBy);
43daffd0e5Sdrh   }else{
449bb61fe7Sdrh     pNew->pEList = pEList;
459bb61fe7Sdrh     pNew->pSrc = pSrc;
469bb61fe7Sdrh     pNew->pWhere = pWhere;
479bb61fe7Sdrh     pNew->pGroupBy = pGroupBy;
489bb61fe7Sdrh     pNew->pHaving = pHaving;
499bb61fe7Sdrh     pNew->pOrderBy = pOrderBy;
509bb61fe7Sdrh     pNew->isDistinct = isDistinct;
5182c3d636Sdrh     pNew->op = TK_SELECT;
529bbca4c1Sdrh     pNew->nLimit = nLimit;
539bbca4c1Sdrh     pNew->nOffset = nOffset;
54daffd0e5Sdrh   }
559bb61fe7Sdrh   return pNew;
569bb61fe7Sdrh }
579bb61fe7Sdrh 
589bb61fe7Sdrh /*
5901f3f253Sdrh ** Given 1 to 3 identifiers preceeding the JOIN keyword, determine the
6001f3f253Sdrh ** type of join.  Return an integer constant that expresses that type
6101f3f253Sdrh ** in terms of the following bit values:
6201f3f253Sdrh **
6301f3f253Sdrh **     JT_INNER
6401f3f253Sdrh **     JT_OUTER
6501f3f253Sdrh **     JT_NATURAL
6601f3f253Sdrh **     JT_LEFT
6701f3f253Sdrh **     JT_RIGHT
6801f3f253Sdrh **
6901f3f253Sdrh ** A full outer join is the combination of JT_LEFT and JT_RIGHT.
7001f3f253Sdrh **
7101f3f253Sdrh ** If an illegal or unsupported join type is seen, then still return
7201f3f253Sdrh ** a join type, but put an error in the pParse structure.
7301f3f253Sdrh */
7401f3f253Sdrh int sqliteJoinType(Parse *pParse, Token *pA, Token *pB, Token *pC){
7501f3f253Sdrh   int jointype = 0;
7601f3f253Sdrh   Token *apAll[3];
7701f3f253Sdrh   Token *p;
7801f3f253Sdrh   static struct {
7901f3f253Sdrh     const char *zKeyword;
8001f3f253Sdrh     int nChar;
8101f3f253Sdrh     int code;
8201f3f253Sdrh   } keywords[] = {
8301f3f253Sdrh     { "natural", 7, JT_NATURAL },
84195e6967Sdrh     { "left",    4, JT_LEFT|JT_OUTER },
85195e6967Sdrh     { "right",   5, JT_RIGHT|JT_OUTER },
86195e6967Sdrh     { "full",    4, JT_LEFT|JT_RIGHT|JT_OUTER },
8701f3f253Sdrh     { "outer",   5, JT_OUTER },
8801f3f253Sdrh     { "inner",   5, JT_INNER },
8901f3f253Sdrh     { "cross",   5, JT_INNER },
9001f3f253Sdrh   };
9101f3f253Sdrh   int i, j;
9201f3f253Sdrh   apAll[0] = pA;
9301f3f253Sdrh   apAll[1] = pB;
9401f3f253Sdrh   apAll[2] = pC;
95195e6967Sdrh   for(i=0; i<3 && apAll[i]; i++){
9601f3f253Sdrh     p = apAll[i];
9701f3f253Sdrh     for(j=0; j<sizeof(keywords)/sizeof(keywords[0]); j++){
9801f3f253Sdrh       if( p->n==keywords[j].nChar
9901f3f253Sdrh           && sqliteStrNICmp(p->z, keywords[j].zKeyword, p->n)==0 ){
10001f3f253Sdrh         jointype |= keywords[j].code;
10101f3f253Sdrh         break;
10201f3f253Sdrh       }
10301f3f253Sdrh     }
10401f3f253Sdrh     if( j>=sizeof(keywords)/sizeof(keywords[0]) ){
10501f3f253Sdrh       jointype |= JT_ERROR;
10601f3f253Sdrh       break;
10701f3f253Sdrh     }
10801f3f253Sdrh   }
109ad2d8307Sdrh   if(
110ad2d8307Sdrh      (jointype & (JT_INNER|JT_OUTER))==(JT_INNER|JT_OUTER) ||
111195e6967Sdrh      (jointype & JT_ERROR)!=0
112ad2d8307Sdrh   ){
11301f3f253Sdrh     static Token dummy = { 0, 0 };
11401f3f253Sdrh     char *zSp1 = " ", *zSp2 = " ";
11501f3f253Sdrh     if( pB==0 ){ pB = &dummy; zSp1 = 0; }
11601f3f253Sdrh     if( pC==0 ){ pC = &dummy; zSp2 = 0; }
11701f3f253Sdrh     sqliteSetNString(&pParse->zErrMsg, "unknown or unsupported join type: ", 0,
11801f3f253Sdrh        pA->z, pA->n, zSp1, 1, pB->z, pB->n, zSp2, 1, pC->z, pC->n, 0);
11901f3f253Sdrh     pParse->nErr++;
12001f3f253Sdrh     jointype = JT_INNER;
121195e6967Sdrh   }else if( jointype & JT_RIGHT ){
122195e6967Sdrh     sqliteSetString(&pParse->zErrMsg,
123195e6967Sdrh       "RIGHT and FULL OUTER JOINs are not currently supported", 0);
124195e6967Sdrh     pParse->nErr++;
125195e6967Sdrh     jointype = JT_INNER;
12601f3f253Sdrh   }
12701f3f253Sdrh   return jointype;
12801f3f253Sdrh }
12901f3f253Sdrh 
13001f3f253Sdrh /*
131ad2d8307Sdrh ** Return the index of a column in a table.  Return -1 if the column
132ad2d8307Sdrh ** is not contained in the table.
133ad2d8307Sdrh */
134ad2d8307Sdrh static int columnIndex(Table *pTab, const char *zCol){
135ad2d8307Sdrh   int i;
136ad2d8307Sdrh   for(i=0; i<pTab->nCol; i++){
137ad2d8307Sdrh     if( sqliteStrICmp(pTab->aCol[i].zName, zCol)==0 ) return i;
138ad2d8307Sdrh   }
139ad2d8307Sdrh   return -1;
140ad2d8307Sdrh }
141ad2d8307Sdrh 
142ad2d8307Sdrh /*
143ad2d8307Sdrh ** Add a term to the WHERE expression in *ppExpr that requires the
144ad2d8307Sdrh ** zCol column to be equal in the two tables pTab1 and pTab2.
145ad2d8307Sdrh */
146ad2d8307Sdrh static void addWhereTerm(
147ad2d8307Sdrh   const char *zCol,        /* Name of the column */
148ad2d8307Sdrh   const Table *pTab1,      /* First table */
149ad2d8307Sdrh   const Table *pTab2,      /* Second table */
150ad2d8307Sdrh   Expr **ppExpr            /* Add the equality term to this expression */
151ad2d8307Sdrh ){
152ad2d8307Sdrh   Token dummy;
153ad2d8307Sdrh   Expr *pE1a, *pE1b, *pE1c;
154ad2d8307Sdrh   Expr *pE2a, *pE2b, *pE2c;
155ad2d8307Sdrh   Expr *pE;
156ad2d8307Sdrh 
157ad2d8307Sdrh   dummy.z = zCol;
158ad2d8307Sdrh   dummy.n = strlen(zCol);
159ad2d8307Sdrh   pE1a = sqliteExpr(TK_ID, 0, 0, &dummy);
1603b167c75Sdrh   pE1a->staticToken = 1;
161ad2d8307Sdrh   pE2a = sqliteExpr(TK_ID, 0, 0, &dummy);
1623b167c75Sdrh   pE2a->staticToken = 1;
163ad2d8307Sdrh   dummy.z = pTab1->zName;
164ad2d8307Sdrh   dummy.n = strlen(dummy.z);
165ad2d8307Sdrh   pE1b = sqliteExpr(TK_ID, 0, 0, &dummy);
1663b167c75Sdrh   pE1b->staticToken = 1;
167ad2d8307Sdrh   dummy.z = pTab2->zName;
168ad2d8307Sdrh   dummy.n = strlen(dummy.z);
169ad2d8307Sdrh   pE2b = sqliteExpr(TK_ID, 0, 0, &dummy);
1703b167c75Sdrh   pE2b->staticToken = 1;
171ad2d8307Sdrh   pE1c = sqliteExpr(TK_DOT, pE1b, pE1a, 0);
172ad2d8307Sdrh   pE2c = sqliteExpr(TK_DOT, pE2b, pE2a, 0);
173ad2d8307Sdrh   pE = sqliteExpr(TK_EQ, pE1c, pE2c, 0);
1741cc093c2Sdrh   pE->isJoinExpr = 1;
175ad2d8307Sdrh   if( *ppExpr ){
176ad2d8307Sdrh     *ppExpr = sqliteExpr(TK_AND, *ppExpr, pE, 0);
177ad2d8307Sdrh   }else{
178ad2d8307Sdrh     *ppExpr = pE;
179ad2d8307Sdrh   }
180ad2d8307Sdrh }
181ad2d8307Sdrh 
182ad2d8307Sdrh /*
1831cc093c2Sdrh ** Set the Expr.isJoinExpr flag on all terms of the given expression.
1841cc093c2Sdrh **
1851cc093c2Sdrh ** The Expr.isJoinExpr flag is used at on terms of an expression to tell
1861cc093c2Sdrh ** the LEFT OUTER JOIN processing logic that this term is part of the
1871cc093c2Sdrh ** join restriction and not a part of the more general WHERE clause.
1881cc093c2Sdrh */
1891cc093c2Sdrh static void setJoinExpr(Expr *p){
1901cc093c2Sdrh   while( p ){
1911cc093c2Sdrh     p->isJoinExpr = 1;
1921cc093c2Sdrh     setJoinExpr(p->pLeft);
1931cc093c2Sdrh     p = p->pRight;
1941cc093c2Sdrh   }
1951cc093c2Sdrh }
1961cc093c2Sdrh 
1971cc093c2Sdrh /*
198ad2d8307Sdrh ** This routine processes the join information for a SELECT statement.
199ad2d8307Sdrh ** ON and USING clauses are converted into extra terms of the WHERE clause.
200ad2d8307Sdrh ** NATURAL joins also create extra WHERE clause terms.
201ad2d8307Sdrh **
202ad2d8307Sdrh ** This routine returns the number of errors encountered.
203ad2d8307Sdrh */
204ad2d8307Sdrh static int sqliteProcessJoin(Parse *pParse, Select *p){
205ad2d8307Sdrh   SrcList *pSrc;
206ad2d8307Sdrh   int i, j;
207ad2d8307Sdrh   pSrc = p->pSrc;
208ad2d8307Sdrh   for(i=0; i<pSrc->nSrc-1; i++){
209ad2d8307Sdrh     struct SrcList_item *pTerm = &pSrc->a[i];
210ad2d8307Sdrh     struct SrcList_item *pOther = &pSrc->a[i+1];
211ad2d8307Sdrh 
212ad2d8307Sdrh     if( pTerm->pTab==0 || pOther->pTab==0 ) continue;
213ad2d8307Sdrh 
214ad2d8307Sdrh     /* When the NATURAL keyword is present, add WHERE clause terms for
215ad2d8307Sdrh     ** every column that the two tables have in common.
216ad2d8307Sdrh     */
217ad2d8307Sdrh     if( pTerm->jointype & JT_NATURAL ){
218ad2d8307Sdrh       Table *pTab;
219ad2d8307Sdrh       if( pTerm->pOn || pTerm->pUsing ){
220ad2d8307Sdrh         sqliteSetString(&pParse->zErrMsg, "a NATURAL join may not have "
221ad2d8307Sdrh            "an ON or USING clause", 0);
222ad2d8307Sdrh         pParse->nErr++;
223ad2d8307Sdrh         return 1;
224ad2d8307Sdrh       }
225ad2d8307Sdrh       pTab = pTerm->pTab;
226ad2d8307Sdrh       for(j=0; j<pTab->nCol; j++){
227ad2d8307Sdrh         if( columnIndex(pOther->pTab, pTab->aCol[j].zName)>=0 ){
228ad2d8307Sdrh           addWhereTerm(pTab->aCol[j].zName, pTab, pOther->pTab, &p->pWhere);
229ad2d8307Sdrh         }
230ad2d8307Sdrh       }
231ad2d8307Sdrh     }
232ad2d8307Sdrh 
233ad2d8307Sdrh     /* Disallow both ON and USING clauses in the same join
234ad2d8307Sdrh     */
235ad2d8307Sdrh     if( pTerm->pOn && pTerm->pUsing ){
236ad2d8307Sdrh       sqliteSetString(&pParse->zErrMsg, "cannot have both ON and USING "
237ad2d8307Sdrh         "clauses in the same join", 0);
238ad2d8307Sdrh       pParse->nErr++;
239ad2d8307Sdrh       return 1;
240ad2d8307Sdrh     }
241ad2d8307Sdrh 
242ad2d8307Sdrh     /* Add the ON clause to the end of the WHERE clause, connected by
243ad2d8307Sdrh     ** and AND operator.
244ad2d8307Sdrh     */
245ad2d8307Sdrh     if( pTerm->pOn ){
2461cc093c2Sdrh       setJoinExpr(pTerm->pOn);
247ad2d8307Sdrh       if( p->pWhere==0 ){
248ad2d8307Sdrh         p->pWhere = pTerm->pOn;
249ad2d8307Sdrh       }else{
250ad2d8307Sdrh         p->pWhere = sqliteExpr(TK_AND, p->pWhere, pTerm->pOn, 0);
251ad2d8307Sdrh       }
252ad2d8307Sdrh       pTerm->pOn = 0;
253ad2d8307Sdrh     }
254ad2d8307Sdrh 
255ad2d8307Sdrh     /* Create extra terms on the WHERE clause for each column named
256ad2d8307Sdrh     ** in the USING clause.  Example: If the two tables to be joined are
257ad2d8307Sdrh     ** A and B and the USING clause names X, Y, and Z, then add this
258ad2d8307Sdrh     ** to the WHERE clause:    A.X=B.X AND A.Y=B.Y AND A.Z=B.Z
259ad2d8307Sdrh     ** Report an error if any column mentioned in the USING clause is
260ad2d8307Sdrh     ** not contained in both tables to be joined.
261ad2d8307Sdrh     */
262ad2d8307Sdrh     if( pTerm->pUsing ){
263ad2d8307Sdrh       IdList *pList;
264ad2d8307Sdrh       int j;
265ad2d8307Sdrh       assert( i<pSrc->nSrc-1 );
266ad2d8307Sdrh       pList = pTerm->pUsing;
267ad2d8307Sdrh       for(j=0; j<pList->nId; j++){
268bf5cd97eSdrh         if( columnIndex(pTerm->pTab, pList->a[j].zName)<0 ||
269bf5cd97eSdrh             columnIndex(pOther->pTab, pList->a[j].zName)<0 ){
270ad2d8307Sdrh           sqliteSetString(&pParse->zErrMsg, "cannot join using column ",
271bf5cd97eSdrh             pList->a[j].zName, " - column not present in both tables", 0);
272ad2d8307Sdrh           pParse->nErr++;
273ad2d8307Sdrh           return 1;
274ad2d8307Sdrh         }
275bf5cd97eSdrh         addWhereTerm(pList->a[j].zName, pTerm->pTab, pOther->pTab, &p->pWhere);
276ad2d8307Sdrh       }
277ad2d8307Sdrh     }
278ad2d8307Sdrh   }
279ad2d8307Sdrh   return 0;
280ad2d8307Sdrh }
281ad2d8307Sdrh 
282ad2d8307Sdrh /*
2839bb61fe7Sdrh ** Delete the given Select structure and all of its substructures.
2849bb61fe7Sdrh */
2859bb61fe7Sdrh void sqliteSelectDelete(Select *p){
28682c3d636Sdrh   if( p==0 ) return;
2879bb61fe7Sdrh   sqliteExprListDelete(p->pEList);
288ad3cab52Sdrh   sqliteSrcListDelete(p->pSrc);
2899bb61fe7Sdrh   sqliteExprDelete(p->pWhere);
2909bb61fe7Sdrh   sqliteExprListDelete(p->pGroupBy);
2919bb61fe7Sdrh   sqliteExprDelete(p->pHaving);
2929bb61fe7Sdrh   sqliteExprListDelete(p->pOrderBy);
29382c3d636Sdrh   sqliteSelectDelete(p->pPrior);
294a76b5dfcSdrh   sqliteFree(p->zSelect);
2959bb61fe7Sdrh   sqliteFree(p);
2969bb61fe7Sdrh }
2979bb61fe7Sdrh 
2989bb61fe7Sdrh /*
2992282792aSdrh ** Delete the aggregate information from the parse structure.
3002282792aSdrh */
3011d83f052Sdrh static void sqliteAggregateInfoReset(Parse *pParse){
3022282792aSdrh   sqliteFree(pParse->aAgg);
3032282792aSdrh   pParse->aAgg = 0;
3042282792aSdrh   pParse->nAgg = 0;
3052282792aSdrh   pParse->useAgg = 0;
3062282792aSdrh }
3072282792aSdrh 
3082282792aSdrh /*
309c926afbcSdrh ** Insert code into "v" that will push the record on the top of the
310c926afbcSdrh ** stack into the sorter.
311c926afbcSdrh */
312c926afbcSdrh static void pushOntoSorter(Parse *pParse, Vdbe *v, ExprList *pOrderBy){
313c926afbcSdrh   char *zSortOrder;
314c926afbcSdrh   int i;
315c926afbcSdrh   zSortOrder = sqliteMalloc( pOrderBy->nExpr + 1 );
316c926afbcSdrh   if( zSortOrder==0 ) return;
317c926afbcSdrh   for(i=0; i<pOrderBy->nExpr; i++){
31838640e15Sdrh     int order = pOrderBy->a[i].sortOrder;
31938640e15Sdrh     int type;
32038640e15Sdrh     int c;
32138640e15Sdrh     if( (order & SQLITE_SO_TYPEMASK)==SQLITE_SO_TEXT ){
32238640e15Sdrh       type = SQLITE_SO_TEXT;
32338640e15Sdrh     }else if( (order & SQLITE_SO_TYPEMASK)==SQLITE_SO_NUM ){
32438640e15Sdrh       type = SQLITE_SO_NUM;
32538640e15Sdrh     }else if( pParse->db->file_format>=3 ){
32638640e15Sdrh       type = sqliteExprType(pOrderBy->a[i].pExpr);
32738640e15Sdrh     }else{
32838640e15Sdrh       type = SQLITE_SO_NUM;
32938640e15Sdrh     }
33038640e15Sdrh     if( (order & SQLITE_SO_DIRMASK)==SQLITE_SO_ASC ){
33138640e15Sdrh       c = type==SQLITE_SO_TEXT ? 'A' : '+';
33238640e15Sdrh     }else{
33338640e15Sdrh       c = type==SQLITE_SO_TEXT ? 'D' : '-';
33438640e15Sdrh     }
33538640e15Sdrh     zSortOrder[i] = c;
336c926afbcSdrh     sqliteExprCode(pParse, pOrderBy->a[i].pExpr);
337c926afbcSdrh   }
338c926afbcSdrh   zSortOrder[pOrderBy->nExpr] = 0;
339c926afbcSdrh   sqliteVdbeAddOp(v, OP_SortMakeKey, pOrderBy->nExpr, 0);
340c926afbcSdrh   sqliteVdbeChangeP3(v, -1, zSortOrder, strlen(zSortOrder));
341c926afbcSdrh   sqliteFree(zSortOrder);
342c926afbcSdrh   sqliteVdbeAddOp(v, OP_SortPut, 0, 0);
343c926afbcSdrh }
344c926afbcSdrh 
345c926afbcSdrh /*
34638640e15Sdrh ** This routine adds a P3 argument to the last VDBE opcode that was
34738640e15Sdrh ** inserted. The P3 argument added is a string suitable for the
34838640e15Sdrh ** OP_MakeKey or OP_MakeIdxKey opcodes.  The string consists of
34938640e15Sdrh ** characters 't' or 'n' depending on whether or not the various
35038640e15Sdrh ** fields of the key to be generated should be treated as numeric
35138640e15Sdrh ** or as text.  See the OP_MakeKey and OP_MakeIdxKey opcode
35238640e15Sdrh ** documentation for additional information about the P3 string.
35338640e15Sdrh ** See also the sqliteAddIdxKeyType() routine.
35438640e15Sdrh */
35538640e15Sdrh void sqliteAddKeyType(Vdbe *v, ExprList *pEList){
35638640e15Sdrh   int nColumn = pEList->nExpr;
35738640e15Sdrh   char *zType = sqliteMalloc( nColumn+1 );
35838640e15Sdrh   int i;
35938640e15Sdrh   if( zType==0 ) return;
36038640e15Sdrh   for(i=0; i<nColumn; i++){
36138640e15Sdrh     zType[i] = sqliteExprType(pEList->a[i].pExpr)==SQLITE_SO_NUM ? 'n' : 't';
36238640e15Sdrh   }
36338640e15Sdrh   zType[i] = 0;
36438640e15Sdrh   sqliteVdbeChangeP3(v, -1, zType, nColumn);
36538640e15Sdrh   sqliteFree(zType);
36638640e15Sdrh }
36738640e15Sdrh 
36838640e15Sdrh /*
3692282792aSdrh ** This routine generates the code for the inside of the inner loop
3702282792aSdrh ** of a SELECT.
37182c3d636Sdrh **
37238640e15Sdrh ** If srcTab and nColumn are both zero, then the pEList expressions
37338640e15Sdrh ** are evaluated in order to get the data for this row.  If nColumn>0
37438640e15Sdrh ** then data is pulled from srcTab and pEList is used only to get the
37538640e15Sdrh ** datatypes for each column.
3762282792aSdrh */
3772282792aSdrh static int selectInnerLoop(
3782282792aSdrh   Parse *pParse,          /* The parser context */
379df199a25Sdrh   Select *p,              /* The complete select statement being coded */
3802282792aSdrh   ExprList *pEList,       /* List of values being extracted */
38182c3d636Sdrh   int srcTab,             /* Pull data from this table */
382967e8b73Sdrh   int nColumn,            /* Number of columns in the source table */
3832282792aSdrh   ExprList *pOrderBy,     /* If not NULL, sort results using this key */
3842282792aSdrh   int distinct,           /* If >=0, make sure results are distinct */
3852282792aSdrh   int eDest,              /* How to dispose of the results */
3862282792aSdrh   int iParm,              /* An argument to the disposal method */
3872282792aSdrh   int iContinue,          /* Jump here to continue with next row */
3882282792aSdrh   int iBreak              /* Jump here to break out of the inner loop */
3892282792aSdrh ){
3902282792aSdrh   Vdbe *v = pParse->pVdbe;
3912282792aSdrh   int i;
39238640e15Sdrh 
393daffd0e5Sdrh   if( v==0 ) return 0;
39438640e15Sdrh   assert( pEList!=0 );
3952282792aSdrh 
396df199a25Sdrh   /* If there was a LIMIT clause on the SELECT statement, then do the check
397df199a25Sdrh   ** to see if this row should be output.
398df199a25Sdrh   */
399df199a25Sdrh   if( pOrderBy==0 ){
400df199a25Sdrh     if( p->nOffset>0 ){
401d11d382cSdrh       int addr = sqliteVdbeCurrentAddr(v);
402d11d382cSdrh       sqliteVdbeAddOp(v, OP_MemIncr, p->nOffset, addr+2);
403d11d382cSdrh       sqliteVdbeAddOp(v, OP_Goto, 0, iContinue);
404df199a25Sdrh     }
405d11d382cSdrh     if( p->nLimit>=0 ){
406d11d382cSdrh       sqliteVdbeAddOp(v, OP_MemIncr, p->nLimit, iBreak);
407df199a25Sdrh     }
408df199a25Sdrh   }
409df199a25Sdrh 
410967e8b73Sdrh   /* Pull the requested columns.
4112282792aSdrh   */
41238640e15Sdrh   if( nColumn>0 ){
413967e8b73Sdrh     for(i=0; i<nColumn; i++){
41499fcd718Sdrh       sqliteVdbeAddOp(v, OP_Column, srcTab, i);
41582c3d636Sdrh     }
41638640e15Sdrh   }else{
41738640e15Sdrh     nColumn = pEList->nExpr;
41838640e15Sdrh     for(i=0; i<pEList->nExpr; i++){
41938640e15Sdrh       sqliteExprCode(pParse, pEList->a[i].pExpr);
42038640e15Sdrh     }
42182c3d636Sdrh   }
4222282792aSdrh 
423daffd0e5Sdrh   /* If the DISTINCT keyword was present on the SELECT statement
424daffd0e5Sdrh   ** and this row has been seen before, then do not make this row
425daffd0e5Sdrh   ** part of the result.
4262282792aSdrh   */
427f5905aa7Sdrh   if( distinct>=0 && pEList && pEList->nExpr>0 ){
4280bd1f4eaSdrh #if NULL_ALWAYS_DISTINCT
4290bd1f4eaSdrh     sqliteVdbeAddOp(v, OP_IsNull, -pEList->nExpr, sqliteVdbeCurrentAddr(v)+7);
4300bd1f4eaSdrh #endif
43199fcd718Sdrh     sqliteVdbeAddOp(v, OP_MakeKey, pEList->nExpr, 1);
43238640e15Sdrh     if( pParse->db->file_format>=3 ) sqliteAddKeyType(v, pEList);
433f5905aa7Sdrh     sqliteVdbeAddOp(v, OP_Distinct, distinct, sqliteVdbeCurrentAddr(v)+3);
43499fcd718Sdrh     sqliteVdbeAddOp(v, OP_Pop, pEList->nExpr+1, 0);
43599fcd718Sdrh     sqliteVdbeAddOp(v, OP_Goto, 0, iContinue);
43699fcd718Sdrh     sqliteVdbeAddOp(v, OP_String, 0, 0);
4376b12545fSdrh     sqliteVdbeAddOp(v, OP_PutStrKey, distinct, 0);
4382282792aSdrh   }
43982c3d636Sdrh 
440c926afbcSdrh   switch( eDest ){
44182c3d636Sdrh     /* In this mode, write each query result to the key of the temporary
44282c3d636Sdrh     ** table iParm.
4432282792aSdrh     */
444c926afbcSdrh     case SRT_Union: {
4450bd1f4eaSdrh       sqliteVdbeAddOp(v, OP_MakeRecord, nColumn, NULL_ALWAYS_DISTINCT);
446f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
4476b12545fSdrh       sqliteVdbeAddOp(v, OP_PutStrKey, iParm, 0);
448c926afbcSdrh       break;
449c926afbcSdrh     }
45082c3d636Sdrh 
4515974a30fSdrh     /* Store the result as data using a unique key.
4525974a30fSdrh     */
453c926afbcSdrh     case SRT_Table:
454c926afbcSdrh     case SRT_TempTable: {
45599fcd718Sdrh       sqliteVdbeAddOp(v, OP_MakeRecord, nColumn, 0);
456c926afbcSdrh       if( pOrderBy ){
457c926afbcSdrh         pushOntoSorter(pParse, v, pOrderBy);
458c926afbcSdrh       }else{
45999fcd718Sdrh         sqliteVdbeAddOp(v, OP_NewRecno, iParm, 0);
46099fcd718Sdrh         sqliteVdbeAddOp(v, OP_Pull, 1, 0);
4616b12545fSdrh         sqliteVdbeAddOp(v, OP_PutIntKey, iParm, 0);
462c926afbcSdrh       }
463c926afbcSdrh       break;
464c926afbcSdrh     }
4655974a30fSdrh 
46682c3d636Sdrh     /* Construct a record from the query result, but instead of
46782c3d636Sdrh     ** saving that record, use it as a key to delete elements from
46882c3d636Sdrh     ** the temporary table iParm.
46982c3d636Sdrh     */
470c926afbcSdrh     case SRT_Except: {
4710bd1f4eaSdrh       int addr;
4720bd1f4eaSdrh       addr = sqliteVdbeAddOp(v, OP_MakeRecord, nColumn, NULL_ALWAYS_DISTINCT);
47399fcd718Sdrh       sqliteVdbeAddOp(v, OP_NotFound, iParm, addr+3);
47499fcd718Sdrh       sqliteVdbeAddOp(v, OP_Delete, iParm, 0);
475c926afbcSdrh       break;
476c926afbcSdrh     }
4772282792aSdrh 
4782282792aSdrh     /* If we are creating a set for an "expr IN (SELECT ...)" construct,
4792282792aSdrh     ** then there should be a single item on the stack.  Write this
4802282792aSdrh     ** item into the set table with bogus data.
4812282792aSdrh     */
482c926afbcSdrh     case SRT_Set: {
483a9f9d1c0Sdrh       int lbl = sqliteVdbeMakeLabel(v);
484967e8b73Sdrh       assert( nColumn==1 );
485a9f9d1c0Sdrh       sqliteVdbeAddOp(v, OP_IsNull, -1, lbl);
486c926afbcSdrh       if( pOrderBy ){
487c926afbcSdrh         pushOntoSorter(pParse, v, pOrderBy);
488c926afbcSdrh       }else{
489a9f9d1c0Sdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
4906b12545fSdrh         sqliteVdbeAddOp(v, OP_PutStrKey, iParm, 0);
491c926afbcSdrh       }
492a9f9d1c0Sdrh       sqliteVdbeResolveLabel(v, lbl);
493c926afbcSdrh       break;
494c926afbcSdrh     }
49582c3d636Sdrh 
4962282792aSdrh     /* If this is a scalar select that is part of an expression, then
4972282792aSdrh     ** store the results in the appropriate memory cell and break out
4982282792aSdrh     ** of the scan loop.
4992282792aSdrh     */
500c926afbcSdrh     case SRT_Mem: {
501967e8b73Sdrh       assert( nColumn==1 );
502c926afbcSdrh       if( pOrderBy ){
503c926afbcSdrh         pushOntoSorter(pParse, v, pOrderBy);
504c926afbcSdrh       }else{
5058721ce4aSdrh         sqliteVdbeAddOp(v, OP_MemStore, iParm, 1);
50699fcd718Sdrh         sqliteVdbeAddOp(v, OP_Goto, 0, iBreak);
507c926afbcSdrh       }
508c926afbcSdrh       break;
509c926afbcSdrh     }
5102282792aSdrh 
511f46f905aSdrh     /* Send the data to the callback function.
512f46f905aSdrh     */
513f46f905aSdrh     case SRT_Callback:
514f46f905aSdrh     case SRT_Sorter: {
515f46f905aSdrh       if( pOrderBy ){
516f46f905aSdrh         sqliteVdbeAddOp(v, OP_SortMakeRec, nColumn, 0);
517f46f905aSdrh         pushOntoSorter(pParse, v, pOrderBy);
518f46f905aSdrh       }else{
519f46f905aSdrh         assert( eDest==SRT_Callback );
520f46f905aSdrh         sqliteVdbeAddOp(v, OP_Callback, nColumn, 0);
521f46f905aSdrh       }
522f46f905aSdrh       break;
523f46f905aSdrh     }
524f46f905aSdrh 
525d7489c39Sdrh     /* Discard the results.  This is used for SELECT statements inside
526d7489c39Sdrh     ** the body of a TRIGGER.  The purpose of such selects is to call
527d7489c39Sdrh     ** user-defined functions that have side effects.  We do not care
528d7489c39Sdrh     ** about the actual results of the select.
529d7489c39Sdrh     */
530c926afbcSdrh     default: {
531f46f905aSdrh       assert( eDest==SRT_Discard );
532f46f905aSdrh       sqliteVdbeAddOp(v, OP_Pop, nColumn, 0);
533c926afbcSdrh       break;
534c926afbcSdrh     }
535c926afbcSdrh   }
53682c3d636Sdrh   return 0;
53782c3d636Sdrh }
53882c3d636Sdrh 
53982c3d636Sdrh /*
540d8bc7086Sdrh ** If the inner loop was generated using a non-null pOrderBy argument,
541d8bc7086Sdrh ** then the results were placed in a sorter.  After the loop is terminated
542d8bc7086Sdrh ** we need to run the sorter and output the results.  The following
543d8bc7086Sdrh ** routine generates the code needed to do that.
544d8bc7086Sdrh */
545c926afbcSdrh static void generateSortTail(
546c926afbcSdrh   Select *p,       /* The SELECT statement */
547c926afbcSdrh   Vdbe *v,         /* Generate code into this VDBE */
548c926afbcSdrh   int nColumn,     /* Number of columns of data */
549c926afbcSdrh   int eDest,       /* Write the sorted results here */
550c926afbcSdrh   int iParm        /* Optional parameter associated with eDest */
551c926afbcSdrh ){
552d8bc7086Sdrh   int end = sqliteVdbeMakeLabel(v);
553d8bc7086Sdrh   int addr;
554f46f905aSdrh   if( eDest==SRT_Sorter ) return;
55599fcd718Sdrh   sqliteVdbeAddOp(v, OP_Sort, 0, 0);
55699fcd718Sdrh   addr = sqliteVdbeAddOp(v, OP_SortNext, 0, end);
557df199a25Sdrh   if( p->nOffset>0 ){
558d11d382cSdrh     sqliteVdbeAddOp(v, OP_MemIncr, p->nOffset, addr+4);
559d11d382cSdrh     sqliteVdbeAddOp(v, OP_Pop, 1, 0);
560d11d382cSdrh     sqliteVdbeAddOp(v, OP_Goto, 0, addr);
561df199a25Sdrh   }
562d11d382cSdrh   if( p->nLimit>=0 ){
563d11d382cSdrh     sqliteVdbeAddOp(v, OP_MemIncr, p->nLimit, end);
564df199a25Sdrh   }
565c926afbcSdrh   switch( eDest ){
566c926afbcSdrh     case SRT_Callback: {
567df199a25Sdrh       sqliteVdbeAddOp(v, OP_SortCallback, nColumn, 0);
568c926afbcSdrh       break;
569c926afbcSdrh     }
570c926afbcSdrh     case SRT_Table:
571c926afbcSdrh     case SRT_TempTable: {
572c926afbcSdrh       sqliteVdbeAddOp(v, OP_NewRecno, iParm, 0);
573c926afbcSdrh       sqliteVdbeAddOp(v, OP_Pull, 1, 0);
574c926afbcSdrh       sqliteVdbeAddOp(v, OP_PutIntKey, iParm, 0);
575c926afbcSdrh       break;
576c926afbcSdrh     }
577c926afbcSdrh     case SRT_Set: {
578c926afbcSdrh       assert( nColumn==1 );
579c926afbcSdrh       sqliteVdbeAddOp(v, OP_IsNull, -1, sqliteVdbeCurrentAddr(v)+3);
580c926afbcSdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
581c926afbcSdrh       sqliteVdbeAddOp(v, OP_PutStrKey, iParm, 0);
582c926afbcSdrh       break;
583c926afbcSdrh     }
584c926afbcSdrh     case SRT_Mem: {
585c926afbcSdrh       assert( nColumn==1 );
586c926afbcSdrh       sqliteVdbeAddOp(v, OP_MemStore, iParm, 1);
587c926afbcSdrh       sqliteVdbeAddOp(v, OP_Goto, 0, end);
588c926afbcSdrh       break;
589c926afbcSdrh     }
590c926afbcSdrh     default: {
591f46f905aSdrh       /* Do nothing */
592c926afbcSdrh       break;
593c926afbcSdrh     }
594c926afbcSdrh   }
59599fcd718Sdrh   sqliteVdbeAddOp(v, OP_Goto, 0, addr);
59699fcd718Sdrh   sqliteVdbeResolveLabel(v, end);
597a8b38d28Sdrh   sqliteVdbeAddOp(v, OP_SortReset, 0, 0);
598d8bc7086Sdrh }
599d8bc7086Sdrh 
600d8bc7086Sdrh /*
60182c3d636Sdrh ** Generate code that will tell the VDBE how many columns there
60282c3d636Sdrh ** are in the result and the name for each column.  This information
60382c3d636Sdrh ** is used to provide "argc" and "azCol[]" values in the callback.
60482c3d636Sdrh */
605832508b7Sdrh static void generateColumnNames(
606832508b7Sdrh   Parse *pParse,      /* Parser context */
607832508b7Sdrh   int base,           /* VDBE cursor corresponding to first entry in pTabList */
608ad3cab52Sdrh   SrcList *pTabList,  /* List of tables */
609832508b7Sdrh   ExprList *pEList    /* Expressions defining the result set */
610832508b7Sdrh ){
611d8bc7086Sdrh   Vdbe *v = pParse->pVdbe;
61282c3d636Sdrh   int i;
613daffd0e5Sdrh   if( pParse->colNamesSet || v==0 || sqlite_malloc_failed ) return;
614d8bc7086Sdrh   pParse->colNamesSet = 1;
615b1363206Sdrh   sqliteVdbeAddOp(v, OP_ColumnCount, pEList->nExpr*2+1, 0);
61682c3d636Sdrh   for(i=0; i<pEList->nExpr; i++){
61782c3d636Sdrh     Expr *p;
618b1363206Sdrh     char *zType = 0;
6191bee3d7bSdrh     int showFullNames;
62082c3d636Sdrh     if( pEList->a[i].zName ){
62182c3d636Sdrh       char *zName = pEList->a[i].zName;
62299fcd718Sdrh       sqliteVdbeAddOp(v, OP_ColumnName, i, 0);
62399fcd718Sdrh       sqliteVdbeChangeP3(v, -1, zName, strlen(zName));
62482c3d636Sdrh       continue;
62582c3d636Sdrh     }
62682c3d636Sdrh     p = pEList->a[i].pExpr;
627daffd0e5Sdrh     if( p==0 ) continue;
6281bee3d7bSdrh     showFullNames = (pParse->db->flags & SQLITE_FullColNames)!=0;
629*fa173a76Sdrh     if( p->op==TK_COLUMN && pTabList ){
630832508b7Sdrh       Table *pTab = pTabList->a[p->iTable - base].pTab;
63197665873Sdrh       char *zCol;
6328aff1015Sdrh       int iCol = p->iColumn;
6338aff1015Sdrh       if( iCol<0 ) iCol = pTab->iPKey;
63497665873Sdrh       assert( iCol==-1 || (iCol>=0 && iCol<pTab->nCol) );
635b1363206Sdrh       if( iCol<0 ){
636b1363206Sdrh         zCol = "_ROWID_";
637b1363206Sdrh         zType = "INTEGER";
638b1363206Sdrh       }else{
639b1363206Sdrh         zCol = pTab->aCol[iCol].zName;
640b1363206Sdrh         zType = pTab->aCol[iCol].zType;
641b1363206Sdrh       }
642*fa173a76Sdrh       if( p->span.z && p->span.z[0] && !showFullNames ){
643*fa173a76Sdrh         int addr = sqliteVdbeAddOp(v,OP_ColumnName, i, 0);
644*fa173a76Sdrh         sqliteVdbeChangeP3(v, -1, p->span.z, p->span.n);
645*fa173a76Sdrh         sqliteVdbeCompressSpace(v, addr);
646*fa173a76Sdrh       }else if( pTabList->nSrc>1 || showFullNames ){
64782c3d636Sdrh         char *zName = 0;
64882c3d636Sdrh         char *zTab;
64982c3d636Sdrh 
650832508b7Sdrh         zTab = pTabList->a[p->iTable - base].zAlias;
65101a34661Sdrh         if( showFullNames || zTab==0 ) zTab = pTab->zName;
65297665873Sdrh         sqliteSetString(&zName, zTab, ".", zCol, 0);
65399fcd718Sdrh         sqliteVdbeAddOp(v, OP_ColumnName, i, 0);
65499fcd718Sdrh         sqliteVdbeChangeP3(v, -1, zName, strlen(zName));
65582c3d636Sdrh         sqliteFree(zName);
65682c3d636Sdrh       }else{
65799fcd718Sdrh         sqliteVdbeAddOp(v, OP_ColumnName, i, 0);
65822f70c32Sdrh         sqliteVdbeChangeP3(v, -1, zCol, 0);
65982c3d636Sdrh       }
660*fa173a76Sdrh     }else if( p->span.z && p->span.z[0] && !showFullNames ){
661*fa173a76Sdrh       int addr = sqliteVdbeAddOp(v,OP_ColumnName, i, 0);
662*fa173a76Sdrh       sqliteVdbeChangeP3(v, -1, p->span.z, p->span.n);
663*fa173a76Sdrh       sqliteVdbeCompressSpace(v, addr);
6641bee3d7bSdrh     }else if( p->span.z && p->span.z[0] ){
6651bee3d7bSdrh       int addr = sqliteVdbeAddOp(v,OP_ColumnName, i, 0);
6661bee3d7bSdrh       sqliteVdbeChangeP3(v, -1, p->span.z, p->span.n);
6671bee3d7bSdrh       sqliteVdbeCompressSpace(v, addr);
6681bee3d7bSdrh     }else{
6691bee3d7bSdrh       char zName[30];
6701bee3d7bSdrh       assert( p->op!=TK_COLUMN || pTabList==0 );
6711bee3d7bSdrh       sprintf(zName, "column%d", i+1);
6721bee3d7bSdrh       sqliteVdbeAddOp(v, OP_ColumnName, i, 0);
6731bee3d7bSdrh       sqliteVdbeChangeP3(v, -1, zName, strlen(zName));
67482c3d636Sdrh     }
675b1363206Sdrh     if( zType==0 ){
676b1363206Sdrh       if( sqliteExprType(p)==SQLITE_SO_TEXT ){
677b1363206Sdrh         zType = "TEXT";
678b1363206Sdrh       }else{
679b1363206Sdrh         zType = "NUMERIC";
680b1363206Sdrh       }
681b1363206Sdrh     }
682b1363206Sdrh     sqliteVdbeAddOp(v, OP_ColumnName, i + pEList->nExpr + 1, 0);
683b1363206Sdrh     sqliteVdbeChangeP3(v, -1, zType, P3_STATIC);
68482c3d636Sdrh   }
68582c3d636Sdrh }
68682c3d636Sdrh 
68782c3d636Sdrh /*
688d8bc7086Sdrh ** Name of the connection operator, used for error messages.
689d8bc7086Sdrh */
690d8bc7086Sdrh static const char *selectOpName(int id){
691d8bc7086Sdrh   char *z;
692d8bc7086Sdrh   switch( id ){
693d8bc7086Sdrh     case TK_ALL:       z = "UNION ALL";   break;
694d8bc7086Sdrh     case TK_INTERSECT: z = "INTERSECT";   break;
695d8bc7086Sdrh     case TK_EXCEPT:    z = "EXCEPT";      break;
696d8bc7086Sdrh     default:           z = "UNION";       break;
697d8bc7086Sdrh   }
698d8bc7086Sdrh   return z;
699d8bc7086Sdrh }
700d8bc7086Sdrh 
701d8bc7086Sdrh /*
70222f70c32Sdrh ** Given a SELECT statement, generate a Table structure that describes
70322f70c32Sdrh ** the result set of that SELECT.
70422f70c32Sdrh */
70522f70c32Sdrh Table *sqliteResultSetOfSelect(Parse *pParse, char *zTabName, Select *pSelect){
70622f70c32Sdrh   Table *pTab;
70722f70c32Sdrh   int i;
70822f70c32Sdrh   ExprList *pEList;
70922f70c32Sdrh   static int fillInColumnList(Parse*, Select*);
71022f70c32Sdrh 
71122f70c32Sdrh   if( fillInColumnList(pParse, pSelect) ){
71222f70c32Sdrh     return 0;
71322f70c32Sdrh   }
71422f70c32Sdrh   pTab = sqliteMalloc( sizeof(Table) );
71522f70c32Sdrh   if( pTab==0 ){
71622f70c32Sdrh     return 0;
71722f70c32Sdrh   }
71822f70c32Sdrh   pTab->zName = zTabName ? sqliteStrDup(zTabName) : 0;
71922f70c32Sdrh   pEList = pSelect->pEList;
72022f70c32Sdrh   pTab->nCol = pEList->nExpr;
721417be79cSdrh   assert( pTab->nCol>0 );
72222f70c32Sdrh   pTab->aCol = sqliteMalloc( sizeof(pTab->aCol[0])*pTab->nCol );
72322f70c32Sdrh   for(i=0; i<pTab->nCol; i++){
72422f70c32Sdrh     Expr *p;
72522f70c32Sdrh     if( pEList->a[i].zName ){
72622f70c32Sdrh       pTab->aCol[i].zName = sqliteStrDup(pEList->a[i].zName);
72722f70c32Sdrh     }else if( (p=pEList->a[i].pExpr)->span.z && p->span.z[0] ){
72822f70c32Sdrh       sqliteSetNString(&pTab->aCol[i].zName, p->span.z, p->span.n, 0);
729d820cb1bSdrh     }else if( p->op==TK_DOT && p->pRight && p->pRight->token.z &&
730d820cb1bSdrh            p->pRight->token.z[0] ){
731d820cb1bSdrh       sqliteSetNString(&pTab->aCol[i].zName,
732d820cb1bSdrh            p->pRight->token.z, p->pRight->token.n, 0);
73322f70c32Sdrh     }else{
73422f70c32Sdrh       char zBuf[30];
73522f70c32Sdrh       sprintf(zBuf, "column%d", i+1);
73622f70c32Sdrh       pTab->aCol[i].zName = sqliteStrDup(zBuf);
73722f70c32Sdrh     }
73822f70c32Sdrh   }
73922f70c32Sdrh   pTab->iPKey = -1;
74022f70c32Sdrh   return pTab;
74122f70c32Sdrh }
74222f70c32Sdrh 
74322f70c32Sdrh /*
744ad2d8307Sdrh ** For the given SELECT statement, do three things.
745d8bc7086Sdrh **
746ad3cab52Sdrh **    (1)  Fill in the pTabList->a[].pTab fields in the SrcList that
747967e8b73Sdrh **         defines the set of tables that should be scanned.
748d8bc7086Sdrh **
749ad2d8307Sdrh **    (2)  Add terms to the WHERE clause to accomodate the NATURAL keyword
750ad2d8307Sdrh **         on joins and the ON and USING clause of joins.
751ad2d8307Sdrh **
752ad2d8307Sdrh **    (3)  Scan the list of columns in the result set (pEList) looking
75354473229Sdrh **         for instances of the "*" operator or the TABLE.* operator.
75454473229Sdrh **         If found, expand each "*" to be every column in every table
75554473229Sdrh **         and TABLE.* to be every column in TABLE.
756d8bc7086Sdrh **
757d8bc7086Sdrh ** Return 0 on success.  If there are problems, leave an error message
758d8bc7086Sdrh ** in pParse and return non-zero.
759d8bc7086Sdrh */
760d8bc7086Sdrh static int fillInColumnList(Parse *pParse, Select *p){
76154473229Sdrh   int i, j, k, rc;
762ad3cab52Sdrh   SrcList *pTabList;
763daffd0e5Sdrh   ExprList *pEList;
764a76b5dfcSdrh   Table *pTab;
765daffd0e5Sdrh 
766daffd0e5Sdrh   if( p==0 || p->pSrc==0 ) return 1;
767daffd0e5Sdrh   pTabList = p->pSrc;
768daffd0e5Sdrh   pEList = p->pEList;
769d8bc7086Sdrh 
770d8bc7086Sdrh   /* Look up every table in the table list.
771d8bc7086Sdrh   */
772ad3cab52Sdrh   for(i=0; i<pTabList->nSrc; i++){
773d8bc7086Sdrh     if( pTabList->a[i].pTab ){
774d8bc7086Sdrh       /* This routine has run before!  No need to continue */
775d8bc7086Sdrh       return 0;
776d8bc7086Sdrh     }
777daffd0e5Sdrh     if( pTabList->a[i].zName==0 ){
77822f70c32Sdrh       /* A sub-query in the FROM clause of a SELECT */
77922f70c32Sdrh       assert( pTabList->a[i].pSelect!=0 );
780ad2d8307Sdrh       if( pTabList->a[i].zAlias==0 ){
781ad2d8307Sdrh         char zFakeName[60];
782ad2d8307Sdrh         sprintf(zFakeName, "sqlite_subquery_%p_",
783ad2d8307Sdrh            (void*)pTabList->a[i].pSelect);
784ad2d8307Sdrh         sqliteSetString(&pTabList->a[i].zAlias, zFakeName, 0);
785ad2d8307Sdrh       }
78622f70c32Sdrh       pTabList->a[i].pTab = pTab =
78722f70c32Sdrh         sqliteResultSetOfSelect(pParse, pTabList->a[i].zAlias,
78822f70c32Sdrh                                         pTabList->a[i].pSelect);
78922f70c32Sdrh       if( pTab==0 ){
790daffd0e5Sdrh         return 1;
791daffd0e5Sdrh       }
79222f70c32Sdrh       pTab->isTransient = 1;
79322f70c32Sdrh     }else{
794a76b5dfcSdrh       /* An ordinary table or view name in the FROM clause */
795a76b5dfcSdrh       pTabList->a[i].pTab = pTab =
796a76b5dfcSdrh         sqliteFindTable(pParse->db, pTabList->a[i].zName);
797a76b5dfcSdrh       if( pTab==0 ){
798d8bc7086Sdrh         sqliteSetString(&pParse->zErrMsg, "no such table: ",
799d8bc7086Sdrh            pTabList->a[i].zName, 0);
800d8bc7086Sdrh         pParse->nErr++;
801d8bc7086Sdrh         return 1;
802d8bc7086Sdrh       }
803a76b5dfcSdrh       if( pTab->pSelect ){
804417be79cSdrh         if( sqliteViewGetColumnNames(pParse, pTab) ){
805417be79cSdrh           return 1;
806417be79cSdrh         }
807ff78bd2fSdrh         pTabList->a[i].pSelect = sqliteSelectDup(pTab->pSelect);
808a76b5dfcSdrh       }
809d8bc7086Sdrh     }
81022f70c32Sdrh   }
811d8bc7086Sdrh 
812ad2d8307Sdrh   /* Process NATURAL keywords, and ON and USING clauses of joins.
813ad2d8307Sdrh   */
814ad2d8307Sdrh   if( sqliteProcessJoin(pParse, p) ) return 1;
815ad2d8307Sdrh 
8167c917d19Sdrh   /* For every "*" that occurs in the column list, insert the names of
81754473229Sdrh   ** all columns in all tables.  And for every TABLE.* insert the names
81854473229Sdrh   ** of all columns in TABLE.  The parser inserted a special expression
8197c917d19Sdrh   ** with the TK_ALL operator for each "*" that it found in the column list.
8207c917d19Sdrh   ** The following code just has to locate the TK_ALL expressions and expand
8217c917d19Sdrh   ** each one to the list of all columns in all tables.
82254473229Sdrh   **
82354473229Sdrh   ** The first loop just checks to see if there are any "*" operators
82454473229Sdrh   ** that need expanding.
825d8bc7086Sdrh   */
8267c917d19Sdrh   for(k=0; k<pEList->nExpr; k++){
82754473229Sdrh     Expr *pE = pEList->a[k].pExpr;
82854473229Sdrh     if( pE->op==TK_ALL ) break;
82954473229Sdrh     if( pE->op==TK_DOT && pE->pRight && pE->pRight->op==TK_ALL
83054473229Sdrh          && pE->pLeft && pE->pLeft->op==TK_ID ) break;
8317c917d19Sdrh   }
83254473229Sdrh   rc = 0;
8337c917d19Sdrh   if( k<pEList->nExpr ){
83454473229Sdrh     /*
83554473229Sdrh     ** If we get here it means the result set contains one or more "*"
83654473229Sdrh     ** operators that need to be expanded.  Loop through each expression
83754473229Sdrh     ** in the result set and expand them one by one.
83854473229Sdrh     */
8397c917d19Sdrh     struct ExprList_item *a = pEList->a;
8407c917d19Sdrh     ExprList *pNew = 0;
8417c917d19Sdrh     for(k=0; k<pEList->nExpr; k++){
84254473229Sdrh       Expr *pE = a[k].pExpr;
84354473229Sdrh       if( pE->op!=TK_ALL &&
84454473229Sdrh            (pE->op!=TK_DOT || pE->pRight==0 || pE->pRight->op!=TK_ALL) ){
84554473229Sdrh         /* This particular expression does not need to be expanded.
84654473229Sdrh         */
8477c917d19Sdrh         pNew = sqliteExprListAppend(pNew, a[k].pExpr, 0);
8487c917d19Sdrh         pNew->a[pNew->nExpr-1].zName = a[k].zName;
8497c917d19Sdrh         a[k].pExpr = 0;
8507c917d19Sdrh         a[k].zName = 0;
8517c917d19Sdrh       }else{
85254473229Sdrh         /* This expression is a "*" or a "TABLE.*" and needs to be
85354473229Sdrh         ** expanded. */
85454473229Sdrh         int tableSeen = 0;      /* Set to 1 when TABLE matches */
85554473229Sdrh         Token *pName;           /* text of name of TABLE */
85654473229Sdrh         if( pE->op==TK_DOT && pE->pLeft ){
85754473229Sdrh           pName = &pE->pLeft->token;
85854473229Sdrh         }else{
85954473229Sdrh           pName = 0;
86054473229Sdrh         }
861ad3cab52Sdrh         for(i=0; i<pTabList->nSrc; i++){
862d8bc7086Sdrh           Table *pTab = pTabList->a[i].pTab;
86354473229Sdrh           char *zTabName = pTabList->a[i].zAlias;
86454473229Sdrh           if( zTabName==0 || zTabName[0]==0 ){
86554473229Sdrh             zTabName = pTab->zName;
86654473229Sdrh           }
86754473229Sdrh           if( pName && (zTabName==0 || zTabName[0]==0 ||
868c754fa54Sdrh                  sqliteStrNICmp(pName->z, zTabName, pName->n)!=0 ||
869c754fa54Sdrh                  zTabName[pName->n]!=0) ){
87054473229Sdrh             continue;
87154473229Sdrh           }
87254473229Sdrh           tableSeen = 1;
873d8bc7086Sdrh           for(j=0; j<pTab->nCol; j++){
87422f70c32Sdrh             Expr *pExpr, *pLeft, *pRight;
875ad2d8307Sdrh             char *zName = pTab->aCol[j].zName;
876ad2d8307Sdrh 
877ad2d8307Sdrh             if( i>0 && (pTabList->a[i-1].jointype & JT_NATURAL)!=0 &&
878ad2d8307Sdrh                 columnIndex(pTabList->a[i-1].pTab, zName)>=0 ){
879ad2d8307Sdrh               /* In a NATURAL join, omit the join columns from the
880ad2d8307Sdrh               ** table on the right */
881ad2d8307Sdrh               continue;
882ad2d8307Sdrh             }
883ad2d8307Sdrh             if( i>0 && sqliteIdListIndex(pTabList->a[i-1].pUsing, zName)>=0 ){
884ad2d8307Sdrh               /* In a join with a USING clause, omit columns in the
885ad2d8307Sdrh               ** using clause from the table on the right. */
886ad2d8307Sdrh               continue;
887ad2d8307Sdrh             }
88822f70c32Sdrh             pRight = sqliteExpr(TK_ID, 0, 0, 0);
88922f70c32Sdrh             if( pRight==0 ) break;
890ad2d8307Sdrh             pRight->token.z = zName;
891ad2d8307Sdrh             pRight->token.n = strlen(zName);
89254473229Sdrh             if( zTabName ){
89322f70c32Sdrh               pLeft = sqliteExpr(TK_ID, 0, 0, 0);
89422f70c32Sdrh               if( pLeft==0 ) break;
89554473229Sdrh               pLeft->token.z = zTabName;
89654473229Sdrh               pLeft->token.n = strlen(zTabName);
89722f70c32Sdrh               pExpr = sqliteExpr(TK_DOT, pLeft, pRight, 0);
89822f70c32Sdrh               if( pExpr==0 ) break;
89922f70c32Sdrh             }else{
90022f70c32Sdrh               pExpr = pRight;
90122f70c32Sdrh               pExpr->span = pExpr->token;
90222f70c32Sdrh             }
9037c917d19Sdrh             pNew = sqliteExprListAppend(pNew, pExpr, 0);
904d8bc7086Sdrh           }
905d8bc7086Sdrh         }
90654473229Sdrh         if( !tableSeen ){
907f5db2d3eSdrh           if( pName ){
90854473229Sdrh             sqliteSetNString(&pParse->zErrMsg, "no such table: ", -1,
90954473229Sdrh               pName->z, pName->n, 0);
910f5db2d3eSdrh           }else{
911f5db2d3eSdrh             sqliteSetString(&pParse->zErrMsg, "no tables specified", 0);
912f5db2d3eSdrh           }
91354473229Sdrh           rc = 1;
91454473229Sdrh         }
9157c917d19Sdrh       }
9167c917d19Sdrh     }
9177c917d19Sdrh     sqliteExprListDelete(pEList);
9187c917d19Sdrh     p->pEList = pNew;
919d8bc7086Sdrh   }
92054473229Sdrh   return rc;
921d8bc7086Sdrh }
922d8bc7086Sdrh 
923d8bc7086Sdrh /*
924ff78bd2fSdrh ** This routine recursively unlinks the Select.pSrc.a[].pTab pointers
925ff78bd2fSdrh ** in a select structure.  It just sets the pointers to NULL.  This
926ff78bd2fSdrh ** routine is recursive in the sense that if the Select.pSrc.a[].pSelect
927ff78bd2fSdrh ** pointer is not NULL, this routine is called recursively on that pointer.
928ff78bd2fSdrh **
929ff78bd2fSdrh ** This routine is called on the Select structure that defines a
930ff78bd2fSdrh ** VIEW in order to undo any bindings to tables.  This is necessary
931ff78bd2fSdrh ** because those tables might be DROPed by a subsequent SQL command.
932ff78bd2fSdrh */
933ff78bd2fSdrh void sqliteSelectUnbind(Select *p){
934ff78bd2fSdrh   int i;
935ad3cab52Sdrh   SrcList *pSrc = p->pSrc;
936ff78bd2fSdrh   Table *pTab;
937ff78bd2fSdrh   if( p==0 ) return;
938ad3cab52Sdrh   for(i=0; i<pSrc->nSrc; i++){
939ff78bd2fSdrh     if( (pTab = pSrc->a[i].pTab)!=0 ){
940ff78bd2fSdrh       if( pTab->isTransient ){
941ff78bd2fSdrh         sqliteDeleteTable(0, pTab);
942ff78bd2fSdrh         sqliteSelectDelete(pSrc->a[i].pSelect);
943ff78bd2fSdrh         pSrc->a[i].pSelect = 0;
944ff78bd2fSdrh       }
945ff78bd2fSdrh       pSrc->a[i].pTab = 0;
946ff78bd2fSdrh       if( pSrc->a[i].pSelect ){
947ff78bd2fSdrh         sqliteSelectUnbind(pSrc->a[i].pSelect);
948ff78bd2fSdrh       }
949ff78bd2fSdrh     }
950ff78bd2fSdrh   }
951ff78bd2fSdrh }
952ff78bd2fSdrh 
953ff78bd2fSdrh /*
954d8bc7086Sdrh ** This routine associates entries in an ORDER BY expression list with
955d8bc7086Sdrh ** columns in a result.  For each ORDER BY expression, the opcode of
956967e8b73Sdrh ** the top-level node is changed to TK_COLUMN and the iColumn value of
957d8bc7086Sdrh ** the top-level node is filled in with column number and the iTable
958d8bc7086Sdrh ** value of the top-level node is filled with iTable parameter.
959d8bc7086Sdrh **
960d8bc7086Sdrh ** If there are prior SELECT clauses, they are processed first.  A match
961d8bc7086Sdrh ** in an earlier SELECT takes precedence over a later SELECT.
962d8bc7086Sdrh **
963d8bc7086Sdrh ** Any entry that does not match is flagged as an error.  The number
964d8bc7086Sdrh ** of errors is returned.
965d8bc7086Sdrh */
966d8bc7086Sdrh static int matchOrderbyToColumn(
967d8bc7086Sdrh   Parse *pParse,          /* A place to leave error messages */
968d8bc7086Sdrh   Select *pSelect,        /* Match to result columns of this SELECT */
969d8bc7086Sdrh   ExprList *pOrderBy,     /* The ORDER BY values to match against columns */
970e4de1febSdrh   int iTable,             /* Insert this value in iTable */
971d8bc7086Sdrh   int mustComplete        /* If TRUE all ORDER BYs must match */
972d8bc7086Sdrh ){
973d8bc7086Sdrh   int nErr = 0;
974d8bc7086Sdrh   int i, j;
975d8bc7086Sdrh   ExprList *pEList;
976d8bc7086Sdrh 
977daffd0e5Sdrh   if( pSelect==0 || pOrderBy==0 ) return 1;
978d8bc7086Sdrh   if( mustComplete ){
979d8bc7086Sdrh     for(i=0; i<pOrderBy->nExpr; i++){ pOrderBy->a[i].done = 0; }
980d8bc7086Sdrh   }
981d8bc7086Sdrh   if( fillInColumnList(pParse, pSelect) ){
982d8bc7086Sdrh     return 1;
983d8bc7086Sdrh   }
984d8bc7086Sdrh   if( pSelect->pPrior ){
98592cd52f5Sdrh     if( matchOrderbyToColumn(pParse, pSelect->pPrior, pOrderBy, iTable, 0) ){
98692cd52f5Sdrh       return 1;
98792cd52f5Sdrh     }
988d8bc7086Sdrh   }
989d8bc7086Sdrh   pEList = pSelect->pEList;
990d8bc7086Sdrh   for(i=0; i<pOrderBy->nExpr; i++){
991d8bc7086Sdrh     Expr *pE = pOrderBy->a[i].pExpr;
992e4de1febSdrh     int iCol = -1;
993d8bc7086Sdrh     if( pOrderBy->a[i].done ) continue;
994e4de1febSdrh     if( sqliteExprIsInteger(pE, &iCol) ){
995e4de1febSdrh       if( iCol<=0 || iCol>pEList->nExpr ){
996e4de1febSdrh         char zBuf[200];
997e4de1febSdrh         sprintf(zBuf,"ORDER BY position %d should be between 1 and %d",
998e4de1febSdrh            iCol, pEList->nExpr);
999e4de1febSdrh         sqliteSetString(&pParse->zErrMsg, zBuf, 0);
1000e4de1febSdrh         pParse->nErr++;
1001e4de1febSdrh         nErr++;
1002e4de1febSdrh         break;
1003e4de1febSdrh       }
1004e4de1febSdrh       iCol--;
1005e4de1febSdrh     }
1006e4de1febSdrh     for(j=0; iCol<0 && j<pEList->nExpr; j++){
10074cfa7934Sdrh       if( pEList->a[j].zName && (pE->op==TK_ID || pE->op==TK_STRING) ){
1008a76b5dfcSdrh         char *zName, *zLabel;
1009a76b5dfcSdrh         zName = pEList->a[j].zName;
1010a76b5dfcSdrh         assert( pE->token.z );
1011a76b5dfcSdrh         zLabel = sqliteStrNDup(pE->token.z, pE->token.n);
1012d8bc7086Sdrh         sqliteDequote(zLabel);
1013d8bc7086Sdrh         if( sqliteStrICmp(zName, zLabel)==0 ){
1014e4de1febSdrh           iCol = j;
1015d8bc7086Sdrh         }
10166e142f54Sdrh         sqliteFree(zLabel);
1017d8bc7086Sdrh       }
1018e4de1febSdrh       if( iCol<0 && sqliteExprCompare(pE, pEList->a[j].pExpr) ){
1019e4de1febSdrh         iCol = j;
1020d8bc7086Sdrh       }
1021e4de1febSdrh     }
1022e4de1febSdrh     if( iCol>=0 ){
1023967e8b73Sdrh       pE->op = TK_COLUMN;
1024e4de1febSdrh       pE->iColumn = iCol;
1025d8bc7086Sdrh       pE->iTable = iTable;
1026d8bc7086Sdrh       pOrderBy->a[i].done = 1;
1027d8bc7086Sdrh     }
1028e4de1febSdrh     if( iCol<0 && mustComplete ){
1029d8bc7086Sdrh       char zBuf[30];
1030d8bc7086Sdrh       sprintf(zBuf,"%d",i+1);
1031d8bc7086Sdrh       sqliteSetString(&pParse->zErrMsg, "ORDER BY term number ", zBuf,
1032d8bc7086Sdrh         " does not match any result column", 0);
1033d8bc7086Sdrh       pParse->nErr++;
1034d8bc7086Sdrh       nErr++;
1035d8bc7086Sdrh       break;
1036d8bc7086Sdrh     }
1037d8bc7086Sdrh   }
1038d8bc7086Sdrh   return nErr;
1039d8bc7086Sdrh }
1040d8bc7086Sdrh 
1041d8bc7086Sdrh /*
1042d8bc7086Sdrh ** Get a VDBE for the given parser context.  Create a new one if necessary.
1043d8bc7086Sdrh ** If an error occurs, return NULL and leave a message in pParse.
1044d8bc7086Sdrh */
1045d8bc7086Sdrh Vdbe *sqliteGetVdbe(Parse *pParse){
1046d8bc7086Sdrh   Vdbe *v = pParse->pVdbe;
1047d8bc7086Sdrh   if( v==0 ){
10484c504391Sdrh     v = pParse->pVdbe = sqliteVdbeCreate(pParse->db);
1049d8bc7086Sdrh   }
1050d8bc7086Sdrh   return v;
1051d8bc7086Sdrh }
1052d8bc7086Sdrh 
1053d8bc7086Sdrh 
1054d8bc7086Sdrh /*
105582c3d636Sdrh ** This routine is called to process a query that is really the union
105682c3d636Sdrh ** or intersection of two or more separate queries.
1057c926afbcSdrh **
1058c926afbcSdrh ** "p" points to the right-most of the two queries.  The results should
1059c926afbcSdrh ** be stored in eDest with parameter iParm.
106082c3d636Sdrh */
106182c3d636Sdrh static int multiSelect(Parse *pParse, Select *p, int eDest, int iParm){
106210e5e3cfSdrh   int rc;             /* Success code from a subroutine */
106310e5e3cfSdrh   Select *pPrior;     /* Another SELECT immediately to our left */
106410e5e3cfSdrh   Vdbe *v;            /* Generate code to this VDBE */
106510e5e3cfSdrh   int base;           /* Baseline value for pParse->nTab */
106682c3d636Sdrh 
1067d8bc7086Sdrh   /* Make sure there is no ORDER BY clause on prior SELECTs.  Only the
1068d8bc7086Sdrh   ** last SELECT in the series may have an ORDER BY.
106982c3d636Sdrh   */
1070daffd0e5Sdrh   if( p==0 || p->pPrior==0 ) return 1;
1071d8bc7086Sdrh   pPrior = p->pPrior;
1072d8bc7086Sdrh   if( pPrior->pOrderBy ){
1073d8bc7086Sdrh     sqliteSetString(&pParse->zErrMsg,"ORDER BY clause should come after ",
1074d8bc7086Sdrh       selectOpName(p->op), " not before", 0);
107582c3d636Sdrh     pParse->nErr++;
107682c3d636Sdrh     return 1;
107782c3d636Sdrh   }
107882c3d636Sdrh 
1079d8bc7086Sdrh   /* Make sure we have a valid query engine.  If not, create a new one.
1080d8bc7086Sdrh   */
1081d8bc7086Sdrh   v = sqliteGetVdbe(pParse);
1082d8bc7086Sdrh   if( v==0 ) return 1;
1083d8bc7086Sdrh 
10841cc3d75fSdrh   /* Create the destination temporary table if necessary
10851cc3d75fSdrh   */
10861cc3d75fSdrh   if( eDest==SRT_TempTable ){
10871cc3d75fSdrh     sqliteVdbeAddOp(v, OP_OpenTemp, iParm, 0);
10881cc3d75fSdrh     eDest = SRT_Table;
10891cc3d75fSdrh   }
10901cc3d75fSdrh 
1091f46f905aSdrh   /* Generate code for the left and right SELECT statements.
1092d8bc7086Sdrh   */
109310e5e3cfSdrh   base = pParse->nTab;
109482c3d636Sdrh   switch( p->op ){
1095f46f905aSdrh     case TK_ALL: {
1096f46f905aSdrh       if( p->pOrderBy==0 ){
1097f46f905aSdrh         rc = sqliteSelect(pParse, pPrior, eDest, iParm, 0, 0, 0);
1098f46f905aSdrh         if( rc ) return rc;
1099f46f905aSdrh         p->pPrior = 0;
1100f46f905aSdrh         rc = sqliteSelect(pParse, p, eDest, iParm, 0, 0, 0);
1101f46f905aSdrh         p->pPrior = pPrior;
1102f46f905aSdrh         if( rc ) return rc;
1103f46f905aSdrh         break;
1104f46f905aSdrh       }
1105f46f905aSdrh       /* For UNION ALL ... ORDER BY fall through to the next case */
1106f46f905aSdrh     }
110782c3d636Sdrh     case TK_EXCEPT:
110882c3d636Sdrh     case TK_UNION: {
1109d8bc7086Sdrh       int unionTab;    /* Cursor number of the temporary table holding result */
1110d8bc7086Sdrh       int op;          /* One of the SRT_ operations to apply to self */
1111d8bc7086Sdrh       int priorOp;     /* The SRT_ operation to apply to prior selects */
1112c926afbcSdrh       ExprList *pOrderBy;  /* The ORDER BY clause for the right SELECT */
111382c3d636Sdrh 
1114d8bc7086Sdrh       priorOp = p->op==TK_ALL ? SRT_Table : SRT_Union;
1115c926afbcSdrh       if( eDest==priorOp && p->pOrderBy==0 ){
1116d8bc7086Sdrh         /* We can reuse a temporary table generated by a SELECT to our
1117c926afbcSdrh         ** right.
1118d8bc7086Sdrh         */
111982c3d636Sdrh         unionTab = iParm;
112082c3d636Sdrh       }else{
1121d8bc7086Sdrh         /* We will need to create our own temporary table to hold the
1122d8bc7086Sdrh         ** intermediate results.
1123d8bc7086Sdrh         */
112482c3d636Sdrh         unionTab = pParse->nTab++;
1125d8bc7086Sdrh         if( p->pOrderBy
1126d8bc7086Sdrh         && matchOrderbyToColumn(pParse, p, p->pOrderBy, unionTab, 1) ){
1127d8bc7086Sdrh           return 1;
1128d8bc7086Sdrh         }
1129d8bc7086Sdrh         if( p->op!=TK_ALL ){
1130c6b52df3Sdrh           sqliteVdbeAddOp(v, OP_OpenTemp, unionTab, 1);
113199fcd718Sdrh           sqliteVdbeAddOp(v, OP_KeyAsData, unionTab, 1);
1132345fda3eSdrh         }else{
113399fcd718Sdrh           sqliteVdbeAddOp(v, OP_OpenTemp, unionTab, 0);
113482c3d636Sdrh         }
1135d8bc7086Sdrh       }
1136d8bc7086Sdrh 
1137d8bc7086Sdrh       /* Code the SELECT statements to our left
1138d8bc7086Sdrh       */
1139832508b7Sdrh       rc = sqliteSelect(pParse, pPrior, priorOp, unionTab, 0, 0, 0);
114082c3d636Sdrh       if( rc ) return rc;
1141d8bc7086Sdrh 
1142d8bc7086Sdrh       /* Code the current SELECT statement
1143d8bc7086Sdrh       */
1144d8bc7086Sdrh       switch( p->op ){
1145d8bc7086Sdrh          case TK_EXCEPT:  op = SRT_Except;   break;
1146d8bc7086Sdrh          case TK_UNION:   op = SRT_Union;    break;
1147d8bc7086Sdrh          case TK_ALL:     op = SRT_Table;    break;
1148d8bc7086Sdrh       }
114982c3d636Sdrh       p->pPrior = 0;
1150c926afbcSdrh       pOrderBy = p->pOrderBy;
1151c926afbcSdrh       p->pOrderBy = 0;
1152832508b7Sdrh       rc = sqliteSelect(pParse, p, op, unionTab, 0, 0, 0);
115382c3d636Sdrh       p->pPrior = pPrior;
1154c926afbcSdrh       p->pOrderBy = pOrderBy;
115582c3d636Sdrh       if( rc ) return rc;
1156d8bc7086Sdrh 
1157d8bc7086Sdrh       /* Convert the data in the temporary table into whatever form
1158d8bc7086Sdrh       ** it is that we currently need.
1159d8bc7086Sdrh       */
1160c926afbcSdrh       if( eDest!=priorOp || unionTab!=iParm ){
11616b56344dSdrh         int iCont, iBreak, iStart;
116282c3d636Sdrh         assert( p->pEList );
116341202ccaSdrh         if( eDest==SRT_Callback ){
1164832508b7Sdrh           generateColumnNames(pParse, p->base, 0, p->pEList);
116541202ccaSdrh         }
116682c3d636Sdrh         iBreak = sqliteVdbeMakeLabel(v);
11676b56344dSdrh         iCont = sqliteVdbeMakeLabel(v);
11686b56344dSdrh         sqliteVdbeAddOp(v, OP_Rewind, unionTab, iBreak);
11696b56344dSdrh         iStart = sqliteVdbeCurrentAddr(v);
117038640e15Sdrh         rc = selectInnerLoop(pParse, p, p->pEList, unionTab, p->pEList->nExpr,
1171d8bc7086Sdrh                              p->pOrderBy, -1, eDest, iParm,
117282c3d636Sdrh                              iCont, iBreak);
117382c3d636Sdrh         if( rc ) return 1;
11746b56344dSdrh         sqliteVdbeResolveLabel(v, iCont);
11756b56344dSdrh         sqliteVdbeAddOp(v, OP_Next, unionTab, iStart);
117699fcd718Sdrh         sqliteVdbeResolveLabel(v, iBreak);
117799fcd718Sdrh         sqliteVdbeAddOp(v, OP_Close, unionTab, 0);
1178d8bc7086Sdrh         if( p->pOrderBy ){
1179c926afbcSdrh           generateSortTail(p, v, p->pEList->nExpr, eDest, iParm);
1180d8bc7086Sdrh         }
118182c3d636Sdrh       }
118282c3d636Sdrh       break;
118382c3d636Sdrh     }
118482c3d636Sdrh     case TK_INTERSECT: {
118582c3d636Sdrh       int tab1, tab2;
11866b56344dSdrh       int iCont, iBreak, iStart;
118782c3d636Sdrh 
1188d8bc7086Sdrh       /* INTERSECT is different from the others since it requires
11896206d50aSdrh       ** two temporary tables.  Hence it has its own case.  Begin
1190d8bc7086Sdrh       ** by allocating the tables we will need.
1191d8bc7086Sdrh       */
119282c3d636Sdrh       tab1 = pParse->nTab++;
119382c3d636Sdrh       tab2 = pParse->nTab++;
1194d8bc7086Sdrh       if( p->pOrderBy && matchOrderbyToColumn(pParse,p,p->pOrderBy,tab1,1) ){
1195d8bc7086Sdrh         return 1;
1196d8bc7086Sdrh       }
1197c6b52df3Sdrh       sqliteVdbeAddOp(v, OP_OpenTemp, tab1, 1);
119899fcd718Sdrh       sqliteVdbeAddOp(v, OP_KeyAsData, tab1, 1);
1199d8bc7086Sdrh 
1200d8bc7086Sdrh       /* Code the SELECTs to our left into temporary table "tab1".
1201d8bc7086Sdrh       */
1202832508b7Sdrh       rc = sqliteSelect(pParse, pPrior, SRT_Union, tab1, 0, 0, 0);
120382c3d636Sdrh       if( rc ) return rc;
1204d8bc7086Sdrh 
1205d8bc7086Sdrh       /* Code the current SELECT into temporary table "tab2"
1206d8bc7086Sdrh       */
1207c6b52df3Sdrh       sqliteVdbeAddOp(v, OP_OpenTemp, tab2, 1);
120899fcd718Sdrh       sqliteVdbeAddOp(v, OP_KeyAsData, tab2, 1);
120982c3d636Sdrh       p->pPrior = 0;
1210832508b7Sdrh       rc = sqliteSelect(pParse, p, SRT_Union, tab2, 0, 0, 0);
121182c3d636Sdrh       p->pPrior = pPrior;
121282c3d636Sdrh       if( rc ) return rc;
1213d8bc7086Sdrh 
1214d8bc7086Sdrh       /* Generate code to take the intersection of the two temporary
1215d8bc7086Sdrh       ** tables.
1216d8bc7086Sdrh       */
121782c3d636Sdrh       assert( p->pEList );
121841202ccaSdrh       if( eDest==SRT_Callback ){
1219832508b7Sdrh         generateColumnNames(pParse, p->base, 0, p->pEList);
122041202ccaSdrh       }
122182c3d636Sdrh       iBreak = sqliteVdbeMakeLabel(v);
12226b56344dSdrh       iCont = sqliteVdbeMakeLabel(v);
12236b56344dSdrh       sqliteVdbeAddOp(v, OP_Rewind, tab1, iBreak);
12246b56344dSdrh       iStart = sqliteVdbeAddOp(v, OP_FullKey, tab1, 0);
122599fcd718Sdrh       sqliteVdbeAddOp(v, OP_NotFound, tab2, iCont);
122638640e15Sdrh       rc = selectInnerLoop(pParse, p, p->pEList, tab1, p->pEList->nExpr,
1227d8bc7086Sdrh                              p->pOrderBy, -1, eDest, iParm,
122882c3d636Sdrh                              iCont, iBreak);
122982c3d636Sdrh       if( rc ) return 1;
12306b56344dSdrh       sqliteVdbeResolveLabel(v, iCont);
12316b56344dSdrh       sqliteVdbeAddOp(v, OP_Next, tab1, iStart);
123299fcd718Sdrh       sqliteVdbeResolveLabel(v, iBreak);
123399fcd718Sdrh       sqliteVdbeAddOp(v, OP_Close, tab2, 0);
123499fcd718Sdrh       sqliteVdbeAddOp(v, OP_Close, tab1, 0);
1235d8bc7086Sdrh       if( p->pOrderBy ){
1236c926afbcSdrh         generateSortTail(p, v, p->pEList->nExpr, eDest, iParm);
1237d8bc7086Sdrh       }
123882c3d636Sdrh       break;
123982c3d636Sdrh     }
124082c3d636Sdrh   }
124182c3d636Sdrh   assert( p->pEList && pPrior->pEList );
124282c3d636Sdrh   if( p->pEList->nExpr!=pPrior->pEList->nExpr ){
1243d8bc7086Sdrh     sqliteSetString(&pParse->zErrMsg, "SELECTs to the left and right of ",
1244d8bc7086Sdrh       selectOpName(p->op), " do not have the same number of result columns", 0);
124582c3d636Sdrh     pParse->nErr++;
124682c3d636Sdrh     return 1;
12472282792aSdrh   }
124810e5e3cfSdrh   pParse->nTab = base;
12492282792aSdrh   return 0;
12502282792aSdrh }
12512282792aSdrh 
12522282792aSdrh /*
1253832508b7Sdrh ** Recursively scan through an expression tree.  For every reference
1254832508b7Sdrh ** to a column in table number iFrom, change that reference to the
1255832508b7Sdrh ** same column in table number iTo.
1256832508b7Sdrh */
1257832508b7Sdrh static void changeTables(Expr *pExpr, int iFrom, int iTo){
1258832508b7Sdrh   if( pExpr==0 ) return;
1259832508b7Sdrh   if( pExpr->op==TK_COLUMN && pExpr->iTable==iFrom ){
1260832508b7Sdrh     pExpr->iTable = iTo;
1261832508b7Sdrh   }else{
12621b2e0329Sdrh     static void changeTablesInList(ExprList*, int, int);
1263832508b7Sdrh     changeTables(pExpr->pLeft, iFrom, iTo);
1264832508b7Sdrh     changeTables(pExpr->pRight, iFrom, iTo);
12651b2e0329Sdrh     changeTablesInList(pExpr->pList, iFrom, iTo);
1266832508b7Sdrh   }
1267832508b7Sdrh }
12681b2e0329Sdrh static void changeTablesInList(ExprList *pList, int iFrom, int iTo){
12691b2e0329Sdrh   if( pList ){
12701b2e0329Sdrh     int i;
12711b2e0329Sdrh     for(i=0; i<pList->nExpr; i++){
12721b2e0329Sdrh       changeTables(pList->a[i].pExpr, iFrom, iTo);
12731b2e0329Sdrh     }
1274832508b7Sdrh   }
1275832508b7Sdrh }
1276832508b7Sdrh 
1277832508b7Sdrh /*
1278832508b7Sdrh ** Scan through the expression pExpr.  Replace every reference to
1279832508b7Sdrh ** a column in table number iTable with a copy of the corresponding
128084e59207Sdrh ** entry in pEList.  (But leave references to the ROWID column
128184e59207Sdrh ** unchanged.)  When making a copy of an expression in pEList, change
128284e59207Sdrh ** references to columns in table iSub into references to table iTable.
1283832508b7Sdrh **
1284832508b7Sdrh ** This routine is part of the flattening procedure.  A subquery
1285832508b7Sdrh ** whose result set is defined by pEList appears as entry in the
1286832508b7Sdrh ** FROM clause of a SELECT such that the VDBE cursor assigned to that
1287832508b7Sdrh ** FORM clause entry is iTable.  This routine make the necessary
1288832508b7Sdrh ** changes to pExpr so that it refers directly to the source table
1289832508b7Sdrh ** of the subquery rather the result set of the subquery.
1290832508b7Sdrh */
1291832508b7Sdrh static void substExpr(Expr *pExpr, int iTable, ExprList *pEList, int iSub){
1292832508b7Sdrh   if( pExpr==0 ) return;
129384e59207Sdrh   if( pExpr->op==TK_COLUMN && pExpr->iTable==iTable && pExpr->iColumn>=0 ){
1294832508b7Sdrh     Expr *pNew;
129584e59207Sdrh     assert( pEList!=0 && pExpr->iColumn<pEList->nExpr );
1296832508b7Sdrh     assert( pExpr->pLeft==0 && pExpr->pRight==0 && pExpr->pList==0 );
1297832508b7Sdrh     pNew = pEList->a[pExpr->iColumn].pExpr;
1298832508b7Sdrh     assert( pNew!=0 );
1299832508b7Sdrh     pExpr->op = pNew->op;
1300832508b7Sdrh     pExpr->pLeft = sqliteExprDup(pNew->pLeft);
1301832508b7Sdrh     pExpr->pRight = sqliteExprDup(pNew->pRight);
1302832508b7Sdrh     pExpr->pList = sqliteExprListDup(pNew->pList);
1303832508b7Sdrh     pExpr->iTable = pNew->iTable;
1304832508b7Sdrh     pExpr->iColumn = pNew->iColumn;
1305832508b7Sdrh     pExpr->iAgg = pNew->iAgg;
13061b2e0329Sdrh     pExpr->token = pNew->token;
1307832508b7Sdrh     if( iSub!=iTable ){
1308832508b7Sdrh       changeTables(pExpr, iSub, iTable);
1309832508b7Sdrh     }
1310832508b7Sdrh   }else{
1311832508b7Sdrh     static void substExprList(ExprList*,int,ExprList*,int);
1312832508b7Sdrh     substExpr(pExpr->pLeft, iTable, pEList, iSub);
1313832508b7Sdrh     substExpr(pExpr->pRight, iTable, pEList, iSub);
1314832508b7Sdrh     substExprList(pExpr->pList, iTable, pEList, iSub);
1315832508b7Sdrh   }
1316832508b7Sdrh }
1317832508b7Sdrh static void
1318832508b7Sdrh substExprList(ExprList *pList, int iTable, ExprList *pEList, int iSub){
1319832508b7Sdrh   int i;
1320832508b7Sdrh   if( pList==0 ) return;
1321832508b7Sdrh   for(i=0; i<pList->nExpr; i++){
1322832508b7Sdrh     substExpr(pList->a[i].pExpr, iTable, pEList, iSub);
1323832508b7Sdrh   }
1324832508b7Sdrh }
1325832508b7Sdrh 
1326832508b7Sdrh /*
13271350b030Sdrh ** This routine attempts to flatten subqueries in order to speed
13281350b030Sdrh ** execution.  It returns 1 if it makes changes and 0 if no flattening
13291350b030Sdrh ** occurs.
13301350b030Sdrh **
13311350b030Sdrh ** To understand the concept of flattening, consider the following
13321350b030Sdrh ** query:
13331350b030Sdrh **
13341350b030Sdrh **     SELECT a FROM (SELECT x+y AS a FROM t1 WHERE z<100) WHERE a>5
13351350b030Sdrh **
13361350b030Sdrh ** The default way of implementing this query is to execute the
13371350b030Sdrh ** subquery first and store the results in a temporary table, then
13381350b030Sdrh ** run the outer query on that temporary table.  This requires two
13391350b030Sdrh ** passes over the data.  Furthermore, because the temporary table
13401350b030Sdrh ** has no indices, the WHERE clause on the outer query cannot be
1341832508b7Sdrh ** optimized.
13421350b030Sdrh **
1343832508b7Sdrh ** This routine attempts to rewrite queries such as the above into
13441350b030Sdrh ** a single flat select, like this:
13451350b030Sdrh **
13461350b030Sdrh **     SELECT x+y AS a FROM t1 WHERE z<100 AND a>5
13471350b030Sdrh **
13481350b030Sdrh ** The code generated for this simpification gives the same result
1349832508b7Sdrh ** but only has to scan the data once.  And because indices might
1350832508b7Sdrh ** exist on the table t1, a complete scan of the data might be
1351832508b7Sdrh ** avoided.
13521350b030Sdrh **
1353832508b7Sdrh ** Flattening is only attempted if all of the following are true:
13541350b030Sdrh **
1355832508b7Sdrh **   (1)  The subquery and the outer query do not both use aggregates.
13561350b030Sdrh **
1357832508b7Sdrh **   (2)  The subquery is not an aggregate or the outer query is not a join.
1358832508b7Sdrh **
1359832508b7Sdrh **   (3)  The subquery is not a join.
1360832508b7Sdrh **
1361832508b7Sdrh **   (4)  The subquery is not DISTINCT or the outer query is not a join.
1362832508b7Sdrh **
1363832508b7Sdrh **   (5)  The subquery is not DISTINCT or the outer query does not use
1364832508b7Sdrh **        aggregates.
1365832508b7Sdrh **
1366832508b7Sdrh **   (6)  The subquery does not use aggregates or the outer query is not
1367832508b7Sdrh **        DISTINCT.
1368832508b7Sdrh **
136908192d5fSdrh **   (7)  The subquery has a FROM clause.
137008192d5fSdrh **
1371df199a25Sdrh **   (8)  The subquery does not use LIMIT or the outer query is not a join.
1372df199a25Sdrh **
1373df199a25Sdrh **   (9)  The subquery does not use LIMIT or the outer query does not use
1374df199a25Sdrh **        aggregates.
1375df199a25Sdrh **
1376df199a25Sdrh **  (10)  The subquery does not use aggregates or the outer query does not
1377df199a25Sdrh **        use LIMIT.
1378df199a25Sdrh **
1379832508b7Sdrh ** In this routine, the "p" parameter is a pointer to the outer query.
1380832508b7Sdrh ** The subquery is p->pSrc->a[iFrom].  isAgg is true if the outer query
1381832508b7Sdrh ** uses aggregates and subqueryIsAgg is true if the subquery uses aggregates.
1382832508b7Sdrh **
1383832508b7Sdrh ** If flattening is not attempted, this routine is a no-op and return 0.
1384832508b7Sdrh ** If flattening is attempted this routine returns 1.
1385832508b7Sdrh **
1386832508b7Sdrh ** All of the expression analysis must occur on both the outer query and
1387832508b7Sdrh ** the subquery before this routine runs.
13881350b030Sdrh */
1389832508b7Sdrh int flattenSubquery(Select *p, int iFrom, int isAgg, int subqueryIsAgg){
13900bb28106Sdrh   Select *pSub;       /* The inner query or "subquery" */
1391ad3cab52Sdrh   SrcList *pSrc;      /* The FROM clause of the outer query */
1392ad3cab52Sdrh   SrcList *pSubSrc;   /* The FROM clause of the subquery */
13930bb28106Sdrh   ExprList *pList;    /* The result set of the outer query */
1394832508b7Sdrh   int i;
1395832508b7Sdrh   int iParent, iSub;
1396832508b7Sdrh   Expr *pWhere;
13971350b030Sdrh 
1398832508b7Sdrh   /* Check to see if flattening is permitted.  Return 0 if not.
1399832508b7Sdrh   */
1400832508b7Sdrh   if( p==0 ) return 0;
1401832508b7Sdrh   pSrc = p->pSrc;
1402ad3cab52Sdrh   assert( pSrc && iFrom>=0 && iFrom<pSrc->nSrc );
1403832508b7Sdrh   pSub = pSrc->a[iFrom].pSelect;
1404832508b7Sdrh   assert( pSub!=0 );
1405832508b7Sdrh   if( isAgg && subqueryIsAgg ) return 0;
1406ad3cab52Sdrh   if( subqueryIsAgg && pSrc->nSrc>1 ) return 0;
1407832508b7Sdrh   pSubSrc = pSub->pSrc;
1408832508b7Sdrh   assert( pSubSrc );
1409ad3cab52Sdrh   if( pSubSrc->nSrc!=1 ) return 0;
1410df199a25Sdrh   if( (pSub->isDistinct || pSub->nLimit>=0) &&  (pSrc->nSrc>1 || isAgg) ){
1411df199a25Sdrh      return 0;
1412df199a25Sdrh   }
1413d11d382cSdrh   if( (p->isDistinct || p->nLimit>=0) && subqueryIsAgg ) return 0;
1414832508b7Sdrh 
14150bb28106Sdrh   /* If we reach this point, it means flattening is permitted for the
1416832508b7Sdrh   ** i-th entry of the FROM clause in the outer query.
1417832508b7Sdrh   */
1418832508b7Sdrh   iParent = p->base + iFrom;
1419832508b7Sdrh   iSub = pSub->base;
1420832508b7Sdrh   substExprList(p->pEList, iParent, pSub->pEList, iSub);
1421832508b7Sdrh   pList = p->pEList;
1422832508b7Sdrh   for(i=0; i<pList->nExpr; i++){
1423832508b7Sdrh     if( pList->a[i].zName==0 ){
1424832508b7Sdrh       Expr *pExpr = pList->a[i].pExpr;
1425832508b7Sdrh       pList->a[i].zName = sqliteStrNDup(pExpr->span.z, pExpr->span.n);
1426832508b7Sdrh     }
1427832508b7Sdrh   }
14281b2e0329Sdrh   if( isAgg ){
1429832508b7Sdrh     substExprList(p->pGroupBy, iParent, pSub->pEList, iSub);
1430832508b7Sdrh     substExpr(p->pHaving, iParent, pSub->pEList, iSub);
14311b2e0329Sdrh   }
1432832508b7Sdrh   substExprList(p->pOrderBy, iParent, pSub->pEList, iSub);
1433832508b7Sdrh   if( pSub->pWhere ){
1434832508b7Sdrh     pWhere = sqliteExprDup(pSub->pWhere);
1435832508b7Sdrh     if( iParent!=iSub ){
1436832508b7Sdrh       changeTables(pWhere, iSub, iParent);
1437832508b7Sdrh     }
1438832508b7Sdrh   }else{
1439832508b7Sdrh     pWhere = 0;
1440832508b7Sdrh   }
1441832508b7Sdrh   if( subqueryIsAgg ){
1442832508b7Sdrh     assert( p->pHaving==0 );
14431b2e0329Sdrh     p->pHaving = p->pWhere;
14441b2e0329Sdrh     p->pWhere = pWhere;
1445832508b7Sdrh     substExpr(p->pHaving, iParent, pSub->pEList, iSub);
14461b2e0329Sdrh     if( pSub->pHaving ){
14471b2e0329Sdrh       Expr *pHaving = sqliteExprDup(pSub->pHaving);
14481b2e0329Sdrh       if( iParent!=iSub ){
14491b2e0329Sdrh         changeTables(pHaving, iSub, iParent);
14501b2e0329Sdrh       }
14511b2e0329Sdrh       if( p->pHaving ){
14521b2e0329Sdrh         p->pHaving = sqliteExpr(TK_AND, p->pHaving, pHaving, 0);
14531b2e0329Sdrh       }else{
14541b2e0329Sdrh         p->pHaving = pHaving;
14551b2e0329Sdrh       }
14561b2e0329Sdrh     }
14571b2e0329Sdrh     assert( p->pGroupBy==0 );
14581b2e0329Sdrh     p->pGroupBy = sqliteExprListDup(pSub->pGroupBy);
14591b2e0329Sdrh     if( iParent!=iSub ){
14601b2e0329Sdrh       changeTablesInList(p->pGroupBy, iSub, iParent);
14611b2e0329Sdrh     }
1462832508b7Sdrh   }else if( p->pWhere==0 ){
1463832508b7Sdrh     p->pWhere = pWhere;
1464832508b7Sdrh   }else{
1465832508b7Sdrh     substExpr(p->pWhere, iParent, pSub->pEList, iSub);
1466832508b7Sdrh     if( pWhere ){
1467832508b7Sdrh       p->pWhere = sqliteExpr(TK_AND, p->pWhere, pWhere, 0);
1468832508b7Sdrh     }
1469832508b7Sdrh   }
1470832508b7Sdrh   p->isDistinct = p->isDistinct || pSub->isDistinct;
1471df199a25Sdrh   if( pSub->nLimit>=0 ){
1472df199a25Sdrh     if( p->nLimit<0 ){
1473df199a25Sdrh       p->nLimit = pSub->nLimit;
1474df199a25Sdrh     }else if( p->nLimit+p->nOffset > pSub->nLimit+pSub->nOffset ){
1475df199a25Sdrh       p->nLimit = pSub->nLimit + pSub->nOffset - p->nOffset;
1476df199a25Sdrh     }
1477df199a25Sdrh   }
1478df199a25Sdrh   p->nOffset += pSub->nOffset;
1479832508b7Sdrh   if( pSrc->a[iFrom].pTab && pSrc->a[iFrom].pTab->isTransient ){
1480832508b7Sdrh     sqliteDeleteTable(0, pSrc->a[iFrom].pTab);
1481832508b7Sdrh   }
1482832508b7Sdrh   pSrc->a[iFrom].pTab = pSubSrc->a[0].pTab;
1483832508b7Sdrh   pSubSrc->a[0].pTab = 0;
1484832508b7Sdrh   pSrc->a[iFrom].pSelect = pSubSrc->a[0].pSelect;
1485832508b7Sdrh   pSubSrc->a[0].pSelect = 0;
1486832508b7Sdrh   sqliteSelectDelete(pSub);
1487832508b7Sdrh   return 1;
14881350b030Sdrh }
14891350b030Sdrh 
14901350b030Sdrh /*
14919562b551Sdrh ** Analyze the SELECT statement passed in as an argument to see if it
14929562b551Sdrh ** is a simple min() or max() query.  If it is and this query can be
14939562b551Sdrh ** satisfied using a single seek to the beginning or end of an index,
14949562b551Sdrh ** then generate the code for this SELECT return 1.  If this is not a
14959562b551Sdrh ** simple min() or max() query, then return 0;
14969562b551Sdrh **
14979562b551Sdrh ** A simply min() or max() query looks like this:
14989562b551Sdrh **
14999562b551Sdrh **    SELECT min(a) FROM table;
15009562b551Sdrh **    SELECT max(a) FROM table;
15019562b551Sdrh **
15029562b551Sdrh ** The query may have only a single table in its FROM argument.  There
15039562b551Sdrh ** can be no GROUP BY or HAVING or WHERE clauses.  The result set must
15049562b551Sdrh ** be the min() or max() of a single column of the table.  The column
15059562b551Sdrh ** in the min() or max() function must be indexed.
15069562b551Sdrh **
15079562b551Sdrh ** The parameters to this routine are the same as for sqliteSelect().
15089562b551Sdrh ** See the header comment on that routine for additional information.
15099562b551Sdrh */
15109562b551Sdrh static int simpleMinMaxQuery(Parse *pParse, Select *p, int eDest, int iParm){
15119562b551Sdrh   Expr *pExpr;
15129562b551Sdrh   int iCol;
15139562b551Sdrh   Table *pTab;
15149562b551Sdrh   Index *pIdx;
15159562b551Sdrh   int base;
15169562b551Sdrh   Vdbe *v;
15179562b551Sdrh   int openOp;
15189562b551Sdrh   int seekOp;
15199562b551Sdrh   int cont;
15209562b551Sdrh   ExprList eList;
15219562b551Sdrh   struct ExprList_item eListItem;
15229562b551Sdrh 
15239562b551Sdrh   /* Check to see if this query is a simple min() or max() query.  Return
15249562b551Sdrh   ** zero if it is  not.
15259562b551Sdrh   */
15269562b551Sdrh   if( p->pGroupBy || p->pHaving || p->pWhere ) return 0;
1527ad3cab52Sdrh   if( p->pSrc->nSrc!=1 ) return 0;
15289562b551Sdrh   if( p->pEList->nExpr!=1 ) return 0;
15299562b551Sdrh   pExpr = p->pEList->a[0].pExpr;
15309562b551Sdrh   if( pExpr->op!=TK_AGG_FUNCTION ) return 0;
15319562b551Sdrh   if( pExpr->pList==0 || pExpr->pList->nExpr!=1 ) return 0;
15320bce8354Sdrh   if( pExpr->token.n!=3 ) return 0;
15330bce8354Sdrh   if( sqliteStrNICmp(pExpr->token.z,"min",3)==0 ){
15340bce8354Sdrh     seekOp = OP_Rewind;
15350bce8354Sdrh   }else if( sqliteStrNICmp(pExpr->token.z,"max",3)==0 ){
15360bce8354Sdrh     seekOp = OP_Last;
15370bce8354Sdrh   }else{
15380bce8354Sdrh     return 0;
15390bce8354Sdrh   }
15409562b551Sdrh   pExpr = pExpr->pList->a[0].pExpr;
15419562b551Sdrh   if( pExpr->op!=TK_COLUMN ) return 0;
15429562b551Sdrh   iCol = pExpr->iColumn;
15439562b551Sdrh   pTab = p->pSrc->a[0].pTab;
15449562b551Sdrh 
15459562b551Sdrh   /* If we get to here, it means the query is of the correct form.
154617f71934Sdrh   ** Check to make sure we have an index and make pIdx point to the
154717f71934Sdrh   ** appropriate index.  If the min() or max() is on an INTEGER PRIMARY
154817f71934Sdrh   ** key column, no index is necessary so set pIdx to NULL.  If no
154917f71934Sdrh   ** usable index is found, return 0.
15509562b551Sdrh   */
15519562b551Sdrh   if( iCol<0 ){
15529562b551Sdrh     pIdx = 0;
15539562b551Sdrh   }else{
15549562b551Sdrh     for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){
15559562b551Sdrh       assert( pIdx->nColumn>=1 );
15569562b551Sdrh       if( pIdx->aiColumn[0]==iCol ) break;
15579562b551Sdrh     }
15589562b551Sdrh     if( pIdx==0 ) return 0;
15599562b551Sdrh   }
15609562b551Sdrh 
156117f71934Sdrh   /* Identify column names if we will be using the callback.  This
15629562b551Sdrh   ** step is skipped if the output is going to a table or a memory cell.
15639562b551Sdrh   */
15649562b551Sdrh   v = sqliteGetVdbe(pParse);
15659562b551Sdrh   if( v==0 ) return 0;
15669562b551Sdrh   if( eDest==SRT_Callback ){
1567832508b7Sdrh     generateColumnNames(pParse, p->base, p->pSrc, p->pEList);
15689562b551Sdrh   }
15699562b551Sdrh 
157017f71934Sdrh   /* Generating code to find the min or the max.  Basically all we have
157117f71934Sdrh   ** to do is find the first or the last entry in the chosen index.  If
157217f71934Sdrh   ** the min() or max() is on the INTEGER PRIMARY KEY, then find the first
157317f71934Sdrh   ** or last entry in the main table.
15749562b551Sdrh   */
15755cf8e8c7Sdrh   if( !pParse->schemaVerified && (pParse->db->flags & SQLITE_InTrans)==0 ){
15765cf8e8c7Sdrh     sqliteVdbeAddOp(v, OP_VerifyCookie, pParse->db->schema_cookie, 0);
15775cf8e8c7Sdrh     pParse->schemaVerified = 1;
15785cf8e8c7Sdrh   }
15799562b551Sdrh   openOp = pTab->isTemp ? OP_OpenAux : OP_Open;
1580832508b7Sdrh   base = p->base;
15819562b551Sdrh   sqliteVdbeAddOp(v, openOp, base, pTab->tnum);
15825cf8e8c7Sdrh   sqliteVdbeChangeP3(v, -1, pTab->zName, P3_STATIC);
15839562b551Sdrh   if( pIdx==0 ){
15849562b551Sdrh     sqliteVdbeAddOp(v, seekOp, base, 0);
15859562b551Sdrh   }else{
15869562b551Sdrh     sqliteVdbeAddOp(v, openOp, base+1, pIdx->tnum);
15875cf8e8c7Sdrh     sqliteVdbeChangeP3(v, -1, pIdx->zName, P3_STATIC);
15889562b551Sdrh     sqliteVdbeAddOp(v, seekOp, base+1, 0);
15899562b551Sdrh     sqliteVdbeAddOp(v, OP_IdxRecno, base+1, 0);
15909562b551Sdrh     sqliteVdbeAddOp(v, OP_Close, base+1, 0);
15919562b551Sdrh     sqliteVdbeAddOp(v, OP_MoveTo, base, 0);
15929562b551Sdrh   }
15935cf8e8c7Sdrh   eList.nExpr = 1;
15945cf8e8c7Sdrh   memset(&eListItem, 0, sizeof(eListItem));
15955cf8e8c7Sdrh   eList.a = &eListItem;
15965cf8e8c7Sdrh   eList.a[0].pExpr = pExpr;
15979562b551Sdrh   cont = sqliteVdbeMakeLabel(v);
159838640e15Sdrh   selectInnerLoop(pParse, p, &eList, 0, 0, 0, -1, eDest, iParm, cont, cont);
15999562b551Sdrh   sqliteVdbeResolveLabel(v, cont);
16009562b551Sdrh   sqliteVdbeAddOp(v, OP_Close, base, 0);
16019562b551Sdrh   return 1;
16029562b551Sdrh }
16039562b551Sdrh 
16049562b551Sdrh /*
16059bb61fe7Sdrh ** Generate code for the given SELECT statement.
16069bb61fe7Sdrh **
1607fef5208cSdrh ** The results are distributed in various ways depending on the
1608fef5208cSdrh ** value of eDest and iParm.
1609fef5208cSdrh **
1610fef5208cSdrh **     eDest Value       Result
1611fef5208cSdrh **     ------------    -------------------------------------------
1612fef5208cSdrh **     SRT_Callback    Invoke the callback for each row of the result.
1613fef5208cSdrh **
1614fef5208cSdrh **     SRT_Mem         Store first result in memory cell iParm
1615fef5208cSdrh **
1616fef5208cSdrh **     SRT_Set         Store results as keys of a table with cursor iParm
1617fef5208cSdrh **
161882c3d636Sdrh **     SRT_Union       Store results as a key in a temporary table iParm
161982c3d636Sdrh **
1620c4a3c779Sdrh **     SRT_Except      Remove results form the temporary table iParm.
1621c4a3c779Sdrh **
1622c4a3c779Sdrh **     SRT_Table       Store results in temporary table iParm
16239bb61fe7Sdrh **
16249bb61fe7Sdrh ** This routine returns the number of errors.  If any errors are
16259bb61fe7Sdrh ** encountered, then an appropriate error message is left in
16269bb61fe7Sdrh ** pParse->zErrMsg.
16279bb61fe7Sdrh **
16289bb61fe7Sdrh ** This routine does NOT free the Select structure passed in.  The
16299bb61fe7Sdrh ** calling function needs to do that.
16301b2e0329Sdrh **
16311b2e0329Sdrh ** The pParent, parentTab, and *pParentAgg fields are filled in if this
16321b2e0329Sdrh ** SELECT is a subquery.  This routine may try to combine this SELECT
16331b2e0329Sdrh ** with its parent to form a single flat query.  In so doing, it might
16341b2e0329Sdrh ** change the parent query from a non-aggregate to an aggregate query.
16351b2e0329Sdrh ** For that reason, the pParentAgg flag is passed as a pointer, so it
16361b2e0329Sdrh ** can be changed.
16379bb61fe7Sdrh */
16389bb61fe7Sdrh int sqliteSelect(
1639cce7d176Sdrh   Parse *pParse,         /* The parser context */
16409bb61fe7Sdrh   Select *p,             /* The SELECT statement being coded. */
164182c3d636Sdrh   int eDest,             /* One of: SRT_Callback Mem Set Union Except */
1642832508b7Sdrh   int iParm,             /* Save result in this memory location, if >=0 */
1643832508b7Sdrh   Select *pParent,       /* Another SELECT for which this is a sub-query */
1644832508b7Sdrh   int parentTab,         /* Index in pParent->pSrc of this query */
16451b2e0329Sdrh   int *pParentAgg        /* True if pParent uses aggregate functions */
1646cce7d176Sdrh ){
1647d8bc7086Sdrh   int i;
1648cce7d176Sdrh   WhereInfo *pWInfo;
1649cce7d176Sdrh   Vdbe *v;
1650cce7d176Sdrh   int isAgg = 0;         /* True for select lists like "count(*)" */
1651a2e00042Sdrh   ExprList *pEList;      /* List of columns to extract. */
1652ad3cab52Sdrh   SrcList *pTabList;     /* List of tables to select from */
16539bb61fe7Sdrh   Expr *pWhere;          /* The WHERE clause.  May be NULL */
16549bb61fe7Sdrh   ExprList *pOrderBy;    /* The ORDER BY clause.  May be NULL */
16552282792aSdrh   ExprList *pGroupBy;    /* The GROUP BY clause.  May be NULL */
16562282792aSdrh   Expr *pHaving;         /* The HAVING clause.  May be NULL */
165719a775c2Sdrh   int isDistinct;        /* True if the DISTINCT keyword is present */
165819a775c2Sdrh   int distinct;          /* Table to use for the distinct set */
165910e5e3cfSdrh   int base;              /* First cursor available for use */
16601d83f052Sdrh   int rc = 1;            /* Value to return from this function */
16619bb61fe7Sdrh 
1662daffd0e5Sdrh   if( sqlite_malloc_failed || pParse->nErr || p==0 ) return 1;
1663daffd0e5Sdrh 
166482c3d636Sdrh   /* If there is are a sequence of queries, do the earlier ones first.
166582c3d636Sdrh   */
166682c3d636Sdrh   if( p->pPrior ){
166782c3d636Sdrh     return multiSelect(pParse, p, eDest, iParm);
166882c3d636Sdrh   }
166982c3d636Sdrh 
167082c3d636Sdrh   /* Make local copies of the parameters for this query.
167182c3d636Sdrh   */
16729bb61fe7Sdrh   pTabList = p->pSrc;
16739bb61fe7Sdrh   pWhere = p->pWhere;
16749bb61fe7Sdrh   pOrderBy = p->pOrderBy;
16752282792aSdrh   pGroupBy = p->pGroupBy;
16762282792aSdrh   pHaving = p->pHaving;
167719a775c2Sdrh   isDistinct = p->isDistinct;
16789bb61fe7Sdrh 
1679832508b7Sdrh   /* Allocate a block of VDBE cursors, one for each table in the FROM clause.
1680832508b7Sdrh   ** The WHERE processing requires that the cursors for the tables in the
1681832508b7Sdrh   ** FROM clause be consecutive.
168210e5e3cfSdrh   */
1683832508b7Sdrh   base = p->base = pParse->nTab;
1684ad3cab52Sdrh   pParse->nTab += pTabList->nSrc;
168510e5e3cfSdrh 
16869bb61fe7Sdrh   /*
16879bb61fe7Sdrh   ** Do not even attempt to generate any code if we have already seen
16889bb61fe7Sdrh   ** errors before this routine starts.
16899bb61fe7Sdrh   */
16901d83f052Sdrh   if( pParse->nErr>0 ) goto select_end;
1691cce7d176Sdrh 
1692d8bc7086Sdrh   /* Look up every table in the table list and create an appropriate
1693d8bc7086Sdrh   ** columnlist in pEList if there isn't one already.  (The parser leaves
1694967e8b73Sdrh   ** a NULL in the p->pEList if the SQL said "SELECT * FROM ...")
1695cce7d176Sdrh   */
1696d8bc7086Sdrh   if( fillInColumnList(pParse, p) ){
16971d83f052Sdrh     goto select_end;
1698cce7d176Sdrh   }
1699ad2d8307Sdrh   pWhere = p->pWhere;
1700d8bc7086Sdrh   pEList = p->pEList;
17011d83f052Sdrh   if( pEList==0 ) goto select_end;
1702cce7d176Sdrh 
17032282792aSdrh   /* If writing to memory or generating a set
17042282792aSdrh   ** only a single column may be output.
170519a775c2Sdrh   */
1706fef5208cSdrh   if( (eDest==SRT_Mem || eDest==SRT_Set) && pEList->nExpr>1 ){
170719a775c2Sdrh     sqliteSetString(&pParse->zErrMsg, "only a single result allowed for "
170819a775c2Sdrh        "a SELECT that is part of an expression", 0);
170919a775c2Sdrh     pParse->nErr++;
17101d83f052Sdrh     goto select_end;
171119a775c2Sdrh   }
171219a775c2Sdrh 
1713c926afbcSdrh   /* ORDER BY is ignored for some destinations.
17142282792aSdrh   */
1715c926afbcSdrh   switch( eDest ){
1716c926afbcSdrh     case SRT_Union:
1717c926afbcSdrh     case SRT_Except:
1718c926afbcSdrh     case SRT_Discard:
1719acd4c695Sdrh       pOrderBy = 0;
1720c926afbcSdrh       break;
1721c926afbcSdrh     default:
1722c926afbcSdrh       break;
17232282792aSdrh   }
17242282792aSdrh 
172510e5e3cfSdrh   /* At this point, we should have allocated all the cursors that we
1726832508b7Sdrh   ** need to handle subquerys and temporary tables.
172710e5e3cfSdrh   **
1728967e8b73Sdrh   ** Resolve the column names and do a semantics check on all the expressions.
17292282792aSdrh   */
17304794b980Sdrh   for(i=0; i<pEList->nExpr; i++){
1731832508b7Sdrh     if( sqliteExprResolveIds(pParse, base, pTabList, 0, pEList->a[i].pExpr) ){
17321d83f052Sdrh       goto select_end;
1733cce7d176Sdrh     }
17342282792aSdrh     if( sqliteExprCheck(pParse, pEList->a[i].pExpr, 1, &isAgg) ){
17351d83f052Sdrh       goto select_end;
1736cce7d176Sdrh     }
1737cce7d176Sdrh   }
1738cce7d176Sdrh   if( pWhere ){
1739832508b7Sdrh     if( sqliteExprResolveIds(pParse, base, pTabList, pEList, pWhere) ){
17401d83f052Sdrh       goto select_end;
1741cce7d176Sdrh     }
1742cce7d176Sdrh     if( sqliteExprCheck(pParse, pWhere, 0, 0) ){
17431d83f052Sdrh       goto select_end;
1744cce7d176Sdrh     }
1745cce7d176Sdrh   }
1746cce7d176Sdrh   if( pOrderBy ){
1747cce7d176Sdrh     for(i=0; i<pOrderBy->nExpr; i++){
17482282792aSdrh       Expr *pE = pOrderBy->a[i].pExpr;
17499208643dSdrh       if( sqliteExprIsConstant(pE) ){
1750e4de1febSdrh         int iCol;
1751e4de1febSdrh         if( sqliteExprIsInteger(pE, &iCol)==0 ){
17529208643dSdrh           sqliteSetString(&pParse->zErrMsg,
1753e4de1febSdrh                "ORDER BY terms must not be non-integer constants", 0);
17549208643dSdrh           pParse->nErr++;
17551d83f052Sdrh           goto select_end;
1756e4de1febSdrh         }else if( iCol<=0 || iCol>pEList->nExpr ){
1757e4de1febSdrh           char zBuf[2000];
1758e4de1febSdrh           sprintf(zBuf,"ORDER BY column number %d out of range - should be "
1759e4de1febSdrh              "between 1 and %d", iCol, pEList->nExpr);
1760e4de1febSdrh           sqliteSetString(&pParse->zErrMsg, zBuf, 0);
1761e4de1febSdrh           pParse->nErr++;
1762e4de1febSdrh           goto select_end;
1763e4de1febSdrh         }
1764e4de1febSdrh         sqliteExprDelete(pE);
1765e4de1febSdrh         pE = pOrderBy->a[i].pExpr = sqliteExprDup(pEList->a[iCol-1].pExpr);
17669208643dSdrh       }
1767832508b7Sdrh       if( sqliteExprResolveIds(pParse, base, pTabList, pEList, pE) ){
17681d83f052Sdrh         goto select_end;
1769cce7d176Sdrh       }
17702282792aSdrh       if( sqliteExprCheck(pParse, pE, isAgg, 0) ){
17711d83f052Sdrh         goto select_end;
1772cce7d176Sdrh       }
1773cce7d176Sdrh     }
1774cce7d176Sdrh   }
17752282792aSdrh   if( pGroupBy ){
17762282792aSdrh     for(i=0; i<pGroupBy->nExpr; i++){
17772282792aSdrh       Expr *pE = pGroupBy->a[i].pExpr;
17789208643dSdrh       if( sqliteExprIsConstant(pE) ){
17799208643dSdrh         sqliteSetString(&pParse->zErrMsg,
17809208643dSdrh              "GROUP BY expressions should not be constant", 0);
17819208643dSdrh         pParse->nErr++;
17821d83f052Sdrh         goto select_end;
17839208643dSdrh       }
1784832508b7Sdrh       if( sqliteExprResolveIds(pParse, base, pTabList, pEList, pE) ){
17851d83f052Sdrh         goto select_end;
17862282792aSdrh       }
17872282792aSdrh       if( sqliteExprCheck(pParse, pE, isAgg, 0) ){
17881d83f052Sdrh         goto select_end;
17892282792aSdrh       }
17902282792aSdrh     }
17912282792aSdrh   }
17922282792aSdrh   if( pHaving ){
17932282792aSdrh     if( pGroupBy==0 ){
1794da93281eSdrh       sqliteSetString(&pParse->zErrMsg, "a GROUP BY clause is required "
1795da93281eSdrh          "before HAVING", 0);
17962282792aSdrh       pParse->nErr++;
17971d83f052Sdrh       goto select_end;
17982282792aSdrh     }
1799832508b7Sdrh     if( sqliteExprResolveIds(pParse, base, pTabList, pEList, pHaving) ){
18001d83f052Sdrh       goto select_end;
18012282792aSdrh     }
1802da93281eSdrh     if( sqliteExprCheck(pParse, pHaving, isAgg, 0) ){
18031d83f052Sdrh       goto select_end;
18042282792aSdrh     }
1805cce7d176Sdrh   }
1806cce7d176Sdrh 
18079562b551Sdrh   /* Check for the special case of a min() or max() function by itself
18089562b551Sdrh   ** in the result set.
18099562b551Sdrh   */
18109562b551Sdrh   if( simpleMinMaxQuery(pParse, p, eDest, iParm) ){
18115cf8e8c7Sdrh     rc = 0;
18129562b551Sdrh     goto select_end;
18139562b551Sdrh   }
18149562b551Sdrh 
1815d820cb1bSdrh   /* Begin generating code.
1816d820cb1bSdrh   */
1817d820cb1bSdrh   v = sqliteGetVdbe(pParse);
1818d820cb1bSdrh   if( v==0 ) goto select_end;
1819d820cb1bSdrh 
18200bb28106Sdrh   /* Identify column names if we will be using in the callback.  This
18210bb28106Sdrh   ** step is skipped if the output is going to a table or a memory cell.
18220bb28106Sdrh   */
18230bb28106Sdrh   if( eDest==SRT_Callback ){
18240bb28106Sdrh     generateColumnNames(pParse, p->base, pTabList, pEList);
18250bb28106Sdrh   }
18260bb28106Sdrh 
18270bb28106Sdrh   /* Set the limiter
18280bb28106Sdrh   */
18290bb28106Sdrh   if( p->nLimit<=0 ){
1830d11d382cSdrh     p->nLimit = -1;
18310bb28106Sdrh     p->nOffset = 0;
18320bb28106Sdrh   }else{
1833d11d382cSdrh     int iMem = pParse->nMem++;
1834d11d382cSdrh     sqliteVdbeAddOp(v, OP_Integer, -p->nLimit, 0);
1835bf5cd97eSdrh     sqliteVdbeAddOp(v, OP_MemStore, iMem, 1);
1836d11d382cSdrh     p->nLimit = iMem;
1837d11d382cSdrh     if( p->nOffset<=0 ){
1838d11d382cSdrh       p->nOffset = 0;
1839d11d382cSdrh     }else{
1840d11d382cSdrh       iMem = pParse->nMem++;
1841d11d382cSdrh       sqliteVdbeAddOp(v, OP_Integer, -p->nOffset, 0);
1842bf5cd97eSdrh       sqliteVdbeAddOp(v, OP_MemStore, iMem, 1);
1843d11d382cSdrh       p->nOffset = iMem;
1844d11d382cSdrh     }
18450bb28106Sdrh   }
18460bb28106Sdrh 
1847d820cb1bSdrh   /* Generate code for all sub-queries in the FROM clause
1848d820cb1bSdrh   */
1849ad3cab52Sdrh   for(i=0; i<pTabList->nSrc; i++){
1850a76b5dfcSdrh     if( pTabList->a[i].pSelect==0 ) continue;
18512d0794e3Sdrh     sqliteSelect(pParse, pTabList->a[i].pSelect, SRT_TempTable, base+i,
18521b2e0329Sdrh                  p, i, &isAgg);
1853832508b7Sdrh     pTabList = p->pSrc;
1854832508b7Sdrh     pWhere = p->pWhere;
1855acd4c695Sdrh     if( eDest==SRT_Callback ){
1856832508b7Sdrh       pOrderBy = p->pOrderBy;
1857acd4c695Sdrh     }
1858832508b7Sdrh     pGroupBy = p->pGroupBy;
1859832508b7Sdrh     pHaving = p->pHaving;
1860832508b7Sdrh     isDistinct = p->isDistinct;
18611b2e0329Sdrh   }
18621b2e0329Sdrh 
18631b2e0329Sdrh   /* Check to see if this is a subquery that can be "flattened" into its parent.
18641b2e0329Sdrh   ** If flattening is a possiblity, do so and return immediately.
18651b2e0329Sdrh   */
18661b2e0329Sdrh   if( pParent && pParentAgg &&
18671b2e0329Sdrh       flattenSubquery(pParent, parentTab, *pParentAgg, isAgg) ){
18681b2e0329Sdrh     if( isAgg ) *pParentAgg = 1;
18691b2e0329Sdrh     return rc;
18701b2e0329Sdrh   }
1871832508b7Sdrh 
18722d0794e3Sdrh   /* If the output is destined for a temporary table, open that table.
18732d0794e3Sdrh   */
18742d0794e3Sdrh   if( eDest==SRT_TempTable ){
18752d0794e3Sdrh     sqliteVdbeAddOp(v, OP_OpenTemp, iParm, 0);
18762d0794e3Sdrh   }
18772d0794e3Sdrh 
18782282792aSdrh   /* Do an analysis of aggregate expressions.
1879efb7251dSdrh   */
1880d820cb1bSdrh   sqliteAggregateInfoReset(pParse);
18812282792aSdrh   if( isAgg ){
18820bce8354Sdrh     assert( pParse->nAgg==0 );
18832282792aSdrh     for(i=0; i<pEList->nExpr; i++){
18842282792aSdrh       if( sqliteExprAnalyzeAggregates(pParse, pEList->a[i].pExpr) ){
18851d83f052Sdrh         goto select_end;
18862282792aSdrh       }
18872282792aSdrh     }
18882282792aSdrh     if( pGroupBy ){
18892282792aSdrh       for(i=0; i<pGroupBy->nExpr; i++){
18902282792aSdrh         if( sqliteExprAnalyzeAggregates(pParse, pGroupBy->a[i].pExpr) ){
18911d83f052Sdrh           goto select_end;
18922282792aSdrh         }
18932282792aSdrh       }
18942282792aSdrh     }
18952282792aSdrh     if( pHaving && sqliteExprAnalyzeAggregates(pParse, pHaving) ){
18961d83f052Sdrh       goto select_end;
18972282792aSdrh     }
1898191b690eSdrh     if( pOrderBy ){
1899191b690eSdrh       for(i=0; i<pOrderBy->nExpr; i++){
1900191b690eSdrh         if( sqliteExprAnalyzeAggregates(pParse, pOrderBy->a[i].pExpr) ){
19011d83f052Sdrh           goto select_end;
1902191b690eSdrh         }
1903191b690eSdrh       }
1904191b690eSdrh     }
1905efb7251dSdrh   }
1906efb7251dSdrh 
19072282792aSdrh   /* Reset the aggregator
1908cce7d176Sdrh   */
1909cce7d176Sdrh   if( isAgg ){
191099fcd718Sdrh     sqliteVdbeAddOp(v, OP_AggReset, 0, pParse->nAgg);
1911e5095355Sdrh     for(i=0; i<pParse->nAgg; i++){
19120bce8354Sdrh       FuncDef *pFunc;
19130bce8354Sdrh       if( (pFunc = pParse->aAgg[i].pFunc)!=0 && pFunc->xFinalize!=0 ){
19141350b030Sdrh         sqliteVdbeAddOp(v, OP_AggInit, 0, i);
19150bce8354Sdrh         sqliteVdbeChangeP3(v, -1, (char*)pFunc, P3_POINTER);
1916e5095355Sdrh       }
1917e5095355Sdrh     }
19181bee3d7bSdrh     if( pGroupBy==0 ){
19191bee3d7bSdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
19201bee3d7bSdrh       sqliteVdbeAddOp(v, OP_AggFocus, 0, 0);
19211bee3d7bSdrh     }
1922cce7d176Sdrh   }
1923cce7d176Sdrh 
192419a775c2Sdrh   /* Initialize the memory cell to NULL
192519a775c2Sdrh   */
1926fef5208cSdrh   if( eDest==SRT_Mem ){
192799fcd718Sdrh     sqliteVdbeAddOp(v, OP_String, 0, 0);
19288721ce4aSdrh     sqliteVdbeAddOp(v, OP_MemStore, iParm, 1);
192919a775c2Sdrh   }
193019a775c2Sdrh 
1931832508b7Sdrh   /* Open a temporary table to use for the distinct set.
1932cce7d176Sdrh   */
193319a775c2Sdrh   if( isDistinct ){
1934832508b7Sdrh     distinct = pParse->nTab++;
1935c6b52df3Sdrh     sqliteVdbeAddOp(v, OP_OpenTemp, distinct, 1);
1936832508b7Sdrh   }else{
1937832508b7Sdrh     distinct = -1;
1938efb7251dSdrh   }
1939832508b7Sdrh 
1940832508b7Sdrh   /* Begin the database scan
1941832508b7Sdrh   */
1942e3184744Sdrh   pWInfo = sqliteWhereBegin(pParse, p->base, pTabList, pWhere, 0, &pOrderBy);
19431d83f052Sdrh   if( pWInfo==0 ) goto select_end;
1944cce7d176Sdrh 
19452282792aSdrh   /* Use the standard inner loop if we are not dealing with
19462282792aSdrh   ** aggregates
1947cce7d176Sdrh   */
1948da9d6c45Sdrh   if( !isAgg ){
1949df199a25Sdrh     if( selectInnerLoop(pParse, p, pEList, 0, 0, pOrderBy, distinct, eDest,
1950df199a25Sdrh                     iParm, pWInfo->iContinue, pWInfo->iBreak) ){
19511d83f052Sdrh        goto select_end;
1952cce7d176Sdrh     }
1953da9d6c45Sdrh   }
1954cce7d176Sdrh 
1955e3184744Sdrh   /* If we are dealing with aggregates, then do the special aggregate
19562282792aSdrh   ** processing.
1957efb7251dSdrh   */
19582282792aSdrh   else{
19592282792aSdrh     if( pGroupBy ){
19601bee3d7bSdrh       int lbl1;
19612282792aSdrh       for(i=0; i<pGroupBy->nExpr; i++){
19622282792aSdrh         sqliteExprCode(pParse, pGroupBy->a[i].pExpr);
1963efb7251dSdrh       }
196499fcd718Sdrh       sqliteVdbeAddOp(v, OP_MakeKey, pGroupBy->nExpr, 0);
196538640e15Sdrh       if( pParse->db->file_format>=3 ) sqliteAddKeyType(v, pGroupBy);
19661bee3d7bSdrh       lbl1 = sqliteVdbeMakeLabel(v);
196799fcd718Sdrh       sqliteVdbeAddOp(v, OP_AggFocus, 0, lbl1);
19682282792aSdrh       for(i=0; i<pParse->nAgg; i++){
19692282792aSdrh         if( pParse->aAgg[i].isAgg ) continue;
19702282792aSdrh         sqliteExprCode(pParse, pParse->aAgg[i].pExpr);
197199fcd718Sdrh         sqliteVdbeAddOp(v, OP_AggSet, 0, i);
19722282792aSdrh       }
19732282792aSdrh       sqliteVdbeResolveLabel(v, lbl1);
19742282792aSdrh     }
19752282792aSdrh     for(i=0; i<pParse->nAgg; i++){
19762282792aSdrh       Expr *pE;
19770bce8354Sdrh       int j;
19782282792aSdrh       if( !pParse->aAgg[i].isAgg ) continue;
19792282792aSdrh       pE = pParse->aAgg[i].pExpr;
19802282792aSdrh       assert( pE->op==TK_AGG_FUNCTION );
19810bce8354Sdrh       if( pE->pList ){
1982e5095355Sdrh         for(j=0; j<pE->pList->nExpr; j++){
1983e5095355Sdrh           sqliteExprCode(pParse, pE->pList->a[j].pExpr);
1984e5095355Sdrh         }
19852282792aSdrh       }
19861350b030Sdrh       sqliteVdbeAddOp(v, OP_Integer, i, 0);
1987f55f25f0Sdrh       sqliteVdbeAddOp(v, OP_AggFunc, 0, pE->pList ? pE->pList->nExpr : 0);
19880bce8354Sdrh       assert( pParse->aAgg[i].pFunc!=0 );
19890bce8354Sdrh       assert( pParse->aAgg[i].pFunc->xStep!=0 );
19900bce8354Sdrh       sqliteVdbeChangeP3(v, -1, (char*)pParse->aAgg[i].pFunc, P3_POINTER);
19912282792aSdrh     }
19922282792aSdrh   }
19932282792aSdrh 
1994cce7d176Sdrh   /* End the database scan loop.
1995cce7d176Sdrh   */
1996cce7d176Sdrh   sqliteWhereEnd(pWInfo);
1997cce7d176Sdrh 
19982282792aSdrh   /* If we are processing aggregates, we need to set up a second loop
19992282792aSdrh   ** over all of the aggregate values and process them.
20002282792aSdrh   */
20012282792aSdrh   if( isAgg ){
20022282792aSdrh     int endagg = sqliteVdbeMakeLabel(v);
20032282792aSdrh     int startagg;
200499fcd718Sdrh     startagg = sqliteVdbeAddOp(v, OP_AggNext, 0, endagg);
20052282792aSdrh     pParse->useAgg = 1;
20062282792aSdrh     if( pHaving ){
2007f5905aa7Sdrh       sqliteExprIfFalse(pParse, pHaving, startagg, 1);
20082282792aSdrh     }
2009df199a25Sdrh     if( selectInnerLoop(pParse, p, pEList, 0, 0, pOrderBy, distinct, eDest,
2010df199a25Sdrh                     iParm, startagg, endagg) ){
20111d83f052Sdrh       goto select_end;
20122282792aSdrh     }
201399fcd718Sdrh     sqliteVdbeAddOp(v, OP_Goto, 0, startagg);
201499fcd718Sdrh     sqliteVdbeResolveLabel(v, endagg);
201599fcd718Sdrh     sqliteVdbeAddOp(v, OP_Noop, 0, 0);
20162282792aSdrh     pParse->useAgg = 0;
20172282792aSdrh   }
20182282792aSdrh 
2019cce7d176Sdrh   /* If there is an ORDER BY clause, then we need to sort the results
2020cce7d176Sdrh   ** and send them to the callback one by one.
2021cce7d176Sdrh   */
2022cce7d176Sdrh   if( pOrderBy ){
2023c926afbcSdrh     generateSortTail(p, v, pEList->nExpr, eDest, iParm);
2024cce7d176Sdrh   }
20256a535340Sdrh 
20266a535340Sdrh 
20276a535340Sdrh   /* Issue a null callback if that is what the user wants.
20286a535340Sdrh   */
20296a535340Sdrh   if( (pParse->db->flags & SQLITE_NullCallback)!=0 && eDest==SRT_Callback ){
20306a535340Sdrh     sqliteVdbeAddOp(v, OP_NullCallback, pEList->nExpr, 0);
20316a535340Sdrh   }
20326a535340Sdrh 
20331d83f052Sdrh   /* The SELECT was successfully coded.   Set the return code to 0
20341d83f052Sdrh   ** to indicate no errors.
20351d83f052Sdrh   */
20361d83f052Sdrh   rc = 0;
20371d83f052Sdrh 
20381d83f052Sdrh   /* Control jumps to here if an error is encountered above, or upon
20391d83f052Sdrh   ** successful coding of the SELECT.
20401d83f052Sdrh   */
20411d83f052Sdrh select_end:
2042832508b7Sdrh   pParse->nTab = base;
20431d83f052Sdrh   sqliteAggregateInfoReset(pParse);
20441d83f052Sdrh   return rc;
2045cce7d176Sdrh }
2046