C++封装sqlite3库

最新推荐文章于 2024-06-13 13:04:41 发布
最新推荐文章于 2024-06-13 13:04:41 发布
阅读量1.1k 收藏 16
点赞数
分类专栏： sqlite3 文章标签： sqlite3 c++
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_40468982/article/details/115741106
版权

SQLite C++ 数据库封装 SQL注入 字符编码转换
关键词由CSDN通过智能技术生成 
#ifndef _SMART_DB_H
#define _SMART_DB_H

#include "sqlite3.h"
#include <string>
#include <iostream>
#include <vector>

/*
* function：封装的sqlite类，具备链接，查询，关闭功能
* des：sql用字符串拼接方式，会有sql注入的风险
*/
class SmartDB
{
public:

    SmartDB();

    /*
    * @function: 创建或打开数据库,如果数据库不存在，
    * 则数据库将被创建并打开，如果创建失败则设置失败标志
    * @param[in] fileName: 数据库文件的位置
    */
    SmartDB(const std::string &fileName);

    /*
    * @des: 析构函数, 释放资源，关闭数据库
    */
    ~SmartDB();

    /*
    * @function: 打开数据库
    * @param[in] fileName: 数据库文件的位置
    * @return bool: 成功打开数据库true, 否则false
    */
    bool Open(const std::string &fileName);

    /*
    * @function: 释放资源，关闭数据库
    * @return bool: 成功关闭数据库true, 否则false
    */
    bool Close();

    /*
    * @function: 得到最后一次错误码
    * @return int: 返回最近一次错误码
    */
    int GetLastErrorCode();

    /*
    * @function: 解析和保存sql语句
    * @des: 不许用拼接字符串，因为有sql注入的安全问题
    * @param[in] strSql: 需要执行的sql语句, 可能带占位符?
    * @return bool: 成功返回true, 否则false
    */
    bool Prepare(const std::string& strSql);

    /*
    * @function: 绑定SQL语句预编译时把一些参数使用占位符（这里是?号）来代替
    * @param[in] current: 占位符的索引
    * @param[in] strData: 占位符问号的具体内容
    */
    void BindValue(int current, const std::string &strData);


    /*
    * @function: 执行sql, 返回函数执行的一个值，执行简单的汇聚函数，如select cout(*), select max(*)等
    * @param[in] strSql: sql语句，有可能占位符"?"
    * @param[in] strArgs: 参数列表，用来填充占位符, 变参
    * @param[out] vecData: 查询出来的全部数据
    * @return int: 最近一次错误码
    */
    int ExecuteScalar(const std::string &strSql, std::vector<std::vector<std::string>>& vecData, const int num, const std::string strFirst, ...);

private:
    /*
    * @function: 将单行所有列数据插入vector中
    * @param[in] iCount: 单行列数
    * @param[out] vecData: 单行数据集
    */
    void InsertLine(std::vector<std::string>& vecData, const int& iCount);

    /*
    * @function: 关闭sqlite句柄
    * @return int: 返回最近一次错误码
    */
    int CloseDBHandle();

private:
    sqlite3 *m_dbHandle = NULL;         // sqlite3数据库句柄
    sqlite3_stmt *m_statement = NULL;   // 一条SQL语句的实例
    int m_code;                         // 记录最近一次的错误码
};

#endif
#include "smartdb.h"
#include <stdarg.h>

SmartDB::SmartDB()
{

}

/*
* @function: 创建或打开数据库,如果数据库不存在，
* 则数据库将被创建并打开，如果创建失败则设置失败标志
* @param[in] fileName: 数据库文件的位置
*/
SmartDB::SmartDB(const std::string &fileName)
{
    this->Open(fileName);
}

/*
* @des: 析构函数, 释放资源，关闭数据库
*/
SmartDB::~SmartDB()
{
    this->Close();
}

/*
* @function: 打开数据库
* @param[in] fileName: 数据库文件的位置
* @return bool: 成功打开数据库true, 否则false
*/
bool SmartDB::Open(const std::string &fileName)
{
    // 打开对应fileName的sqlie数据库
    this->m_code = sqlite3_open(fileName.data(), &m_dbHandle);
    return (SQLITE_OK == this->m_code);     // 是否正常打开
}

