drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 1 | /* |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 2 | ** 2005-07-08 |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 3 | ** |
| 4 | ** The author disclaims copyright to this source code. In place of |
| 5 | ** a legal notice, here is a blessing: |
| 6 | ** |
| 7 | ** May you do good and not evil. |
| 8 | ** May you find forgiveness for yourself and forgive others. |
| 9 | ** May you share freely, never taking more than you give. |
| 10 | ** |
| 11 | ************************************************************************* |
| 12 | ** This file contains code associated with the ANALYZE command. |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 13 | ** |
| 14 | ** The ANALYZE command gather statistics about the content of tables |
| 15 | ** and indices. These statistics are made available to the query planner |
| 16 | ** to help it make better decisions about how to perform queries. |
| 17 | ** |
| 18 | ** The following system tables are or have been supported: |
| 19 | ** |
| 20 | ** CREATE TABLE sqlite_stat1(tbl, idx, stat); |
| 21 | ** CREATE TABLE sqlite_stat2(tbl, idx, sampleno, sample); |
drh | f404c86 | 2011-08-13 15:25:10 +0000 | [diff] [blame] | 22 | ** CREATE TABLE sqlite_stat3(tbl, idx, nEq, nLt, nDLt, sample); |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 23 | ** CREATE TABLE sqlite_stat4(tbl, idx, nEq, nLt, nDLt, sample); |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 24 | ** |
| 25 | ** Additional tables might be added in future releases of SQLite. |
| 26 | ** The sqlite_stat2 table is not created or used unless the SQLite version |
drh | d3ed734 | 2011-09-21 00:09:41 +0000 | [diff] [blame] | 27 | ** is between 3.6.18 and 3.7.8, inclusive, and unless SQLite is compiled |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 28 | ** with SQLITE_ENABLE_STAT2. The sqlite_stat2 table is deprecated. |
drh | f7b5496 | 2013-05-28 12:11:54 +0000 | [diff] [blame] | 29 | ** The sqlite_stat2 table is superseded by sqlite_stat3, which is only |
drh | d3ed734 | 2011-09-21 00:09:41 +0000 | [diff] [blame] | 30 | ** created and used by SQLite versions 3.7.9 and later and with |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 31 | ** SQLITE_ENABLE_STAT3 defined. The functionality of sqlite_stat3 |
| 32 | ** is a superset of sqlite_stat2. The sqlite_stat4 is an enhanced |
| 33 | ** version of sqlite_stat3 and is only available when compiled with |
drh | 0ae4f14 | 2013-10-14 14:21:59 +0000 | [diff] [blame] | 34 | ** SQLITE_ENABLE_STAT4 and in SQLite versions 3.8.1 and later. It is |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 35 | ** not possible to enable both STAT3 and STAT4 at the same time. If they |
mistachkin | 79e8445 | 2013-08-30 19:59:48 +0000 | [diff] [blame] | 36 | ** are both enabled, then STAT4 takes precedence. |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 37 | ** |
peter.d.reid | 60ec914 | 2014-09-06 16:39:46 +0000 | [diff] [blame] | 38 | ** For most applications, sqlite_stat1 provides all the statistics required |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 39 | ** for the query planner to make good choices. |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 40 | ** |
| 41 | ** Format of sqlite_stat1: |
| 42 | ** |
| 43 | ** There is normally one row per index, with the index identified by the |
| 44 | ** name in the idx column. The tbl column is the name of the table to |
| 45 | ** which the index belongs. In each such row, the stat column will be |
| 46 | ** a string consisting of a list of integers. The first integer in this |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 47 | ** list is the number of rows in the index. (This is the same as the |
| 48 | ** number of rows in the table, except for partial indices.) The second |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 49 | ** integer is the average number of rows in the index that have the same |
| 50 | ** value in the first column of the index. The third integer is the average |
| 51 | ** number of rows in the index that have the same value for the first two |
| 52 | ** columns. The N-th integer (for N>1) is the average number of rows in |
| 53 | ** the index which have the same value for the first N-1 columns. For |
| 54 | ** a K-column index, there will be K+1 integers in the stat column. If |
| 55 | ** the index is unique, then the last integer will be 1. |
| 56 | ** |
| 57 | ** The list of integers in the stat column can optionally be followed |
| 58 | ** by the keyword "unordered". The "unordered" keyword, if it is present, |
| 59 | ** must be separated from the last integer by a single space. If the |
| 60 | ** "unordered" keyword is present, then the query planner assumes that |
| 61 | ** the index is unordered and will not use the index for a range query. |
| 62 | ** |
| 63 | ** If the sqlite_stat1.idx column is NULL, then the sqlite_stat1.stat |
| 64 | ** column contains a single integer which is the (estimated) number of |
| 65 | ** rows in the table identified by sqlite_stat1.tbl. |
| 66 | ** |
| 67 | ** Format of sqlite_stat2: |
| 68 | ** |
| 69 | ** The sqlite_stat2 is only created and is only used if SQLite is compiled |
| 70 | ** with SQLITE_ENABLE_STAT2 and if the SQLite version number is between |
drh | d3ed734 | 2011-09-21 00:09:41 +0000 | [diff] [blame] | 71 | ** 3.6.18 and 3.7.8. The "stat2" table contains additional information |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 72 | ** about the distribution of keys within an index. The index is identified by |
| 73 | ** the "idx" column and the "tbl" column is the name of the table to which |
| 74 | ** the index belongs. There are usually 10 rows in the sqlite_stat2 |
| 75 | ** table for each index. |
| 76 | ** |
dan | 23e7c4d | 2011-08-15 12:02:21 +0000 | [diff] [blame] | 77 | ** The sqlite_stat2 entries for an index that have sampleno between 0 and 9 |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 78 | ** inclusive are samples of the left-most key value in the index taken at |
| 79 | ** evenly spaced points along the index. Let the number of samples be S |
| 80 | ** (10 in the standard build) and let C be the number of rows in the index. |
| 81 | ** Then the sampled rows are given by: |
| 82 | ** |
| 83 | ** rownumber = (i*C*2 + C)/(S*2) |
| 84 | ** |
| 85 | ** For i between 0 and S-1. Conceptually, the index space is divided into |
| 86 | ** S uniform buckets and the samples are the middle row from each bucket. |
| 87 | ** |
| 88 | ** The format for sqlite_stat2 is recorded here for legacy reference. This |
| 89 | ** version of SQLite does not support sqlite_stat2. It neither reads nor |
| 90 | ** writes the sqlite_stat2 table. This version of SQLite only supports |
| 91 | ** sqlite_stat3. |
| 92 | ** |
| 93 | ** Format for sqlite_stat3: |
| 94 | ** |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 95 | ** The sqlite_stat3 format is a subset of sqlite_stat4. Hence, the |
| 96 | ** sqlite_stat4 format will be described first. Further information |
| 97 | ** about sqlite_stat3 follows the sqlite_stat4 description. |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 98 | ** |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 99 | ** Format for sqlite_stat4: |
| 100 | ** |
| 101 | ** As with sqlite_stat2, the sqlite_stat4 table contains histogram data |
| 102 | ** to aid the query planner in choosing good indices based on the values |
| 103 | ** that indexed columns are compared against in the WHERE clauses of |
| 104 | ** queries. |
| 105 | ** |
| 106 | ** The sqlite_stat4 table contains multiple entries for each index. |
drh | d3ed734 | 2011-09-21 00:09:41 +0000 | [diff] [blame] | 107 | ** The idx column names the index and the tbl column is the table of the |
| 108 | ** index. If the idx and tbl columns are the same, then the sample is |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 109 | ** of the INTEGER PRIMARY KEY. The sample column is a blob which is the |
drh | 0ae4f14 | 2013-10-14 14:21:59 +0000 | [diff] [blame] | 110 | ** binary encoding of a key from the index. The nEq column is a |
| 111 | ** list of integers. The first integer is the approximate number |
| 112 | ** of entries in the index whose left-most column exactly matches |
| 113 | ** the left-most column of the sample. The second integer in nEq |
| 114 | ** is the approximate number of entries in the index where the |
| 115 | ** first two columns match the first two columns of the sample. |
dan | 84d4fcc | 2013-08-09 19:04:07 +0000 | [diff] [blame] | 116 | ** And so forth. nLt is another list of integers that show the approximate |
| 117 | ** number of entries that are strictly less than the sample. The first |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 118 | ** integer in nLt contains the number of entries in the index where the |
| 119 | ** left-most column is less than the left-most column of the sample. |
| 120 | ** The K-th integer in the nLt entry is the number of index entries |
| 121 | ** where the first K columns are less than the first K columns of the |
| 122 | ** sample. The nDLt column is like nLt except that it contains the |
| 123 | ** number of distinct entries in the index that are less than the |
| 124 | ** sample. |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 125 | ** |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 126 | ** There can be an arbitrary number of sqlite_stat4 entries per index. |
dan | 84d4fcc | 2013-08-09 19:04:07 +0000 | [diff] [blame] | 127 | ** The ANALYZE command will typically generate sqlite_stat4 tables |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 128 | ** that contain between 10 and 40 samples which are distributed across |
| 129 | ** the key space, though not uniformly, and which include samples with |
drh | c8af850 | 2013-08-09 14:07:55 +0000 | [diff] [blame] | 130 | ** large nEq values. |
| 131 | ** |
| 132 | ** Format for sqlite_stat3 redux: |
| 133 | ** |
| 134 | ** The sqlite_stat3 table is like sqlite_stat4 except that it only |
| 135 | ** looks at the left-most column of the index. The sqlite_stat3.sample |
| 136 | ** column contains the actual value of the left-most column instead |
| 137 | ** of a blob encoding of the complete index key as is found in |
| 138 | ** sqlite_stat4.sample. The nEq, nLt, and nDLt entries of sqlite_stat3 |
| 139 | ** all contain just a single integer which is the same as the first |
| 140 | ** integer in the equivalent columns in sqlite_stat4. |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 141 | */ |
| 142 | #ifndef SQLITE_OMIT_ANALYZE |
| 143 | #include "sqliteInt.h" |
| 144 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 145 | #if defined(SQLITE_ENABLE_STAT4) |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 146 | # define IsStat4 1 |
| 147 | # define IsStat3 0 |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 148 | #elif defined(SQLITE_ENABLE_STAT3) |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 149 | # define IsStat4 0 |
| 150 | # define IsStat3 1 |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 151 | #else |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 152 | # define IsStat4 0 |
| 153 | # define IsStat3 0 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 154 | # undef SQLITE_STAT4_SAMPLES |
| 155 | # define SQLITE_STAT4_SAMPLES 1 |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 156 | #endif |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 157 | #define IsStat34 (IsStat3+IsStat4) /* 1 for STAT3 or STAT4. 0 otherwise */ |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 158 | |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 159 | /* |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 160 | ** This routine generates code that opens the sqlite_statN tables. |
| 161 | ** The sqlite_stat1 table is always relevant. sqlite_stat2 is now |
| 162 | ** obsolete. sqlite_stat3 and sqlite_stat4 are only opened when |
| 163 | ** appropriate compile-time options are provided. |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 164 | ** |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 165 | ** If the sqlite_statN tables do not previously exist, it is created. |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 166 | ** |
| 167 | ** Argument zWhere may be a pointer to a buffer containing a table name, |
| 168 | ** or it may be a NULL pointer. If it is not NULL, then all entries in |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 169 | ** the sqlite_statN tables associated with the named table are deleted. |
| 170 | ** If zWhere==0, then code is generated to delete all stat table entries. |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 171 | */ |
| 172 | static void openStatTable( |
| 173 | Parse *pParse, /* Parsing context */ |
| 174 | int iDb, /* The database we are looking in */ |
| 175 | int iStatCur, /* Open the sqlite_stat1 table on this cursor */ |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 176 | const char *zWhere, /* Delete entries for this table or index */ |
| 177 | const char *zWhereType /* Either "tbl" or "idx" */ |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 178 | ){ |
dan | 558814f | 2010-06-02 05:53:53 +0000 | [diff] [blame] | 179 | static const struct { |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 180 | const char *zName; |
| 181 | const char *zCols; |
| 182 | } aTable[] = { |
| 183 | { "sqlite_stat1", "tbl,idx,stat" }, |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 184 | #if defined(SQLITE_ENABLE_STAT4) |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 185 | { "sqlite_stat4", "tbl,idx,neq,nlt,ndlt,sample" }, |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 186 | { "sqlite_stat3", 0 }, |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 187 | #elif defined(SQLITE_ENABLE_STAT3) |
| 188 | { "sqlite_stat3", "tbl,idx,neq,nlt,ndlt,sample" }, |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 189 | { "sqlite_stat4", 0 }, |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 190 | #else |
| 191 | { "sqlite_stat3", 0 }, |
| 192 | { "sqlite_stat4", 0 }, |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 193 | #endif |
| 194 | }; |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 195 | int i; |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 196 | sqlite3 *db = pParse->db; |
| 197 | Db *pDb; |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 198 | Vdbe *v = sqlite3GetVdbe(pParse); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 199 | int aRoot[ArraySize(aTable)]; |
| 200 | u8 aCreateTbl[ArraySize(aTable)]; |
| 201 | |
drh | cf1be45 | 2007-05-12 12:08:51 +0000 | [diff] [blame] | 202 | if( v==0 ) return; |
drh | 1fee73e | 2007-08-29 04:00:57 +0000 | [diff] [blame] | 203 | assert( sqlite3BtreeHoldsAllMutexes(db) ); |
| 204 | assert( sqlite3VdbeDb(v)==db ); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 205 | pDb = &db->aDb[iDb]; |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 206 | |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 207 | /* Create new statistic tables if they do not exist, or clear them |
| 208 | ** if they do already exist. |
| 209 | */ |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 210 | for(i=0; i<ArraySize(aTable); i++){ |
| 211 | const char *zTab = aTable[i].zName; |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 212 | Table *pStat; |
drh | 69c3382 | 2016-08-18 14:33:11 +0000 | [diff] [blame] | 213 | if( (pStat = sqlite3FindTable(db, zTab, pDb->zDbSName))==0 ){ |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 214 | if( aTable[i].zCols ){ |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 215 | /* The sqlite_statN table does not exist. Create it. Note that a |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 216 | ** side-effect of the CREATE TABLE statement is to leave the rootpage |
| 217 | ** of the new table in register pParse->regRoot. This is important |
| 218 | ** because the OpenWrite opcode below will be needing it. */ |
| 219 | sqlite3NestedParse(pParse, |
drh | 69c3382 | 2016-08-18 14:33:11 +0000 | [diff] [blame] | 220 | "CREATE TABLE %Q.%s(%s)", pDb->zDbSName, zTab, aTable[i].zCols |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 221 | ); |
| 222 | aRoot[i] = pParse->regRoot; |
| 223 | aCreateTbl[i] = OPFLAG_P2ISREG; |
| 224 | } |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 225 | }else{ |
| 226 | /* The table already exists. If zWhere is not NULL, delete all entries |
| 227 | ** associated with the table zWhere. If zWhere is NULL, delete the |
| 228 | ** entire contents of the table. */ |
| 229 | aRoot[i] = pStat->tnum; |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 230 | aCreateTbl[i] = 0; |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 231 | sqlite3TableLock(pParse, iDb, aRoot[i], 1, zTab); |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 232 | if( zWhere ){ |
| 233 | sqlite3NestedParse(pParse, |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 234 | "DELETE FROM %Q.%s WHERE %s=%Q", |
drh | 69c3382 | 2016-08-18 14:33:11 +0000 | [diff] [blame] | 235 | pDb->zDbSName, zTab, zWhereType, zWhere |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 236 | ); |
dan | 614efe2 | 2018-01-12 16:44:29 +0000 | [diff] [blame] | 237 | #ifdef SQLITE_ENABLE_PREUPDATE_HOOK |
| 238 | }else if( db->xPreUpdateCallback ){ |
| 239 | sqlite3NestedParse(pParse, "DELETE FROM %Q.%s", pDb->zDbSName, zTab); |
| 240 | #endif |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 241 | }else{ |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 242 | /* The sqlite_stat[134] table already exists. Delete all rows. */ |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 243 | sqlite3VdbeAddOp2(v, OP_Clear, aRoot[i], iDb); |
| 244 | } |
| 245 | } |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 246 | } |
| 247 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 248 | /* Open the sqlite_stat[134] tables for writing. */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 249 | for(i=0; aTable[i].zCols; i++){ |
| 250 | assert( i<ArraySize(aTable) ); |
drh | 261c02d | 2013-10-25 14:46:15 +0000 | [diff] [blame] | 251 | sqlite3VdbeAddOp4Int(v, OP_OpenWrite, iStatCur+i, aRoot[i], iDb, 3); |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 252 | sqlite3VdbeChangeP5(v, aCreateTbl[i]); |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 253 | VdbeComment((v, aTable[i].zName)); |
danielk1977 | c00da10 | 2006-01-07 13:21:04 +0000 | [diff] [blame] | 254 | } |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 255 | } |
| 256 | |
| 257 | /* |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 258 | ** Recommended number of samples for sqlite_stat4 |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 259 | */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 260 | #ifndef SQLITE_STAT4_SAMPLES |
| 261 | # define SQLITE_STAT4_SAMPLES 24 |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 262 | #endif |
| 263 | |
| 264 | /* |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 265 | ** Three SQL functions - stat_init(), stat_push(), and stat_get() - |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 266 | ** share an instance of the following structure to hold their state |
| 267 | ** information. |
| 268 | */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 269 | typedef struct Stat4Accum Stat4Accum; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 270 | typedef struct Stat4Sample Stat4Sample; |
| 271 | struct Stat4Sample { |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 272 | tRowcnt *anEq; /* sqlite_stat4.nEq */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 273 | tRowcnt *anDLt; /* sqlite_stat4.nDLt */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 274 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 275 | tRowcnt *anLt; /* sqlite_stat4.nLt */ |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 276 | union { |
| 277 | i64 iRowid; /* Rowid in main table of the key */ |
| 278 | u8 *aRowid; /* Key for WITHOUT ROWID tables */ |
| 279 | } u; |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 280 | u32 nRowid; /* Sizeof aRowid[] */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 281 | u8 isPSample; /* True if a periodic sample */ |
| 282 | int iCol; /* If !isPSample, the reason for inclusion */ |
| 283 | u32 iHash; /* Tiebreaker hash */ |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 284 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 285 | }; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 286 | struct Stat4Accum { |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 287 | tRowcnt nRow; /* Number of rows in the entire table */ |
| 288 | tRowcnt nPSample; /* How often to do a periodic sample */ |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 289 | int nCol; /* Number of columns in index + pk/rowid */ |
| 290 | int nKeyCol; /* Number of index columns w/o the pk/rowid */ |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 291 | int mxSample; /* Maximum number of samples to accumulate */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 292 | Stat4Sample current; /* Current row as a Stat4Sample */ |
drh | f404c86 | 2011-08-13 15:25:10 +0000 | [diff] [blame] | 293 | u32 iPrn; /* Pseudo-random number used for sampling */ |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 294 | Stat4Sample *aBest; /* Array of nCol best samples */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 295 | int iMin; /* Index in a[] of entry with minimum score */ |
| 296 | int nSample; /* Current number of samples */ |
dan | bd1d270 | 2017-02-22 19:27:51 +0000 | [diff] [blame] | 297 | int nMaxEqZero; /* Max leading 0 in anEq[] for any a[] entry */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 298 | int iGet; /* Index of current sample accessed by stat_get() */ |
| 299 | Stat4Sample *a; /* Array of mxSample Stat4Sample objects */ |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 300 | sqlite3 *db; /* Database connection, for malloc() */ |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 301 | }; |
| 302 | |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 303 | /* Reclaim memory used by a Stat4Sample |
| 304 | */ |
| 305 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
| 306 | static void sampleClear(sqlite3 *db, Stat4Sample *p){ |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 307 | assert( db!=0 ); |
| 308 | if( p->nRowid ){ |
| 309 | sqlite3DbFree(db, p->u.aRowid); |
| 310 | p->nRowid = 0; |
| 311 | } |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 312 | } |
| 313 | #endif |
| 314 | |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 315 | /* Initialize the BLOB value of a ROWID |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 316 | */ |
| 317 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 318 | static void sampleSetRowid(sqlite3 *db, Stat4Sample *p, int n, const u8 *pData){ |
| 319 | assert( db!=0 ); |
| 320 | if( p->nRowid ) sqlite3DbFree(db, p->u.aRowid); |
drh | 575fad6 | 2016-02-05 13:38:36 +0000 | [diff] [blame] | 321 | p->u.aRowid = sqlite3DbMallocRawNN(db, n); |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 322 | if( p->u.aRowid ){ |
| 323 | p->nRowid = n; |
| 324 | memcpy(p->u.aRowid, pData, n); |
| 325 | }else{ |
| 326 | p->nRowid = 0; |
| 327 | } |
| 328 | } |
| 329 | #endif |
| 330 | |
| 331 | /* Initialize the INTEGER value of a ROWID. |
| 332 | */ |
| 333 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
| 334 | static void sampleSetRowidInt64(sqlite3 *db, Stat4Sample *p, i64 iRowid){ |
| 335 | assert( db!=0 ); |
| 336 | if( p->nRowid ) sqlite3DbFree(db, p->u.aRowid); |
| 337 | p->nRowid = 0; |
| 338 | p->u.iRowid = iRowid; |
| 339 | } |
| 340 | #endif |
| 341 | |
| 342 | |
| 343 | /* |
| 344 | ** Copy the contents of object (*pFrom) into (*pTo). |
| 345 | */ |
| 346 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
| 347 | static void sampleCopy(Stat4Accum *p, Stat4Sample *pTo, Stat4Sample *pFrom){ |
| 348 | pTo->isPSample = pFrom->isPSample; |
| 349 | pTo->iCol = pFrom->iCol; |
| 350 | pTo->iHash = pFrom->iHash; |
| 351 | memcpy(pTo->anEq, pFrom->anEq, sizeof(tRowcnt)*p->nCol); |
| 352 | memcpy(pTo->anLt, pFrom->anLt, sizeof(tRowcnt)*p->nCol); |
| 353 | memcpy(pTo->anDLt, pFrom->anDLt, sizeof(tRowcnt)*p->nCol); |
| 354 | if( pFrom->nRowid ){ |
| 355 | sampleSetRowid(p->db, pTo, pFrom->nRowid, pFrom->u.aRowid); |
| 356 | }else{ |
| 357 | sampleSetRowidInt64(p->db, pTo, pFrom->u.iRowid); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 358 | } |
| 359 | } |
| 360 | #endif |
| 361 | |
| 362 | /* |
| 363 | ** Reclaim all memory of a Stat4Accum structure. |
| 364 | */ |
| 365 | static void stat4Destructor(void *pOld){ |
| 366 | Stat4Accum *p = (Stat4Accum*)pOld; |
| 367 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
| 368 | int i; |
| 369 | for(i=0; i<p->nCol; i++) sampleClear(p->db, p->aBest+i); |
| 370 | for(i=0; i<p->mxSample; i++) sampleClear(p->db, p->a+i); |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 371 | sampleClear(p->db, &p->current); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 372 | #endif |
| 373 | sqlite3DbFree(p->db, p); |
| 374 | } |
| 375 | |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 376 | /* |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 377 | ** Implementation of the stat_init(N,K,C) SQL function. The three parameters |
| 378 | ** are: |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 379 | ** N: The number of columns in the index including the rowid/pk (note 1) |
| 380 | ** K: The number of columns in the index excluding the rowid/pk. |
| 381 | ** C: The number of rows in the index (note 2) |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 382 | ** |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 383 | ** Note 1: In the special case of the covering index that implements a |
| 384 | ** WITHOUT ROWID table, N is the number of PRIMARY KEY columns, not the |
| 385 | ** total number of columns in the table. |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 386 | ** |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 387 | ** Note 2: C is only used for STAT3 and STAT4. |
| 388 | ** |
| 389 | ** For indexes on ordinary rowid tables, N==K+1. But for indexes on |
| 390 | ** WITHOUT ROWID tables, N=K+P where P is the number of columns in the |
| 391 | ** PRIMARY KEY of the table. The covering index that implements the |
| 392 | ** original WITHOUT ROWID table as N==K as a special case. |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 393 | ** |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 394 | ** This routine allocates the Stat4Accum object in heap memory. The return |
peter.d.reid | 60ec914 | 2014-09-06 16:39:46 +0000 | [diff] [blame] | 395 | ** value is a pointer to the Stat4Accum object. The datatype of the |
drh | f8ede57 | 2014-09-01 23:06:44 +0000 | [diff] [blame] | 396 | ** return value is BLOB, but it is really just a pointer to the Stat4Accum |
| 397 | ** object. |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 398 | */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 399 | static void statInit( |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 400 | sqlite3_context *context, |
| 401 | int argc, |
| 402 | sqlite3_value **argv |
| 403 | ){ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 404 | Stat4Accum *p; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 405 | int nCol; /* Number of columns in index being sampled */ |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 406 | int nKeyCol; /* Number of key columns */ |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 407 | int nColUp; /* nCol rounded up for alignment */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 408 | int n; /* Bytes of space to allocate */ |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 409 | sqlite3 *db; /* Database connection */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 410 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 411 | int mxSample = SQLITE_STAT4_SAMPLES; |
| 412 | #endif |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 413 | |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 414 | /* Decode the three function arguments */ |
drh | 6825719 | 2011-08-16 17:06:21 +0000 | [diff] [blame] | 415 | UNUSED_PARAMETER(argc); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 416 | nCol = sqlite3_value_int(argv[0]); |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 417 | assert( nCol>0 ); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 418 | nColUp = sizeof(tRowcnt)<8 ? (nCol+1)&~1 : nCol; |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 419 | nKeyCol = sqlite3_value_int(argv[1]); |
| 420 | assert( nKeyCol<=nCol ); |
| 421 | assert( nKeyCol>0 ); |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 422 | |
| 423 | /* Allocate the space required for the Stat4Accum object */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 424 | n = sizeof(*p) |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 425 | + sizeof(tRowcnt)*nColUp /* Stat4Accum.anEq */ |
| 426 | + sizeof(tRowcnt)*nColUp /* Stat4Accum.anDLt */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 427 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 428 | + sizeof(tRowcnt)*nColUp /* Stat4Accum.anLt */ |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 429 | + sizeof(Stat4Sample)*(nCol+mxSample) /* Stat4Accum.aBest[], a[] */ |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 430 | + sizeof(tRowcnt)*3*nColUp*(nCol+mxSample) |
| 431 | #endif |
| 432 | ; |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 433 | db = sqlite3_context_db_handle(context); |
| 434 | p = sqlite3DbMallocZero(db, n); |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 435 | if( p==0 ){ |
| 436 | sqlite3_result_error_nomem(context); |
| 437 | return; |
| 438 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 439 | |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 440 | p->db = db; |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 441 | p->nRow = 0; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 442 | p->nCol = nCol; |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 443 | p->nKeyCol = nKeyCol; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 444 | p->current.anDLt = (tRowcnt*)&p[1]; |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 445 | p->current.anEq = &p->current.anDLt[nColUp]; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 446 | |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 447 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 448 | { |
| 449 | u8 *pSpace; /* Allocated space not yet assigned */ |
| 450 | int i; /* Used to iterate through p->aSample[] */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 451 | |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 452 | p->iGet = -1; |
| 453 | p->mxSample = mxSample; |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 454 | p->nPSample = (tRowcnt)(sqlite3_value_int64(argv[2])/(mxSample/3+1) + 1); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 455 | p->current.anLt = &p->current.anEq[nColUp]; |
drh | 4081d5d | 2015-01-01 23:02:01 +0000 | [diff] [blame] | 456 | p->iPrn = 0x689e962d*(u32)nCol ^ 0xd0944565*(u32)sqlite3_value_int(argv[2]); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 457 | |
| 458 | /* Set up the Stat4Accum.a[] and aBest[] arrays */ |
| 459 | p->a = (struct Stat4Sample*)&p->current.anLt[nColUp]; |
| 460 | p->aBest = &p->a[mxSample]; |
| 461 | pSpace = (u8*)(&p->a[mxSample+nCol]); |
| 462 | for(i=0; i<(mxSample+nCol); i++){ |
| 463 | p->a[i].anEq = (tRowcnt *)pSpace; pSpace += (sizeof(tRowcnt) * nColUp); |
| 464 | p->a[i].anLt = (tRowcnt *)pSpace; pSpace += (sizeof(tRowcnt) * nColUp); |
| 465 | p->a[i].anDLt = (tRowcnt *)pSpace; pSpace += (sizeof(tRowcnt) * nColUp); |
| 466 | } |
| 467 | assert( (pSpace - (u8*)p)==n ); |
| 468 | |
| 469 | for(i=0; i<nCol; i++){ |
| 470 | p->aBest[i].iCol = i; |
| 471 | } |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 472 | } |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 473 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 474 | |
drh | f8ede57 | 2014-09-01 23:06:44 +0000 | [diff] [blame] | 475 | /* Return a pointer to the allocated object to the caller. Note that |
| 476 | ** only the pointer (the 2nd parameter) matters. The size of the object |
| 477 | ** (given by the 3rd parameter) is never used and can be any positive |
| 478 | ** value. */ |
drh | 975e076 | 2014-09-01 22:34:54 +0000 | [diff] [blame] | 479 | sqlite3_result_blob(context, p, sizeof(*p), stat4Destructor); |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 480 | } |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 481 | static const FuncDef statInitFuncdef = { |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 482 | 2+IsStat34, /* nArg */ |
drh | d36e104 | 2013-09-06 13:10:12 +0000 | [diff] [blame] | 483 | SQLITE_UTF8, /* funcFlags */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 484 | 0, /* pUserData */ |
| 485 | 0, /* pNext */ |
drh | 2d80151 | 2016-01-14 22:19:58 +0000 | [diff] [blame] | 486 | statInit, /* xSFunc */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 487 | 0, /* xFinalize */ |
dan | 980cf25 | 2018-06-10 07:42:35 +0000 | [diff] [blame] | 488 | 0, 0, /* xValue, xInverse */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 489 | "stat_init", /* zName */ |
drh | 80738d9 | 2016-02-15 00:34:16 +0000 | [diff] [blame] | 490 | {0} |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 491 | }; |
| 492 | |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 493 | #ifdef SQLITE_ENABLE_STAT4 |
| 494 | /* |
| 495 | ** pNew and pOld are both candidate non-periodic samples selected for |
| 496 | ** the same column (pNew->iCol==pOld->iCol). Ignoring this column and |
| 497 | ** considering only any trailing columns and the sample hash value, this |
| 498 | ** function returns true if sample pNew is to be preferred over pOld. |
| 499 | ** In other words, if we assume that the cardinalities of the selected |
| 500 | ** column for pNew and pOld are equal, is pNew to be preferred over pOld. |
| 501 | ** |
| 502 | ** This function assumes that for each argument sample, the contents of |
| 503 | ** the anEq[] array from pSample->anEq[pSample->iCol+1] onwards are valid. |
| 504 | */ |
| 505 | static int sampleIsBetterPost( |
| 506 | Stat4Accum *pAccum, |
| 507 | Stat4Sample *pNew, |
| 508 | Stat4Sample *pOld |
| 509 | ){ |
| 510 | int nCol = pAccum->nCol; |
| 511 | int i; |
| 512 | assert( pNew->iCol==pOld->iCol ); |
| 513 | for(i=pNew->iCol+1; i<nCol; i++){ |
| 514 | if( pNew->anEq[i]>pOld->anEq[i] ) return 1; |
| 515 | if( pNew->anEq[i]<pOld->anEq[i] ) return 0; |
| 516 | } |
| 517 | if( pNew->iHash>pOld->iHash ) return 1; |
| 518 | return 0; |
| 519 | } |
| 520 | #endif |
| 521 | |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 522 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 523 | /* |
| 524 | ** Return true if pNew is to be preferred over pOld. |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 525 | ** |
| 526 | ** This function assumes that for each argument sample, the contents of |
| 527 | ** the anEq[] array from pSample->anEq[pSample->iCol] onwards are valid. |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 528 | */ |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 529 | static int sampleIsBetter( |
| 530 | Stat4Accum *pAccum, |
| 531 | Stat4Sample *pNew, |
| 532 | Stat4Sample *pOld |
| 533 | ){ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 534 | tRowcnt nEqNew = pNew->anEq[pNew->iCol]; |
| 535 | tRowcnt nEqOld = pOld->anEq[pOld->iCol]; |
| 536 | |
| 537 | assert( pOld->isPSample==0 && pNew->isPSample==0 ); |
| 538 | assert( IsStat4 || (pNew->iCol==0 && pOld->iCol==0) ); |
| 539 | |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 540 | if( (nEqNew>nEqOld) ) return 1; |
| 541 | #ifdef SQLITE_ENABLE_STAT4 |
| 542 | if( nEqNew==nEqOld ){ |
| 543 | if( pNew->iCol<pOld->iCol ) return 1; |
| 544 | return (pNew->iCol==pOld->iCol && sampleIsBetterPost(pAccum, pNew, pOld)); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 545 | } |
| 546 | return 0; |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 547 | #else |
| 548 | return (nEqNew==nEqOld && pNew->iHash>pOld->iHash); |
| 549 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 550 | } |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 551 | |
| 552 | /* |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 553 | ** Copy the contents of sample *pNew into the p->a[] array. If necessary, |
| 554 | ** remove the least desirable sample from p->a[] to make room. |
| 555 | */ |
| 556 | static void sampleInsert(Stat4Accum *p, Stat4Sample *pNew, int nEqZero){ |
drh | d269461 | 2013-11-04 22:04:17 +0000 | [diff] [blame] | 557 | Stat4Sample *pSample = 0; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 558 | int i; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 559 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 560 | assert( IsStat4 || nEqZero==0 ); |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 561 | |
drh | 30f0704 | 2013-09-04 02:07:38 +0000 | [diff] [blame] | 562 | #ifdef SQLITE_ENABLE_STAT4 |
dan | bd1d270 | 2017-02-22 19:27:51 +0000 | [diff] [blame] | 563 | /* Stat4Accum.nMaxEqZero is set to the maximum number of leading 0 |
| 564 | ** values in the anEq[] array of any sample in Stat4Accum.a[]. In |
| 565 | ** other words, if nMaxEqZero is n, then it is guaranteed that there |
| 566 | ** are no samples with Stat4Sample.anEq[m]==0 for (m>=n). */ |
dan | bd1d270 | 2017-02-22 19:27:51 +0000 | [diff] [blame] | 567 | if( nEqZero>p->nMaxEqZero ){ |
| 568 | p->nMaxEqZero = nEqZero; |
| 569 | } |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 570 | if( pNew->isPSample==0 ){ |
dan | 1f616ad | 2013-08-15 14:39:09 +0000 | [diff] [blame] | 571 | Stat4Sample *pUpgrade = 0; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 572 | assert( pNew->anEq[pNew->iCol]>0 ); |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 573 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 574 | /* This sample is being added because the prefix that ends in column |
| 575 | ** iCol occurs many times in the table. However, if we have already |
| 576 | ** added a sample that shares this prefix, there is no need to add |
dan | 1f616ad | 2013-08-15 14:39:09 +0000 | [diff] [blame] | 577 | ** this one. Instead, upgrade the priority of the highest priority |
| 578 | ** existing sample that shares this prefix. */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 579 | for(i=p->nSample-1; i>=0; i--){ |
| 580 | Stat4Sample *pOld = &p->a[i]; |
| 581 | if( pOld->anEq[pNew->iCol]==0 ){ |
dan | 1f616ad | 2013-08-15 14:39:09 +0000 | [diff] [blame] | 582 | if( pOld->isPSample ) return; |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 583 | assert( pOld->iCol>pNew->iCol ); |
| 584 | assert( sampleIsBetter(p, pNew, pOld) ); |
| 585 | if( pUpgrade==0 || sampleIsBetter(p, pOld, pUpgrade) ){ |
dan | 1f616ad | 2013-08-15 14:39:09 +0000 | [diff] [blame] | 586 | pUpgrade = pOld; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 587 | } |
dan | 1f28ead | 2013-08-07 16:15:32 +0000 | [diff] [blame] | 588 | } |
| 589 | } |
dan | 1f616ad | 2013-08-15 14:39:09 +0000 | [diff] [blame] | 590 | if( pUpgrade ){ |
| 591 | pUpgrade->iCol = pNew->iCol; |
| 592 | pUpgrade->anEq[pUpgrade->iCol] = pNew->anEq[pUpgrade->iCol]; |
| 593 | goto find_new_min; |
| 594 | } |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 595 | } |
drh | 30f0704 | 2013-09-04 02:07:38 +0000 | [diff] [blame] | 596 | #endif |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 597 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 598 | /* If necessary, remove sample iMin to make room for the new sample. */ |
| 599 | if( p->nSample>=p->mxSample ){ |
| 600 | Stat4Sample *pMin = &p->a[p->iMin]; |
dan | c55521a | 2013-08-05 05:34:30 +0000 | [diff] [blame] | 601 | tRowcnt *anEq = pMin->anEq; |
dan | c55521a | 2013-08-05 05:34:30 +0000 | [diff] [blame] | 602 | tRowcnt *anLt = pMin->anLt; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 603 | tRowcnt *anDLt = pMin->anDLt; |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 604 | sampleClear(p->db, pMin); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 605 | memmove(pMin, &pMin[1], sizeof(p->a[0])*(p->nSample-p->iMin-1)); |
drh | f404c86 | 2011-08-13 15:25:10 +0000 | [diff] [blame] | 606 | pSample = &p->a[p->nSample-1]; |
drh | d269461 | 2013-11-04 22:04:17 +0000 | [diff] [blame] | 607 | pSample->nRowid = 0; |
dan | c55521a | 2013-08-05 05:34:30 +0000 | [diff] [blame] | 608 | pSample->anEq = anEq; |
| 609 | pSample->anDLt = anDLt; |
| 610 | pSample->anLt = anLt; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 611 | p->nSample = p->mxSample-1; |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 612 | } |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 613 | |
dan | b49d104 | 2013-09-02 18:58:11 +0000 | [diff] [blame] | 614 | /* The "rows less-than" for the rowid column must be greater than that |
| 615 | ** for the last sample in the p->a[] array. Otherwise, the samples would |
| 616 | ** be out of order. */ |
| 617 | #ifdef SQLITE_ENABLE_STAT4 |
| 618 | assert( p->nSample==0 |
| 619 | || pNew->anLt[p->nCol-1] > p->a[p->nSample-1].anLt[p->nCol-1] ); |
| 620 | #endif |
| 621 | |
| 622 | /* Insert the new sample */ |
| 623 | pSample = &p->a[p->nSample]; |
| 624 | sampleCopy(p, pSample, pNew); |
| 625 | p->nSample++; |
| 626 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 627 | /* Zero the first nEqZero entries in the anEq[] array. */ |
| 628 | memset(pSample->anEq, 0, sizeof(tRowcnt)*nEqZero); |
| 629 | |
mistachkin | c2cfb51 | 2013-09-04 04:04:08 +0000 | [diff] [blame] | 630 | #ifdef SQLITE_ENABLE_STAT4 |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 631 | find_new_min: |
mistachkin | c2cfb51 | 2013-09-04 04:04:08 +0000 | [diff] [blame] | 632 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 633 | if( p->nSample>=p->mxSample ){ |
| 634 | int iMin = -1; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 635 | for(i=0; i<p->mxSample; i++){ |
| 636 | if( p->a[i].isPSample ) continue; |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 637 | if( iMin<0 || sampleIsBetter(p, &p->a[iMin], &p->a[i]) ){ |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 638 | iMin = i; |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 639 | } |
| 640 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 641 | assert( iMin>=0 ); |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 642 | p->iMin = iMin; |
| 643 | } |
| 644 | } |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 645 | #endif /* SQLITE_ENABLE_STAT3_OR_STAT4 */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 646 | |
| 647 | /* |
| 648 | ** Field iChng of the index being scanned has changed. So at this point |
| 649 | ** p->current contains a sample that reflects the previous row of the |
| 650 | ** index. The value of anEq[iChng] and subsequent anEq[] elements are |
| 651 | ** correct at this point. |
| 652 | */ |
| 653 | static void samplePushPrevious(Stat4Accum *p, int iChng){ |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 654 | #ifdef SQLITE_ENABLE_STAT4 |
| 655 | int i; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 656 | |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 657 | /* Check if any samples from the aBest[] array should be pushed |
| 658 | ** into IndexSample.a[] at this point. */ |
| 659 | for(i=(p->nCol-2); i>=iChng; i--){ |
| 660 | Stat4Sample *pBest = &p->aBest[i]; |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 661 | pBest->anEq[i] = p->current.anEq[i]; |
| 662 | if( p->nSample<p->mxSample || sampleIsBetter(p, pBest, &p->a[p->iMin]) ){ |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 663 | sampleInsert(p, pBest, i); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 664 | } |
| 665 | } |
| 666 | |
dan | bd1d270 | 2017-02-22 19:27:51 +0000 | [diff] [blame] | 667 | /* Check that no sample contains an anEq[] entry with an index of |
| 668 | ** p->nMaxEqZero or greater set to zero. */ |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 669 | for(i=p->nSample-1; i>=0; i--){ |
| 670 | int j; |
dan | bd1d270 | 2017-02-22 19:27:51 +0000 | [diff] [blame] | 671 | for(j=p->nMaxEqZero; j<p->nCol; j++) assert( p->a[i].anEq[j]>0 ); |
| 672 | } |
| 673 | |
| 674 | /* Update the anEq[] fields of any samples already collected. */ |
| 675 | if( iChng<p->nMaxEqZero ){ |
| 676 | for(i=p->nSample-1; i>=0; i--){ |
| 677 | int j; |
| 678 | for(j=iChng; j<p->nCol; j++){ |
| 679 | if( p->a[i].anEq[j]==0 ) p->a[i].anEq[j] = p->current.anEq[j]; |
| 680 | } |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 681 | } |
dan | bd1d270 | 2017-02-22 19:27:51 +0000 | [diff] [blame] | 682 | p->nMaxEqZero = iChng; |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 683 | } |
| 684 | #endif |
| 685 | |
| 686 | #if defined(SQLITE_ENABLE_STAT3) && !defined(SQLITE_ENABLE_STAT4) |
| 687 | if( iChng==0 ){ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 688 | tRowcnt nLt = p->current.anLt[0]; |
| 689 | tRowcnt nEq = p->current.anEq[0]; |
| 690 | |
| 691 | /* Check if this is to be a periodic sample. If so, add it. */ |
| 692 | if( (nLt/p->nPSample)!=(nLt+nEq)/p->nPSample ){ |
| 693 | p->current.isPSample = 1; |
| 694 | sampleInsert(p, &p->current, 0); |
| 695 | p->current.isPSample = 0; |
| 696 | }else |
| 697 | |
| 698 | /* Or if it is a non-periodic sample. Add it in this case too. */ |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 699 | if( p->nSample<p->mxSample |
| 700 | || sampleIsBetter(p, &p->current, &p->a[p->iMin]) |
| 701 | ){ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 702 | sampleInsert(p, &p->current, 0); |
| 703 | } |
| 704 | } |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 705 | #endif |
drh | 4f99189 | 2013-10-11 15:05:05 +0000 | [diff] [blame] | 706 | |
| 707 | #ifndef SQLITE_ENABLE_STAT3_OR_STAT4 |
| 708 | UNUSED_PARAMETER( p ); |
| 709 | UNUSED_PARAMETER( iChng ); |
| 710 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 711 | } |
| 712 | |
| 713 | /* |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 714 | ** Implementation of the stat_push SQL function: stat_push(P,C,R) |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 715 | ** Arguments: |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 716 | ** |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 717 | ** P Pointer to the Stat4Accum object created by stat_init() |
| 718 | ** C Index of left-most column to differ from previous row |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 719 | ** R Rowid for the current row. Might be a key record for |
| 720 | ** WITHOUT ROWID tables. |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 721 | ** |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 722 | ** This SQL function always returns NULL. It's purpose it to accumulate |
| 723 | ** statistical data and/or samples in the Stat4Accum object about the |
| 724 | ** index being analyzed. The stat_get() SQL function will later be used to |
| 725 | ** extract relevant information for constructing the sqlite_statN tables. |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 726 | ** |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 727 | ** The R parameter is only used for STAT3 and STAT4 |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 728 | */ |
| 729 | static void statPush( |
| 730 | sqlite3_context *context, |
| 731 | int argc, |
| 732 | sqlite3_value **argv |
| 733 | ){ |
| 734 | int i; |
| 735 | |
| 736 | /* The three function arguments */ |
| 737 | Stat4Accum *p = (Stat4Accum*)sqlite3_value_blob(argv[0]); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 738 | int iChng = sqlite3_value_int(argv[1]); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 739 | |
drh | 4f99189 | 2013-10-11 15:05:05 +0000 | [diff] [blame] | 740 | UNUSED_PARAMETER( argc ); |
| 741 | UNUSED_PARAMETER( context ); |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 742 | assert( p->nCol>0 ); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 743 | assert( iChng<p->nCol ); |
| 744 | |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 745 | if( p->nRow==0 ){ |
dan | b49d104 | 2013-09-02 18:58:11 +0000 | [diff] [blame] | 746 | /* This is the first call to this function. Do initialization. */ |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 747 | for(i=0; i<p->nCol; i++) p->current.anEq[i] = 1; |
| 748 | }else{ |
| 749 | /* Second and subsequent calls get processed here */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 750 | samplePushPrevious(p, iChng); |
| 751 | |
| 752 | /* Update anDLt[], anLt[] and anEq[] to reflect the values that apply |
| 753 | ** to the current row of the index. */ |
| 754 | for(i=0; i<iChng; i++){ |
| 755 | p->current.anEq[i]++; |
| 756 | } |
| 757 | for(i=iChng; i<p->nCol; i++){ |
| 758 | p->current.anDLt[i]++; |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 759 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 760 | p->current.anLt[i] += p->current.anEq[i]; |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 761 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 762 | p->current.anEq[i] = 1; |
| 763 | } |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 764 | } |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 765 | p->nRow++; |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 766 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 767 | if( sqlite3_value_type(argv[2])==SQLITE_INTEGER ){ |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 768 | sampleSetRowidInt64(p->db, &p->current, sqlite3_value_int64(argv[2])); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 769 | }else{ |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 770 | sampleSetRowid(p->db, &p->current, sqlite3_value_bytes(argv[2]), |
| 771 | sqlite3_value_blob(argv[2])); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 772 | } |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 773 | p->current.iHash = p->iPrn = p->iPrn*1103515245 + 12345; |
| 774 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 775 | |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 776 | #ifdef SQLITE_ENABLE_STAT4 |
| 777 | { |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 778 | tRowcnt nLt = p->current.anLt[p->nCol-1]; |
| 779 | |
| 780 | /* Check if this is to be a periodic sample. If so, add it. */ |
| 781 | if( (nLt/p->nPSample)!=(nLt+1)/p->nPSample ){ |
| 782 | p->current.isPSample = 1; |
| 783 | p->current.iCol = 0; |
| 784 | sampleInsert(p, &p->current, p->nCol-1); |
| 785 | p->current.isPSample = 0; |
| 786 | } |
| 787 | |
| 788 | /* Update the aBest[] array. */ |
| 789 | for(i=0; i<(p->nCol-1); i++){ |
| 790 | p->current.iCol = i; |
dan | b13af4c | 2013-09-03 14:43:12 +0000 | [diff] [blame] | 791 | if( i>=iChng || sampleIsBetterPost(p, &p->current, &p->aBest[i]) ){ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 792 | sampleCopy(p, &p->aBest[i], &p->current); |
| 793 | } |
| 794 | } |
| 795 | } |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 796 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 797 | } |
| 798 | static const FuncDef statPushFuncdef = { |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 799 | 2+IsStat34, /* nArg */ |
drh | d36e104 | 2013-09-06 13:10:12 +0000 | [diff] [blame] | 800 | SQLITE_UTF8, /* funcFlags */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 801 | 0, /* pUserData */ |
| 802 | 0, /* pNext */ |
drh | 2d80151 | 2016-01-14 22:19:58 +0000 | [diff] [blame] | 803 | statPush, /* xSFunc */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 804 | 0, /* xFinalize */ |
dan | 980cf25 | 2018-06-10 07:42:35 +0000 | [diff] [blame] | 805 | 0, 0, /* xValue, xInverse */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 806 | "stat_push", /* zName */ |
drh | 80738d9 | 2016-02-15 00:34:16 +0000 | [diff] [blame] | 807 | {0} |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 808 | }; |
| 809 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 810 | #define STAT_GET_STAT1 0 /* "stat" column of stat1 table */ |
| 811 | #define STAT_GET_ROWID 1 /* "rowid" column of stat[34] entry */ |
| 812 | #define STAT_GET_NEQ 2 /* "neq" column of stat[34] entry */ |
| 813 | #define STAT_GET_NLT 3 /* "nlt" column of stat[34] entry */ |
| 814 | #define STAT_GET_NDLT 4 /* "ndlt" column of stat[34] entry */ |
| 815 | |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 816 | /* |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 817 | ** Implementation of the stat_get(P,J) SQL function. This routine is |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 818 | ** used to query statistical information that has been gathered into |
| 819 | ** the Stat4Accum object by prior calls to stat_push(). The P parameter |
drh | f8ede57 | 2014-09-01 23:06:44 +0000 | [diff] [blame] | 820 | ** has type BLOB but it is really just a pointer to the Stat4Accum object. |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 821 | ** The content to returned is determined by the parameter J |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 822 | ** which is one of the STAT_GET_xxxx values defined above. |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 823 | ** |
drh | 35497fc | 2017-02-01 01:34:15 +0000 | [diff] [blame] | 824 | ** The stat_get(P,J) function is not available to generic SQL. It is |
| 825 | ** inserted as part of a manually constructed bytecode program. (See |
| 826 | ** the callStatGet() routine below.) It is guaranteed that the P |
| 827 | ** parameter will always be a poiner to a Stat4Accum object, never a |
| 828 | ** NULL. |
| 829 | ** |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 830 | ** If neither STAT3 nor STAT4 are enabled, then J is always |
| 831 | ** STAT_GET_STAT1 and is hence omitted and this routine becomes |
| 832 | ** a one-parameter function, stat_get(P), that always returns the |
| 833 | ** stat1 table entry information. |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 834 | */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 835 | static void statGet( |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 836 | sqlite3_context *context, |
| 837 | int argc, |
| 838 | sqlite3_value **argv |
| 839 | ){ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 840 | Stat4Accum *p = (Stat4Accum*)sqlite3_value_blob(argv[0]); |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 841 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 842 | /* STAT3 and STAT4 have a parameter on this routine. */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 843 | int eCall = sqlite3_value_int(argv[1]); |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 844 | assert( argc==2 ); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 845 | assert( eCall==STAT_GET_STAT1 || eCall==STAT_GET_NEQ |
| 846 | || eCall==STAT_GET_ROWID || eCall==STAT_GET_NLT |
| 847 | || eCall==STAT_GET_NDLT |
| 848 | ); |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 849 | if( eCall==STAT_GET_STAT1 ) |
| 850 | #else |
| 851 | assert( argc==1 ); |
| 852 | #endif |
| 853 | { |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 854 | /* Return the value to store in the "stat" column of the sqlite_stat1 |
| 855 | ** table for this index. |
| 856 | ** |
| 857 | ** The value is a string composed of a list of integers describing |
| 858 | ** the index. The first integer in the list is the total number of |
| 859 | ** entries in the index. There is one additional integer in the list |
| 860 | ** for each indexed column. This additional integer is an estimate of |
| 861 | ** the number of rows matched by a stabbing query on the index using |
| 862 | ** a key with the corresponding number of fields. In other words, |
| 863 | ** if the index is on columns (a,b) and the sqlite_stat1 value is |
| 864 | ** "100 10 2", then SQLite estimates that: |
| 865 | ** |
| 866 | ** * the index contains 100 rows, |
| 867 | ** * "WHERE a=?" matches 10 rows, and |
| 868 | ** * "WHERE a=? AND b=?" matches 2 rows. |
| 869 | ** |
| 870 | ** If D is the count of distinct values and K is the total number of |
| 871 | ** rows, then each estimate is computed as: |
| 872 | ** |
| 873 | ** I = (K+D-1)/D |
| 874 | */ |
| 875 | char *z; |
| 876 | int i; |
| 877 | |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 878 | char *zRet = sqlite3MallocZero( (p->nKeyCol+1)*25 ); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 879 | if( zRet==0 ){ |
| 880 | sqlite3_result_error_nomem(context); |
| 881 | return; |
| 882 | } |
| 883 | |
drh | 26586e7 | 2013-10-09 19:07:22 +0000 | [diff] [blame] | 884 | sqlite3_snprintf(24, zRet, "%llu", (u64)p->nRow); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 885 | z = zRet + sqlite3Strlen30(zRet); |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 886 | for(i=0; i<p->nKeyCol; i++){ |
drh | 26586e7 | 2013-10-09 19:07:22 +0000 | [diff] [blame] | 887 | u64 nDistinct = p->current.anDLt[i] + 1; |
| 888 | u64 iVal = (p->nRow + nDistinct - 1) / nDistinct; |
| 889 | sqlite3_snprintf(24, z, " %llu", iVal); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 890 | z += sqlite3Strlen30(z); |
| 891 | assert( p->current.anEq[i] ); |
| 892 | } |
| 893 | assert( z[0]=='\0' && z>zRet ); |
| 894 | |
| 895 | sqlite3_result_text(context, zRet, -1, sqlite3_free); |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 896 | } |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 897 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 898 | else if( eCall==STAT_GET_ROWID ){ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 899 | if( p->iGet<0 ){ |
| 900 | samplePushPrevious(p, 0); |
| 901 | p->iGet = 0; |
| 902 | } |
| 903 | if( p->iGet<p->nSample ){ |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 904 | Stat4Sample *pS = p->a + p->iGet; |
| 905 | if( pS->nRowid==0 ){ |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 906 | sqlite3_result_int64(context, pS->u.iRowid); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 907 | }else{ |
drh | da475b8 | 2013-11-04 21:44:54 +0000 | [diff] [blame] | 908 | sqlite3_result_blob(context, pS->u.aRowid, pS->nRowid, |
| 909 | SQLITE_TRANSIENT); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 910 | } |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 911 | } |
| 912 | }else{ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 913 | tRowcnt *aCnt = 0; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 914 | |
| 915 | assert( p->iGet<p->nSample ); |
| 916 | switch( eCall ){ |
| 917 | case STAT_GET_NEQ: aCnt = p->a[p->iGet].anEq; break; |
| 918 | case STAT_GET_NLT: aCnt = p->a[p->iGet].anLt; break; |
| 919 | default: { |
| 920 | aCnt = p->a[p->iGet].anDLt; |
| 921 | p->iGet++; |
| 922 | break; |
| 923 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 924 | } |
| 925 | |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 926 | if( IsStat3 ){ |
| 927 | sqlite3_result_int64(context, (i64)aCnt[0]); |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 928 | }else{ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 929 | char *zRet = sqlite3MallocZero(p->nCol * 25); |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 930 | if( zRet==0 ){ |
| 931 | sqlite3_result_error_nomem(context); |
| 932 | }else{ |
| 933 | int i; |
| 934 | char *z = zRet; |
| 935 | for(i=0; i<p->nCol; i++){ |
drh | 26586e7 | 2013-10-09 19:07:22 +0000 | [diff] [blame] | 936 | sqlite3_snprintf(24, z, "%llu ", (u64)aCnt[i]); |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 937 | z += sqlite3Strlen30(z); |
| 938 | } |
| 939 | assert( z[0]=='\0' && z>zRet ); |
| 940 | z[-1] = '\0'; |
| 941 | sqlite3_result_text(context, zRet, -1, sqlite3_free); |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 942 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 943 | } |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 944 | } |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 945 | #endif /* SQLITE_ENABLE_STAT3_OR_STAT4 */ |
drh | 4f99189 | 2013-10-11 15:05:05 +0000 | [diff] [blame] | 946 | #ifndef SQLITE_DEBUG |
| 947 | UNUSED_PARAMETER( argc ); |
| 948 | #endif |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 949 | } |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 950 | static const FuncDef statGetFuncdef = { |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 951 | 1+IsStat34, /* nArg */ |
drh | d36e104 | 2013-09-06 13:10:12 +0000 | [diff] [blame] | 952 | SQLITE_UTF8, /* funcFlags */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 953 | 0, /* pUserData */ |
| 954 | 0, /* pNext */ |
drh | 2d80151 | 2016-01-14 22:19:58 +0000 | [diff] [blame] | 955 | statGet, /* xSFunc */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 956 | 0, /* xFinalize */ |
dan | 980cf25 | 2018-06-10 07:42:35 +0000 | [diff] [blame] | 957 | 0, 0, /* xValue, xInverse */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 958 | "stat_get", /* zName */ |
drh | 80738d9 | 2016-02-15 00:34:16 +0000 | [diff] [blame] | 959 | {0} |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 960 | }; |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 961 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 962 | static void callStatGet(Vdbe *v, int regStat4, int iParam, int regOut){ |
| 963 | assert( regOut!=regStat4 && regOut!=regStat4+1 ); |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 964 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 965 | sqlite3VdbeAddOp2(v, OP_Integer, iParam, regStat4+1); |
drh | 4f99189 | 2013-10-11 15:05:05 +0000 | [diff] [blame] | 966 | #elif SQLITE_DEBUG |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 967 | assert( iParam==STAT_GET_STAT1 ); |
drh | 4f99189 | 2013-10-11 15:05:05 +0000 | [diff] [blame] | 968 | #else |
| 969 | UNUSED_PARAMETER( iParam ); |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 970 | #endif |
drh | 9b34abe | 2016-01-16 15:12:35 +0000 | [diff] [blame] | 971 | sqlite3VdbeAddOp4(v, OP_Function0, 0, regStat4, regOut, |
| 972 | (char*)&statGetFuncdef, P4_FUNCDEF); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 973 | sqlite3VdbeChangeP5(v, 1 + IsStat34); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 974 | } |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 975 | |
| 976 | /* |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 977 | ** Generate code to do an analysis of all indices associated with |
| 978 | ** a single table. |
| 979 | */ |
| 980 | static void analyzeOneTable( |
| 981 | Parse *pParse, /* Parser context */ |
| 982 | Table *pTab, /* Table whose indices are to be analyzed */ |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 983 | Index *pOnlyIdx, /* If not NULL, only analyze this one index */ |
drh | dfe88ec | 2008-11-03 20:55:06 +0000 | [diff] [blame] | 984 | int iStatCur, /* Index of VdbeCursor that writes the sqlite_stat1 table */ |
dan | c612970 | 2013-08-05 19:04:07 +0000 | [diff] [blame] | 985 | int iMem, /* Available memory locations begin here */ |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 986 | int iTab /* Next available cursor */ |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 987 | ){ |
dan | 0f9a34e | 2009-08-19 16:34:31 +0000 | [diff] [blame] | 988 | sqlite3 *db = pParse->db; /* Database handle */ |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 989 | Index *pIdx; /* An index to being analyzed */ |
| 990 | int iIdxCur; /* Cursor open on index being analyzed */ |
dan | c612970 | 2013-08-05 19:04:07 +0000 | [diff] [blame] | 991 | int iTabCur; /* Table cursor */ |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 992 | Vdbe *v; /* The virtual machine being built up */ |
| 993 | int i; /* Loop counter */ |
drh | f6cf1ff | 2011-03-30 14:54:05 +0000 | [diff] [blame] | 994 | int jZeroRows = -1; /* Jump from here if number of rows is zero */ |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 995 | int iDb; /* Index of database containing pTab */ |
drh | 721dfcf | 2013-08-01 04:39:17 +0000 | [diff] [blame] | 996 | u8 needTableCnt = 1; /* True to count the table */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 997 | int regNewRowid = iMem++; /* Rowid for the inserted record */ |
| 998 | int regStat4 = iMem++; /* Register to hold Stat4Accum object */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 999 | int regChng = iMem++; /* Index of changed index field */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1000 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1001 | int regRowid = iMem++; /* Rowid argument passed to stat_push() */ |
| 1002 | #endif |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1003 | int regTemp = iMem++; /* Temporary use register */ |
dan | e275dc3 | 2009-08-18 16:24:58 +0000 | [diff] [blame] | 1004 | int regTabname = iMem++; /* Register containing table name */ |
| 1005 | int regIdxname = iMem++; /* Register containing index name */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1006 | int regStat1 = iMem++; /* Value for the stat column of sqlite_stat1 */ |
| 1007 | int regPrev = iMem; /* MUST BE LAST (see below) */ |
dan | 614efe2 | 2018-01-12 16:44:29 +0000 | [diff] [blame] | 1008 | #ifdef SQLITE_ENABLE_PREUPDATE_HOOK |
| 1009 | Table *pStat1 = 0; |
| 1010 | #endif |
dan | 68c4dbb | 2009-08-20 09:11:06 +0000 | [diff] [blame] | 1011 | |
drh | f0459fc | 2013-08-15 16:15:00 +0000 | [diff] [blame] | 1012 | pParse->nMem = MAX(pParse->nMem, iMem); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1013 | v = sqlite3GetVdbe(pParse); |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1014 | if( v==0 || NEVER(pTab==0) ){ |
| 1015 | return; |
| 1016 | } |
drh | f39d29c | 2010-09-28 17:34:46 +0000 | [diff] [blame] | 1017 | if( pTab->tnum==0 ){ |
| 1018 | /* Do not gather statistics on views or virtual tables */ |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1019 | return; |
| 1020 | } |
drh | cedfecf | 2018-03-23 12:59:10 +0000 | [diff] [blame] | 1021 | if( sqlite3_strlike("sqlite\\_%", pTab->zName, '\\')==0 ){ |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1022 | /* Do not gather statistics on system tables */ |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1023 | return; |
| 1024 | } |
dan | 0f9a34e | 2009-08-19 16:34:31 +0000 | [diff] [blame] | 1025 | assert( sqlite3BtreeHoldsAllMutexes(db) ); |
| 1026 | iDb = sqlite3SchemaToIndex(db, pTab->pSchema); |
danielk1977 | da18423 | 2006-01-05 11:34:32 +0000 | [diff] [blame] | 1027 | assert( iDb>=0 ); |
drh | 2120608 | 2011-04-04 18:22:02 +0000 | [diff] [blame] | 1028 | assert( sqlite3SchemaMutexHeld(db, iDb, 0) ); |
drh | e6e0496 | 2005-07-23 02:17:03 +0000 | [diff] [blame] | 1029 | #ifndef SQLITE_OMIT_AUTHORIZATION |
| 1030 | if( sqlite3AuthCheck(pParse, SQLITE_ANALYZE, pTab->zName, 0, |
drh | 69c3382 | 2016-08-18 14:33:11 +0000 | [diff] [blame] | 1031 | db->aDb[iDb].zDbSName ) ){ |
drh | e6e0496 | 2005-07-23 02:17:03 +0000 | [diff] [blame] | 1032 | return; |
| 1033 | } |
| 1034 | #endif |
| 1035 | |
dan | 614efe2 | 2018-01-12 16:44:29 +0000 | [diff] [blame] | 1036 | #ifdef SQLITE_ENABLE_PREUPDATE_HOOK |
| 1037 | if( db->xPreUpdateCallback ){ |
| 1038 | pStat1 = (Table*)sqlite3DbMallocZero(db, sizeof(Table) + 13); |
| 1039 | if( pStat1==0 ) return; |
| 1040 | pStat1->zName = (char*)&pStat1[1]; |
| 1041 | memcpy(pStat1->zName, "sqlite_stat1", 13); |
| 1042 | pStat1->nCol = 3; |
| 1043 | pStat1->iPKey = -1; |
| 1044 | sqlite3VdbeAddOp4(pParse->pVdbe, OP_Noop, 0, 0, 0,(char*)pStat1,P4_DYNBLOB); |
| 1045 | } |
| 1046 | #endif |
| 1047 | |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1048 | /* Establish a read-lock on the table at the shared-cache level. |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1049 | ** Open a read-only cursor on the table. Also allocate a cursor number |
| 1050 | ** to use for scanning indexes (iIdxCur). No index cursor is opened at |
| 1051 | ** this time though. */ |
danielk1977 | c00da10 | 2006-01-07 13:21:04 +0000 | [diff] [blame] | 1052 | sqlite3TableLock(pParse, iDb, pTab->tnum, 0, pTab->zName); |
dan | c612970 | 2013-08-05 19:04:07 +0000 | [diff] [blame] | 1053 | iTabCur = iTab++; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1054 | iIdxCur = iTab++; |
dan | c612970 | 2013-08-05 19:04:07 +0000 | [diff] [blame] | 1055 | pParse->nTab = MAX(pParse->nTab, iTab); |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1056 | sqlite3OpenTable(pParse, iTabCur, iDb, pTab, OP_OpenRead); |
drh | 076e85f | 2015-09-03 13:46:12 +0000 | [diff] [blame] | 1057 | sqlite3VdbeLoadString(v, regTabname, pTab->zName); |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1058 | |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1059 | for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){ |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1060 | int nCol; /* Number of columns in pIdx. "N" */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1061 | int addrRewind; /* Address of "OP_Rewind iIdxCur" */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1062 | int addrNextRow; /* Address of "next_row:" */ |
drh | ce95d11 | 2013-11-02 19:34:38 +0000 | [diff] [blame] | 1063 | const char *zIdxName; /* Name of the index */ |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1064 | int nColTest; /* Number of columns to test for changes */ |
danielk1977 | b3bf556 | 2006-01-10 17:58:23 +0000 | [diff] [blame] | 1065 | |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 1066 | if( pOnlyIdx && pOnlyIdx!=pIdx ) continue; |
drh | 721dfcf | 2013-08-01 04:39:17 +0000 | [diff] [blame] | 1067 | if( pIdx->pPartIdxWhere==0 ) needTableCnt = 0; |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1068 | if( !HasRowid(pTab) && IsPrimaryKeyIndex(pIdx) ){ |
| 1069 | nCol = pIdx->nKeyCol; |
| 1070 | zIdxName = pTab->zName; |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1071 | nColTest = nCol - 1; |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1072 | }else{ |
| 1073 | nCol = pIdx->nColumn; |
| 1074 | zIdxName = pIdx->zName; |
drh | 6861b8a | 2014-07-23 19:37:21 +0000 | [diff] [blame] | 1075 | nColTest = pIdx->uniqNotNull ? pIdx->nKeyCol-1 : nCol-1; |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1076 | } |
dan | e275dc3 | 2009-08-18 16:24:58 +0000 | [diff] [blame] | 1077 | |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1078 | /* Populate the register containing the index name. */ |
drh | 076e85f | 2015-09-03 13:46:12 +0000 | [diff] [blame] | 1079 | sqlite3VdbeLoadString(v, regIdxname, zIdxName); |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1080 | VdbeComment((v, "Analysis for %s.%s", pTab->zName, zIdxName)); |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 1081 | |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1082 | /* |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1083 | ** Pseudo-code for loop that calls stat_push(): |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1084 | ** |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1085 | ** Rewind csr |
| 1086 | ** if eof(csr) goto end_of_scan; |
| 1087 | ** regChng = 0 |
| 1088 | ** goto chng_addr_0; |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1089 | ** |
dan | dd6e1f1 | 2013-08-10 19:08:30 +0000 | [diff] [blame] | 1090 | ** next_row: |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1091 | ** regChng = 0 |
| 1092 | ** if( idx(0) != regPrev(0) ) goto chng_addr_0 |
| 1093 | ** regChng = 1 |
| 1094 | ** if( idx(1) != regPrev(1) ) goto chng_addr_1 |
| 1095 | ** ... |
| 1096 | ** regChng = N |
| 1097 | ** goto chng_addr_N |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1098 | ** |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1099 | ** chng_addr_0: |
| 1100 | ** regPrev(0) = idx(0) |
| 1101 | ** chng_addr_1: |
| 1102 | ** regPrev(1) = idx(1) |
| 1103 | ** ... |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1104 | ** |
drh | 9d79332 | 2014-07-24 19:54:20 +0000 | [diff] [blame] | 1105 | ** endDistinctTest: |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1106 | ** regRowid = idx(rowid) |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1107 | ** stat_push(P, regChng, regRowid) |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1108 | ** Next csr |
| 1109 | ** if !eof(csr) goto next_row; |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1110 | ** |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1111 | ** end_of_scan: |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1112 | */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1113 | |
| 1114 | /* Make sure there are enough memory cells allocated to accommodate |
| 1115 | ** the regPrev array and a trailing rowid (the rowid slot is required |
| 1116 | ** when building a record to insert into the sample column of |
| 1117 | ** the sqlite_stat4 table. */ |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1118 | pParse->nMem = MAX(pParse->nMem, regPrev+nColTest); |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 1119 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1120 | /* Open a read-only cursor on the index being analyzed. */ |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1121 | assert( iDb==sqlite3SchemaToIndex(db, pIdx->pSchema) ); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1122 | sqlite3VdbeAddOp3(v, OP_OpenRead, iIdxCur, pIdx->tnum, iDb); |
drh | 2ec2fb2 | 2013-11-06 19:59:23 +0000 | [diff] [blame] | 1123 | sqlite3VdbeSetP4KeyInfo(pParse, pIdx); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1124 | VdbeComment((v, "%s", pIdx->zName)); |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1125 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1126 | /* Invoke the stat_init() function. The arguments are: |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1127 | ** |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1128 | ** (1) the number of columns in the index including the rowid |
| 1129 | ** (or for a WITHOUT ROWID table, the number of PK columns), |
| 1130 | ** (2) the number of columns in the key without the rowid/pk |
| 1131 | ** (3) the number of rows in the index, |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1132 | ** |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1133 | ** |
| 1134 | ** The third argument is only used for STAT3 and STAT4 |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1135 | */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1136 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1137 | sqlite3VdbeAddOp2(v, OP_Count, iIdxCur, regStat4+3); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1138 | #endif |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1139 | sqlite3VdbeAddOp2(v, OP_Integer, nCol, regStat4+1); |
| 1140 | sqlite3VdbeAddOp2(v, OP_Integer, pIdx->nKeyCol, regStat4+2); |
drh | 9b34abe | 2016-01-16 15:12:35 +0000 | [diff] [blame] | 1141 | sqlite3VdbeAddOp4(v, OP_Function0, 0, regStat4+1, regStat4, |
| 1142 | (char*)&statInitFuncdef, P4_FUNCDEF); |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1143 | sqlite3VdbeChangeP5(v, 2+IsStat34); |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1144 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1145 | /* Implementation of the following: |
| 1146 | ** |
| 1147 | ** Rewind csr |
| 1148 | ** if eof(csr) goto end_of_scan; |
| 1149 | ** regChng = 0 |
| 1150 | ** goto next_push_0; |
| 1151 | ** |
dan | dd6e1f1 | 2013-08-10 19:08:30 +0000 | [diff] [blame] | 1152 | */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1153 | addrRewind = sqlite3VdbeAddOp1(v, OP_Rewind, iIdxCur); |
drh | 688852a | 2014-02-17 22:40:43 +0000 | [diff] [blame] | 1154 | VdbeCoverage(v); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1155 | sqlite3VdbeAddOp2(v, OP_Integer, 0, regChng); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1156 | addrNextRow = sqlite3VdbeCurrentAddr(v); |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1157 | |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1158 | if( nColTest>0 ){ |
drh | 9d79332 | 2014-07-24 19:54:20 +0000 | [diff] [blame] | 1159 | int endDistinctTest = sqlite3VdbeMakeLabel(v); |
drh | b12879f | 2014-07-24 20:25:16 +0000 | [diff] [blame] | 1160 | int *aGotoChng; /* Array of jump instruction addresses */ |
drh | 575fad6 | 2016-02-05 13:38:36 +0000 | [diff] [blame] | 1161 | aGotoChng = sqlite3DbMallocRawNN(db, sizeof(int)*nColTest); |
drh | b12879f | 2014-07-24 20:25:16 +0000 | [diff] [blame] | 1162 | if( aGotoChng==0 ) continue; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1163 | |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1164 | /* |
| 1165 | ** next_row: |
| 1166 | ** regChng = 0 |
| 1167 | ** if( idx(0) != regPrev(0) ) goto chng_addr_0 |
| 1168 | ** regChng = 1 |
| 1169 | ** if( idx(1) != regPrev(1) ) goto chng_addr_1 |
| 1170 | ** ... |
| 1171 | ** regChng = N |
drh | 9d79332 | 2014-07-24 19:54:20 +0000 | [diff] [blame] | 1172 | ** goto endDistinctTest |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1173 | */ |
| 1174 | sqlite3VdbeAddOp0(v, OP_Goto); |
| 1175 | addrNextRow = sqlite3VdbeCurrentAddr(v); |
drh | 5f1d1d9 | 2014-07-31 22:59:04 +0000 | [diff] [blame] | 1176 | if( nColTest==1 && pIdx->nKeyCol==1 && IsUniqueIndex(pIdx) ){ |
drh | 9d79332 | 2014-07-24 19:54:20 +0000 | [diff] [blame] | 1177 | /* For a single-column UNIQUE index, once we have found a non-NULL |
| 1178 | ** row, we know that all the rest will be distinct, so skip |
| 1179 | ** subsequent distinctness tests. */ |
| 1180 | sqlite3VdbeAddOp2(v, OP_NotNull, regPrev, endDistinctTest); |
| 1181 | VdbeCoverage(v); |
| 1182 | } |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1183 | for(i=0; i<nColTest; i++){ |
| 1184 | char *pColl = (char*)sqlite3LocateCollSeq(pParse, pIdx->azColl[i]); |
| 1185 | sqlite3VdbeAddOp2(v, OP_Integer, i, regChng); |
| 1186 | sqlite3VdbeAddOp3(v, OP_Column, iIdxCur, i, regTemp); |
| 1187 | aGotoChng[i] = |
| 1188 | sqlite3VdbeAddOp4(v, OP_Ne, regTemp, 0, regPrev+i, pColl, P4_COLLSEQ); |
| 1189 | sqlite3VdbeChangeP5(v, SQLITE_NULLEQ); |
| 1190 | VdbeCoverage(v); |
| 1191 | } |
| 1192 | sqlite3VdbeAddOp2(v, OP_Integer, nColTest, regChng); |
drh | 076e85f | 2015-09-03 13:46:12 +0000 | [diff] [blame] | 1193 | sqlite3VdbeGoto(v, endDistinctTest); |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1194 | |
| 1195 | |
| 1196 | /* |
| 1197 | ** chng_addr_0: |
| 1198 | ** regPrev(0) = idx(0) |
| 1199 | ** chng_addr_1: |
| 1200 | ** regPrev(1) = idx(1) |
| 1201 | ** ... |
| 1202 | */ |
| 1203 | sqlite3VdbeJumpHere(v, addrNextRow-1); |
| 1204 | for(i=0; i<nColTest; i++){ |
| 1205 | sqlite3VdbeJumpHere(v, aGotoChng[i]); |
| 1206 | sqlite3VdbeAddOp3(v, OP_Column, iIdxCur, i, regPrev+i); |
| 1207 | } |
drh | 9d79332 | 2014-07-24 19:54:20 +0000 | [diff] [blame] | 1208 | sqlite3VdbeResolveLabel(v, endDistinctTest); |
drh | b12879f | 2014-07-24 20:25:16 +0000 | [diff] [blame] | 1209 | sqlite3DbFree(db, aGotoChng); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1210 | } |
drh | 553818a | 2014-07-23 18:36:55 +0000 | [diff] [blame] | 1211 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1212 | /* |
| 1213 | ** chng_addr_N: |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1214 | ** regRowid = idx(rowid) // STAT34 only |
| 1215 | ** stat_push(P, regChng, regRowid) // 3rd parameter STAT34 only |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1216 | ** Next csr |
| 1217 | ** if !eof(csr) goto next_row; |
| 1218 | */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1219 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1220 | assert( regRowid==(regStat4+2) ); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 1221 | if( HasRowid(pTab) ){ |
| 1222 | sqlite3VdbeAddOp2(v, OP_IdxRowid, iIdxCur, regRowid); |
| 1223 | }else{ |
| 1224 | Index *pPk = sqlite3PrimaryKeyIndex(pIdx->pTable); |
| 1225 | int j, k, regKey; |
| 1226 | regKey = sqlite3GetTempRange(pParse, pPk->nKeyCol); |
| 1227 | for(j=0; j<pPk->nKeyCol; j++){ |
| 1228 | k = sqlite3ColumnOfIndex(pIdx, pPk->aiColumn[j]); |
drh | 49297c8 | 2017-03-21 18:56:52 +0000 | [diff] [blame] | 1229 | assert( k>=0 && k<pIdx->nColumn ); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 1230 | sqlite3VdbeAddOp3(v, OP_Column, iIdxCur, k, regKey+j); |
| 1231 | VdbeComment((v, "%s", pTab->aCol[pPk->aiColumn[j]].zName)); |
| 1232 | } |
| 1233 | sqlite3VdbeAddOp3(v, OP_MakeRecord, regKey, pPk->nKeyCol, regRowid); |
| 1234 | sqlite3ReleaseTempRange(pParse, regKey, pPk->nKeyCol); |
| 1235 | } |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1236 | #endif |
| 1237 | assert( regChng==(regStat4+1) ); |
drh | 9b34abe | 2016-01-16 15:12:35 +0000 | [diff] [blame] | 1238 | sqlite3VdbeAddOp4(v, OP_Function0, 1, regStat4, regTemp, |
| 1239 | (char*)&statPushFuncdef, P4_FUNCDEF); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1240 | sqlite3VdbeChangeP5(v, 2+IsStat34); |
drh | 688852a | 2014-02-17 22:40:43 +0000 | [diff] [blame] | 1241 | sqlite3VdbeAddOp2(v, OP_Next, iIdxCur, addrNextRow); VdbeCoverage(v); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1242 | |
| 1243 | /* Add the entry to the stat1 table. */ |
| 1244 | callStatGet(v, regStat4, STAT_GET_STAT1, regStat1); |
drh | 4583c37 | 2014-09-19 20:13:25 +0000 | [diff] [blame] | 1245 | assert( "BBB"[0]==SQLITE_AFF_TEXT ); |
| 1246 | sqlite3VdbeAddOp4(v, OP_MakeRecord, regTabname, 3, regTemp, "BBB", 0); |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 1247 | sqlite3VdbeAddOp2(v, OP_NewRowid, iStatCur, regNewRowid); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1248 | sqlite3VdbeAddOp3(v, OP_Insert, iStatCur, regTemp, regNewRowid); |
dan | 614efe2 | 2018-01-12 16:44:29 +0000 | [diff] [blame] | 1249 | #ifdef SQLITE_ENABLE_PREUPDATE_HOOK |
| 1250 | sqlite3VdbeChangeP4(v, -1, (char*)pStat1, P4_TABLE); |
| 1251 | #endif |
drh | 2d401ab | 2008-01-10 23:50:11 +0000 | [diff] [blame] | 1252 | sqlite3VdbeChangeP5(v, OPFLAG_APPEND); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1253 | |
| 1254 | /* Add the entries to the stat3 or stat4 table. */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1255 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1256 | { |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1257 | int regEq = regStat1; |
| 1258 | int regLt = regStat1+1; |
| 1259 | int regDLt = regStat1+2; |
| 1260 | int regSample = regStat1+3; |
| 1261 | int regCol = regStat1+4; |
| 1262 | int regSampleRowid = regCol + nCol; |
| 1263 | int addrNext; |
| 1264 | int addrIsNull; |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 1265 | u8 seekOp = HasRowid(pTab) ? OP_NotExists : OP_NotFound; |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1266 | |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1267 | pParse->nMem = MAX(pParse->nMem, regCol+nCol); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1268 | |
| 1269 | addrNext = sqlite3VdbeCurrentAddr(v); |
| 1270 | callStatGet(v, regStat4, STAT_GET_ROWID, regSampleRowid); |
| 1271 | addrIsNull = sqlite3VdbeAddOp1(v, OP_IsNull, regSampleRowid); |
drh | 688852a | 2014-02-17 22:40:43 +0000 | [diff] [blame] | 1272 | VdbeCoverage(v); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1273 | callStatGet(v, regStat4, STAT_GET_NEQ, regEq); |
| 1274 | callStatGet(v, regStat4, STAT_GET_NLT, regLt); |
| 1275 | callStatGet(v, regStat4, STAT_GET_NDLT, regDLt); |
drh | ebe25af | 2013-11-02 18:46:04 +0000 | [diff] [blame] | 1276 | sqlite3VdbeAddOp4Int(v, seekOp, iTabCur, addrNext, regSampleRowid, 0); |
drh | 6ccbd27 | 2018-07-10 17:10:44 +0000 | [diff] [blame] | 1277 | VdbeCoverage(v); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1278 | #ifdef SQLITE_ENABLE_STAT3 |
drh | 1f9ca2c | 2015-08-25 16:57:52 +0000 | [diff] [blame] | 1279 | sqlite3ExprCodeLoadIndexColumn(pParse, pIdx, iTabCur, 0, regSample); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1280 | #else |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1281 | for(i=0; i<nCol; i++){ |
drh | 1f9ca2c | 2015-08-25 16:57:52 +0000 | [diff] [blame] | 1282 | sqlite3ExprCodeLoadIndexColumn(pParse, pIdx, iTabCur, i, regCol+i); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1283 | } |
drh | ec9e55d | 2014-06-30 17:07:39 +0000 | [diff] [blame] | 1284 | sqlite3VdbeAddOp3(v, OP_MakeRecord, regCol, nCol, regSample); |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1285 | #endif |
drh | 57bf4a8 | 2014-02-17 14:59:22 +0000 | [diff] [blame] | 1286 | sqlite3VdbeAddOp3(v, OP_MakeRecord, regTabname, 6, regTemp); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1287 | sqlite3VdbeAddOp2(v, OP_NewRowid, iStatCur+1, regNewRowid); |
| 1288 | sqlite3VdbeAddOp3(v, OP_Insert, iStatCur+1, regTemp, regNewRowid); |
drh | fe70510 | 2014-03-06 13:38:37 +0000 | [diff] [blame] | 1289 | sqlite3VdbeAddOp2(v, OP_Goto, 1, addrNext); /* P1==1 for end-of-loop */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1290 | sqlite3VdbeJumpHere(v, addrIsNull); |
| 1291 | } |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1292 | #endif /* SQLITE_ENABLE_STAT3_OR_STAT4 */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1293 | |
drh | 9fecc54 | 2013-08-27 20:16:48 +0000 | [diff] [blame] | 1294 | /* End of analysis */ |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1295 | sqlite3VdbeJumpHere(v, addrRewind); |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1296 | } |
| 1297 | |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1298 | |
drh | 721dfcf | 2013-08-01 04:39:17 +0000 | [diff] [blame] | 1299 | /* Create a single sqlite_stat1 entry containing NULL as the index |
| 1300 | ** name and the row count as the content. |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1301 | */ |
drh | 721dfcf | 2013-08-01 04:39:17 +0000 | [diff] [blame] | 1302 | if( pOnlyIdx==0 && needTableCnt ){ |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1303 | VdbeComment((v, "%s", pTab->zName)); |
dan | e043201 | 2013-08-05 18:00:56 +0000 | [diff] [blame] | 1304 | sqlite3VdbeAddOp2(v, OP_Count, iTabCur, regStat1); |
drh | 688852a | 2014-02-17 22:40:43 +0000 | [diff] [blame] | 1305 | jZeroRows = sqlite3VdbeAddOp1(v, OP_IfNot, regStat1); VdbeCoverage(v); |
drh | 721dfcf | 2013-08-01 04:39:17 +0000 | [diff] [blame] | 1306 | sqlite3VdbeAddOp2(v, OP_Null, 0, regIdxname); |
drh | 4583c37 | 2014-09-19 20:13:25 +0000 | [diff] [blame] | 1307 | assert( "BBB"[0]==SQLITE_AFF_TEXT ); |
| 1308 | sqlite3VdbeAddOp4(v, OP_MakeRecord, regTabname, 3, regTemp, "BBB", 0); |
drh | 721dfcf | 2013-08-01 04:39:17 +0000 | [diff] [blame] | 1309 | sqlite3VdbeAddOp2(v, OP_NewRowid, iStatCur, regNewRowid); |
dan | f00e902 | 2013-08-14 19:54:12 +0000 | [diff] [blame] | 1310 | sqlite3VdbeAddOp3(v, OP_Insert, iStatCur, regTemp, regNewRowid); |
drh | 721dfcf | 2013-08-01 04:39:17 +0000 | [diff] [blame] | 1311 | sqlite3VdbeChangeP5(v, OPFLAG_APPEND); |
dan | 3739f29 | 2018-01-17 20:57:20 +0000 | [diff] [blame] | 1312 | #ifdef SQLITE_ENABLE_PREUPDATE_HOOK |
| 1313 | sqlite3VdbeChangeP4(v, -1, (char*)pStat1, P4_TABLE); |
| 1314 | #endif |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1315 | sqlite3VdbeJumpHere(v, jZeroRows); |
| 1316 | } |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1317 | } |
| 1318 | |
drh | 72052a7 | 2017-02-17 16:26:34 +0000 | [diff] [blame] | 1319 | |
| 1320 | /* |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1321 | ** Generate code that will cause the most recent index analysis to |
| 1322 | ** be loaded into internal hash tables where is can be used. |
drh | 72052a7 | 2017-02-17 16:26:34 +0000 | [diff] [blame] | 1323 | */ |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1324 | static void loadAnalysis(Parse *pParse, int iDb){ |
drh | 182e84c | 2017-02-18 02:19:02 +0000 | [diff] [blame] | 1325 | Vdbe *v = sqlite3GetVdbe(pParse); |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1326 | if( v ){ |
| 1327 | sqlite3VdbeAddOp1(v, OP_LoadAnalysis, iDb); |
| 1328 | } |
| 1329 | } |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1330 | |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1331 | /* |
| 1332 | ** Generate code that will do an analysis of an entire database |
| 1333 | */ |
| 1334 | static void analyzeDatabase(Parse *pParse, int iDb){ |
| 1335 | sqlite3 *db = pParse->db; |
| 1336 | Schema *pSchema = db->aDb[iDb].pSchema; /* Schema of database iDb */ |
| 1337 | HashElem *k; |
| 1338 | int iStatCur; |
| 1339 | int iMem; |
| 1340 | int iTab; |
| 1341 | |
| 1342 | sqlite3BeginWriteOperation(pParse, 0, iDb); |
| 1343 | iStatCur = pParse->nTab; |
| 1344 | pParse->nTab += 3; |
| 1345 | openStatTable(pParse, iDb, iStatCur, 0, 0); |
| 1346 | iMem = pParse->nMem+1; |
| 1347 | iTab = pParse->nTab; |
| 1348 | assert( sqlite3SchemaMutexHeld(db, iDb, 0) ); |
| 1349 | for(k=sqliteHashFirst(&pSchema->tblHash); k; k=sqliteHashNext(k)){ |
| 1350 | Table *pTab = (Table*)sqliteHashData(k); |
| 1351 | analyzeOneTable(pParse, pTab, 0, iStatCur, iMem, iTab); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1352 | } |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1353 | loadAnalysis(pParse, iDb); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1354 | } |
| 1355 | |
| 1356 | /* |
| 1357 | ** Generate code that will do an analysis of a single table in |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 1358 | ** a database. If pOnlyIdx is not NULL then it is a single index |
| 1359 | ** in pTab that should be analyzed. |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1360 | */ |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 1361 | static void analyzeTable(Parse *pParse, Table *pTab, Index *pOnlyIdx){ |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1362 | int iDb; |
| 1363 | int iStatCur; |
| 1364 | |
| 1365 | assert( pTab!=0 ); |
drh | 1fee73e | 2007-08-29 04:00:57 +0000 | [diff] [blame] | 1366 | assert( sqlite3BtreeHoldsAllMutexes(pParse->db) ); |
danielk1977 | da18423 | 2006-01-05 11:34:32 +0000 | [diff] [blame] | 1367 | iDb = sqlite3SchemaToIndex(pParse->db, pTab->pSchema); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1368 | sqlite3BeginWriteOperation(pParse, 0, iDb); |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 1369 | iStatCur = pParse->nTab; |
drh | ade3add | 2011-08-13 00:58:05 +0000 | [diff] [blame] | 1370 | pParse->nTab += 3; |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 1371 | if( pOnlyIdx ){ |
| 1372 | openStatTable(pParse, iDb, iStatCur, pOnlyIdx->zName, "idx"); |
| 1373 | }else{ |
| 1374 | openStatTable(pParse, iDb, iStatCur, pTab->zName, "tbl"); |
| 1375 | } |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1376 | analyzeOneTable(pParse, pTab, pOnlyIdx, iStatCur,pParse->nMem+1,pParse->nTab); |
| 1377 | loadAnalysis(pParse, iDb); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1378 | } |
| 1379 | |
| 1380 | /* |
| 1381 | ** Generate code for the ANALYZE command. The parser calls this routine |
| 1382 | ** when it recognizes an ANALYZE command. |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 1383 | ** |
| 1384 | ** ANALYZE -- 1 |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1385 | ** ANALYZE <database> -- 2 |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 1386 | ** ANALYZE ?<database>.?<tablename> -- 3 |
| 1387 | ** |
| 1388 | ** Form 1 causes all indices in all attached databases to be analyzed. |
| 1389 | ** Form 2 analyzes all indices the single database named. |
| 1390 | ** Form 3 analyzes all indices associated with the named table. |
| 1391 | */ |
drh | 182e84c | 2017-02-18 02:19:02 +0000 | [diff] [blame] | 1392 | void sqlite3Analyze(Parse *pParse, Token *pName1, Token *pName2){ |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1393 | sqlite3 *db = pParse->db; |
| 1394 | int iDb; |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1395 | int i; |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1396 | char *z, *zDb; |
| 1397 | Table *pTab; |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 1398 | Index *pIdx; |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1399 | Token *pTableName; |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1400 | Vdbe *v; |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1401 | |
| 1402 | /* Read the database schema. If an error occurs, leave an error message |
| 1403 | ** and code in pParse and return NULL. */ |
drh | 1fee73e | 2007-08-29 04:00:57 +0000 | [diff] [blame] | 1404 | assert( sqlite3BtreeHoldsAllMutexes(pParse->db) ); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1405 | if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){ |
| 1406 | return; |
| 1407 | } |
| 1408 | |
drh | 05800a1 | 2009-04-16 17:45:47 +0000 | [diff] [blame] | 1409 | assert( pName2!=0 || pName1==0 ); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1410 | if( pName1==0 ){ |
| 1411 | /* Form 1: Analyze everything */ |
drh | 4a54bb5 | 2017-02-18 15:58:52 +0000 | [diff] [blame] | 1412 | for(i=0; i<db->nDb; i++){ |
| 1413 | if( i==1 ) continue; /* Do not analyze the TEMP database */ |
| 1414 | analyzeDatabase(pParse, i); |
| 1415 | } |
drh | bce0414 | 2017-02-23 00:58:36 +0000 | [diff] [blame] | 1416 | }else if( pName2->n==0 && (iDb = sqlite3FindDb(db, pName1))>=0 ){ |
| 1417 | /* Analyze the schema named as the argument */ |
| 1418 | analyzeDatabase(pParse, iDb); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1419 | }else{ |
drh | bce0414 | 2017-02-23 00:58:36 +0000 | [diff] [blame] | 1420 | /* Form 3: Analyze the table or index named as an argument */ |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1421 | iDb = sqlite3TwoPartName(pParse, pName1, pName2, &pTableName); |
| 1422 | if( iDb>=0 ){ |
drh | bce0414 | 2017-02-23 00:58:36 +0000 | [diff] [blame] | 1423 | zDb = pName2->n ? db->aDb[iDb].zDbSName : 0; |
drh | 1743575 | 2007-08-16 04:30:38 +0000 | [diff] [blame] | 1424 | z = sqlite3NameFromToken(db, pTableName); |
drh | cf1be45 | 2007-05-12 12:08:51 +0000 | [diff] [blame] | 1425 | if( z ){ |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 1426 | if( (pIdx = sqlite3FindIndex(db, z, zDb))!=0 ){ |
| 1427 | analyzeTable(pParse, pIdx->pTable, pIdx); |
| 1428 | }else if( (pTab = sqlite3LocateTable(pParse, 0, z, zDb))!=0 ){ |
| 1429 | analyzeTable(pParse, pTab, 0); |
drh | cf1be45 | 2007-05-12 12:08:51 +0000 | [diff] [blame] | 1430 | } |
drh | a071bc5 | 2011-03-31 02:03:28 +0000 | [diff] [blame] | 1431 | sqlite3DbFree(db, z); |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1432 | } |
drh | bce0414 | 2017-02-23 00:58:36 +0000 | [diff] [blame] | 1433 | } |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1434 | } |
drh | bce0414 | 2017-02-23 00:58:36 +0000 | [diff] [blame] | 1435 | if( db->nSqlExec==0 && (v = sqlite3GetVdbe(pParse))!=0 ){ |
| 1436 | sqlite3VdbeAddOp0(v, OP_Expire); |
| 1437 | } |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 1438 | } |
| 1439 | |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1440 | /* |
| 1441 | ** Used to pass information from the analyzer reader through to the |
| 1442 | ** callback routine. |
| 1443 | */ |
| 1444 | typedef struct analysisInfo analysisInfo; |
| 1445 | struct analysisInfo { |
| 1446 | sqlite3 *db; |
| 1447 | const char *zDatabase; |
| 1448 | }; |
| 1449 | |
| 1450 | /* |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1451 | ** The first argument points to a nul-terminated string containing a |
| 1452 | ** list of space separated integers. Read the first nOut of these into |
| 1453 | ** the array aOut[]. |
| 1454 | */ |
| 1455 | static void decodeIntArray( |
drh | c28c4e5 | 2013-10-03 19:21:41 +0000 | [diff] [blame] | 1456 | char *zIntArray, /* String containing int array to decode */ |
| 1457 | int nOut, /* Number of slots in aOut[] */ |
| 1458 | tRowcnt *aOut, /* Store integers here */ |
dan | cfc9df7 | 2014-04-25 15:01:01 +0000 | [diff] [blame] | 1459 | LogEst *aLog, /* Or, if aOut==0, here */ |
drh | c28c4e5 | 2013-10-03 19:21:41 +0000 | [diff] [blame] | 1460 | Index *pIndex /* Handle extra flags for this index, if not NULL */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1461 | ){ |
| 1462 | char *z = zIntArray; |
| 1463 | int c; |
| 1464 | int i; |
| 1465 | tRowcnt v; |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1466 | |
drh | 6d57bbf | 2013-08-28 11:43:49 +0000 | [diff] [blame] | 1467 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
| 1468 | if( z==0 ) z = ""; |
| 1469 | #else |
drh | 85d117b | 2014-10-06 18:33:49 +0000 | [diff] [blame] | 1470 | assert( z!=0 ); |
drh | 6d57bbf | 2013-08-28 11:43:49 +0000 | [diff] [blame] | 1471 | #endif |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1472 | for(i=0; *z && i<nOut; i++){ |
| 1473 | v = 0; |
| 1474 | while( (c=z[0])>='0' && c<='9' ){ |
| 1475 | v = v*10 + c - '0'; |
| 1476 | z++; |
| 1477 | } |
drh | c5f246e | 2014-05-01 20:24:21 +0000 | [diff] [blame] | 1478 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 85d117b | 2014-10-06 18:33:49 +0000 | [diff] [blame] | 1479 | if( aOut ) aOut[i] = v; |
| 1480 | if( aLog ) aLog[i] = sqlite3LogEst(v); |
drh | c5f246e | 2014-05-01 20:24:21 +0000 | [diff] [blame] | 1481 | #else |
| 1482 | assert( aOut==0 ); |
| 1483 | UNUSED_PARAMETER(aOut); |
drh | 85d117b | 2014-10-06 18:33:49 +0000 | [diff] [blame] | 1484 | assert( aLog!=0 ); |
| 1485 | aLog[i] = sqlite3LogEst(v); |
drh | c5f246e | 2014-05-01 20:24:21 +0000 | [diff] [blame] | 1486 | #endif |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1487 | if( *z==' ' ) z++; |
| 1488 | } |
drh | 1f1fc0c | 2013-10-08 23:16:48 +0000 | [diff] [blame] | 1489 | #ifndef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 0da10d32 | 2014-11-22 21:37:00 +0000 | [diff] [blame] | 1490 | assert( pIndex!=0 ); { |
drh | 1f1fc0c | 2013-10-08 23:16:48 +0000 | [diff] [blame] | 1491 | #else |
drh | 0da10d32 | 2014-11-22 21:37:00 +0000 | [diff] [blame] | 1492 | if( pIndex ){ |
drh | 1f1fc0c | 2013-10-08 23:16:48 +0000 | [diff] [blame] | 1493 | #endif |
drh | 0da10d32 | 2014-11-22 21:37:00 +0000 | [diff] [blame] | 1494 | pIndex->bUnordered = 0; |
| 1495 | pIndex->noSkipScan = 0; |
| 1496 | while( z[0] ){ |
| 1497 | if( sqlite3_strglob("unordered*", z)==0 ){ |
| 1498 | pIndex->bUnordered = 1; |
| 1499 | }else if( sqlite3_strglob("sz=[0-9]*", z)==0 ){ |
| 1500 | pIndex->szIdxRow = sqlite3LogEst(sqlite3Atoi(z+3)); |
| 1501 | }else if( sqlite3_strglob("noskipscan*", z)==0 ){ |
| 1502 | pIndex->noSkipScan = 1; |
| 1503 | } |
drh | dbd9486 | 2014-07-23 23:57:42 +0000 | [diff] [blame] | 1504 | #ifdef SQLITE_ENABLE_COSTMULT |
drh | 0da10d32 | 2014-11-22 21:37:00 +0000 | [diff] [blame] | 1505 | else if( sqlite3_strglob("costmult=[0-9]*",z)==0 ){ |
| 1506 | pIndex->pTable->costMult = sqlite3LogEst(sqlite3Atoi(z+9)); |
| 1507 | } |
drh | dbd9486 | 2014-07-23 23:57:42 +0000 | [diff] [blame] | 1508 | #endif |
drh | 0da10d32 | 2014-11-22 21:37:00 +0000 | [diff] [blame] | 1509 | while( z[0]!=0 && z[0]!=' ' ) z++; |
| 1510 | while( z[0]==' ' ) z++; |
| 1511 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1512 | } |
| 1513 | } |
| 1514 | |
| 1515 | /* |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1516 | ** This callback is invoked once for each index when reading the |
| 1517 | ** sqlite_stat1 table. |
| 1518 | ** |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1519 | ** argv[0] = name of the table |
| 1520 | ** argv[1] = name of the index (might be NULL) |
| 1521 | ** argv[2] = results of analysis - on integer for each column |
| 1522 | ** |
| 1523 | ** Entries for which argv[1]==NULL simply record the number of rows in |
| 1524 | ** the table. |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1525 | */ |
danielk1977 | 62c14b3 | 2008-11-19 09:05:26 +0000 | [diff] [blame] | 1526 | static int analysisLoader(void *pData, int argc, char **argv, char **NotUsed){ |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1527 | analysisInfo *pInfo = (analysisInfo*)pData; |
| 1528 | Index *pIndex; |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1529 | Table *pTable; |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1530 | const char *z; |
| 1531 | |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1532 | assert( argc==3 ); |
danielk1977 | f3d3c27 | 2008-11-19 16:52:44 +0000 | [diff] [blame] | 1533 | UNUSED_PARAMETER2(NotUsed, argc); |
| 1534 | |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1535 | if( argv==0 || argv[0]==0 || argv[2]==0 ){ |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1536 | return 0; |
| 1537 | } |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1538 | pTable = sqlite3FindTable(pInfo->db, argv[0], pInfo->zDatabase); |
| 1539 | if( pTable==0 ){ |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1540 | return 0; |
| 1541 | } |
drh | c2b23e7 | 2013-11-13 15:32:15 +0000 | [diff] [blame] | 1542 | if( argv[1]==0 ){ |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1543 | pIndex = 0; |
drh | c2b23e7 | 2013-11-13 15:32:15 +0000 | [diff] [blame] | 1544 | }else if( sqlite3_stricmp(argv[0],argv[1])==0 ){ |
| 1545 | pIndex = sqlite3PrimaryKeyIndex(pTable); |
| 1546 | }else{ |
| 1547 | pIndex = sqlite3FindIndex(pInfo->db, argv[1], pInfo->zDatabase); |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1548 | } |
drh | 1556405 | 2010-09-25 22:32:56 +0000 | [diff] [blame] | 1549 | z = argv[2]; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1550 | |
| 1551 | if( pIndex ){ |
dan | 4eed053 | 2015-04-20 15:13:08 +0000 | [diff] [blame] | 1552 | tRowcnt *aiRowEst = 0; |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1553 | int nCol = pIndex->nKeyCol+1; |
| 1554 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
dan | 4eed053 | 2015-04-20 15:13:08 +0000 | [diff] [blame] | 1555 | /* Index.aiRowEst may already be set here if there are duplicate |
| 1556 | ** sqlite_stat1 entries for this index. In that case just clobber |
| 1557 | ** the old data with the new instead of allocating a new array. */ |
| 1558 | if( pIndex->aiRowEst==0 ){ |
| 1559 | pIndex->aiRowEst = (tRowcnt*)sqlite3MallocZero(sizeof(tRowcnt) * nCol); |
drh | 4a642b6 | 2016-02-05 01:55:27 +0000 | [diff] [blame] | 1560 | if( pIndex->aiRowEst==0 ) sqlite3OomFault(pInfo->db); |
dan | 4eed053 | 2015-04-20 15:13:08 +0000 | [diff] [blame] | 1561 | } |
| 1562 | aiRowEst = pIndex->aiRowEst; |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1563 | #endif |
drh | 25df48d | 2014-07-22 14:58:12 +0000 | [diff] [blame] | 1564 | pIndex->bUnordered = 0; |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1565 | decodeIntArray((char*)z, nCol, aiRowEst, pIndex->aiRowLogEst, pIndex); |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1566 | pIndex->hasStat1 = 1; |
| 1567 | if( pIndex->pPartIdxWhere==0 ){ |
| 1568 | pTable->nRowLogEst = pIndex->aiRowLogEst[0]; |
| 1569 | pTable->tabFlags |= TF_HasStat1; |
| 1570 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1571 | }else{ |
drh | 52f6c91 | 2013-10-06 22:12:41 +0000 | [diff] [blame] | 1572 | Index fakeIdx; |
| 1573 | fakeIdx.szIdxRow = pTable->szTabRow; |
drh | dbd9486 | 2014-07-23 23:57:42 +0000 | [diff] [blame] | 1574 | #ifdef SQLITE_ENABLE_COSTMULT |
| 1575 | fakeIdx.pTable = pTable; |
| 1576 | #endif |
dan | cfc9df7 | 2014-04-25 15:01:01 +0000 | [diff] [blame] | 1577 | decodeIntArray((char*)z, 1, 0, &pTable->nRowLogEst, &fakeIdx); |
drh | 52f6c91 | 2013-10-06 22:12:41 +0000 | [diff] [blame] | 1578 | pTable->szTabRow = fakeIdx.szIdxRow; |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1579 | pTable->tabFlags |= TF_HasStat1; |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1580 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1581 | |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1582 | return 0; |
| 1583 | } |
| 1584 | |
| 1585 | /* |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1586 | ** If the Index.aSample variable is not NULL, delete the aSample[] array |
| 1587 | ** and its contents. |
| 1588 | */ |
dan | d46def7 | 2010-07-24 11:28:28 +0000 | [diff] [blame] | 1589 | void sqlite3DeleteIndexSamples(sqlite3 *db, Index *pIdx){ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1590 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1591 | if( pIdx->aSample ){ |
| 1592 | int j; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1593 | for(j=0; j<pIdx->nSample; j++){ |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1594 | IndexSample *p = &pIdx->aSample[j]; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1595 | sqlite3DbFree(db, p->p); |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1596 | } |
drh | 8e3937f | 2011-08-18 13:45:23 +0000 | [diff] [blame] | 1597 | sqlite3DbFree(db, pIdx->aSample); |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1598 | } |
drh | 8e3937f | 2011-08-18 13:45:23 +0000 | [diff] [blame] | 1599 | if( db && db->pnBytesFreed==0 ){ |
| 1600 | pIdx->nSample = 0; |
| 1601 | pIdx->aSample = 0; |
| 1602 | } |
shane | cea72b2 | 2009-09-07 04:38:36 +0000 | [diff] [blame] | 1603 | #else |
drh | 43b18e1 | 2010-08-17 19:40:08 +0000 | [diff] [blame] | 1604 | UNUSED_PARAMETER(db); |
shane | cea72b2 | 2009-09-07 04:38:36 +0000 | [diff] [blame] | 1605 | UNUSED_PARAMETER(pIdx); |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1606 | #endif /* SQLITE_ENABLE_STAT3_OR_STAT4 */ |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1607 | } |
| 1608 | |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1609 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1610 | /* |
| 1611 | ** Populate the pIdx->aAvgEq[] array based on the samples currently |
| 1612 | ** stored in pIdx->aSample[]. |
| 1613 | */ |
| 1614 | static void initAvgEq(Index *pIdx){ |
| 1615 | if( pIdx ){ |
| 1616 | IndexSample *aSample = pIdx->aSample; |
| 1617 | IndexSample *pFinal = &aSample[pIdx->nSample-1]; |
| 1618 | int iCol; |
dan | 39caccf | 2014-07-01 11:54:02 +0000 | [diff] [blame] | 1619 | int nCol = 1; |
| 1620 | if( pIdx->nSampleCol>1 ){ |
| 1621 | /* If this is stat4 data, then calculate aAvgEq[] values for all |
| 1622 | ** sample columns except the last. The last is always set to 1, as |
| 1623 | ** once the trailing PK fields are considered all index keys are |
| 1624 | ** unique. */ |
| 1625 | nCol = pIdx->nSampleCol-1; |
| 1626 | pIdx->aAvgEq[nCol] = 1; |
| 1627 | } |
| 1628 | for(iCol=0; iCol<nCol; iCol++){ |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1629 | int nSample = pIdx->nSample; |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1630 | int i; /* Used to iterate through samples */ |
| 1631 | tRowcnt sumEq = 0; /* Sum of the nEq values */ |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1632 | tRowcnt avgEq = 0; |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1633 | tRowcnt nRow; /* Number of rows in index */ |
| 1634 | i64 nSum100 = 0; /* Number of terms contributing to sumEq */ |
| 1635 | i64 nDist100; /* Number of distinct values in index */ |
| 1636 | |
dan | 5cca94e | 2014-12-05 21:04:26 +0000 | [diff] [blame] | 1637 | if( !pIdx->aiRowEst || iCol>=pIdx->nKeyCol || pIdx->aiRowEst[iCol+1]==0 ){ |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1638 | nRow = pFinal->anLt[iCol]; |
| 1639 | nDist100 = (i64)100 * pFinal->anDLt[iCol]; |
| 1640 | nSample--; |
| 1641 | }else{ |
| 1642 | nRow = pIdx->aiRowEst[0]; |
| 1643 | nDist100 = ((i64)100 * pIdx->aiRowEst[0]) / pIdx->aiRowEst[iCol+1]; |
| 1644 | } |
drh | 9f07cf7 | 2014-10-22 15:27:05 +0000 | [diff] [blame] | 1645 | pIdx->nRowEst0 = nRow; |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1646 | |
| 1647 | /* Set nSum to the number of distinct (iCol+1) field prefixes that |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1648 | ** occur in the stat4 table for this index. Set sumEq to the sum of |
| 1649 | ** the nEq values for column iCol for the same set (adding the value |
| 1650 | ** only once where there exist duplicate prefixes). */ |
| 1651 | for(i=0; i<nSample; i++){ |
| 1652 | if( i==(pIdx->nSample-1) |
| 1653 | || aSample[i].anDLt[iCol]!=aSample[i+1].anDLt[iCol] |
| 1654 | ){ |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1655 | sumEq += aSample[i].anEq[iCol]; |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1656 | nSum100 += 100; |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1657 | } |
| 1658 | } |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1659 | |
drh | 785d8ed | 2017-01-11 14:15:29 +0000 | [diff] [blame] | 1660 | if( nDist100>nSum100 && sumEq<nRow ){ |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1661 | avgEq = ((i64)100 * (nRow - sumEq))/(nDist100 - nSum100); |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1662 | } |
| 1663 | if( avgEq==0 ) avgEq = 1; |
| 1664 | pIdx->aAvgEq[iCol] = avgEq; |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1665 | } |
| 1666 | } |
| 1667 | } |
| 1668 | |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1669 | /* |
drh | ce95d11 | 2013-11-02 19:34:38 +0000 | [diff] [blame] | 1670 | ** Look up an index by name. Or, if the name of a WITHOUT ROWID table |
| 1671 | ** is supplied instead, find the PRIMARY KEY index for that table. |
| 1672 | */ |
| 1673 | static Index *findIndexOrPrimaryKey( |
| 1674 | sqlite3 *db, |
| 1675 | const char *zName, |
| 1676 | const char *zDb |
| 1677 | ){ |
| 1678 | Index *pIdx = sqlite3FindIndex(db, zName, zDb); |
| 1679 | if( pIdx==0 ){ |
| 1680 | Table *pTab = sqlite3FindTable(db, zName, zDb); |
| 1681 | if( pTab && !HasRowid(pTab) ) pIdx = sqlite3PrimaryKeyIndex(pTab); |
| 1682 | } |
| 1683 | return pIdx; |
| 1684 | } |
| 1685 | |
| 1686 | /* |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1687 | ** Load the content from either the sqlite_stat4 or sqlite_stat3 table |
| 1688 | ** into the relevant Index.aSample[] arrays. |
| 1689 | ** |
| 1690 | ** Arguments zSql1 and zSql2 must point to SQL statements that return |
| 1691 | ** data equivalent to the following (statements are different for stat3, |
| 1692 | ** see the caller of this function for details): |
| 1693 | ** |
| 1694 | ** zSql1: SELECT idx,count(*) FROM %Q.sqlite_stat4 GROUP BY idx |
| 1695 | ** zSql2: SELECT idx,neq,nlt,ndlt,sample FROM %Q.sqlite_stat4 |
| 1696 | ** |
| 1697 | ** where %Q is replaced with the database name before the SQL is executed. |
| 1698 | */ |
| 1699 | static int loadStatTbl( |
| 1700 | sqlite3 *db, /* Database handle */ |
| 1701 | int bStat3, /* Assume single column records only */ |
| 1702 | const char *zSql1, /* SQL statement 1 (see above) */ |
| 1703 | const char *zSql2, /* SQL statement 2 (see above) */ |
| 1704 | const char *zDb /* Database name (e.g. "main") */ |
| 1705 | ){ |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1706 | int rc; /* Result codes from subroutines */ |
| 1707 | sqlite3_stmt *pStmt = 0; /* An SQL statement being run */ |
| 1708 | char *zSql; /* Text of the SQL statement */ |
| 1709 | Index *pPrevIdx = 0; /* Previous index in the loop */ |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1710 | IndexSample *pSample; /* A slot in pIdx->aSample[] */ |
| 1711 | |
drh | 4a642b6 | 2016-02-05 01:55:27 +0000 | [diff] [blame] | 1712 | assert( db->lookaside.bDisable ); |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1713 | zSql = sqlite3MPrintf(db, zSql1, zDb); |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1714 | if( !zSql ){ |
mistachkin | fad3039 | 2016-02-13 23:43:46 +0000 | [diff] [blame] | 1715 | return SQLITE_NOMEM_BKPT; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1716 | } |
| 1717 | rc = sqlite3_prepare(db, zSql, -1, &pStmt, 0); |
| 1718 | sqlite3DbFree(db, zSql); |
| 1719 | if( rc ) return rc; |
| 1720 | |
| 1721 | while( sqlite3_step(pStmt)==SQLITE_ROW ){ |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1722 | int nIdxCol = 1; /* Number of columns in stat4 records */ |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1723 | |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1724 | char *zIndex; /* Index name */ |
| 1725 | Index *pIdx; /* Pointer to the index object */ |
| 1726 | int nSample; /* Number of samples */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1727 | int nByte; /* Bytes of space required */ |
| 1728 | int i; /* Bytes of space required */ |
| 1729 | tRowcnt *pSpace; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1730 | |
| 1731 | zIndex = (char *)sqlite3_column_text(pStmt, 0); |
| 1732 | if( zIndex==0 ) continue; |
| 1733 | nSample = sqlite3_column_int(pStmt, 1); |
drh | ce95d11 | 2013-11-02 19:34:38 +0000 | [diff] [blame] | 1734 | pIdx = findIndexOrPrimaryKey(db, zIndex, zDb); |
dan | 86f69d9 | 2013-08-12 17:31:32 +0000 | [diff] [blame] | 1735 | assert( pIdx==0 || bStat3 || pIdx->nSample==0 ); |
| 1736 | /* Index.nSample is non-zero at this point if data has already been |
| 1737 | ** loaded from the stat4 table. In this case ignore stat3 data. */ |
| 1738 | if( pIdx==0 || pIdx->nSample ) continue; |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1739 | if( bStat3==0 ){ |
dan | 39caccf | 2014-07-01 11:54:02 +0000 | [diff] [blame] | 1740 | assert( !HasRowid(pIdx->pTable) || pIdx->nColumn==pIdx->nKeyCol+1 ); |
| 1741 | if( !HasRowid(pIdx->pTable) && IsPrimaryKeyIndex(pIdx) ){ |
| 1742 | nIdxCol = pIdx->nKeyCol; |
| 1743 | }else{ |
| 1744 | nIdxCol = pIdx->nColumn; |
| 1745 | } |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1746 | } |
dan | 8ad169a | 2013-08-12 20:14:04 +0000 | [diff] [blame] | 1747 | pIdx->nSampleCol = nIdxCol; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1748 | nByte = sizeof(IndexSample) * nSample; |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1749 | nByte += sizeof(tRowcnt) * nIdxCol * 3 * nSample; |
dan | 39caccf | 2014-07-01 11:54:02 +0000 | [diff] [blame] | 1750 | nByte += nIdxCol * sizeof(tRowcnt); /* Space for Index.aAvgEq[] */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1751 | |
| 1752 | pIdx->aSample = sqlite3DbMallocZero(db, nByte); |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1753 | if( pIdx->aSample==0 ){ |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1754 | sqlite3_finalize(pStmt); |
mistachkin | fad3039 | 2016-02-13 23:43:46 +0000 | [diff] [blame] | 1755 | return SQLITE_NOMEM_BKPT; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1756 | } |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1757 | pSpace = (tRowcnt*)&pIdx->aSample[nSample]; |
dan | 39caccf | 2014-07-01 11:54:02 +0000 | [diff] [blame] | 1758 | pIdx->aAvgEq = pSpace; pSpace += nIdxCol; |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1759 | for(i=0; i<nSample; i++){ |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1760 | pIdx->aSample[i].anEq = pSpace; pSpace += nIdxCol; |
| 1761 | pIdx->aSample[i].anLt = pSpace; pSpace += nIdxCol; |
| 1762 | pIdx->aSample[i].anDLt = pSpace; pSpace += nIdxCol; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1763 | } |
| 1764 | assert( ((u8*)pSpace)-nByte==(u8*)(pIdx->aSample) ); |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1765 | } |
drh | 8e3937f | 2011-08-18 13:45:23 +0000 | [diff] [blame] | 1766 | rc = sqlite3_finalize(pStmt); |
| 1767 | if( rc ) return rc; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1768 | |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1769 | zSql = sqlite3MPrintf(db, zSql2, zDb); |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1770 | if( !zSql ){ |
mistachkin | fad3039 | 2016-02-13 23:43:46 +0000 | [diff] [blame] | 1771 | return SQLITE_NOMEM_BKPT; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1772 | } |
| 1773 | rc = sqlite3_prepare(db, zSql, -1, &pStmt, 0); |
| 1774 | sqlite3DbFree(db, zSql); |
| 1775 | if( rc ) return rc; |
| 1776 | |
| 1777 | while( sqlite3_step(pStmt)==SQLITE_ROW ){ |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1778 | char *zIndex; /* Index name */ |
| 1779 | Index *pIdx; /* Pointer to the index object */ |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1780 | int nCol = 1; /* Number of columns in index */ |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1781 | |
| 1782 | zIndex = (char *)sqlite3_column_text(pStmt, 0); |
| 1783 | if( zIndex==0 ) continue; |
drh | ce95d11 | 2013-11-02 19:34:38 +0000 | [diff] [blame] | 1784 | pIdx = findIndexOrPrimaryKey(db, zIndex, zDb); |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1785 | if( pIdx==0 ) continue; |
dan | 86f69d9 | 2013-08-12 17:31:32 +0000 | [diff] [blame] | 1786 | /* This next condition is true if data has already been loaded from |
| 1787 | ** the sqlite_stat4 table. In this case ignore stat3 data. */ |
dan | 0adbed8 | 2013-08-15 19:56:32 +0000 | [diff] [blame] | 1788 | nCol = pIdx->nSampleCol; |
| 1789 | if( bStat3 && nCol>1 ) continue; |
| 1790 | if( pIdx!=pPrevIdx ){ |
| 1791 | initAvgEq(pPrevIdx); |
| 1792 | pPrevIdx = pIdx; |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1793 | } |
dan | c367d4c | 2013-08-16 14:09:43 +0000 | [diff] [blame] | 1794 | pSample = &pIdx->aSample[pIdx->nSample]; |
dan | cfc9df7 | 2014-04-25 15:01:01 +0000 | [diff] [blame] | 1795 | decodeIntArray((char*)sqlite3_column_text(pStmt,1),nCol,pSample->anEq,0,0); |
| 1796 | decodeIntArray((char*)sqlite3_column_text(pStmt,2),nCol,pSample->anLt,0,0); |
| 1797 | decodeIntArray((char*)sqlite3_column_text(pStmt,3),nCol,pSample->anDLt,0,0); |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1798 | |
dan | c367d4c | 2013-08-16 14:09:43 +0000 | [diff] [blame] | 1799 | /* Take a copy of the sample. Add two 0x00 bytes the end of the buffer. |
| 1800 | ** This is in case the sample record is corrupted. In that case, the |
| 1801 | ** sqlite3VdbeRecordCompare() may read up to two varints past the |
| 1802 | ** end of the allocated buffer before it realizes it is dealing with |
| 1803 | ** a corrupt record. Adding the two 0x00 bytes prevents this from causing |
| 1804 | ** a buffer overread. */ |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1805 | pSample->n = sqlite3_column_bytes(pStmt, 4); |
dan | c367d4c | 2013-08-16 14:09:43 +0000 | [diff] [blame] | 1806 | pSample->p = sqlite3DbMallocZero(db, pSample->n + 2); |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1807 | if( pSample->p==0 ){ |
| 1808 | sqlite3_finalize(pStmt); |
mistachkin | fad3039 | 2016-02-13 23:43:46 +0000 | [diff] [blame] | 1809 | return SQLITE_NOMEM_BKPT; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1810 | } |
dan | 895decf | 2016-12-30 14:15:56 +0000 | [diff] [blame] | 1811 | if( pSample->n ){ |
| 1812 | memcpy(pSample->p, sqlite3_column_blob(pStmt, 4), pSample->n); |
| 1813 | } |
dan | c367d4c | 2013-08-16 14:09:43 +0000 | [diff] [blame] | 1814 | pIdx->nSample++; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1815 | } |
drh | 61b3440 | 2013-08-16 13:34:50 +0000 | [diff] [blame] | 1816 | rc = sqlite3_finalize(pStmt); |
| 1817 | if( rc==SQLITE_OK ) initAvgEq(pPrevIdx); |
| 1818 | return rc; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1819 | } |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1820 | |
| 1821 | /* |
| 1822 | ** Load content from the sqlite_stat4 and sqlite_stat3 tables into |
| 1823 | ** the Index.aSample[] arrays of all indices. |
| 1824 | */ |
| 1825 | static int loadStat4(sqlite3 *db, const char *zDb){ |
| 1826 | int rc = SQLITE_OK; /* Result codes from subroutines */ |
| 1827 | |
drh | 4a642b6 | 2016-02-05 01:55:27 +0000 | [diff] [blame] | 1828 | assert( db->lookaside.bDisable ); |
dan | 0106e37 | 2013-08-12 16:34:32 +0000 | [diff] [blame] | 1829 | if( sqlite3FindTable(db, "sqlite_stat4", zDb) ){ |
| 1830 | rc = loadStatTbl(db, 0, |
| 1831 | "SELECT idx,count(*) FROM %Q.sqlite_stat4 GROUP BY idx", |
| 1832 | "SELECT idx,neq,nlt,ndlt,sample FROM %Q.sqlite_stat4", |
| 1833 | zDb |
| 1834 | ); |
| 1835 | } |
| 1836 | |
| 1837 | if( rc==SQLITE_OK && sqlite3FindTable(db, "sqlite_stat3", zDb) ){ |
| 1838 | rc = loadStatTbl(db, 1, |
| 1839 | "SELECT idx,count(*) FROM %Q.sqlite_stat3 GROUP BY idx", |
| 1840 | "SELECT idx,neq,nlt,ndlt,sqlite_record(sample) FROM %Q.sqlite_stat3", |
| 1841 | zDb |
| 1842 | ); |
| 1843 | } |
| 1844 | |
| 1845 | return rc; |
| 1846 | } |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1847 | #endif /* SQLITE_ENABLE_STAT3_OR_STAT4 */ |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1848 | |
| 1849 | /* |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 1850 | ** Load the content of the sqlite_stat1 and sqlite_stat3/4 tables. The |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1851 | ** contents of sqlite_stat1 are used to populate the Index.aiRowEst[] |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 1852 | ** arrays. The contents of sqlite_stat3/4 are used to populate the |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1853 | ** Index.aSample[] arrays. |
| 1854 | ** |
| 1855 | ** If the sqlite_stat1 table is not present in the database, SQLITE_ERROR |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 1856 | ** is returned. In this case, even if SQLITE_ENABLE_STAT3/4 was defined |
| 1857 | ** during compilation and the sqlite_stat3/4 table is present, no data is |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1858 | ** read from it. |
| 1859 | ** |
drh | 92707ac | 2013-08-17 18:57:15 +0000 | [diff] [blame] | 1860 | ** If SQLITE_ENABLE_STAT3/4 was defined during compilation and the |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1861 | ** sqlite_stat4 table is not present in the database, SQLITE_ERROR is |
dan | 85c165c | 2009-08-19 14:34:54 +0000 | [diff] [blame] | 1862 | ** returned. However, in this case, data is read from the sqlite_stat1 |
| 1863 | ** table (if it is present) before returning. |
| 1864 | ** |
| 1865 | ** If an OOM error occurs, this function always sets db->mallocFailed. |
| 1866 | ** This means if the caller does not care about other errors, the return |
| 1867 | ** code may be ignored. |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1868 | */ |
drh | cf1be45 | 2007-05-12 12:08:51 +0000 | [diff] [blame] | 1869 | int sqlite3AnalysisLoad(sqlite3 *db, int iDb){ |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1870 | analysisInfo sInfo; |
| 1871 | HashElem *i; |
| 1872 | char *zSql; |
drh | 51b55a3 | 2016-04-04 12:38:05 +0000 | [diff] [blame] | 1873 | int rc = SQLITE_OK; |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1874 | Schema *pSchema = db->aDb[iDb].pSchema; |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1875 | |
drh | ff0587c | 2007-08-29 17:43:19 +0000 | [diff] [blame] | 1876 | assert( iDb>=0 && iDb<db->nDb ); |
| 1877 | assert( db->aDb[iDb].pBt!=0 ); |
drh | 1fee73e | 2007-08-29 04:00:57 +0000 | [diff] [blame] | 1878 | |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1879 | /* Clear any prior statistics */ |
drh | 2120608 | 2011-04-04 18:22:02 +0000 | [diff] [blame] | 1880 | assert( sqlite3SchemaMutexHeld(db, iDb, 0) ); |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1881 | for(i=sqliteHashFirst(&pSchema->tblHash); i; i=sqliteHashNext(i)){ |
| 1882 | Table *pTab = sqliteHashData(i); |
| 1883 | pTab->tabFlags &= ~TF_HasStat1; |
| 1884 | } |
| 1885 | for(i=sqliteHashFirst(&pSchema->idxHash); i; i=sqliteHashNext(i)){ |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1886 | Index *pIdx = sqliteHashData(i); |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1887 | pIdx->hasStat1 = 0; |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1888 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
dan | d46def7 | 2010-07-24 11:28:28 +0000 | [diff] [blame] | 1889 | sqlite3DeleteIndexSamples(db, pIdx); |
| 1890 | pIdx->aSample = 0; |
drh | faacf17 | 2011-08-12 01:51:45 +0000 | [diff] [blame] | 1891 | #endif |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1892 | } |
| 1893 | |
drh | f6661a8 | 2016-04-01 12:35:22 +0000 | [diff] [blame] | 1894 | /* Load new statistics out of the sqlite_stat1 table */ |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1895 | sInfo.db = db; |
drh | 69c3382 | 2016-08-18 14:33:11 +0000 | [diff] [blame] | 1896 | sInfo.zDatabase = db->aDb[iDb].zDbSName; |
drh | f6661a8 | 2016-04-01 12:35:22 +0000 | [diff] [blame] | 1897 | if( sqlite3FindTable(db, "sqlite_stat1", sInfo.zDatabase)!=0 ){ |
| 1898 | zSql = sqlite3MPrintf(db, |
| 1899 | "SELECT tbl,idx,stat FROM %Q.sqlite_stat1", sInfo.zDatabase); |
| 1900 | if( zSql==0 ){ |
| 1901 | rc = SQLITE_NOMEM_BKPT; |
| 1902 | }else{ |
| 1903 | rc = sqlite3_exec(db, zSql, analysisLoader, &sInfo, 0); |
| 1904 | sqlite3DbFree(db, zSql); |
| 1905 | } |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1906 | } |
| 1907 | |
drh | f6661a8 | 2016-04-01 12:35:22 +0000 | [diff] [blame] | 1908 | /* Set appropriate defaults on all indexes not in the sqlite_stat1 table */ |
| 1909 | assert( sqlite3SchemaMutexHeld(db, iDb, 0) ); |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1910 | for(i=sqliteHashFirst(&pSchema->idxHash); i; i=sqliteHashNext(i)){ |
drh | f6661a8 | 2016-04-01 12:35:22 +0000 | [diff] [blame] | 1911 | Index *pIdx = sqliteHashData(i); |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1912 | if( !pIdx->hasStat1 ) sqlite3DefaultRowEst(pIdx); |
drh | f16ce3b | 2009-02-13 16:59:53 +0000 | [diff] [blame] | 1913 | } |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 1914 | |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1915 | /* Load the statistics from the sqlite_stat4 table. */ |
drh | 1435a9a | 2013-08-27 23:15:44 +0000 | [diff] [blame] | 1916 | #ifdef SQLITE_ENABLE_STAT3_OR_STAT4 |
drh | 72d03a6 | 2018-07-20 19:24:02 +0000 | [diff] [blame] | 1917 | if( rc==SQLITE_OK ){ |
drh | 4a642b6 | 2016-02-05 01:55:27 +0000 | [diff] [blame] | 1918 | db->lookaside.bDisable++; |
dan | f52bb8d | 2013-08-03 20:24:58 +0000 | [diff] [blame] | 1919 | rc = loadStat4(db, sInfo.zDatabase); |
drh | 4a642b6 | 2016-02-05 01:55:27 +0000 | [diff] [blame] | 1920 | db->lookaside.bDisable--; |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 1921 | } |
drh | 33bec3f | 2017-02-17 13:38:15 +0000 | [diff] [blame] | 1922 | for(i=sqliteHashFirst(&pSchema->idxHash); i; i=sqliteHashNext(i)){ |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1923 | Index *pIdx = sqliteHashData(i); |
drh | 75b170b | 2014-10-04 00:07:44 +0000 | [diff] [blame] | 1924 | sqlite3_free(pIdx->aiRowEst); |
dan | 43085d7 | 2014-10-03 19:16:53 +0000 | [diff] [blame] | 1925 | pIdx->aiRowEst = 0; |
| 1926 | } |
dan | 69188d9 | 2009-08-19 08:18:32 +0000 | [diff] [blame] | 1927 | #endif |
dan | 02fa469 | 2009-08-17 17:06:58 +0000 | [diff] [blame] | 1928 | |
dan | e275dc3 | 2009-08-18 16:24:58 +0000 | [diff] [blame] | 1929 | if( rc==SQLITE_NOMEM ){ |
drh | 4a642b6 | 2016-02-05 01:55:27 +0000 | [diff] [blame] | 1930 | sqlite3OomFault(db); |
dan | e275dc3 | 2009-08-18 16:24:58 +0000 | [diff] [blame] | 1931 | } |
drh | cf1be45 | 2007-05-12 12:08:51 +0000 | [diff] [blame] | 1932 | return rc; |
drh | 497e446 | 2005-07-23 03:18:40 +0000 | [diff] [blame] | 1933 | } |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 1934 | |
drh | ff2d5ea | 2005-07-23 00:41:48 +0000 | [diff] [blame] | 1935 | |
drh | 9f18e8a | 2005-07-08 12:13:04 +0000 | [diff] [blame] | 1936 | #endif /* SQLITE_OMIT_ANALYZE */ |