mirror of
https://github.com/tursodatabase/libsql.git
synced 2024-12-15 15:39:52 +00:00
234 lines
6.9 KiB
C
234 lines
6.9 KiB
C
/*
|
|
** 2018-12-04
|
|
**
|
|
** The author disclaims copyright to this source code. In place of
|
|
** a legal notice, here is a blessing:
|
|
**
|
|
** May you do good and not evil.
|
|
** May you find forgiveness for yourself and forgive others.
|
|
** May you share freely, never taking more than you give.
|
|
**
|
|
*************************************************************************
|
|
**
|
|
** This file implements a utility program used to help determine which
|
|
** indexes in a database schema are used and unused, and how often specific
|
|
** indexes are used.
|
|
*/
|
|
#include "sqlite3.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <assert.h>
|
|
#include <string.h>
|
|
|
|
static void usage(const char *argv0){
|
|
printf("Usage: %s [OPTIONS] DATABASE LOG\n\n", argv0);
|
|
printf(
|
|
"DATABASE is an SQLite database against which various statements\n"
|
|
"have been run. The SQL text is stored in LOG. LOG is an SQLite\n"
|
|
"database with this schema:\n"
|
|
"\n"
|
|
" CREATE TABLE sqllog(sql TEXT);\n"
|
|
"\n"
|
|
"This utility program analyzes statements contained in LOG and prints\n"
|
|
"a report showing how many times each index in DATABASE is used by the\n"
|
|
"statements in LOG.\n"
|
|
"\n"
|
|
"DATABASE only needs to contain the schema used by the statements in\n"
|
|
"LOG. The content can be removed from DATABASE.\n"
|
|
);
|
|
printf(
|
|
"\nOPTIONS:\n\n"
|
|
" --progress N Show a progress message after every N input rows\n"
|
|
" -q Omit error message when parsing log entries\n"
|
|
" --using NAME Print SQL statements that use index NAME\n"
|
|
);
|
|
printf("\nAnalysis will be done by SQLite version %s dated %.20s\n"
|
|
"checkin number %.40s. Different versions\n"
|
|
"of SQLite might use different indexes.\n",
|
|
sqlite3_libversion(), sqlite3_sourceid(), sqlite3_sourceid()+21);
|
|
exit(1);
|
|
}
|
|
|
|
int main(int argc, char **argv){
|
|
sqlite3 *db = 0; /* The main database */
|
|
sqlite3_stmt *pStmt = 0; /* a query */
|
|
char *zSql;
|
|
int nErr = 0;
|
|
int rc;
|
|
int bQuiet = 0;
|
|
int i, j;
|
|
const char *zUsing = 0;
|
|
sqlite3_stmt *pIncrCnt = 0;
|
|
int nRow = 0;
|
|
int iProgress = 0;
|
|
|
|
for(i=j=1; i<argc; i++){
|
|
const char *z = argv[i];
|
|
if( z[0]=='-' ){
|
|
z++;
|
|
if( z[0]=='-' ) z++;
|
|
if( strcmp(z,"progress")==0 ){
|
|
if( i+1<argc ){
|
|
iProgress = strtol(argv[++i],0,0);
|
|
continue;
|
|
}
|
|
printf("The --progress option requires an argument\n");
|
|
exit(0);
|
|
}
|
|
if( strcmp(z,"q")==0 ){
|
|
bQuiet = 1;
|
|
continue;
|
|
}
|
|
if( strcmp(z,"using")==0 ){
|
|
if( i+1<argc ){
|
|
zUsing = argv[++i];
|
|
continue;
|
|
}
|
|
printf("The --using option requires an argument\n");
|
|
exit(0);
|
|
}
|
|
if( strcmp(z, "help")==0 || strcmp(z, "?")==0 ){
|
|
usage(argv[0]);
|
|
}
|
|
printf("Unknown command-line option: \"%s\"\n", argv[i]);
|
|
exit(0);
|
|
}else{
|
|
if( j<i ) argv[j++] = argv[i];
|
|
}
|
|
}
|
|
argc = j;
|
|
|
|
if( argc!=3 ) usage(argv[0]);
|
|
rc = sqlite3_open_v2(argv[1], &db, SQLITE_OPEN_READONLY, 0);
|
|
if( rc ){
|
|
printf("Cannot open \"%s\" for reading: %s\n", argv[1], sqlite3_errmsg(db));
|
|
goto errorOut;
|
|
}
|
|
rc = sqlite3_prepare_v2(db, "SELECT * FROM sqlite_schema", -1, &pStmt, 0);
|
|
if( rc ){
|
|
printf("Cannot read the schema from \"%s\" - %s\n", argv[1],
|
|
sqlite3_errmsg(db));
|
|
goto errorOut;
|
|
}
|
|
sqlite3_finalize(pStmt);
|
|
pStmt = 0;
|
|
rc = sqlite3_exec(db,
|
|
"CREATE TABLE temp.idxu(\n"
|
|
" tbl TEXT COLLATE nocase,\n"
|
|
" idx TEXT COLLATE nocase,\n"
|
|
" cnt INT,\n"
|
|
" PRIMARY KEY(idx)\n"
|
|
") WITHOUT ROWID;", 0, 0, 0);
|
|
if( rc ){
|
|
printf("Cannot create the result table - %s\n",
|
|
sqlite3_errmsg(db));
|
|
goto errorOut;
|
|
}
|
|
rc = sqlite3_exec(db,
|
|
"INSERT INTO temp.idxu(tbl,idx,cnt)"
|
|
" SELECT tbl_name, name, 0 FROM sqlite_schema"
|
|
" WHERE type='index' AND sql IS NOT NULL", 0, 0, 0);
|
|
|
|
/* Open the LOG database */
|
|
zSql = sqlite3_mprintf("ATTACH %Q AS log", argv[2]);
|
|
rc = sqlite3_exec(db, zSql, 0, 0, 0);
|
|
sqlite3_free(zSql);
|
|
if( rc ){
|
|
printf("Cannot open the LOG database \"%s\" - %s\n",
|
|
argv[2], sqlite3_errmsg(db));
|
|
goto errorOut;
|
|
}
|
|
rc = sqlite3_prepare_v2(db,
|
|
"SELECT sql, rowid FROM log.sqllog"
|
|
" WHERE upper(substr(sql,1,5)) NOT IN ('BEGIN','COMMI','ROLLB','PRAGM')",
|
|
-1, &pStmt, 0);
|
|
if( rc ){
|
|
printf("Cannot read the SQLLOG table in the LOG database \"%s\" - %s\n",
|
|
argv[2], sqlite3_errmsg(db));
|
|
goto errorOut;
|
|
}
|
|
|
|
rc = sqlite3_prepare_v2(db,
|
|
"UPDATE temp.idxu SET cnt=cnt+1 WHERE idx=?1",
|
|
-1, &pIncrCnt, 0);
|
|
if( rc ){
|
|
printf("Cannot prepare a statement to increment a counter for "
|
|
"indexes used\n");
|
|
goto errorOut;
|
|
}
|
|
|
|
/* Update the counts based on LOG */
|
|
while( sqlite3_step(pStmt)==SQLITE_ROW ){
|
|
const char *zLog = (const char*)sqlite3_column_text(pStmt, 0);
|
|
sqlite3_stmt *pS2;
|
|
if( zLog==0 ) continue;
|
|
zSql = sqlite3_mprintf("EXPLAIN QUERY PLAN %s", zLog);
|
|
rc = sqlite3_prepare_v2(db, zSql, -1, &pS2, 0);
|
|
sqlite3_free(zSql);
|
|
if( rc ){
|
|
if( !bQuiet ){
|
|
printf("Cannot compile LOG entry %d (%s): %s\n",
|
|
sqlite3_column_int(pStmt, 1), zLog, sqlite3_errmsg(db));
|
|
fflush(stdout);
|
|
}
|
|
nErr++;
|
|
}else{
|
|
nRow++;
|
|
if( iProgress>0 && (nRow%iProgress)==0 ){
|
|
printf("%d...\n", nRow);
|
|
fflush(stdout);
|
|
}
|
|
while( sqlite3_step(pS2)==SQLITE_ROW ){
|
|
const char *zExplain = (const char*)sqlite3_column_text(pS2,3);
|
|
const char *z1, *z2;
|
|
int n;
|
|
/* printf("EXPLAIN: %s\n", zExplain); */
|
|
z1 = strstr(zExplain, " USING INDEX ");
|
|
if( z1==0 ) continue;
|
|
z1 += 13;
|
|
for(z2=z1+1; z2[0] && z2[1]!='('; z2++){}
|
|
n = z2 - z1;
|
|
if( zUsing && sqlite3_strnicmp(zUsing, z1, n)==0 ){
|
|
printf("Using %s:\n%s\n", zUsing, zLog);
|
|
fflush(stdout);
|
|
}
|
|
sqlite3_bind_text(pIncrCnt,1,z1,n,SQLITE_STATIC);
|
|
sqlite3_step(pIncrCnt);
|
|
sqlite3_reset(pIncrCnt);
|
|
}
|
|
}
|
|
sqlite3_finalize(pS2);
|
|
}
|
|
sqlite3_finalize(pStmt);
|
|
|
|
/* Generate the report */
|
|
rc = sqlite3_prepare_v2(db,
|
|
"SELECT tbl, idx, cnt, "
|
|
" (SELECT group_concat(name,',') FROM pragma_index_info(idx))"
|
|
" FROM temp.idxu, main.sqlite_schema"
|
|
" WHERE temp.idxu.tbl=main.sqlite_schema.tbl_name"
|
|
" AND temp.idxu.idx=main.sqlite_schema.name"
|
|
" ORDER BY cnt DESC, tbl, idx",
|
|
-1, &pStmt, 0);
|
|
if( rc ){
|
|
printf("Cannot query the result table - %s\n",
|
|
sqlite3_errmsg(db));
|
|
goto errorOut;
|
|
}
|
|
while( sqlite3_step(pStmt)==SQLITE_ROW ){
|
|
printf("%10d %s on %s(%s)\n",
|
|
sqlite3_column_int(pStmt, 2),
|
|
sqlite3_column_text(pStmt, 1),
|
|
sqlite3_column_text(pStmt, 0),
|
|
sqlite3_column_text(pStmt, 3));
|
|
}
|
|
sqlite3_finalize(pStmt);
|
|
pStmt = 0;
|
|
|
|
errorOut:
|
|
sqlite3_finalize(pIncrCnt);
|
|
sqlite3_finalize(pStmt);
|
|
sqlite3_close(db);
|
|
return nErr;
|
|
}
|