SQLite3 使用容器vector批量插入记录

本文展示了一个使用C++在SQLite数据库中批量执行SQL语句的函数实现。该函数首先开始一个事务,然后遍历传入的SQL语句向量,逐个执行每个SQL语句,最后提交事务。如果执行过程中遇到错误,会尝试继续执行后续的SQL语句,并在最后统一返回错误信息。
int SQL_EXE(std::vector<std::string> vtExeSql)
{
	if (NULL == m_pDB)
	{
		return -1;
	}
 
	char* pErrMsg = NULL;
	int iRet = SQLITE_OK;
 
	iRet = sqlite3_exec(m_pDB, "BEGIN", NULL, NULL, &pErrMsg);
	if (SQLITE_OK != iRet)
	{
		if (NULL != pErrMsg)
		{
			sqlite3_free(pErrMsg);
		}
 
		return -1;
	}

	std::vector<std::string>::const_iterator iter = vtExeSql.begin();
	for (; iter != vtExeSql.end(); ++iter)
	{
		std::string strSql = *iter;
		if (!(strSql.empty()))
		{
			iRet = sqlite3_exec(m_pDB, strSql.c_str(), NULL, NULL, &pErrMsg);
			if (SQLITE_OK != iRet)
			{
				continue;
			}
		}
	}
 
	iRet = sqlite3_exec(m_pDB, "COMMIT", NULL, NULL, &pErrMsg);
	if (SQLITE_OK != iRet)
	{
		iRet = -1;
	}
 
	if (NULL != pErrMsg)
	{
		sqlite3_free(pErrMsg);
	}
 
	return iRet;
}

 

#include <sqlite3.h> #include <iostream> // 创建表的 SQL 语句(已省略) int main() { sqlite3* db; int rc = sqlite3_open("test.db", &db); if (rc != SQLITE_OK) { std::cerr << "无法打开数据库: " << sqlite3_errmsg(db) << std::endl; return -1; } // 创建表 char* errMsg = nullptr; sqlite3_exec(db, createProvinces, nullptr, nullptr, &errMsg); sqlite3_exec(db, createCities, nullptr, nullptr, &errMsg); sqlite3_exec(db, createCounties, nullptr, nullptr, &errMsg); sqlite3_exec(db, createTowns, nullptr, nullptr, &errMsg); sqlite3_exec(db, createCommunities, nullptr, nullptr, &errMsg); // 开启事务 sqlite3_exec(db, "BEGIN TRANSACTION", nullptr, nullptr, &errMsg); // 插入省份 const char* insertProvinceSQL = "INSERT INTO Provinces (ProvinceName) VALUES (?);"; sqlite3_stmt* stmt; sqlite3_prepare_v2(db, insertProvinceSQL, -1, &stmt, nullptr); sqlite3_bind_text(stmt, 1, "河北省", -1, SQLITE_STATIC); sqlite3_step(stmt); sqlite3_reset(stmt); int lastProvinceId = sqlite3_last_insert_rowid(db); // 插入城市 const char* insertCitySQL = "INSERT INTO Cities (CityName, ProvinceID) VALUES (?, ?);"; sqlite3_prepare_v2(db, insertCitySQL, -1, &stmt, nullptr); sqlite3_bind_text(stmt, 1, "石家庄市", -1, SQLITE_STATIC); sqlite3_bind_int(stmt, 2, lastProvinceId); sqlite3_step(stmt); sqlite3_reset(stmt); int lastCityId = sqlite3_last_insert_rowid(db); // 插入县 const char* insertCountySQL = "INSERT INTO Counties (CountyName, CityID) VALUES (?, ?);"; sqlite3_prepare_v2(db, insertCountySQL, -1, &stmt, nullptr); sqlite3_bind_text(stmt, 1, "正定县", -1, SQLITE_STATIC); sqlite3_bind_int(stmt, 2, lastCityId); sqlite3_step(stmt); sqlite3_reset(stmt); int lastCountyId = sqlite3_last_insert_rowid(db); // 插入镇 const char* insertTownSQL = "INSERT INTO Towns (TownName, CountyID) VALUES (?, ?);"; sqlite3_prepare_v2(db, insertTownSQL, -1, &stmt, nullptr); sqlite3_bind_text(stmt, 1, "正定镇", -1, SQLITE_STATIC); sqlite3_bind_int(stmt, 2, lastCountyId); sqlite3_step(stmt); sqlite3_reset(stmt); int lastTownId = sqlite3_last_insert_rowid(db); // 插入社区 const char* insertCommunitySQL = "INSERT INTO Communities (CommunityName, TownID) VALUES (?, ?);"; sqlite3_prepare_v2(db, insertCommunitySQL, -1, &stmt, nullptr); sqlite3_bind_text(stmt, 1, "某社区", -1, SQLITE_STATIC); sqlite3_bind_int(stmt, 2, lastTownId); sqlite3_step(stmt); sqlite3_finalize(stmt); // 提交事务 sqlite3_exec(db, "COMMIT", nullptr, nullptr, &errMsg); // 清理资源 sqlite3_close(db); return 0; } 能否添加事务和for循环插入,有几十万条数据那不是要把人给写死
07-15
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值