1. 程式人生 > >sqlite3如何判斷一個表是否已經存在於資料庫中c++

sqlite3如何判斷一個表是否已經存在於資料庫中c++

網上找了很多,都不能使用,無奈自己不是專門搞資料庫的,不過最終在某個回撥函式帖的啟示下,自己修改了一下程式碼,終於可以使用了。

實在是沒有想到這麼簡單的一個問題困擾了那麼多人,這裡分享一下程式碼,也感謝原作者(忘記地方了)對我的啟示。

int sqlite3_callback_func( void* pHandle,int iRet, char** szSrc, char** szDst )
{
	//... 
	if ( 1 == iRet )
	{
		int iTableExist = atoi(*(szSrc));  //此處返回值為查詢到同名表的個數,沒有則為0,否則大於0
		if (pHandle != nullptr)
		{
			int* pRes = (int*)pHandle;
			*pRes = iTableExist;
		}
		// szDst 指向的內容為"count(*)"
	}

	return 0; //返回值一定要寫,否則下次呼叫 sqlite3_exec(...) 時會返回 SQLITE_ABORT
}

bool SQLite::IsTableExist(const std::string& strTableName)
{
	std::string strFindTable = "SELECT COUNT(*) FROM sqlite_master where type ='table' and name ='" + strTableName + "'";

	char* sErrMsg = nullptr;

	//void *pHandle = ***;
	int nTableNums = 0;
	if (sqlite3_exec(m_db, strFindTable.c_str(), &sqlite3_callback_func, &nTableNums, &sErrMsg) != SQLITE_OK)
	{
		return false;
	}
	//回撥函式無返回值,則此處第一次時返回SQLITE_OK, 第n次會返回SQLITE_ABORT

	return nTableNums > 0;
}

個人測試成功,歡迎交流,如果有興趣,歡迎加入Q群271888395。