[sqlite3]_[初级]_[使用正则表达式REGEXP查询]

场景

1.在一些数据库db文件中,随着版本更新可能会有不同的表名, 但是这些表名一般都符合某种规则, 所以适配不同版本的数据库表时, 最好的办法是动态获取表名.

2.或者在查询Text数据里需要匹配某个email,某个数值, 都可以用正则表达式.

说明

1.REGEXP 操作符是sqlite3的一个调用regexp()函数的特殊语法. 默认情况下没有定义这个regexp()用户函数, 所以在使用REGEXP操作符时会返回一个错误信息,

SQLITE_ERROR = 1

. 如果在运行时定义了一个regexp函数的话, 那么使用语法

X REGEXP Y

操作语句会以这样的形式调用 regexp(Y,X).

regexp(zPattern, zString)

2.用户定义的regexp(Y,X)函数使用

sqlite3_create_function

创建.

int sqlite3_create_function(
sqlite3 *db,
const char *zFunctionName,
int nArg,
int eTextRep,
void *pApp,
void (*xFunc)(sqlite3_context*,int,sqlite3_value**),
void (*xStep)(sqlite3_context*,int,sqlite3_value**),
void (*xFinal)(sqlite3_context*)
);

3.因为我们的regexp(Y,X)只有2个参数,所以nArg的值是2. 如果 xFunc,xStep,XFinal 三个都为NULL,会删除这个用户自定义函数.

db: sqlite3指针.
zFunctionName: "regexp"
nArg: 2
eTextRep: 指定regexp参数的文本编码,一般是 SQLITE_UTF8 | SQLITE_DETERMINISTIC,
SQLITE_DETERMINISTIC的意思确定性,当给同样的输入时,输出是一样的. 但是有些函数却不一样, 比如random()就不是确定性的.
pApp：用户自定义userdata, 通过 ``` void *sqlite3_user_data(sqlite3_context*) ``` 获取.
xFunc: 用户自定义的regexp函数指针.
xStep: NULL
xFinal: NULL

4.

void (*xFunc)(sqlite3_context*,int,sqlite3_value**)

函数定义里通过调用

void sqlite3_result_int(sqlite3_context*, int)

来实现返回值. 这里我们返回非0 为true, 0为false.

例子

// test-sqlite3-regex.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"

#include <Windows.h>
#include <memory>
#include <string>
#include <algorithm>
#include <regex>
#include <iostream>
#include <assert.h>

#include "sqlite3.h"

void Sqlite3Regexp(sqlite3_context* context, int argc, sqlite3_value** values) {

std::cout << (const char*)sqlite3_user_data(context) << std::endl;

int ret;
std::string reg = (char*)sqlite3_value_text(values[0]);
std::string text = (char*)sqlite3_value_text(values[1]);
std::transform(text.begin(),text.end(),text.begin(),::toupper);
std::transform(reg.begin(),reg.end(),reg.begin(),::toupper);

if ( argc != 2 || reg.size() == 0 || text.size() == 0) {
sqlite3_result_error(context, "SQL function regexp() called with invalid arguments.\n", -1);
return;
}
std::regex pattern(reg);
std::smatch color_match;
sqlite3_result_int(context, std::regex_search(text, color_match, pattern));
}

bool TestSqlite3Regexp(){
sqlite3* db = NULL;
sqlite3_stmt* query = NULL;
int ret = 0;

// 创建DB文件
// 如果不需要创建db文件,使用sqlite3_open即可
ret = sqlite3_open_v2("test.db",&db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL);
if (SQLITE_OK != ret)
return false;

std::shared_ptr<sqlite3> sp_db(db,[](sqlite3* db){
sqlite3_close(db);
});

// 创建表
std::string create = "CREATE TABLE IF NOT EXISTS X_12PHOTO (userid INTEGER PRIMARY KEY, ipaddr"
" TEXT,username TEXT,useradd TEXT,userphone INTEGER,age INTEGER, "
"time TEXT NOT NULL DEFAULT"
" (NOW()));";
// 注意,如果使用了sqlite3_stmt不关闭的话调用sqlite3_close是返回 SQLITE_BUSY 失败的.
sqlite3_stmt *stmt = NULL;
ret = sqlite3_prepare(db, create.c_str(), create.size(), &stmt, NULL);
if(SQLITE_OK != ret)
return false;

if (sqlite3_step(stmt) != SQLITE_DONE){
sqlite3_finalize(stmt);
return false;
}
sqlite3_finalize(stmt);

// 获取数据库匹配的表
static const char* sql = "SELECT name FROM sqlite_master WHERE type=\"table\" AND name REGEXP 'X_[0-9]+PHOTO'";
sqlite3_create_function(db, "regexp", 2, SQLITE_UTF8 | SQLITE_DETERMINISTIC,(void*)sql, &Sqlite3Regexp,NULL,NULL);
ret = sqlite3_prepare(db, sql, strlen(sql), &stmt, NULL);
if(SQLITE_OK != ret){
return false;
}

int flag = 0;
while(sqlite3_step(stmt) == SQLITE_ROW ){
auto table_sql = (const char*)sqlite3_column_text(stmt,0);
std::cout << table_sql << std::endl;
}
sqlite3_finalize(stmt);

sqlite3_create_function(db, "regexp", 2, SQLITE_UTF8 | SQLITE_DETERMINISTIC,0, NULL,NULL,NULL);
::DeleteFile(L"test.db");

return true;
}

int _tmain(int argc, _TCHAR* argv[])
{
assert(TestSqlite3Regexp());
system("pause");
return 0;
}

输出:

SELECT name FROM sqlite_master WHERE type="table" AND name REGEXP 'X_[0-9]+PHOTO'
X_12PHOTO

参考

1.lang_expr

2.create_function

3.sqlite-in-c-and-supporting-regexp

4.icu.c