/*
* @function: 关闭sqlite句柄
* @return int: 返回最近一次错误码
*/
int SmartDB::CloseDBHandle()
{
    this->m_code = sqlite3_close(this->m_dbHandle);

    while (this->m_code == SQLITE_BUSY)     // 如果sqlite状态繁忙
    {
        this->m_code = SQLITE_OK;
        // 循环取得sql语句实例
        sqlite3_stmt *stmt = sqlite3_next_stmt(this->m_dbHandle, NULL);
        if (stmt == NULL)
        {
            break;
        }

        // 销毁sql语句实例，防止内存泄漏
        this->m_code = sqlite3_finalize(stmt);
        if (this->m_code == SQLITE_OK)          // 成功销毁
        {
            // 关闭sqlite句柄
            this->m_code = sqlite3_close(this->m_dbHandle);
        }
    }
    return this->m_code;
}

/*
* @function: 释放资源，关闭数据库
* @return bool: 成功关闭数据库true, 否则false
*/
bool SmartDB::Close()
{
    if (this->m_dbHandle == NULL)
    {
        return true;
    }

    sqlite3_finalize(this->m_statement);    // 销毁sql语句实例，防止内存泄漏
    this->m_code = this->CloseDBHandle();         // 关闭sqlite句柄
    this->m_statement = nullptr;
    this->m_dbHandle = nullptr;
    return (SQLITE_OK == this->m_code);     // 是否正常关闭
}

/*
* @function: 得到最后一次错误码
* @return int: 返回最近一次错误码
*/
int SmartDB::GetLastErrorCode()
{
    return this->m_code;
}


/*
* @function: 解析和保存sql语句
* @des: 不许用拼接字符串，因为有sql注入的安全问题
* @param[in] strSql: 需要执行的sql语句, 可能带占位符?
* @return bool: 成功返回true, 否则false
*/
bool SmartDB::Prepare(const std::string& strSql)
{
    // 执行sql脚本语句
    this->m_code = sqlite3_prepare_v2(this->m_dbHandle, strSql.data(), strlen(strSql.data()), &this->m_statement, NULL);
    if (m_code != SQLITE_OK)
    {
        return false;
    }
    return true;
}


/*
* @function: 绑定SQL语句预编译时把一些参数使用占位符（这里是?号）来代替
* @param[in] current: 占位符的索引
* @param[in] strData: 占位符问号的具体内容
*/
void SmartDB::BindValue(int current, const std::string &strData)
{
    // 绑定参数
    this->m_code = sqlite3_bind_text(this->m_statement, current, strData.data(), strlen(strData.data()), NULL);
}



/*
* @function: 执行sql, 返回函数执行的一个值，执行简单的汇聚函数，如select cout(*), select max(*)等
* @param[in] strSql: sql语句，有可能占位符"?"
* @param[in] strArgs: 参数列表，用来填充占位符, 变参
* @param[out] vecData: 查询出来的全部数据
* @return int: 最近一次错误码
*/
int SmartDB::ExecuteScalar(const std::string &strSql, std::vector<std::vector<std::string>>& vecData, const int num, const std::string strFirst, ...)
{
    // 准备执行sql脚本语句
    if (this->Prepare(strSql) == false)
    {
        return this->m_code;
    }
    
    if (num > 0) {
        int current = 1;
        BindValue(current, strFirst);
        if (this->m_code != SQLITE_OK)
        {
            return this->m_code;
        }

        // 绑定参数
        va_list ap;
        va_start(ap, strFirst);
        std::string strSrc = "";
        for (int i = 1; i < num; ++i)
        {
            current += 1;
            strSrc = va_arg(ap, const std::string);
            BindValue(current, strSrc);
            if (this->m_code != SQLITE_OK)
            {
                return this->m_code;
            }
        }
        va_end(ap);
    }
    

    // 获取列数
    int iCount = sqlite3_column_count(this->m_statement);
    
    while (true)
    {
        // 执行sql语句，获取每行的数据
        this->m_code = sqlite3_step(this->m_statement);          
        if (this->m_code == SQLITE_ROW)
        {
            // 插入每一列的值
            std::vector<std::string> vecTmp;
            this->InsertLine(vecTmp, iCount);
            vecData.push_back(vecTmp);
        }
        else if (this->m_code == SQLITE_DONE)    // 数据行都已经获取，跳出循环
        {
            break;
        }
        else       // 出错
        {
            break;
        }            
        
    }

    // 重置一个准备语句对象到它的初始状态，然后准备被重新执行
    sqlite3_reset(this->m_statement);

    return this->m_code;
}

