织梦CMS - 轻松建站从此开始!

罗索

SQLite 用户自定义函数举例

落鹤生 发布于 2012-02-14 10:53 点击:次 
SQLite 用户自定义函数示例实现代码
TAG:

#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\n", azColName[i], argv[i] ? argv[i] : "NULL");
   }
   printf("\n");
   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\n", argv[0]);
   exit(1);
   }
   rc = sqlite3_open(argv[1], &db);
   if( rc ){
   fprintf(stderr, "Can't open database: %s\n", 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\n", zErrMsg);
   }

   sqlite3_close(db);
   return 0;

}

(cr0_3)
本站文章除注明转载外,均为本站原创或编译欢迎任何形式的转载,但请务必注明出处,尊重他人劳动,同学习共成长。转载请注明:文章转载自:罗索实验室 [http://www.rosoo.net/a/201202/15701.html]
本文出处:百度博客 作者:cr0_3 原文
顶一下
(0)
0%
踩一下
(0)
0%
------分隔线----------------------------
发表评论
请自觉遵守互联网相关的政策法规,严禁发布色情、暴力、反动的言论。
评价:
表情:
用户名: 验证码:点击我更换图片
栏目列表
将本文分享到微信
织梦二维码生成器
推荐内容