当前位置：首页 > news >正文

SQLiteC/C++接口详细介绍之sqlite3类（十五）

news 来源：原创 2025/5/13 23:25:08

返回目录：SQLite—免费开源数据库系列文章目录

上一篇：SQLiteC/C++接口详细介绍之sqlite3类（十四）

下一篇：SQLiteC/C++接口详细介绍之sqlite3类（十六）

47.sqlite3_set_authorizer

用法：在SQLite的访问控制中注册一个授权回调函数，用于限制或禁止某些SQL语句的执行。

函数原型如下：

int sqlite3_set_authorizer(
  sqlite3*,
  int (*xAuth)(void*,int,const char*,const char*,const char*,const char*),
  void* pUserData
);

- 第一个参数是数据库连接的指针
- 第二个参数是一个回调函数指针，该回调函数是访问授权的核心，用于审批或拒绝SQL语句的执行
- 第三个参数是用户带的指针，该指针可以在回调函数中使用

回调函数的原型如下：

int my_authorizer_callback(void* pArg, int eCode, const char* zDb, const char* zPtr1, const char* zPtr2, const char* zPtr3);

- 第一个参数是user-data指针
- 第二个参数是一个SQLITE_AUTH_*常量，指示某个SQL语句属于何种类型操作，比如是SELECT、INSERT等等。
- 后面的参数zDb，zPtr1，zPtr2，zPtr3指向与用户访问有关的数据库、表、列或者索引的名称。这四个参数不一定全部被用到，取决于具体的SQL语句类型。

返回值如下：

- 如果回调函数返回SQLITE_DENY，则表示不允许执行这条SQL语句。
- 如果回调函数返回SQLITE_IGNORE，则表示通过SQL语句的执行，但是不允许修改具体的数据库数据。
- 如果回调函数返回SQLITE_OK，则表示允许执行这条SQL语句。

下面是一个简单的示例，回调函数阻止DROP TABLE、DROP VIEW和DELETE语句的执行：

#include <sqlite3.h>
#include <stdio.h>
int authorizer_callback(void* pArg, int eCode, const char* zDb, const char* zPtr1, const char* zPtr2, const char* zPtr3) {
    if (eCode == SQLITE_DROP_TABLE || eCode == SQLITE_DROP_VIEW || eCode == SQLITE_DELETE) {
        printf("Unauthorized Operation\n");
        return SQLITE_DENY;
    }
    return SQLITE_OK;
}
int main() {
    sqlite3 *db;
    sqlite3_open(":memory:", &db);
    sqlite3_set_authorizer(db, authorizer_callback, NULL);
    sqlite3_exec(db, "CREATE TABLE test (id INTEGER PRIMARY KEY, value TEXT)", NULL, NULL, NULL);
    sqlite3_exec(db, "INSERT INTO test VALUES (1, 'hello')", NULL, NULL, NULL);
    sqlite3_exec(db, "DELETE FROM test WHERE id = 1", NULL, NULL, NULL);
    sqlite3_exec(db, "DROP TABLE test", NULL, NULL, NULL);
    sqlite3_close(db);
    return 0;
}

当执行DROP TABLE、DROP VIEW和DELETE语句时，回调函数将阻止执行，并输出"Unauthorized Operation"。需要注意的是，要在执行SQL语句之前设置授权回调函数，否则将无法生效。

48.sqlite3_set_clientdata

sqlite3_set_clientdata函数用于在SQLite的会话中为客户端数据设置指针，该指针可以在SQLite操作期间使用，并永久存储。这个指针通常被用于存储客户端上下文数据，比如在回调函数中访问。

函数原型如下：

void sqlite3_set_clientdata(sqlite3_context*, void*);

函数使用较为简单，第一个参数是指向sqlite3_context结构体的指针，这个指针通常由回调函数提供。第二个参数是一个void类型的指针，这个指针将与sqlite3_context结构体关联。

下面是一个示例，展示如何在sqlite3_create_function()回调函数中使用sqlite3_set_clientdata()来存储客户端上下文数据：

#include <sqlite3.h>
#include <stdio.h>
static void my_function(
    sqlite3_context *context,
    int argc,
    sqlite3_value **argv
) {
    int count = (int)(intptr_t)sqlite3_get_auxdata(context, 0);  // 从上下文中获取计数器
    count++;                                                      // 计数器加1
    sqlite3_set_auxdata(context, 0, (void*)(intptr_t)count);      // 将计数器放回上下文中
    sqlite3_result_int(context, count);                           // 返回计数器值
}
int main() {
    sqlite3 *db;
    sqlite3_open(":memory:", &db);
    sqlite3_create_function(db, "my_function", -1, SQLITE_ANY, NULL, my_function, NULL, NULL);
    void *pUserData = malloc(sizeof(int));
    *(int*)pUserData = 0;
    sqlite3_set_auxdata(db, 0, pUserData);  // 将计数器存储在数据库连接中
    sqlite3_context *context = sqlite3_malloc(sizeof(sqlite3_context));
    sqlite3_set_clientdata(context, db);   // 将数据库连接与上下文关联
    sqlite3_exec(db, "SELECT my_function()", NULL, NULL, NULL);
    sqlite3_exec(db, "SELECT my_function()", NULL, NULL, NULL);
    int count = (int)(intptr_t)sqlite3_get_auxdata(db, 0);
    printf("function called %d times\n", count);
    sqlite3_close(db);
    return 0;
}

该示例通过sqlite3_set_auxdata()函数在数据库连接中存储一个计数器。每次调用my_function()回调函数时，它将自加该计数器并返回该计数器。用sqlite3_set_clientdata()函数将上下文与数据库连接关联。最后，从数据库连接中获取计数器并输出结果。

49.sqlite3_set_last_insert_rowid

sqlite3_set_last_insert_rowid函数用于手动设置最后插入的行ID。通常情况下，SQLite会自动维护最后插入的行ID，无需手动设置。但是，有时候可能需要在想要插入的ID不是递增的情况下手动指定插入行ID。

函数原型如下：

void sqlite3_set_last_insert_rowid(sqlite3*, sqlite3_int64);

第一个参数是指向sqlite3结构体的指针，第二个参数是一个sqlite3_int64类型的整数，代表要设置的最后插入的行ID。

下面是一个示例，展示如何手动设置最后插入的行ID：

#include <sqlite3.h>
#include <stdio.h>
int main() {
    sqlite3 *db;
    sqlite3_open(":memory:", &db);
    sqlite3_exec(db, "CREATE TABLE test (id INTEGER PRIMARY KEY, value TEXT)", NULL, NULL, NULL);
    sqlite3_int64 rowid = 10;
    sqlite3_set_last_insert_rowid(db, rowid);   // 设置最后插入的行ID
    sqlite3_exec(db, "INSERT INTO test (id, value) VALUES (11, 'hello')", NULL, NULL, NULL);
    rowid = sqlite3_last_insert_rowid(db);      // 获取最后插入的行ID
    printf("last_insert_rowid = %lld\n", rowid);
    sqlite3_close(db);
    return 0;
}

在该示例中，我们使用sqlite3_set_last_insert_rowid()函数手动设置了最后插入的行ID为10，然后插入一条ID为11的数据。最后，我们获取最后插入的行ID并将其输出。输出结果为：last_insert_rowid = 10。