/*
* @function: 将单行数据插入vector中
* @param[in] iCount: 单行列数
* @param[out] vecData: 单行数据集
*/
void SmartDB::InsertLine(std::vector<std::string>& vecData, const int& iCount)
{
    for (int i = 0; i < iCount; ++i)
    {
        // 这里需要先判断当前列的类型，列的类型不同，调用的api函数不同
        // 获取实际的数据值
        int coltype = sqlite3_column_type(this->m_statement, i);
        if (coltype == SQLITE_TEXT)
        {
            const char* val = (const char*)sqlite3_column_text(this->m_statement, i);
            vecData.push_back(std::string(val));
        }
        else if (coltype == SQLITE_NULL)
        {
            vecData.push_back(std::string(""));
        }
        std::cout << sqlite3_column_name(this->m_statement, i) << "\n";
    }
}

因为sqlite是utf-8编码，需要转码,vs

#include "smartdb.h"
#include <vector>
#include <stdio.h>
#include <windows.h>


void ConvertGBKToUtf8(std::string& amp, std::string strGBK)
{
    int len = MultiByteToWideChar(CP_ACP, 0, (LPCSTR)strGBK.c_str(), -1, NULL, 0);
    unsigned short * wszUtf8 = new unsigned short[len + 1];
    memset(wszUtf8, 0, len * 2 + 2);
    MultiByteToWideChar(CP_ACP, 0, (LPCSTR)strGBK.c_str(), -1, (LPWSTR)wszUtf8, len);
    len = WideCharToMultiByte(CP_UTF8, 0, (LPCWSTR)wszUtf8, -1, NULL, 0, NULL, NULL);
    char *szUtf8 = new char[len + 1];
    memset(szUtf8, 0, len + 1);
    WideCharToMultiByte(CP_UTF8, 0, (LPCWSTR)wszUtf8, -1, szUtf8, len, NULL, NULL);
    //strGBK = szUtf8; 
    amp = szUtf8;
    delete[] szUtf8;
    delete[] wszUtf8;
}


void ConvertUtf8ToGBK(std::string&amp, std::string strUtf8)
{
    int len = MultiByteToWideChar(CP_UTF8, 0, (LPCSTR)strUtf8.c_str(), -1, NULL, 0);
    unsigned short * wszGBK = new unsigned short[len + 1];
    memset(wszGBK, 0, len * 2 + 2);
    MultiByteToWideChar(CP_UTF8, 0, (LPCSTR)strUtf8.c_str(), -1, (LPWSTR)wszGBK, len);
    len = WideCharToMultiByte(CP_ACP, 0, (LPCWSTR)wszGBK, -1, NULL, 0, NULL, NULL);
    char *szGBK = new char[len + 1];
    memset(szGBK, 0, len + 1);
    WideCharToMultiByte(CP_ACP, 0, (LPCWSTR)wszGBK, -1, szGBK, len, NULL, NULL);
    //strUtf8 = szGBK; 
    amp = szGBK;
    delete[] szGBK;
    delete[] wszGBK;
}

int main()
{
    SmartDB dB;
    if (!dB.Open("FastFile.db"))
    {
        return 0;
    }

    std::string strSql = "select cn, funcname from typelanguage where funcname = ?;";
    ConvertGBKToUtf8(strSql, strSql);
    std::vector<std::vector<std::string>> vecDst;
    std::string str = "car_mode";
    ConvertGBKToUtf8(str, str);
    std::string str2 = "1";
    dB.ExecuteScalar(strSql, vecDst, 1, str);

    for (int i = 0; i < vecDst.size(); ++i)
    {
        for (int j = 0; j < vecDst[i].size(); ++j)
        {
            std::string strDst = "";
            ConvertUtf8ToGBK(strDst, vecDst[i][j]);
            std::cout << strDst << "\n";
        }
    }



    getchar();
}

 

