修复笔误

mensong · Nov 8, 2023 · 6b85992 · 6b85992
1 parent 3a67568
commit 6b85992
Show file tree

Hide file tree

Showing 4 changed files with 149 additions and 84 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,3 +3,5 @@
 /SQLit3/tmp
 /x64
 /Win32
+/ConsoleApplicationTest/1.db
+/ConsoleApplicationTest/2.db
diff --git a/ConsoleApplicationTest/ConsoleApplicationTest.cpp b/ConsoleApplicationTest/ConsoleApplicationTest.cpp
@@ -4,52 +4,51 @@
 #include <iostream>
 #include "..\SQLit3\SQLit3.h"
 
-int main()
+int test_normal()
 {
-    Database* db = SQLit3::Ins().CreateDatabase();
-	bool b = db->Open("F:\\1.db", "", Database::SQLIT3_OPEN_DEFAULT);
-    if (!b)
-    {
-        std::cout << "打开数据库失败"<< db->GetLastErrorCode()
+	Database* db = SQLit3::Ins().CreateDatabase();
+	bool b = db->Open("1.db", "12345678", Database::SQLIT3_OPEN_DEFAULT);
+	if (!b)
+	{
+		std::cout << "打开数据库失败" << db->GetLastErrorCode()
 			<< ":" << db->GetLastErrorMsg() << std::endl;
-        return 1;
-    }
-	b = db->Execute("select 1");
+		return 1;
+	}
 	//b = db->ResetPassword("12345678");
 
-	__int64 maxId = 0;
-	b = db->QueryInt64("SELECT MAX(id) FROM Url2FileCache", &maxId);
+	b = db->Execute("CREATE TABLE IF NOT EXISTS t1 (\"a\" TEXT,\"b\" integer,\"c\" real);");
+	b = db->Execute("INSERT INTO t1 (\"a\", \"b\", \"c\") VALUES ('abc', 1, 2.0);");
+	__int64 changedRow = db->GetChangeRowCount();
 
-    b = db->Execute("CREATE TABLE IF NOT EXISTS \"t1\" (\"a\" TEXT,\"b\" integer,\"c\" real);");
-    b = db->Execute("INSERT INTO \"main\".\"t1\" (\"a\", \"b\", \"c\") VALUES ('abc', 1, 2.0);");
-    __int64 changedRow = db->GetChangeRowCount();
+	__int64 countRows = 0;
+	b = db->QueryInt64("SELECT COUNT(1) FROM t1", &countRows);
 
-    SqlStatement* stmt = db->StatementPrepare("SELECT a,c FROM \"main\".\"t1\" where b=? or b=$cv");
-    if (stmt)
-    {
+	SqlStatement* stmt = db->StatementPrepare("SELECT a,c FROM t1 where b=? or b=$cv");
+	if (stmt)
+	{
 		b = stmt->BindInt(0, 1);
 		int idx = stmt->GetParamIndexByName("$cv");
 		b = stmt->BindDouble(idx, 2);
 
 		int colCount = stmt->GetColumnCount();
-		while (stmt->Next() == SqlStatement::SQLIT3_READ_HAS_ROW)
+		while (stmt->Next() == SqlStatement::SQLIT3_EXEC_HAS_ROW)
 		{
 			for (int i = 0; i < colCount; i++)
-			{				
-                const char* columnName = stmt->GetColumnName(i);
-                auto type = stmt->GetType(i);
+			{
+				const char* columnName = stmt->GetColumnName(i);
+				auto type = stmt->GetType(i);
 				switch (type)
 				{
 				case SqlStatement::SQLIT3_VALUE_INTEGER:
 				{
 					__int64 v = stmt->GetInt64(i);
-					std::cout << columnName << "=" << v << std::endl;
+					std::cout << columnName << "=" << v << "\t";
 					break;
 				}
 				case SqlStatement::SQLIT3_VALUE_FLOAT:
 				{
 					double v = stmt->GetDouble(i);
-					std::cout << columnName << "=" << v << std::endl;
+					std::cout << columnName << "=" << v << "\t";
 					break;
 				}
 				case SqlStatement::SQLIT3_VALUE_BLOB:
@@ -59,32 +58,97 @@ int main()
 					char* pBuf = new char[dataLen + 1];
 					memcpy(pBuf, data, dataLen);
 					pBuf[dataLen] = '\0';
-					std::cout << columnName << "=" << pBuf << std::endl;
+					std::cout << columnName << "=" << pBuf << "\t";
 					delete[] pBuf;
 					break;
 				}
 				case SqlStatement::SQLIT3_VALUE_NULL:
 				{
-					std::cout << columnName << "=<NULL>" << std::endl;
+					std::cout << columnName << "=<NULL>" << "\t";
 					break;
 				}
-				case SqlStatement::SQLIT3_VALUE_TEXT:					
+				case SqlStatement::SQLIT3_VALUE_TEXT:
 				default:
 				{
 					int dataLen = 0;
 					auto data = stmt->GetText(i, &dataLen);
-					std::cout << columnName << "=" << data << std::endl;
+					std::cout << columnName << "=" << data << "\t";
 					break;
 				}
 				}
 			}
+
+			std::cout << std::endl;
 		}
 		db->StatementFinalize(stmt);
-    }
+	}
 
 
-    b = db->Close();
+	b = db->Close();
 	SQLit3::Ins().DestoryDatabase(db);
+	return 0;
+}
+
+#include <fstream>
+
+int test_blob(const char* exefile)
+{
+	Database* db = SQLit3::Ins().CreateDatabase();
+	bool b = db->Open("2.db", "12345678", Database::SQLIT3_OPEN_DEFAULT);
+	if (!b)
+	{
+		std::cout << "打开数据库失败" << db->GetLastErrorCode()
+			<< ":" << db->GetLastErrorMsg() << std::endl;
+		return 1;
+	}
+	//b = db->ResetPassword("12345678");
+
+	b = db->Execute("CREATE TABLE IF NOT EXISTS test_blob (\"a\" integer NOT NULL,\"b\" blob, PRIMARY KEY (\"a\"));");
+
+	__int64 id = 0;
+	db->QueryInt64("SELECT MAX(a) FROM test_blob", &id);
+
+	SqlStatement* stmt = db->StatementPrepare("INSERT INTO test_blob(\"a\", \"b\") VALUES (?, ?);");
+	if (stmt)
+	{
+		stmt->BindInt64(0, id + 1);
+
+		// 打开二进制文件
+		std::ifstream inFile(exefile, std::ios::binary);
+		// 检查文件是否打开成功
+		if (inFile)
+		{
+			inFile.seekg(0, std::ios::end);
+			std::streampos ps = inFile.tellg();
+			inFile.seekg(0, std::ios_base::beg);
+			// 读取文件内容
+			char* buffer = new char[ps];
+			inFile.read(buffer, ps);
+			// 关闭文件
+			inFile.close();
+
+			stmt->BindBlob(1, buffer, ps);
+			SqlStatement::ExecStatus es = stmt->Next();
+			delete[] buffer;
+			__int64 changedRow = db->GetChangeRowCount();
+		}
+
+		db->StatementFinalize(stmt);
+	}
+
+	const void* data = NULL;
+	int dataLen = 0;
+	b = db->QueryBlob("SELECT b FROM test_blob LIMIT 0,1", &data, &dataLen);
+
+	b = db->Close();
+	SQLit3::Ins().DestoryDatabase(db);
+	return 0;
+}
+
+int main(int argc, char** argv)
+{
+	test_normal();
+	test_blob(argv[0]);
 
     return 0;
 }