diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-05 17:28:19 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-05-05 17:28:19 +0000 |
commit | 18657a960e125336f704ea058e25c27bd3900dcb (patch) | |
tree | 17b438b680ed45a996d7b59951e6aa34023783f2 /test/kvtest.c | |
parent | Initial commit. (diff) | |
download | sqlite3-18657a960e125336f704ea058e25c27bd3900dcb.tar.xz sqlite3-18657a960e125336f704ea058e25c27bd3900dcb.zip |
Adding upstream version 3.40.1.upstream/3.40.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'test/kvtest.c')
-rw-r--r-- | test/kvtest.c | 1137 |
1 files changed, 1137 insertions, 0 deletions
diff --git a/test/kvtest.c b/test/kvtest.c new file mode 100644 index 0000000..04dc010 --- /dev/null +++ b/test/kvtest.c @@ -0,0 +1,1137 @@ +/* +** 2016-12-28 +** +** 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 "key-value" performance test for SQLite. The +** purpose is to compare the speed of SQLite for accessing large BLOBs +** versus reading those same BLOB values out of individual files in the +** filesystem. +** +** Run "kvtest" with no arguments for on-line help, or see comments below. +** +** HOW TO COMPILE: +** +** (1) Gather this source file and a recent SQLite3 amalgamation with its +** header into the working directory. You should have: +** +** kvtest.c >--- this file +** sqlite3.c \___ SQLite +** sqlite3.h / amlagamation & header +** +** (2) Run you compiler against the two C source code files. +** +** (a) On linux or mac: +** +** OPTS="-DSQLITE_THREADSAFE=0 -DSQLITE_OMIT_LOAD_EXTENSION" +** gcc -Os -I. $OPTS kvtest.c sqlite3.c -o kvtest +** +** The $OPTS options can be omitted. The $OPTS merely omit +** the need to link against -ldl and -lpthread, or whatever +** the equivalent libraries are called on your system. +** +** (b) Windows with MSVC: +** +** cl -I. kvtest.c sqlite3.c +** +** USAGE: +** +** (1) Create a test database by running "kvtest init" with appropriate +** options. See the help message for available options. +** +** (2) Construct the corresponding pile-of-files database on disk using +** the "kvtest export" command. +** +** (3) Run tests using "kvtest run" against either the SQLite database or +** the pile-of-files database and with appropriate options. +** +** For example: +** +** ./kvtest init x1.db --count 100000 --size 10000 +** mkdir x1 +** ./kvtest export x1.db x1 +** ./kvtest run x1.db --count 10000 --max-id 1000000 +** ./kvtest run x1 --count 10000 --max-id 1000000 +*/ +static const char zHelp[] = +"Usage: kvtest COMMAND ARGS...\n" +"\n" +" kvtest init DBFILE --count N --size M --pagesize X\n" +"\n" +" Generate a new test database file named DBFILE containing N\n" +" BLOBs each of size M bytes. The page size of the new database\n" +" file will be X. Additional options:\n" +"\n" +" --variance V Randomly vary M by plus or minus V\n" +"\n" +" kvtest export DBFILE DIRECTORY [--tree]\n" +"\n" +" Export all the blobs in the kv table of DBFILE into separate\n" +" files in DIRECTORY. DIRECTORY is created if it does not previously\n" +" exist. If the --tree option is used, then the blobs are written\n" +" into a hierarchy of directories, using names like 00/00/00,\n" +" 00/00/01, 00/00/02, and so forth. Without the --tree option, all\n" +" files are in the top-level directory with names like 000000, 000001,\n" +" 000002, and so forth.\n" +"\n" +" kvtest stat DBFILE [options]\n" +"\n" +" Display summary information about DBFILE. Options:\n" +"\n" +" --vacuum Run VACUUM on the database file\n" +"\n" +" kvtest run DBFILE [options]\n" +"\n" +" Run a performance test. DBFILE can be either the name of a\n" +" database or a directory containing sample files. Options:\n" +"\n" +" --asc Read blobs in ascending order\n" +" --blob-api Use the BLOB API\n" +" --cache-size N Database cache size\n" +" --count N Read N blobs\n" +" --desc Read blobs in descending order\n" +" --fsync Synchronous file writes\n" +" --integrity-check Run \"PRAGMA integrity_check\" after test\n" +" --max-id N Maximum blob key to use\n" +" --mmap N Mmap as much as N bytes of DBFILE\n" +" --multitrans Each read or write in its own transaction\n" +" --nocheckpoint Omit the checkpoint on WAL mode writes\n" +" --nosync Set \"PRAGMA synchronous=OFF\"\n" +" --jmode MODE Set MODE journal mode prior to starting\n" +" --random Read blobs in a random order\n" +" --start N Start reading with this blob key\n" +" --stats Output operating stats before exiting\n" +" --update Do an overwrite test\n" +; + +/* Reference resources used */ +#include <stdio.h> +#include <stdlib.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <assert.h> +#include <string.h> +#include "sqlite3.h" + +#ifndef _WIN32 +# include <unistd.h> +#else + /* Provide Windows equivalent for the needed parts of unistd.h */ +# include <direct.h> +# include <io.h> +# define R_OK 2 +# define S_ISREG(m) (((m) & S_IFMT) == S_IFREG) +# define S_ISDIR(m) (((m) & S_IFMT) == S_IFDIR) +# define access _access +#endif + +#if !defined(_MSC_VER) +# include <stdint.h> +#endif + +/* +** The following macros are used to cast pointers to integers and +** integers to pointers. The way you do this varies from one compiler +** to the next, so we have developed the following set of #if statements +** to generate appropriate macros for a wide range of compilers. +** +** The correct "ANSI" way to do this is to use the intptr_t type. +** Unfortunately, that typedef is not available on all compilers, or +** if it is available, it requires an #include of specific headers +** that vary from one machine to the next. +** +** Ticket #3860: The llvm-gcc-4.2 compiler from Apple chokes on +** the ((void*)&((char*)0)[X]) construct. But MSVC chokes on ((void*)(X)). +** So we have to define the macros in different ways depending on the +** compiler. +*/ +#if defined(__PTRDIFF_TYPE__) /* This case should work for GCC */ +# define SQLITE_INT_TO_PTR(X) ((void*)(__PTRDIFF_TYPE__)(X)) +# define SQLITE_PTR_TO_INT(X) ((sqlite3_int64)(__PTRDIFF_TYPE__)(X)) +#else +# define SQLITE_INT_TO_PTR(X) ((void*)(intptr_t)(X)) +# define SQLITE_PTR_TO_INT(X) ((sqlite3_int64)(intptr_t)(X)) +#endif + +/* +** Show thqe help text and quit. +*/ +static void showHelp(void){ + fprintf(stdout, "%s", zHelp); + exit(1); +} + +/* +** Show an error message an quit. +*/ +static void fatalError(const char *zFormat, ...){ + va_list ap; + fprintf(stdout, "ERROR: "); + va_start(ap, zFormat); + vfprintf(stdout, zFormat, ap); + va_end(ap); + fprintf(stdout, "\n"); + exit(1); +} + +/* +** Return the value of a hexadecimal digit. Return -1 if the input +** is not a hex digit. +*/ +static int hexDigitValue(char c){ + if( c>='0' && c<='9' ) return c - '0'; + if( c>='a' && c<='f' ) return c - 'a' + 10; + if( c>='A' && c<='F' ) return c - 'A' + 10; + return -1; +} + +/* +** Interpret zArg as an integer value, possibly with suffixes. +*/ +static int integerValue(const char *zArg){ + int v = 0; + static const struct { char *zSuffix; int iMult; } aMult[] = { + { "KiB", 1024 }, + { "MiB", 1024*1024 }, + { "GiB", 1024*1024*1024 }, + { "KB", 1000 }, + { "MB", 1000000 }, + { "GB", 1000000000 }, + { "K", 1000 }, + { "M", 1000000 }, + { "G", 1000000000 }, + }; + int i; + int isNeg = 0; + if( zArg[0]=='-' ){ + isNeg = 1; + zArg++; + }else if( zArg[0]=='+' ){ + zArg++; + } + if( zArg[0]=='0' && zArg[1]=='x' ){ + int x; + zArg += 2; + while( (x = hexDigitValue(zArg[0]))>=0 ){ + v = (v<<4) + x; + zArg++; + } + }else{ + while( zArg[0]>='0' && zArg[0]<='9' ){ + v = v*10 + zArg[0] - '0'; + zArg++; + } + } + for(i=0; i<sizeof(aMult)/sizeof(aMult[0]); i++){ + if( sqlite3_stricmp(aMult[i].zSuffix, zArg)==0 ){ + v *= aMult[i].iMult; + break; + } + } + return isNeg? -v : v; +} + + +/* +** Check the filesystem object zPath. Determine what it is: +** +** PATH_DIR A single directory holding many files +** PATH_TREE A directory hierarchy with files at the leaves +** PATH_DB An SQLite database +** PATH_NEXIST Does not exist +** PATH_OTHER Something else +** +** PATH_DIR means all of the separate files are grouped together +** into a single directory with names like 000000, 000001, 000002, and +** so forth. PATH_TREE means there is a hierarchy of directories so +** that no single directory has too many entries. The files have names +** like 00/00/00, 00/00/01, 00/00/02 and so forth. The decision between +** PATH_DIR and PATH_TREE is determined by the presence of a subdirectory +** named "00" at the top-level. +*/ +#define PATH_DIR 1 +#define PATH_TREE 2 +#define PATH_DB 3 +#define PATH_NEXIST 0 +#define PATH_OTHER 99 +static int pathType(const char *zPath){ + struct stat x; + int rc; + if( access(zPath,R_OK) ) return PATH_NEXIST; + memset(&x, 0, sizeof(x)); + rc = stat(zPath, &x); + if( rc<0 ) return PATH_OTHER; + if( S_ISDIR(x.st_mode) ){ + char *zLayer1 = sqlite3_mprintf("%s/00", zPath); + memset(&x, 0, sizeof(x)); + rc = stat(zLayer1, &x); + sqlite3_free(zLayer1); + if( rc<0 ) return PATH_DIR; + if( S_ISDIR(x.st_mode) ) return PATH_TREE; + return PATH_DIR; + } + if( (x.st_size%512)==0 ) return PATH_DB; + return PATH_OTHER; +} + +/* +** Return the size of a file in bytes. Or return -1 if the +** named object is not a regular file or does not exist. +*/ +static sqlite3_int64 fileSize(const char *zPath){ + struct stat x; + int rc; + memset(&x, 0, sizeof(x)); + rc = stat(zPath, &x); + if( rc<0 ) return -1; + if( !S_ISREG(x.st_mode) ) return -1; + return x.st_size; +} + +/* +** A Pseudo-random number generator with a fixed seed. Use this so +** that the same sequence of "random" numbers are generated on each +** run, for repeatability. +*/ +static unsigned int randInt(void){ + static unsigned int x = 0x333a13cd; + static unsigned int y = 0xecb2adea; + x = (x>>1) ^ ((1+~(x&1)) & 0xd0000001); + y = y*1103515245 + 12345; + return x^y; +} + +/* +** Do database initialization. +*/ +static int initMain(int argc, char **argv){ + char *zDb; + int i, rc; + int nCount = 1000; + int sz = 10000; + int iVariance = 0; + int pgsz = 4096; + sqlite3 *db; + char *zSql; + char *zErrMsg = 0; + + assert( strcmp(argv[1],"init")==0 ); + assert( argc>=3 ); + zDb = argv[2]; + for(i=3; i<argc; i++){ + char *z = argv[i]; + if( z[0]!='-' ) fatalError("unknown argument: \"%s\"", z); + if( z[1]=='-' ) z++; + if( strcmp(z, "-count")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + nCount = integerValue(argv[++i]); + if( nCount<1 ) fatalError("the --count must be positive"); + continue; + } + if( strcmp(z, "-size")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + sz = integerValue(argv[++i]); + if( sz<1 ) fatalError("the --size must be positive"); + continue; + } + if( strcmp(z, "-variance")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + iVariance = integerValue(argv[++i]); + continue; + } + if( strcmp(z, "-pagesize")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + pgsz = integerValue(argv[++i]); + if( pgsz<512 || pgsz>65536 || ((pgsz-1)&pgsz)!=0 ){ + fatalError("the --pagesize must be power of 2 between 512 and 65536"); + } + continue; + } + fatalError("unknown option: \"%s\"", argv[i]); + } + rc = sqlite3_open(zDb, &db); + if( rc ){ + fatalError("cannot open database \"%s\": %s", zDb, sqlite3_errmsg(db)); + } + zSql = sqlite3_mprintf( + "DROP TABLE IF EXISTS kv;\n" + "PRAGMA page_size=%d;\n" + "VACUUM;\n" + "BEGIN;\n" + "CREATE TABLE kv(k INTEGER PRIMARY KEY, v BLOB);\n" + "WITH RECURSIVE c(x) AS (VALUES(1) UNION ALL SELECT x+1 FROM c WHERE x<%d)" + " INSERT INTO kv(k,v) SELECT x, randomblob(%d+(random()%%(%d))) FROM c;\n" + "COMMIT;\n", + pgsz, nCount, sz, iVariance+1 + ); + rc = sqlite3_exec(db, zSql, 0, 0, &zErrMsg); + if( rc ) fatalError("database create failed: %s", zErrMsg); + sqlite3_free(zSql); + sqlite3_close(db); + return 0; +} + +/* +** Analyze an existing database file. Report its content. +*/ +static int statMain(int argc, char **argv){ + char *zDb; + int i, rc; + sqlite3 *db; + char *zSql; + sqlite3_stmt *pStmt; + int doVacuum = 0; + + assert( strcmp(argv[1],"stat")==0 ); + assert( argc>=3 ); + zDb = argv[2]; + for(i=3; i<argc; i++){ + char *z = argv[i]; + if( z[0]!='-' ) fatalError("unknown argument: \"%s\"", z); + if( z[1]=='-' ) z++; + if( strcmp(z, "-vacuum")==0 ){ + doVacuum = 1; + continue; + } + fatalError("unknown option: \"%s\"", argv[i]); + } + rc = sqlite3_open(zDb, &db); + if( rc ){ + fatalError("cannot open database \"%s\": %s", zDb, sqlite3_errmsg(db)); + } + if( doVacuum ){ + printf("Vacuuming...."); fflush(stdout); + sqlite3_exec(db, "VACUUM", 0, 0, 0); + printf(" done\n"); + } + zSql = sqlite3_mprintf( + "SELECT count(*), min(length(v)), max(length(v)), avg(length(v))" + " FROM kv" + ); + rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0); + if( rc ) fatalError("cannot prepare SQL [%s]: %s", zSql, sqlite3_errmsg(db)); + sqlite3_free(zSql); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + printf("Number of entries: %8d\n", sqlite3_column_int(pStmt, 0)); + printf("Average value size: %8d\n", sqlite3_column_int(pStmt, 3)); + printf("Minimum value size: %8d\n", sqlite3_column_int(pStmt, 1)); + printf("Maximum value size: %8d\n", sqlite3_column_int(pStmt, 2)); + }else{ + printf("No rows\n"); + } + sqlite3_finalize(pStmt); + zSql = sqlite3_mprintf("PRAGMA page_size"); + rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0); + if( rc ) fatalError("cannot prepare SQL [%s]: %s", zSql, sqlite3_errmsg(db)); + sqlite3_free(zSql); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + printf("Page-size: %8d\n", sqlite3_column_int(pStmt, 0)); + } + sqlite3_finalize(pStmt); + zSql = sqlite3_mprintf("PRAGMA page_count"); + rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0); + if( rc ) fatalError("cannot prepare SQL [%s]: %s", zSql, sqlite3_errmsg(db)); + sqlite3_free(zSql); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + printf("Page-count: %8d\n", sqlite3_column_int(pStmt, 0)); + } + sqlite3_finalize(pStmt); + zSql = sqlite3_mprintf("PRAGMA freelist_count"); + rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0); + if( rc ) fatalError("cannot prepare SQL [%s]: %s", zSql, sqlite3_errmsg(db)); + sqlite3_free(zSql); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + printf("Freelist-count: %8d\n", sqlite3_column_int(pStmt, 0)); + } + sqlite3_finalize(pStmt); + rc = sqlite3_prepare_v2(db, "PRAGMA integrity_check(10)", -1, &pStmt, 0); + if( rc ) fatalError("cannot prepare integrity check: %s", sqlite3_errmsg(db)); + while( sqlite3_step(pStmt)==SQLITE_ROW ){ + printf("Integrity-check: %s\n", sqlite3_column_text(pStmt, 0)); + } + sqlite3_finalize(pStmt); + sqlite3_close(db); + return 0; +} + +/* +** remember(V,PTR) +** +** Return the integer value V. Also save the value of V in a +** C-language variable whose address is PTR. +*/ +static void rememberFunc( + sqlite3_context *pCtx, + int argc, + sqlite3_value **argv +){ + sqlite3_int64 v; + sqlite3_int64 ptr; + assert( argc==2 ); + v = sqlite3_value_int64(argv[0]); + ptr = sqlite3_value_int64(argv[1]); + *(sqlite3_int64*)SQLITE_INT_TO_PTR(ptr) = v; + sqlite3_result_int64(pCtx, v); +} + +/* +** Make sure a directory named zDir exists. +*/ +static void kvtest_mkdir(const char *zDir){ +#if defined(_WIN32) + (void)mkdir(zDir); +#else + (void)mkdir(zDir, 0755); +#endif +} + +/* +** Export the kv table to individual files in the filesystem +*/ +static int exportMain(int argc, char **argv){ + char *zDb; + char *zDir; + sqlite3 *db; + sqlite3_stmt *pStmt; + int rc; + int ePathType; + int nFN; + char *zFN; + char *zTail; + size_t nWrote; + int i; + + assert( strcmp(argv[1],"export")==0 ); + assert( argc>=3 ); + if( argc<4 ) fatalError("Usage: kvtest export DATABASE DIRECTORY [OPTIONS]"); + zDb = argv[2]; + zDir = argv[3]; + kvtest_mkdir(zDir); + for(i=4; i<argc; i++){ + const char *z = argv[i]; + if( z[0]=='-' && z[1]=='-' ) z++; + if( strcmp(z,"-tree")==0 ){ + zFN = sqlite3_mprintf("%s/00", zDir); + kvtest_mkdir(zFN); + sqlite3_free(zFN); + continue; + } + fatalError("unknown argument: \"%s\"\n", argv[i]); + } + ePathType = pathType(zDir); + if( ePathType!=PATH_DIR && ePathType!=PATH_TREE ){ + fatalError("object \"%s\" is not a directory", zDir); + } + rc = sqlite3_open(zDb, &db); + if( rc ){ + fatalError("cannot open database \"%s\": %s", zDb, sqlite3_errmsg(db)); + } + rc = sqlite3_prepare_v2(db, "SELECT k, v FROM kv ORDER BY k", -1, &pStmt, 0); + if( rc ){ + fatalError("prepare_v2 failed: %s\n", sqlite3_errmsg(db)); + } + nFN = (int)strlen(zDir); + zFN = sqlite3_mprintf("%s/00/00/00.extra---------------------", zDir); + if( zFN==0 ){ + fatalError("malloc failed\n"); + } + zTail = zFN + nFN + 1; + while( sqlite3_step(pStmt)==SQLITE_ROW ){ + int iKey = sqlite3_column_int(pStmt, 0); + sqlite3_int64 nData = sqlite3_column_bytes(pStmt, 1); + const void *pData = sqlite3_column_blob(pStmt, 1); + FILE *out; + if( ePathType==PATH_DIR ){ + sqlite3_snprintf(20, zTail, "%06d", iKey); + }else{ + sqlite3_snprintf(20, zTail, "%02d", iKey/10000); + kvtest_mkdir(zFN); + sqlite3_snprintf(20, zTail, "%02d/%02d", iKey/10000, (iKey/100)%100); + kvtest_mkdir(zFN); + sqlite3_snprintf(20, zTail, "%02d/%02d/%02d", + iKey/10000, (iKey/100)%100, iKey%100); + } + out = fopen(zFN, "wb"); + nWrote = fwrite(pData, 1, (size_t)nData, out); + fclose(out); + printf("\r%s ", zTail); fflush(stdout); + if( nWrote!=(size_t)nData ){ + fatalError("Wrote only %d of %d bytes to %s\n", + (int)nWrote, nData, zFN); + } + } + sqlite3_finalize(pStmt); + sqlite3_close(db); + sqlite3_free(zFN); + printf("\n"); + return 0; +} + +/* +** Read the content of file zName into memory obtained from sqlite3_malloc64() +** and return a pointer to the buffer. The caller is responsible for freeing +** the memory. +** +** If parameter pnByte is not NULL, (*pnByte) is set to the number of bytes +** read. +** +** For convenience, a nul-terminator byte is always appended to the data read +** from the file before the buffer is returned. This byte is not included in +** the final value of (*pnByte), if applicable. +** +** NULL is returned if any error is encountered. The final value of *pnByte +** is undefined in this case. +*/ +static unsigned char *readFile(const char *zName, sqlite3_int64 *pnByte){ + FILE *in; /* FILE from which to read content of zName */ + sqlite3_int64 nIn; /* Size of zName in bytes */ + size_t nRead; /* Number of bytes actually read */ + unsigned char *pBuf; /* Content read from disk */ + + nIn = fileSize(zName); + if( nIn<0 ) return 0; + in = fopen(zName, "rb"); + if( in==0 ) return 0; + pBuf = sqlite3_malloc64( nIn ); + if( pBuf==0 ) return 0; + nRead = fread(pBuf, (size_t)nIn, 1, in); + fclose(in); + if( nRead!=1 ){ + sqlite3_free(pBuf); + return 0; + } + if( pnByte ) *pnByte = nIn; + return pBuf; +} + +/* +** Overwrite a file with randomness. Do not change the size of the +** file. +*/ +static void updateFile(const char *zName, sqlite3_int64 *pnByte, int doFsync){ + FILE *out; /* FILE from which to read content of zName */ + sqlite3_int64 sz; /* Size of zName in bytes */ + size_t nWritten; /* Number of bytes actually read */ + unsigned char *pBuf; /* Content to store on disk */ + const char *zMode = "wb"; /* Mode for fopen() */ + + sz = fileSize(zName); + if( sz<0 ){ + fatalError("No such file: \"%s\"", zName); + } + *pnByte = sz; + if( sz==0 ) return; + pBuf = sqlite3_malloc64( sz ); + if( pBuf==0 ){ + fatalError("Cannot allocate %lld bytes\n", sz); + } + sqlite3_randomness((int)sz, pBuf); +#if defined(_WIN32) + if( doFsync ) zMode = "wbc"; +#endif + out = fopen(zName, zMode); + if( out==0 ){ + fatalError("Cannot open \"%s\" for writing\n", zName); + } + nWritten = fwrite(pBuf, 1, (size_t)sz, out); + if( doFsync ){ +#if defined(_WIN32) + fflush(out); +#else + fsync(fileno(out)); +#endif + } + fclose(out); + if( nWritten!=(size_t)sz ){ + fatalError("Wrote only %d of %d bytes to \"%s\"\n", + (int)nWritten, (int)sz, zName); + } + sqlite3_free(pBuf); +} + +/* +** Return the current time in milliseconds since the beginning of +** the Julian epoch. +*/ +static sqlite3_int64 timeOfDay(void){ + static sqlite3_vfs *clockVfs = 0; + sqlite3_int64 t; + if( clockVfs==0 ) clockVfs = sqlite3_vfs_find(0); + if( clockVfs->iVersion>=2 && clockVfs->xCurrentTimeInt64!=0 ){ + clockVfs->xCurrentTimeInt64(clockVfs, &t); + }else{ + double r; + clockVfs->xCurrentTime(clockVfs, &r); + t = (sqlite3_int64)(r*86400000.0); + } + return t; +} + +#ifdef __linux__ +/* +** Attempt to display I/O stats on Linux using /proc/PID/io +*/ +static void displayLinuxIoStats(FILE *out){ + FILE *in; + char z[200]; + sqlite3_snprintf(sizeof(z), z, "/proc/%d/io", getpid()); + in = fopen(z, "rb"); + if( in==0 ) return; + while( fgets(z, sizeof(z), in)!=0 ){ + static const struct { + const char *zPattern; + const char *zDesc; + } aTrans[] = { + { "rchar: ", "Bytes received by read():" }, + { "wchar: ", "Bytes sent to write():" }, + { "syscr: ", "Read() system calls:" }, + { "syscw: ", "Write() system calls:" }, + { "read_bytes: ", "Bytes read from storage:" }, + { "write_bytes: ", "Bytes written to storage:" }, + { "cancelled_write_bytes: ", "Cancelled write bytes:" }, + }; + int i; + for(i=0; i<sizeof(aTrans)/sizeof(aTrans[0]); i++){ + int n = (int)strlen(aTrans[i].zPattern); + if( strncmp(aTrans[i].zPattern, z, n)==0 ){ + fprintf(out, "%-36s %s", aTrans[i].zDesc, &z[n]); + break; + } + } + } + fclose(in); +} +#endif + +/* +** Display memory stats. +*/ +static int display_stats( + sqlite3 *db, /* Database to query */ + int bReset /* True to reset SQLite stats */ +){ + int iCur; + int iHiwtr; + FILE *out = stdout; + + fprintf(out, "\n"); + + iHiwtr = iCur = -1; + sqlite3_status(SQLITE_STATUS_MEMORY_USED, &iCur, &iHiwtr, bReset); + fprintf(out, + "Memory Used: %d (max %d) bytes\n", + iCur, iHiwtr); + iHiwtr = iCur = -1; + sqlite3_status(SQLITE_STATUS_MALLOC_COUNT, &iCur, &iHiwtr, bReset); + fprintf(out, "Number of Outstanding Allocations: %d (max %d)\n", + iCur, iHiwtr); + iHiwtr = iCur = -1; + sqlite3_status(SQLITE_STATUS_PAGECACHE_USED, &iCur, &iHiwtr, bReset); + fprintf(out, + "Number of Pcache Pages Used: %d (max %d) pages\n", + iCur, iHiwtr); + iHiwtr = iCur = -1; + sqlite3_status(SQLITE_STATUS_PAGECACHE_OVERFLOW, &iCur, &iHiwtr, bReset); + fprintf(out, + "Number of Pcache Overflow Bytes: %d (max %d) bytes\n", + iCur, iHiwtr); + iHiwtr = iCur = -1; + sqlite3_status(SQLITE_STATUS_MALLOC_SIZE, &iCur, &iHiwtr, bReset); + fprintf(out, "Largest Allocation: %d bytes\n", + iHiwtr); + iHiwtr = iCur = -1; + sqlite3_status(SQLITE_STATUS_PAGECACHE_SIZE, &iCur, &iHiwtr, bReset); + fprintf(out, "Largest Pcache Allocation: %d bytes\n", + iHiwtr); + + iHiwtr = iCur = -1; + sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_USED, &iCur, &iHiwtr, bReset); + fprintf(out, "Pager Heap Usage: %d bytes\n", + iCur); + iHiwtr = iCur = -1; + sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_HIT, &iCur, &iHiwtr, 1); + fprintf(out, "Page cache hits: %d\n", iCur); + iHiwtr = iCur = -1; + sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_MISS, &iCur, &iHiwtr, 1); + fprintf(out, "Page cache misses: %d\n", iCur); + iHiwtr = iCur = -1; + sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_WRITE, &iCur, &iHiwtr, 1); + fprintf(out, "Page cache writes: %d\n", iCur); + iHiwtr = iCur = -1; + +#ifdef __linux__ + displayLinuxIoStats(out); +#endif + + return 0; +} + +/* Blob access order */ +#define ORDER_ASC 1 +#define ORDER_DESC 2 +#define ORDER_RANDOM 3 + + +/* +** Run a performance test +*/ +static int runMain(int argc, char **argv){ + int eType; /* Is zDb a database or a directory? */ + char *zDb; /* Database or directory name */ + int i; /* Loop counter */ + int rc; /* Return code from SQLite calls */ + int nCount = 1000; /* Number of blob fetch operations */ + int nExtra = 0; /* Extra cycles */ + int iKey = 1; /* Next blob key */ + int iMax = 0; /* Largest allowed key */ + int iPagesize = 0; /* Database page size */ + int iCache = 1000; /* Database cache size in kibibytes */ + int bBlobApi = 0; /* Use the incremental blob I/O API */ + int bStats = 0; /* Print stats before exiting */ + int eOrder = ORDER_ASC; /* Access order */ + int isUpdateTest = 0; /* Do in-place updates rather than reads */ + int doIntegrityCk = 0; /* Run PRAGMA integrity_check after the test */ + int noSync = 0; /* Disable synchronous mode */ + int doFsync = 0; /* Update disk files synchronously */ + int doMultiTrans = 0; /* Each operation in its own transaction */ + int noCheckpoint = 0; /* Omit the checkpoint in WAL mode */ + sqlite3 *db = 0; /* Database connection */ + sqlite3_stmt *pStmt = 0; /* Prepared statement for SQL access */ + sqlite3_blob *pBlob = 0; /* Handle for incremental Blob I/O */ + sqlite3_int64 tmStart; /* Start time */ + sqlite3_int64 tmElapsed; /* Elapsed time */ + int mmapSize = 0; /* --mmap N argument */ + sqlite3_int64 nData = 0; /* Bytes of data */ + sqlite3_int64 nTotal = 0; /* Total data read */ + unsigned char *pData = 0; /* Content of the blob */ + sqlite3_int64 nAlloc = 0; /* Space allocated for pData[] */ + const char *zJMode = 0; /* Journal mode */ + + + assert( strcmp(argv[1],"run")==0 ); + assert( argc>=3 ); + zDb = argv[2]; + eType = pathType(zDb); + if( eType==PATH_OTHER ) fatalError("unknown object type: \"%s\"", zDb); + if( eType==PATH_NEXIST ) fatalError("object does not exist: \"%s\"", zDb); + for(i=3; i<argc; i++){ + char *z = argv[i]; + if( z[0]!='-' ) fatalError("unknown argument: \"%s\"", z); + if( z[1]=='-' ) z++; + if( strcmp(z, "-asc")==0 ){ + eOrder = ORDER_ASC; + continue; + } + if( strcmp(z, "-blob-api")==0 ){ + bBlobApi = 1; + continue; + } + if( strcmp(z, "-cache-size")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + iCache = integerValue(argv[++i]); + continue; + } + if( strcmp(z, "-count")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + nCount = integerValue(argv[++i]); + if( nCount<1 ) fatalError("the --count must be positive"); + continue; + } + if( strcmp(z, "-desc")==0 ){ + eOrder = ORDER_DESC; + continue; + } + if( strcmp(z, "-fsync")==0 ){ + doFsync = 1; + continue; + } + if( strcmp(z, "-integrity-check")==0 ){ + doIntegrityCk = 1; + continue; + } + if( strcmp(z, "-jmode")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + zJMode = argv[++i]; + continue; + } + if( strcmp(z, "-mmap")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + mmapSize = integerValue(argv[++i]); + if( nCount<0 ) fatalError("the --mmap must be non-negative"); + continue; + } + if( strcmp(z, "-max-id")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + iMax = integerValue(argv[++i]); + continue; + } + if( strcmp(z, "-multitrans")==0 ){ + doMultiTrans = 1; + continue; + } + if( strcmp(z, "-nocheckpoint")==0 ){ + noCheckpoint = 1; + continue; + } + if( strcmp(z, "-nosync")==0 ){ + noSync = 1; + continue; + } + if( strcmp(z, "-random")==0 ){ + eOrder = ORDER_RANDOM; + continue; + } + if( strcmp(z, "-start")==0 ){ + if( i==argc-1 ) fatalError("missing argument on \"%s\"", argv[i]); + iKey = integerValue(argv[++i]); + if( iKey<1 ) fatalError("the --start must be positive"); + continue; + } + if( strcmp(z, "-stats")==0 ){ + bStats = 1; + continue; + } + if( strcmp(z, "-update")==0 ){ + isUpdateTest = 1; + continue; + } + fatalError("unknown option: \"%s\"", argv[i]); + } + if( eType==PATH_DB ){ + /* Recover any prior crashes prior to starting the timer */ + sqlite3_open(zDb, &db); + sqlite3_exec(db, "SELECT rowid FROM sqlite_schema LIMIT 1", 0, 0, 0); + sqlite3_close(db); + db = 0; + } + tmStart = timeOfDay(); + if( eType==PATH_DB ){ + char *zSql; + rc = sqlite3_open(zDb, &db); + if( rc ){ + fatalError("cannot open database \"%s\": %s", zDb, sqlite3_errmsg(db)); + } + zSql = sqlite3_mprintf("PRAGMA mmap_size=%d", mmapSize); + sqlite3_exec(db, zSql, 0, 0, 0); + sqlite3_free(zSql); + zSql = sqlite3_mprintf("PRAGMA cache_size=%d", iCache); + sqlite3_exec(db, zSql, 0, 0, 0); + sqlite3_free(zSql); + if( noSync ){ + sqlite3_exec(db, "PRAGMA synchronous=OFF", 0, 0, 0); + } + pStmt = 0; + sqlite3_prepare_v2(db, "PRAGMA page_size", -1, &pStmt, 0); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + iPagesize = sqlite3_column_int(pStmt, 0); + } + sqlite3_finalize(pStmt); + sqlite3_prepare_v2(db, "PRAGMA cache_size", -1, &pStmt, 0); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + iCache = sqlite3_column_int(pStmt, 0); + }else{ + iCache = 0; + } + sqlite3_finalize(pStmt); + pStmt = 0; + if( zJMode ){ + zSql = sqlite3_mprintf("PRAGMA journal_mode=%Q", zJMode); + sqlite3_exec(db, zSql, 0, 0, 0); + sqlite3_free(zSql); + if( noCheckpoint ){ + sqlite3_exec(db, "PRAGMA wal_autocheckpoint=0", 0, 0, 0); + } + } + sqlite3_prepare_v2(db, "PRAGMA journal_mode", -1, &pStmt, 0); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + zJMode = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0)); + }else{ + zJMode = "???"; + } + sqlite3_finalize(pStmt); + if( iMax<=0 ){ + sqlite3_prepare_v2(db, "SELECT max(k) FROM kv", -1, &pStmt, 0); + if( sqlite3_step(pStmt)==SQLITE_ROW ){ + iMax = sqlite3_column_int(pStmt, 0); + } + sqlite3_finalize(pStmt); + } + pStmt = 0; + if( !doMultiTrans ) sqlite3_exec(db, "BEGIN", 0, 0, 0); + } + if( iMax<=0 ) iMax = 1000; + for(i=0; i<nCount; i++){ + if( eType==PATH_DIR || eType==PATH_TREE ){ + /* CASE 1: Reading or writing blobs out of separate files */ + char *zKey; + if( eType==PATH_DIR ){ + zKey = sqlite3_mprintf("%s/%06d", zDb, iKey); + }else{ + zKey = sqlite3_mprintf("%s/%02d/%02d/%02d", zDb, iKey/10000, + (iKey/100)%100, iKey%100); + } + nData = 0; + if( isUpdateTest ){ + updateFile(zKey, &nData, doFsync); + }else{ + pData = readFile(zKey, &nData); + sqlite3_free(pData); + } + sqlite3_free(zKey); + }else if( bBlobApi ){ + /* CASE 2: Reading from database using the incremental BLOB I/O API */ + if( pBlob==0 ){ + rc = sqlite3_blob_open(db, "main", "kv", "v", iKey, + isUpdateTest, &pBlob); + if( rc ){ + fatalError("could not open sqlite3_blob handle: %s", + sqlite3_errmsg(db)); + } + }else{ + rc = sqlite3_blob_reopen(pBlob, iKey); + } + if( rc==SQLITE_OK ){ + nData = sqlite3_blob_bytes(pBlob); + if( nAlloc<nData+1 ){ + nAlloc = nData+100; + pData = sqlite3_realloc64(pData, nAlloc); + } + if( pData==0 ) fatalError("cannot allocate %d bytes", nData+1); + if( isUpdateTest ){ + sqlite3_randomness((int)nData, pData); + rc = sqlite3_blob_write(pBlob, pData, (int)nData, 0); + if( rc!=SQLITE_OK ){ + fatalError("could not write the blob at %d: %s", iKey, + sqlite3_errmsg(db)); + } + }else{ + rc = sqlite3_blob_read(pBlob, pData, (int)nData, 0); + if( rc!=SQLITE_OK ){ + fatalError("could not read the blob at %d: %s", iKey, + sqlite3_errmsg(db)); + } + } + } + }else{ + /* CASE 3: Reading from database using SQL */ + if( pStmt==0 ){ + if( isUpdateTest ){ + sqlite3_create_function(db, "remember", 2, SQLITE_UTF8, 0, + rememberFunc, 0, 0); + + rc = sqlite3_prepare_v2(db, + "UPDATE kv SET v=randomblob(remember(length(v),?2))" + " WHERE k=?1", -1, &pStmt, 0); + sqlite3_bind_int64(pStmt, 2, SQLITE_PTR_TO_INT(&nData)); + }else{ + rc = sqlite3_prepare_v2(db, + "SELECT v FROM kv WHERE k=?1", -1, &pStmt, 0); + } + if( rc ){ + fatalError("cannot prepare query: %s", sqlite3_errmsg(db)); + } + }else{ + sqlite3_reset(pStmt); + } + sqlite3_bind_int(pStmt, 1, iKey); + nData = 0; + rc = sqlite3_step(pStmt); + if( rc==SQLITE_ROW ){ + nData = sqlite3_column_bytes(pStmt, 0); + pData = (unsigned char*)sqlite3_column_blob(pStmt, 0); + } + } + if( eOrder==ORDER_ASC ){ + iKey++; + if( iKey>iMax ) iKey = 1; + }else if( eOrder==ORDER_DESC ){ + iKey--; + if( iKey<=0 ) iKey = iMax; + }else{ + iKey = (randInt()%iMax)+1; + } + nTotal += nData; + if( nData==0 ){ nCount++; nExtra++; } + } + if( nAlloc ) sqlite3_free(pData); + if( pStmt ) sqlite3_finalize(pStmt); + if( pBlob ) sqlite3_blob_close(pBlob); + if( bStats ){ + display_stats(db, 0); + } + if( db ){ + if( !doMultiTrans ) sqlite3_exec(db, "COMMIT", 0, 0, 0); + if( !noCheckpoint ){ + sqlite3_close(db); + db = 0; + } + } + tmElapsed = timeOfDay() - tmStart; + if( db && noCheckpoint ){ + sqlite3_close(db); + db = 0; + } + if( nExtra ){ + printf("%d cycles due to %d misses\n", nCount, nExtra); + } + if( eType==PATH_DB ){ + printf("SQLite version: %s\n", sqlite3_libversion()); + if( doIntegrityCk ){ + sqlite3_open(zDb, &db); + sqlite3_prepare_v2(db, "PRAGMA integrity_check", -1, &pStmt, 0); + while( sqlite3_step(pStmt)==SQLITE_ROW ){ + printf("integrity-check: %s\n", sqlite3_column_text(pStmt, 0)); + } + sqlite3_finalize(pStmt); + sqlite3_close(db); + db = 0; + } + } + printf("--count %d --max-id %d", nCount-nExtra, iMax); + switch( eOrder ){ + case ORDER_RANDOM: printf(" --random\n"); break; + case ORDER_DESC: printf(" --desc\n"); break; + default: printf(" --asc\n"); break; + } + if( eType==PATH_DB ){ + printf("--cache-size %d --jmode %s\n", iCache, zJMode); + printf("--mmap %d%s\n", mmapSize, bBlobApi ? " --blob-api" : ""); + if( noSync ) printf("--nosync\n"); + } + if( iPagesize ) printf("Database page size: %d\n", iPagesize); + printf("Total elapsed time: %.3f\n", tmElapsed/1000.0); + if( isUpdateTest ){ + printf("Microseconds per BLOB write: %.3f\n", tmElapsed*1000.0/nCount); + printf("Content write rate: %.1f MB/s\n", nTotal/(1000.0*tmElapsed)); + }else{ + printf("Microseconds per BLOB read: %.3f\n", tmElapsed*1000.0/nCount); + printf("Content read rate: %.1f MB/s\n", nTotal/(1000.0*tmElapsed)); + } + return 0; +} + + +int main(int argc, char **argv){ + if( argc<3 ) showHelp(); + if( strcmp(argv[1],"init")==0 ){ + return initMain(argc, argv); + } + if( strcmp(argv[1],"export")==0 ){ + return exportMain(argc, argv); + } + if( strcmp(argv[1],"run")==0 ){ + return runMain(argc, argv); + } + if( strcmp(argv[1],"stat")==0 ){ + return statMain(argc, argv); + } + showHelp(); + return 0; +} |