风贰中
关注
  • 0
    点赞
  • 踩
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
C++封装SQLite实例<一>
坚持理想 面对现实
03-22 4818
第一次结识SQLite是在上学期，有一个软工的同学在看有关安卓应用的书籍，正好在实验室碰到了他，他正在看的就是SQLite这个数据库在安卓上的使用。从那以后就再也没有了解过有关SQLite的信息，上星期学长们的CDIO组在做一个游戏，要用到这个数据库，正好我也在这个组里，于是就分配我去学习如何封装这个数据库中的原生态函数，以便于访问数据库。学了一个星期，根据外文网站上的一个例子逐步的去拆解过程函数
C++封装版sqlite3
11-21
C版sqlite3再封装成C++版，简化创建数据库，增加动态参数绑定快捷函数，查询辅助类，支持更改表字段，删除表字段, 加入异常抛出，事务开始提交回滚
sqlite3-入门日记4-实现C++类封装
weixin_30340353的博客
03-28 229
一、前言： 今天试了下如何用C++类实现接口封装，感觉蛮好 。用于封装的类主要有两个，SQLiteStatement类和SQLiteWrapper类，是一个老外写的。我看了下源码，主要是对C接口进行了封装，好处自然不用说，可以重用。很佩服老外的技巧，在这里就引用下他们的代码供大家分享下他们的思想。 源代码链接：http://www.adp-gmbh.ch/sqlite/wrapper...
Sqlite3入门和c/c++下使用
最新发布
fg247778718的博客
06-13 796
1. 创建数据表格create table 表名(字段名 数据类型， 字段名 数据类型)；2. 插入数据insert into 表名 values('字段数据'，'字段数据'，'字段数据'，'字段数据' );insert into student values('20200101', '张三', '广州','25648');insert into student values('20200102', '李四', '广州','25645');3. 查询数据（1） 查询。
C++ Sqlite封装
田螺姑娘的博客
12-05 986
  在我们的日常快速搭建一个小型的项目时，总会进行存盘操作（比如日志、配置文件信息），难免会对文件进行操作，如Json、二进制这类文件。但是使用Json这类文件，读取配置信息和存入信息难免过于繁琐。故可以使用Sqlite这类轻量级的数据库，跨平台且引入即可使用。
