sqlite自定义函数，zlib库压缩数据

 SQLite 内置函数比较有限，有时通过添加自定义函数（User-Defined Fuctions）的方式可以实现一些通过普通 SQL 操作无法实现或者实现很麻烦的功能；也可以替换 SQLite 原有的内置函数，使其符合我们的要求。

本文以 自定义 数据压缩，解压函数（采用zlib数据库压缩数据）为例讲解 sqlite自定义函数。

1. 添加自定义接口函数

1.1 头文件加入自定义静态函数（源代码函数是加在 类 里面的）

static void sqlite_compress(sqlite3_context *context, int argc, sqlite3_value **argv);

static void sqlite_decompress(sqlite3_context *context, int argc, sqlite3_value **argv);

1.2 实现函数

压缩函数实现：

void SQLiteWrapper::sqlite_compress(sqlite3_context *context, int argc, sqlite3_value **argv)
{
    const char *sourceData = (const char *)sqlite3_value_text(argv[0]);   //传递来的 需要处理的原数据
    fprintf(stderr,"the source data size == %d\n",sqlite3_value_bytes(argv[0]));
    uLong sourceDataLen = sqlite3_value_bytes(argv[0])+1;
    uLongf compressLen = 524288;

    Bytef destData[compressLen];
    memset(destData,0,compressLen);
    int rc = compress(destData,&compressLen,(const Bytef*)sourceData,sourceDataLen);   //调用zlib库压缩数据
    if(rc == Z_OK){
        fprintf(stderr,"compress data is succedd!!!!!!!compress after size=%d\n",compressLen);
        sqlite3_result_text(context,(const char*)destData,compressLen, SQLITE_TRANSIENT);  //返回压缩后的数据
    }else{
        fprintf(stderr,"compress data is faile!!!!!!!!!!!!the error value is == %d\n",rc);
        fprintf(stderr,"compress data is == %s\n",sourceData);
        exit(0);
    }

}

解压函数实现：

void SQLiteWrapper::sqlite_decompress(sqlite3_context *context, int argc, sqlite3_value **argv)
{
    char *result = (char *)sqlite3_value_text(argv[0]);    //获取需要处理的数据
    fprintf(stderr,"read database data ,data size =====%d\n",sqlite3_value_bytes(argv[0]));
    uLong sourceDataLen = sqlite3_value_bytes(argv[0]);
    uLongf uncompressLen = 524288;
    Bytef destData[uncompressLen];
    memset(destData,0,uncompressLen);
    int rc = uncompress(destData,&uncompressLen,(const Bytef*)result,sourceDataLen); //解压数据
    std::string ss = (char *)destData;
    if(rc == Z_OK){
        fprintf(stderr,"uncompress data is succedd!!!!!!!uncompress data size==%d\n",uncompressLen);
        sqlite3_result_text(context,ss.c_str(),ss.size(), SQLITE_TRANSIENT);   //返回解压后的数据
    }else{
        fprintf(stderr,"uncompress data is faile!!!!!!!!!!!!the error value is == %d\n",rc);
        exit(0);
    }

}

2.把自定义函数注册到sqlite里面，成为sqlite内置函数

int SQLiteWrapper::open(const std::string &dbpath)
{
    close();
    int rc = sqlite3_open(dbpath.c_str(), &db);
    if (rc) {
        fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
        close();
    }else{
        fprintf(stderr, "database open is successed!\n");
        sqlite3_create_function(db, "compressdata", -1, SQLITE_UTF8, NULL, &sqlite_compress, NULL, NULL);
        sqlite3_create_function(db, "decompressdata", -1, SQLITE_UTF8, NULL,&sqlite_decompress, NULL, NULL);
    }

    return rc;

}

函数 compressdata，，decompressdata，，就是注册到sqlite的函数，成为sqlite3的内置函数，可以直接调用，，

功能相当于自定义的sqlite_compress，，sqlite_decompress函数功能

3.自定义函数在sqlite中应用

压缩函数调用：

void SQLiteWrapper::insertCompressTabData(const std::string &sourceData,int idValue)
{
    int len = sourceData.size()+1;
    fprintf(stderr, "%s ,,the source data size ==== %d\n",__FUNCTION__,len);

    string query_info = "INSERT INTO translation(id,content) VALUES(?,compressdata(?))";
    sqlite3_stmt *pStmt;
    int rc = sqlite3_prepare(db, query_info.c_str(), -1, &pStmt, 0);
    sqlite3_exec(db, "BEGIN", 0, 0, 0);
    if (rc ) {
        fprintf(stderr, "Error:sqlite3_prepare ====== %s\n", sqlite3_errmsg(db));
        sqlite3_finalize(pStmt);
        return ;
    }
    sqlite3_bind_int(pStmt,1,idValue);
    sqlite3_bind_text(pStmt,2,sourceData.c_str(),-1, SQLITE_STATIC);

    int rr;
    rr = sqlite3_step(pStmt);
    if( rr == SQLITE_OK ){
        fprintf(stderr,"insert table translation Error: %s\n", sqlite3_errmsg(db));
    }
    sqlite3_reset(pStmt);
    sqlite3_finalize(pStmt);

    sqlite3_exec(db,"COMMIT",0,0,0);

}

在执行插入 语句 是传递给变量的数据 直接压缩了，存储到数据库的数据就 压缩后的数据

解压函数使用：

void SQLiteWrapper::selectCompressTab(int id)
{
    strstream stream;
    string intToString;
    stream << id;
    stream >> intToString;

    string query = "select decompressdata(content) from translation where id=%;";
    int pos = query.rfind('%');
    query.replace(pos,1,intToString);
    fprintf(stderr,"select table of translation ,,the execute query = %s\n",query.c_str());
    executeQuery(query);

}

从数据读取的压缩数据直接解压了，，得到的搜索数据就是解压后的数据。