Annotation of embedaddon/sqlite3/ext/fts2/fts2_tokenizer.c, revision 1.1
1.1 ! misho 1: /*
! 2: ** 2007 June 22
! 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: **
! 13: ** This is part of an SQLite module implementing full-text search.
! 14: ** This particular file implements the generic tokenizer interface.
! 15: */
! 16:
! 17: /*
! 18: ** The code in this file is only compiled if:
! 19: **
! 20: ** * The FTS2 module is being built as an extension
! 21: ** (in which case SQLITE_CORE is not defined), or
! 22: **
! 23: ** * The FTS2 module is being built into the core of
! 24: ** SQLite (in which case SQLITE_ENABLE_FTS2 is defined).
! 25: */
! 26: #if !defined(SQLITE_CORE) || defined(SQLITE_ENABLE_FTS2)
! 27:
! 28:
! 29: #include "sqlite3.h"
! 30: #include "sqlite3ext.h"
! 31: SQLITE_EXTENSION_INIT1
! 32:
! 33: #include "fts2_hash.h"
! 34: #include "fts2_tokenizer.h"
! 35: #include <assert.h>
! 36:
! 37: /*
! 38: ** Implementation of the SQL scalar function for accessing the underlying
! 39: ** hash table. This function may be called as follows:
! 40: **
! 41: ** SELECT <function-name>(<key-name>);
! 42: ** SELECT <function-name>(<key-name>, <pointer>);
! 43: **
! 44: ** where <function-name> is the name passed as the second argument
! 45: ** to the sqlite3Fts2InitHashTable() function (e.g. 'fts2_tokenizer').
! 46: **
! 47: ** If the <pointer> argument is specified, it must be a blob value
! 48: ** containing a pointer to be stored as the hash data corresponding
! 49: ** to the string <key-name>. If <pointer> is not specified, then
! 50: ** the string <key-name> must already exist in the has table. Otherwise,
! 51: ** an error is returned.
! 52: **
! 53: ** Whether or not the <pointer> argument is specified, the value returned
! 54: ** is a blob containing the pointer stored as the hash data corresponding
! 55: ** to string <key-name> (after the hash-table is updated, if applicable).
! 56: */
! 57: static void scalarFunc(
! 58: sqlite3_context *context,
! 59: int argc,
! 60: sqlite3_value **argv
! 61: ){
! 62: fts2Hash *pHash;
! 63: void *pPtr = 0;
! 64: const unsigned char *zName;
! 65: int nName;
! 66:
! 67: assert( argc==1 || argc==2 );
! 68:
! 69: pHash = (fts2Hash *)sqlite3_user_data(context);
! 70:
! 71: zName = sqlite3_value_text(argv[0]);
! 72: nName = sqlite3_value_bytes(argv[0])+1;
! 73:
! 74: if( argc==2 ){
! 75: void *pOld;
! 76: int n = sqlite3_value_bytes(argv[1]);
! 77: if( n!=sizeof(pPtr) ){
! 78: sqlite3_result_error(context, "argument type mismatch", -1);
! 79: return;
! 80: }
! 81: pPtr = *(void **)sqlite3_value_blob(argv[1]);
! 82: pOld = sqlite3Fts2HashInsert(pHash, (void *)zName, nName, pPtr);
! 83: if( pOld==pPtr ){
! 84: sqlite3_result_error(context, "out of memory", -1);
! 85: return;
! 86: }
! 87: }else{
! 88: pPtr = sqlite3Fts2HashFind(pHash, zName, nName);
! 89: if( !pPtr ){
! 90: char *zErr = sqlite3_mprintf("unknown tokenizer: %s", zName);
! 91: sqlite3_result_error(context, zErr, -1);
! 92: sqlite3_free(zErr);
! 93: return;
! 94: }
! 95: }
! 96:
! 97: sqlite3_result_blob(context, (void *)&pPtr, sizeof(pPtr), SQLITE_TRANSIENT);
! 98: }
! 99:
! 100: #ifdef SQLITE_TEST
! 101:
! 102: #include <tcl.h>
! 103: #include <string.h>
! 104:
! 105: /*
! 106: ** Implementation of a special SQL scalar function for testing tokenizers
! 107: ** designed to be used in concert with the Tcl testing framework. This
! 108: ** function must be called with two arguments:
! 109: **
! 110: ** SELECT <function-name>(<key-name>, <input-string>);
! 111: ** SELECT <function-name>(<key-name>, <pointer>);
! 112: **
! 113: ** where <function-name> is the name passed as the second argument
! 114: ** to the sqlite3Fts2InitHashTable() function (e.g. 'fts2_tokenizer')
! 115: ** concatenated with the string '_test' (e.g. 'fts2_tokenizer_test').
! 116: **
! 117: ** The return value is a string that may be interpreted as a Tcl
! 118: ** list. For each token in the <input-string>, three elements are
! 119: ** added to the returned list. The first is the token position, the
! 120: ** second is the token text (folded, stemmed, etc.) and the third is the
! 121: ** substring of <input-string> associated with the token. For example,
! 122: ** using the built-in "simple" tokenizer:
! 123: **
! 124: ** SELECT fts_tokenizer_test('simple', 'I don't see how');
! 125: **
! 126: ** will return the string:
! 127: **
! 128: ** "{0 i I 1 dont don't 2 see see 3 how how}"
! 129: **
! 130: */
! 131: static void testFunc(
! 132: sqlite3_context *context,
! 133: int argc,
! 134: sqlite3_value **argv
! 135: ){
! 136: fts2Hash *pHash;
! 137: sqlite3_tokenizer_module *p;
! 138: sqlite3_tokenizer *pTokenizer = 0;
! 139: sqlite3_tokenizer_cursor *pCsr = 0;
! 140:
! 141: const char *zErr = 0;
! 142:
! 143: const char *zName;
! 144: int nName;
! 145: const char *zInput;
! 146: int nInput;
! 147:
! 148: const char *zArg = 0;
! 149:
! 150: const char *zToken;
! 151: int nToken;
! 152: int iStart;
! 153: int iEnd;
! 154: int iPos;
! 155:
! 156: Tcl_Obj *pRet;
! 157:
! 158: assert( argc==2 || argc==3 );
! 159:
! 160: nName = sqlite3_value_bytes(argv[0]);
! 161: zName = (const char *)sqlite3_value_text(argv[0]);
! 162: nInput = sqlite3_value_bytes(argv[argc-1]);
! 163: zInput = (const char *)sqlite3_value_text(argv[argc-1]);
! 164:
! 165: if( argc==3 ){
! 166: zArg = (const char *)sqlite3_value_text(argv[1]);
! 167: }
! 168:
! 169: pHash = (fts2Hash *)sqlite3_user_data(context);
! 170: p = (sqlite3_tokenizer_module *)sqlite3Fts2HashFind(pHash, zName, nName+1);
! 171:
! 172: if( !p ){
! 173: char *zErr = sqlite3_mprintf("unknown tokenizer: %s", zName);
! 174: sqlite3_result_error(context, zErr, -1);
! 175: sqlite3_free(zErr);
! 176: return;
! 177: }
! 178:
! 179: pRet = Tcl_NewObj();
! 180: Tcl_IncrRefCount(pRet);
! 181:
! 182: if( SQLITE_OK!=p->xCreate(zArg ? 1 : 0, &zArg, &pTokenizer) ){
! 183: zErr = "error in xCreate()";
! 184: goto finish;
! 185: }
! 186: pTokenizer->pModule = p;
! 187: if( SQLITE_OK!=p->xOpen(pTokenizer, zInput, nInput, &pCsr) ){
! 188: zErr = "error in xOpen()";
! 189: goto finish;
! 190: }
! 191: pCsr->pTokenizer = pTokenizer;
! 192:
! 193: while( SQLITE_OK==p->xNext(pCsr, &zToken, &nToken, &iStart, &iEnd, &iPos) ){
! 194: Tcl_ListObjAppendElement(0, pRet, Tcl_NewIntObj(iPos));
! 195: Tcl_ListObjAppendElement(0, pRet, Tcl_NewStringObj(zToken, nToken));
! 196: zToken = &zInput[iStart];
! 197: nToken = iEnd-iStart;
! 198: Tcl_ListObjAppendElement(0, pRet, Tcl_NewStringObj(zToken, nToken));
! 199: }
! 200:
! 201: if( SQLITE_OK!=p->xClose(pCsr) ){
! 202: zErr = "error in xClose()";
! 203: goto finish;
! 204: }
! 205: if( SQLITE_OK!=p->xDestroy(pTokenizer) ){
! 206: zErr = "error in xDestroy()";
! 207: goto finish;
! 208: }
! 209:
! 210: finish:
! 211: if( zErr ){
! 212: sqlite3_result_error(context, zErr, -1);
! 213: }else{
! 214: sqlite3_result_text(context, Tcl_GetString(pRet), -1, SQLITE_TRANSIENT);
! 215: }
! 216: Tcl_DecrRefCount(pRet);
! 217: }
! 218:
! 219: static
! 220: int registerTokenizer(
! 221: sqlite3 *db,
! 222: char *zName,
! 223: const sqlite3_tokenizer_module *p
! 224: ){
! 225: int rc;
! 226: sqlite3_stmt *pStmt;
! 227: const char zSql[] = "SELECT fts2_tokenizer(?, ?)";
! 228:
! 229: rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0);
! 230: if( rc!=SQLITE_OK ){
! 231: return rc;
! 232: }
! 233:
! 234: sqlite3_bind_text(pStmt, 1, zName, -1, SQLITE_STATIC);
! 235: sqlite3_bind_blob(pStmt, 2, &p, sizeof(p), SQLITE_STATIC);
! 236: sqlite3_step(pStmt);
! 237:
! 238: return sqlite3_finalize(pStmt);
! 239: }
! 240:
! 241: static
! 242: int queryFts2Tokenizer(
! 243: sqlite3 *db,
! 244: char *zName,
! 245: const sqlite3_tokenizer_module **pp
! 246: ){
! 247: int rc;
! 248: sqlite3_stmt *pStmt;
! 249: const char zSql[] = "SELECT fts2_tokenizer(?)";
! 250:
! 251: *pp = 0;
! 252: rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0);
! 253: if( rc!=SQLITE_OK ){
! 254: return rc;
! 255: }
! 256:
! 257: sqlite3_bind_text(pStmt, 1, zName, -1, SQLITE_STATIC);
! 258: if( SQLITE_ROW==sqlite3_step(pStmt) ){
! 259: if( sqlite3_column_type(pStmt, 0)==SQLITE_BLOB ){
! 260: memcpy(pp, sqlite3_column_blob(pStmt, 0), sizeof(*pp));
! 261: }
! 262: }
! 263:
! 264: return sqlite3_finalize(pStmt);
! 265: }
! 266:
! 267: void sqlite3Fts2SimpleTokenizerModule(sqlite3_tokenizer_module const**ppModule);
! 268:
! 269: /*
! 270: ** Implementation of the scalar function fts2_tokenizer_internal_test().
! 271: ** This function is used for testing only, it is not included in the
! 272: ** build unless SQLITE_TEST is defined.
! 273: **
! 274: ** The purpose of this is to test that the fts2_tokenizer() function
! 275: ** can be used as designed by the C-code in the queryFts2Tokenizer and
! 276: ** registerTokenizer() functions above. These two functions are repeated
! 277: ** in the README.tokenizer file as an example, so it is important to
! 278: ** test them.
! 279: **
! 280: ** To run the tests, evaluate the fts2_tokenizer_internal_test() scalar
! 281: ** function with no arguments. An assert() will fail if a problem is
! 282: ** detected. i.e.:
! 283: **
! 284: ** SELECT fts2_tokenizer_internal_test();
! 285: **
! 286: */
! 287: static void intTestFunc(
! 288: sqlite3_context *context,
! 289: int argc,
! 290: sqlite3_value **argv
! 291: ){
! 292: int rc;
! 293: const sqlite3_tokenizer_module *p1;
! 294: const sqlite3_tokenizer_module *p2;
! 295: sqlite3 *db = (sqlite3 *)sqlite3_user_data(context);
! 296:
! 297: /* Test the query function */
! 298: sqlite3Fts2SimpleTokenizerModule(&p1);
! 299: rc = queryFts2Tokenizer(db, "simple", &p2);
! 300: assert( rc==SQLITE_OK );
! 301: assert( p1==p2 );
! 302: rc = queryFts2Tokenizer(db, "nosuchtokenizer", &p2);
! 303: assert( rc==SQLITE_ERROR );
! 304: assert( p2==0 );
! 305: assert( 0==strcmp(sqlite3_errmsg(db), "unknown tokenizer: nosuchtokenizer") );
! 306:
! 307: /* Test the storage function */
! 308: rc = registerTokenizer(db, "nosuchtokenizer", p1);
! 309: assert( rc==SQLITE_OK );
! 310: rc = queryFts2Tokenizer(db, "nosuchtokenizer", &p2);
! 311: assert( rc==SQLITE_OK );
! 312: assert( p2==p1 );
! 313:
! 314: sqlite3_result_text(context, "ok", -1, SQLITE_STATIC);
! 315: }
! 316:
! 317: #endif
! 318:
! 319: /*
! 320: ** Set up SQL objects in database db used to access the contents of
! 321: ** the hash table pointed to by argument pHash. The hash table must
! 322: ** been initialised to use string keys, and to take a private copy
! 323: ** of the key when a value is inserted. i.e. by a call similar to:
! 324: **
! 325: ** sqlite3Fts2HashInit(pHash, FTS2_HASH_STRING, 1);
! 326: **
! 327: ** This function adds a scalar function (see header comment above
! 328: ** scalarFunc() in this file for details) and, if ENABLE_TABLE is
! 329: ** defined at compilation time, a temporary virtual table (see header
! 330: ** comment above struct HashTableVtab) to the database schema. Both
! 331: ** provide read/write access to the contents of *pHash.
! 332: **
! 333: ** The third argument to this function, zName, is used as the name
! 334: ** of both the scalar and, if created, the virtual table.
! 335: */
! 336: int sqlite3Fts2InitHashTable(
! 337: sqlite3 *db,
! 338: fts2Hash *pHash,
! 339: const char *zName
! 340: ){
! 341: int rc = SQLITE_OK;
! 342: void *p = (void *)pHash;
! 343: const int any = SQLITE_ANY;
! 344: char *zTest = 0;
! 345: char *zTest2 = 0;
! 346:
! 347: #ifdef SQLITE_TEST
! 348: void *pdb = (void *)db;
! 349: zTest = sqlite3_mprintf("%s_test", zName);
! 350: zTest2 = sqlite3_mprintf("%s_internal_test", zName);
! 351: if( !zTest || !zTest2 ){
! 352: rc = SQLITE_NOMEM;
! 353: }
! 354: #endif
! 355:
! 356: if( rc!=SQLITE_OK
! 357: || (rc = sqlite3_create_function(db, zName, 1, any, p, scalarFunc, 0, 0))
! 358: || (rc = sqlite3_create_function(db, zName, 2, any, p, scalarFunc, 0, 0))
! 359: #ifdef SQLITE_TEST
! 360: || (rc = sqlite3_create_function(db, zTest, 2, any, p, testFunc, 0, 0))
! 361: || (rc = sqlite3_create_function(db, zTest, 3, any, p, testFunc, 0, 0))
! 362: || (rc = sqlite3_create_function(db, zTest2, 0, any, pdb, intTestFunc, 0, 0))
! 363: #endif
! 364: );
! 365:
! 366: sqlite3_free(zTest);
! 367: sqlite3_free(zTest2);
! 368: return rc;
! 369: }
! 370:
! 371: #endif /* !defined(SQLITE_CORE) || defined(SQLITE_ENABLE_FTS2) */
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>