conan入门(十九):封装第三方开源库cpp_redis示例
10km的专栏
03-15 2506
conan: 封装第三方开源库cpp_redis示例 在我的之前关于conan系列博客中，已经介绍过如何引用Conan中央仓库(conan-center)预定义的第三库(《conan入门(五):conan 交叉编译引用第三方库示例》),以及如何将自己的项目封装为Conan包以供第三方引用(《conan入门(七):将自己的项目生成conan包》)。 但是这个世界实际是复杂的，虽然Conan中央仓库已经收录了上千个常用的开源第三方库，但仍然有更多的第三库并没被Conan中央仓库收录，你不能指望你所需要的任何第三
C++学习记录002——创建C++动态链接库封装类（函数）给MFC调用（初学者详细步骤）
SYW的博客
10-29 1935
工具：VS2019 项目：MFC调用C++动态链接库的项目的DLL 一、创建DLL 如图，选好要创建的项目 取个名字，选好地址，点击“创建” 在解决方案资源管理器中，右击“头文件”——>"添加“——>"新建项"， 选择”头文件(.h)"——>取个名字——>点击“添加” ...
关系型数据库-sqlite3（c/c++的API接口）
CC的博客
10-02 1903
关系型数据库-sqlite3（c/c++的API接口）
Sqlite3封装库文件
01-31
在这个“Sqlite3封装库文件”中，包含了在Visual Studio环境下开发C++应用程序时所需的一些关键组件，特别是针对MFC（Microsoft Foundation Classes）框架的应用。 1. **sqlite3.dll**：这是一个动态链接库文件，...
Sqlite3 C++ 简单单例数据库操作类封装
02-07
`CSqlite3`类应该封装SQLite3的C API，如`sqlite3_open()`, `sqlite3_exec()`, `sqlite3_close()`等，以提供更友好的C++接口。 为了保证线程安全，单例类的实例化通常会在第一次调用`getInstance()`时进行，使用...
基于标准库的C++ sqlite3常用功能封装和使用
03-19
本教程将详细介绍如何基于C++标准库封装SQLite3的常用功能，包括数据的增、删、改、查以及事务处理。 首先，为了在C++中使用SQLite3，你需要包含相应的头文件`<sqlite3.h>`。安装SQLite3库后，这个头文件通常位于...
自己用C++封装sqlite3 for vc
02-21
自己用C++封装sqlite3 for vc，内含Sqlite3 官方库
sqlite3API的封装
07-17
sqlite3的API的进一步封装，及实例调用。 还有SQLite权威指南.pdf,SQLite学习手册_中文全本.pdf,全面解析_SQLite.pdf 和sqlite3图形化工具
c++ SQLite 封装类以及例子
01-11
c++ SQLite 封装类以及例子，一个好用的代码，欢迎下载。可以直接使用。
SQLiteC/C++接口详细介绍sqlite3_stmt类简介
03-18 1467
SQLite拥有超过225个API，本系列文章将深入介绍 SQLite C/C++ 数据库接口的基础知识，包括如何连接和打开 SQLite 数据库、执行查询和事务管理、提取查询结果集等等，帮助开发者更好地了解和使用 SQLite 数据库接口。本文详细介绍库文件打开关闭接口。
sqlite3：sqlite3_bind 函数
热门推荐
猪哥的专栏
05-28 1万+
该函数组用于绑定变量值到 prepare 语句中，也就是给 sqlite3_stmt变量赋值。前面的文章讲过，我们一定是先通过sqlite3_prepare_v2函数创建并初始化一个 sqlite3_stmt 变量语句，然后使用sqlite3_bind_xxx函数对 这个 sql语句变量进行绑定参数。 常用的sqlite3_bind函数： int sqlite3_bind_int(s...
C++封装SQLite实例<三>
weixin_34194359的博客
09-04 81
前一篇博客中介绍的是怎样依据sqlite3_get_table()函数来获取一张表的内容，就是一股脑的把表中的内容所有存储起来放在一个一维数组中，这其中的规则已经介绍过了。接下来讲的是怎样依据一个SQL查询语句来挑选出合乎选择条件的记录，被挑选出的记录事先是不知道的。你不知道这个查询的结果中有几行更不知道每一行中详细的记录值是什么，所以没办法用给定行值和列值的方式来获取数据，所有之前的用数...
使用 C++ 类进行封装的 SQLite3
weixin_33953384的博客
09-28 705
sqlite3pp 该库对SQLite3 API使用C++类进行包装，支持几乎所有SQLite3功能（打开数据库、查询、执行命令、事物等）。 示例代码如下： database 数据库 sqlite3pp::database db("test.db"); db.execute("INSERT INTO contacts (name, phone) VAL...
C++ Sqlite3的基本使用
weixin_30696427的博客
05-30 558
｜SQLite3简介 SQLite3只是一个轻型的嵌入式数据库引擎，占用资源非常低，处理速度比Mysql还快，专门用于移动设备上进行适量的数据存取，它只是一个文件，不需要服务器进程。 SQL语句是SQL操作的指令，我们用C/C++访问数据库时，需要用char*即C字符串来保存SQL语句，然后调用相应sqlite3库的函数，传入C字符串，来执行SQL指令。 常用术语：表(table)、字段(...
c++封装sqlite3
05-25
SQLite是一种轻型的嵌入式关系型数据库，它不需要独立的服务器进程，以文件的形式存储数据，因此非常适合嵌入到其他应用程序中使用。在C++中，我们可以通过SQLite提供的C接口来操作SQLite数据库，也可以通过封装SQLite的方式来简化操作。 下面是一个简单的C++封装SQLite3的示例： ```cpp #include <sqlite3.h> #include <string> class SQLiteDB { public: SQLiteDB(const std::string& dbPath) : dbPath_(dbPath), db_(nullptr) {} ~SQLiteDB() { close(); } bool open() { if (db_) { return true; } int result = sqlite3_open(dbPath_.c_str(), &db_); if (result != SQLITE_OK) { return false; } return true; } void close() { if (db_) { sqlite3_close(db_); db_ = nullptr; } } bool exec(const std::string& sql) { char* error = nullptr; int result = sqlite3_exec(db_, sql.c_str(), nullptr, nullptr, &error); if (result != SQLITE_OK) { sqlite3_free(error); return false; } return true; } private: std::string dbPath_; sqlite3* db_; }; ``` 以上代码定义了一个SQLiteDB类，它包含了打开和关闭数据库的方法，以及执行SQL语句的方法。在构造函数中，我们需要传入数据库文件的路径，通过sqlite3_open函数打开数据库，如果打开失败则返回false。在析构函数中，我们需要关闭数据库。在exec方法中，我们通过sqlite3_exec函数执行SQL语句，如果执行失败则返回false。 使用示例： ```cpp int main() { SQLiteDB db("test.db"); if (!db.open()) { return -1; } if (!db.exec("CREATE TABLE IF NOT EXISTS user (id INTEGER PRIMARY KEY, name TEXT)")) { return -1; } if (!db.exec("INSERT INTO user (name) VALUES ('Tom')")) { return -1; } db.close(); return 0; } ``` 以上代码创建了一个名为test.db的SQLite数据库文件，并在其中创建了一个名为user的表，然后插入了一行数据。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值