SQLite 用户自定义函数举例

#include <stdio.h> #include <string.h> #include <stdlib.h> #include <time.h> #include <malloc.h> #include <assert.h> #include <sqlite3.h>#define SELECT_QUERY "SELECT host_id, geturl(desc) FROM host_table; void host2url_func( sqlite3_context * ctx, int argc, sqlite3_value ** argv ) { 　if ( argc != 1 ) return; 　char * httpurl = NULL; 　switch( sqlite3_value_type( argv[ 0 ] ) ) 　{ 　 case SQLITE_NULL: 　 { 　　sqlite3_result_text( ctx, "NULL", 4, SQLITE_STATIC ); 　　break; 　 } 　 case SQLITE_TEXT: 　 { 　　httpurl = ( char * ) malloc( strlen( sqlite3_value_text( argv[0] ) ) + 8 ); 　　if ( httpurl == 0 ) return; 　　sprintf( httpurl, "http://%s", sqlite3_value_text( argv[ 0 ] ) ); 　　sqlite3_result_text( ctx, httpurl, strlen( httpurl ), SQLITE_TRANSIENT ); 　　free( httpurl ); 　　httpurl = NULL; 　　break; 　 } 　 default: 　　sqlite3_result_text( ctx, "about:blank", 11, SQLITE_STATIC ); 　} } static int callback(void *NotUsed, int argc, char **argv, char **azColName) { 　 int i; 　 for(i=0; i <argc; i++){ 　　　printf( "%s = %s ", azColName[i], argv[i] ? argv[i] : "NULL"); 　 } 　 printf( " "); 　 return 0; } int main( int argc, char **argv ) { 　 sqlite3 *db; 　 char *zErrMsg = 0; 　 int rc; 　 if( argc!=3 ) 　 { 　 fprintf(stderr, "Usage: %s DATABASE SQL-STATEMENT ", argv[0]); 　 exit(1); 　 } 　 rc = sqlite3_open(argv[1], &db); 　 if( rc ){ 　 fprintf(stderr, "Can't open database: %s ", sqlite3_errmsg(db)); 　 sqlite3_close(db); 　 exit(1); 　 } 　 　 sqlite3_create_function( db, "geturl", 1, SQLITE_ANY, NULL, host2url_func, NULL, NULL ); 　 rc = sqlite3_exec(db, argv[2], callback, 0, &zErrMsg); 　 if( rc!=SQLITE_OK ){ 　 fprintf(stderr, "SQL error: %s ", zErrMsg); 　 } 　 sqlite3_close(db); 　 return 0; }