4 ** The author disclaims copyright to this source code. In place of
5 ** a legal notice, here is a blessing:
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.
11 *************************************************************************
12 ** Code for testing the virtual table xBestIndex method and the query
20 ** This module exports a single tcl command - [register_tcl_module]. When
21 ** invoked, it registers a special virtual table module with a database
24 ** The virtual table is currently read-only. And always returns zero rows.
25 ** It is created with a single argument - the name of a Tcl command - as
28 ** CREATE VIRTUAL TABLE x1 USING tcl(tcl_command);
30 ** The command [tcl_command] is invoked when the table is first created (or
31 ** connected), when the xBestIndex() method is invoked and when the xFilter()
32 ** method is called. When it is created (or connected), it is invoked as
35 ** tcl_command xConnect
37 ** In this case the return value of the script is passed to the
38 ** sqlite3_declare_vtab() function to create the virtual table schema.
40 ** When the xBestIndex() method is called by SQLite, the Tcl command is
43 ** tcl_command xBestIndex CONSTRAINTS ORDERBY MASK
45 ** where CONSTRAINTS is a tcl representation of the aConstraints[] array,
46 ** ORDERBY is a representation of the contents of the aOrderBy[] array and
47 ** MASK is a copy of sqlite3_index_info.colUsed. For example if the virtual
48 ** table is declared as:
50 ** CREATE TABLE x1(a, b, c)
54 ** SELECT * FROM x1 WHERE a=? AND c<? ORDER BY b, c;
56 ** then the Tcl command is:
58 ** tcl_command xBestIndex \
59 ** {{op eq column 0 usable 1} {op lt column 2 usable 1}} \
60 ** {{column 1 desc 0} {column 2 desc 0}} \
63 ** The return value of the script is a list of key-value pairs used to
64 ** populate the output fields of the sqlite3_index_info structure. Possible
65 ** keys and the usage of the accompanying values are:
67 ** "orderby" (value of orderByConsumed flag)
68 ** "cost" (value of estimatedCost field)
69 ** "rows" (value of estimatedRows field)
70 ** "use" (index of used constraint in aConstraint[])
71 ** "omit" (like "use", but also sets omit flag)
72 ** "idxnum" (value of idxNum field)
73 ** "idxstr" (value of idxStr field)
75 ** Refer to code below for further details.
77 ** When SQLite calls the xFilter() method, this module invokes the following
80 ** tcl_command xFilter IDXNUM IDXSTR ARGLIST
82 ** IDXNUM and IDXSTR are the values of the idxNum and idxStr parameters
83 ** passed to xFilter. ARGLIST is a Tcl list containing each of the arguments
84 ** passed to xFilter in text form.
86 ** As with xBestIndex(), the return value of the script is interpreted as a
87 ** list of key-value pairs. There is currently only one key defined - "sql".
88 ** The value must be the full text of an SQL statement that returns the data
89 ** for the current scan. The leftmost column returned by the SELECT is assumed
90 ** to contain the rowid. Other columns must follow, in order from left to
95 #include "sqliteInt.h"
96 #if defined(INCLUDE_SQLITE_TCL_H)
97 # include "sqlite_tcl.h"
102 #ifndef SQLITE_OMIT_VIRTUALTABLE
104 typedef struct tcl_vtab tcl_vtab
;
105 typedef struct tcl_cursor tcl_cursor
;
108 ** A fs virtual-table object
117 /* A tcl cursor object */
119 sqlite3_vtab_cursor base
;
120 sqlite3_stmt
*pStmt
; /* Read data from here */
124 ** Dequote string z in place.
126 static void tclDequote(char *z
){
129 /* Set stack variable q to the close-quote character */
130 if( q
=='[' || q
=='\'' || q
=='"' || q
=='`' ){
133 if( q
=='[' ) q
= ']';
135 while( ALWAYS(z
[iIn
]) ){
138 /* Character iIn was the close quote. */
142 /* Character iIn and iIn+1 form an escaped quote character. Skip
143 ** the input cursor past both and copy a single quote character
144 ** to the output buffer. */
149 z
[iOut
++] = z
[iIn
++];
158 ** This function is the implementation of both the xConnect and xCreate
159 ** methods of the fs virtual table.
161 ** The argv[] array contains the following:
163 ** argv[0] -> module name ("fs")
164 ** argv[1] -> database name
165 ** argv[2] -> table name
166 ** argv[...] -> other module argument fields.
168 static int tclConnect(
171 int argc
, const char *const*argv
,
172 sqlite3_vtab
**ppVtab
,
175 Tcl_Interp
*interp
= (Tcl_Interp
*)pAux
;
178 Tcl_Obj
*pScript
= 0;
182 *pzErr
= sqlite3_mprintf("wrong number of arguments");
186 zCmd
= sqlite3_malloc64(strlen(argv
[3])+1);
187 pTab
= (tcl_vtab
*)sqlite3_malloc64(sizeof(tcl_vtab
));
189 memcpy(zCmd
, argv
[3], strlen(argv
[3])+1);
191 memset(pTab
, 0, sizeof(tcl_vtab
));
193 pTab
->pCmd
= Tcl_NewStringObj(zCmd
, -1);
194 pTab
->interp
= interp
;
196 Tcl_IncrRefCount(pTab
->pCmd
);
198 pScript
= Tcl_DuplicateObj(pTab
->pCmd
);
199 Tcl_IncrRefCount(pScript
);
200 Tcl_ListObjAppendElement(interp
, pScript
, Tcl_NewStringObj("xConnect", -1));
202 rc
= Tcl_EvalObjEx(interp
, pScript
, TCL_EVAL_GLOBAL
);
204 *pzErr
= sqlite3_mprintf("%s", Tcl_GetStringResult(interp
));
207 rc
= sqlite3_declare_vtab(db
, Tcl_GetStringResult(interp
));
219 *ppVtab
= &pTab
->base
;
223 /* The xDisconnect and xDestroy methods are also the same */
224 static int tclDisconnect(sqlite3_vtab
*pVtab
){
225 tcl_vtab
*pTab
= (tcl_vtab
*)pVtab
;
226 Tcl_DecrRefCount(pTab
->pCmd
);
232 ** Open a new tcl cursor.
234 static int tclOpen(sqlite3_vtab
*pVTab
, sqlite3_vtab_cursor
**ppCursor
){
236 pCur
= sqlite3_malloc(sizeof(tcl_cursor
));
237 if( pCur
==0 ) return SQLITE_NOMEM
;
238 memset(pCur
, 0, sizeof(tcl_cursor
));
239 *ppCursor
= &pCur
->base
;
244 ** Close a tcl cursor.
246 static int tclClose(sqlite3_vtab_cursor
*cur
){
247 tcl_cursor
*pCur
= (tcl_cursor
*)cur
;
249 sqlite3_finalize(pCur
->pStmt
);
255 static int tclNext(sqlite3_vtab_cursor
*pVtabCursor
){
256 tcl_cursor
*pCsr
= (tcl_cursor
*)pVtabCursor
;
258 tcl_vtab
*pTab
= (tcl_vtab
*)(pVtabCursor
->pVtab
);
259 int rc
= sqlite3_step(pCsr
->pStmt
);
260 if( rc
!=SQLITE_ROW
){
262 rc
= sqlite3_finalize(pCsr
->pStmt
);
265 zErr
= sqlite3_errmsg(pTab
->db
);
266 pTab
->base
.zErrMsg
= sqlite3_mprintf("%s", zErr
);
273 static int tclFilter(
274 sqlite3_vtab_cursor
*pVtabCursor
,
275 int idxNum
, const char *idxStr
,
276 int argc
, sqlite3_value
**argv
278 tcl_cursor
*pCsr
= (tcl_cursor
*)pVtabCursor
;
279 tcl_vtab
*pTab
= (tcl_vtab
*)(pVtabCursor
->pVtab
);
280 Tcl_Interp
*interp
= pTab
->interp
;
286 pScript
= Tcl_DuplicateObj(pTab
->pCmd
);
287 Tcl_IncrRefCount(pScript
);
288 Tcl_ListObjAppendElement(interp
, pScript
, Tcl_NewStringObj("xFilter", -1));
289 Tcl_ListObjAppendElement(interp
, pScript
, Tcl_NewIntObj(idxNum
));
291 Tcl_ListObjAppendElement(interp
, pScript
, Tcl_NewStringObj(idxStr
, -1));
293 Tcl_ListObjAppendElement(interp
, pScript
, Tcl_NewStringObj("", -1));
297 Tcl_IncrRefCount(pArg
);
298 for(ii
=0; ii
<argc
; ii
++){
299 const char *zVal
= (const char*)sqlite3_value_text(argv
[ii
]);
304 pVal
= Tcl_NewStringObj(zVal
, -1);
306 Tcl_ListObjAppendElement(interp
, pArg
, pVal
);
308 Tcl_ListObjAppendElement(interp
, pScript
, pArg
);
309 Tcl_DecrRefCount(pArg
);
311 rc
= Tcl_EvalObjEx(interp
, pScript
, TCL_EVAL_GLOBAL
);
313 const char *zErr
= Tcl_GetStringResult(interp
);
315 pTab
->base
.zErrMsg
= sqlite3_mprintf("%s", zErr
);
317 /* Analyze the scripts return value. The return value should be a tcl
318 ** list object with an even number of elements. The first element of each
319 ** pair must be one of:
321 ** "sql" (SQL statement to return data)
323 Tcl_Obj
*pRes
= Tcl_GetObjResult(interp
);
324 Tcl_Obj
**apElem
= 0;
326 rc
= Tcl_ListObjGetElements(interp
, pRes
, &nElem
, &apElem
);
328 const char *zErr
= Tcl_GetStringResult(interp
);
330 pTab
->base
.zErrMsg
= sqlite3_mprintf("%s", zErr
);
332 for(ii
=0; rc
==SQLITE_OK
&& ii
<nElem
; ii
+=2){
333 const char *zCmd
= Tcl_GetString(apElem
[ii
]);
334 Tcl_Obj
*p
= apElem
[ii
+1];
335 if( sqlite3_stricmp("sql", zCmd
)==0 ){
336 const char *zSql
= Tcl_GetString(p
);
337 rc
= sqlite3_prepare_v2(pTab
->db
, zSql
, -1, &pCsr
->pStmt
, 0);
339 const char *zErr
= sqlite3_errmsg(pTab
->db
);
340 pTab
->base
.zErrMsg
= sqlite3_mprintf("unexpected: %s", zErr
);
344 pTab
->base
.zErrMsg
= sqlite3_mprintf("unexpected: %s", zCmd
);
351 rc
= tclNext(pVtabCursor
);
356 static int tclColumn(
357 sqlite3_vtab_cursor
*pVtabCursor
,
358 sqlite3_context
*ctx
,
361 tcl_cursor
*pCsr
= (tcl_cursor
*)pVtabCursor
;
362 sqlite3_result_value(ctx
, sqlite3_column_value(pCsr
->pStmt
, i
+1));
366 static int tclRowid(sqlite3_vtab_cursor
*pVtabCursor
, sqlite_int64
*pRowid
){
367 tcl_cursor
*pCsr
= (tcl_cursor
*)pVtabCursor
;
368 *pRowid
= sqlite3_column_int64(pCsr
->pStmt
, 0);
372 static int tclEof(sqlite3_vtab_cursor
*pVtabCursor
){
373 tcl_cursor
*pCsr
= (tcl_cursor
*)pVtabCursor
;
374 return (pCsr
->pStmt
==0);
377 static int tclBestIndex(sqlite3_vtab
*tab
, sqlite3_index_info
*pIdxInfo
){
378 tcl_vtab
*pTab
= (tcl_vtab
*)tab
;
379 Tcl_Interp
*interp
= pTab
->interp
;
385 pScript
= Tcl_DuplicateObj(pTab
->pCmd
);
386 Tcl_IncrRefCount(pScript
);
387 Tcl_ListObjAppendElement(interp
, pScript
, Tcl_NewStringObj("xBestIndex", -1));
390 Tcl_IncrRefCount(pArg
);
391 for(ii
=0; ii
<pIdxInfo
->nConstraint
; ii
++){
392 struct sqlite3_index_constraint
const *pCons
= &pIdxInfo
->aConstraint
[ii
];
393 Tcl_Obj
*pElem
= Tcl_NewObj();
394 const char *zOp
= "?";
396 Tcl_IncrRefCount(pElem
);
399 case SQLITE_INDEX_CONSTRAINT_EQ
:
401 case SQLITE_INDEX_CONSTRAINT_GT
:
403 case SQLITE_INDEX_CONSTRAINT_LE
:
405 case SQLITE_INDEX_CONSTRAINT_LT
:
407 case SQLITE_INDEX_CONSTRAINT_GE
:
409 case SQLITE_INDEX_CONSTRAINT_MATCH
:
410 zOp
= "match"; break;
411 case SQLITE_INDEX_CONSTRAINT_LIKE
:
413 case SQLITE_INDEX_CONSTRAINT_GLOB
:
415 case SQLITE_INDEX_CONSTRAINT_REGEXP
:
416 zOp
= "regexp"; break;
417 case SQLITE_INDEX_CONSTRAINT_NE
:
419 case SQLITE_INDEX_CONSTRAINT_ISNOT
:
420 zOp
= "isnot"; break;
421 case SQLITE_INDEX_CONSTRAINT_ISNOTNULL
:
422 zOp
= "isnotnull"; break;
423 case SQLITE_INDEX_CONSTRAINT_ISNULL
:
424 zOp
= "isnull"; break;
425 case SQLITE_INDEX_CONSTRAINT_IS
:
429 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewStringObj("op", -1));
430 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewStringObj(zOp
, -1));
431 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewStringObj("column", -1));
432 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewIntObj(pCons
->iColumn
));
433 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewStringObj("usable", -1));
434 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewIntObj(pCons
->usable
));
436 Tcl_ListObjAppendElement(0, pArg
, pElem
);
437 Tcl_DecrRefCount(pElem
);
440 Tcl_ListObjAppendElement(0, pScript
, pArg
);
441 Tcl_DecrRefCount(pArg
);
444 Tcl_IncrRefCount(pArg
);
445 for(ii
=0; ii
<pIdxInfo
->nOrderBy
; ii
++){
446 struct sqlite3_index_orderby
const *pOrder
= &pIdxInfo
->aOrderBy
[ii
];
447 Tcl_Obj
*pElem
= Tcl_NewObj();
448 Tcl_IncrRefCount(pElem
);
450 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewStringObj("column", -1));
451 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewIntObj(pOrder
->iColumn
));
452 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewStringObj("desc", -1));
453 Tcl_ListObjAppendElement(0, pElem
, Tcl_NewIntObj(pOrder
->desc
));
455 Tcl_ListObjAppendElement(0, pArg
, pElem
);
456 Tcl_DecrRefCount(pElem
);
459 Tcl_ListObjAppendElement(0, pScript
, pArg
);
460 Tcl_DecrRefCount(pArg
);
462 Tcl_ListObjAppendElement(0, pScript
, Tcl_NewWideIntObj(pIdxInfo
->colUsed
));
464 rc
= Tcl_EvalObjEx(interp
, pScript
, TCL_EVAL_GLOBAL
);
465 Tcl_DecrRefCount(pScript
);
467 const char *zErr
= Tcl_GetStringResult(interp
);
469 pTab
->base
.zErrMsg
= sqlite3_mprintf("%s", zErr
);
471 /* Analyze the scripts return value. The return value should be a tcl
472 ** list object with an even number of elements. The first element of each
473 ** pair must be one of:
475 ** "orderby" (value of orderByConsumed flag)
476 ** "cost" (value of estimatedCost field)
477 ** "rows" (value of estimatedRows field)
478 ** "use" (index of used constraint in aConstraint[])
479 ** "idxnum" (value of idxNum field)
480 ** "idxstr" (value of idxStr field)
481 ** "omit" (index of omitted constraint in aConstraint[])
483 Tcl_Obj
*pRes
= Tcl_GetObjResult(interp
);
484 Tcl_Obj
**apElem
= 0;
486 rc
= Tcl_ListObjGetElements(interp
, pRes
, &nElem
, &apElem
);
488 const char *zErr
= Tcl_GetStringResult(interp
);
490 pTab
->base
.zErrMsg
= sqlite3_mprintf("%s", zErr
);
493 for(ii
=0; rc
==SQLITE_OK
&& ii
<nElem
; ii
+=2){
494 const char *zCmd
= Tcl_GetString(apElem
[ii
]);
495 Tcl_Obj
*p
= apElem
[ii
+1];
496 if( sqlite3_stricmp("cost", zCmd
)==0 ){
497 rc
= Tcl_GetDoubleFromObj(interp
, p
, &pIdxInfo
->estimatedCost
);
499 if( sqlite3_stricmp("orderby", zCmd
)==0 ){
500 rc
= Tcl_GetIntFromObj(interp
, p
, &pIdxInfo
->orderByConsumed
);
502 if( sqlite3_stricmp("idxnum", zCmd
)==0 ){
503 rc
= Tcl_GetIntFromObj(interp
, p
, &pIdxInfo
->idxNum
);
505 if( sqlite3_stricmp("idxstr", zCmd
)==0 ){
506 sqlite3_free(pIdxInfo
->idxStr
);
507 pIdxInfo
->idxStr
= sqlite3_mprintf("%s", Tcl_GetString(p
));
508 pIdxInfo
->needToFreeIdxStr
= 1;
510 if( sqlite3_stricmp("rows", zCmd
)==0 ){
512 rc
= Tcl_GetWideIntFromObj(interp
, p
, &x
);
513 pIdxInfo
->estimatedRows
= (tRowcnt
)x
;
515 if( sqlite3_stricmp("use", zCmd
)==0
516 || sqlite3_stricmp("omit", zCmd
)==0
519 rc
= Tcl_GetIntFromObj(interp
, p
, &iCons
);
521 if( iCons
<0 || iCons
>=pIdxInfo
->nConstraint
){
523 pTab
->base
.zErrMsg
= sqlite3_mprintf("unexpected: %d", iCons
);
525 int bOmit
= (zCmd
[0]=='o' || zCmd
[0]=='O');
526 pIdxInfo
->aConstraintUsage
[iCons
].argvIndex
= iArgv
++;
527 pIdxInfo
->aConstraintUsage
[iCons
].omit
= bOmit
;
532 pTab
->base
.zErrMsg
= sqlite3_mprintf("unexpected: %s", zCmd
);
534 if( rc
!=SQLITE_OK
&& pTab
->base
.zErrMsg
==0 ){
535 const char *zErr
= Tcl_GetStringResult(interp
);
536 pTab
->base
.zErrMsg
= sqlite3_mprintf("%s", zErr
);
546 ** A virtual table module that provides read-only access to a
547 ** Tcl global variable namespace.
549 static sqlite3_module tclModule
= {
556 tclOpen
, /* xOpen - open a cursor */
557 tclClose
, /* xClose - close a cursor */
558 tclFilter
, /* xFilter - configure scan constraints */
559 tclNext
, /* xNext - advance a cursor */
560 tclEof
, /* xEof - check for end of scan */
561 tclColumn
, /* xColumn - read data */
562 tclRowid
, /* xRowid - read data */
573 ** Decode a pointer to an sqlite3 object.
575 extern int getDbPointer(Tcl_Interp
*interp
, const char *zA
, sqlite3
**ppDb
);
578 ** Register the echo virtual table module.
580 static int SQLITE_TCLAPI
register_tcl_module(
581 ClientData clientData
, /* Pointer to sqlite3_enable_XXX function */
582 Tcl_Interp
*interp
, /* The TCL interpreter that invoked this command */
583 int objc
, /* Number of arguments */
584 Tcl_Obj
*CONST objv
[] /* Command arguments */
588 Tcl_WrongNumArgs(interp
, 1, objv
, "DB");
591 if( getDbPointer(interp
, Tcl_GetString(objv
[1]), &db
) ) return TCL_ERROR
;
592 #ifndef SQLITE_OMIT_VIRTUALTABLE
593 sqlite3_create_module(db
, "tcl", &tclModule
, (void *)interp
);
602 ** Register commands with the TCL interpreter.
604 int Sqlitetesttcl_Init(Tcl_Interp
*interp
){
605 #ifndef SQLITE_OMIT_VIRTUALTABLE
608 Tcl_ObjCmdProc
*xProc
;
611 { "register_tcl_module", register_tcl_module
, 0 },
614 for(i
=0; i
<sizeof(aObjCmd
)/sizeof(aObjCmd
[0]); i
++){
615 Tcl_CreateObjCommand(interp
, aObjCmd
[i
].zName
,
616 aObjCmd
[i
].xProc
, aObjCmd
[i
].clientData
, 0);