added sqllite

database/SqliteDatabase.cpp

@@ -1,71 +1,136 @@
 #include "SqliteDatabase.h"
+#include "sqlite3.h"
+#include <iostream>
+using namespace std;
 
-SqliteDatabase::SqliteDatabase() : Database
+SqliteDatabase::SqliteDatabase(const char* host) : Database(host,"","")
 {
-
+	mConnection=NULL;
+	mCurrentStmt=NULL;
 }
 
 void SqliteDatabase::connect()
 {
-
+	
+	char *zErrMsg = 0;
+	int rc = sqlite3_open(mHost.c_str(), &mConnection);
+	if( rc )
+	{
+		cout << "Can't open database: " << mHost << " " << rc << endl;
+		sqlite3_close(mConnection);
+	}
 }
+
 void SqliteDatabase::disconnect()
 {
-
+	sqlite3_finalize(mCurrentStmt);
+	sqlite3_close(mConnection);
 }
 
 // returns true if the query went ok
 bool SqliteDatabase::executeSQL(const char* sql)
 {
+	sqlite3_finalize(mCurrentStmt);
+	char *zErrMsg = 0;
+	int rc=sqlite3_prepare_v2(mConnection,sql,-1,&mCurrentStmt,NULL);
+	if( rc!=SQLITE_OK )
+	{
+		cout << "SQL error:" << rc << endl;
+		return(false);
+	}
+	rc=sqlite3_step(mCurrentStmt);
+	if(rc==SQLITE_ROW)
+	{
+		mMoreRows=true;
+	}else if(rc==SQLITE_DONE)
+	{
+		mMoreRows=false;
+	}else
+	{
+		mMoreRows=false;
+		cout << "SQL error:" << rc << endl;
+		return(false);
+	}
 
+	return(true);
 }
 
 // tells you how many rows were changed by an update or insert
 int SqliteDatabase::getNumRowsAffected()
 {
+	// TODO: SqliteDatabase::getNumRowsAffected()
+	return(0);
+}
 
+int SqliteDatabase::getLastInsertID()
+{
+	return(sqlite3_last_insert_rowid(mConnection));
 }
 
 // returns false if there are no results
 bool SqliteDatabase::startIterRows()
 {
-
+	return(mMoreRows);
 }
+
 void SqliteDatabase::endIterRows()
 {
-
+	sqlite3_finalize(mCurrentStmt);
+	mCurrentStmt=NULL;
 }
 
 // call this after you executeSQL
 // will return false if there are no more rows
 bool SqliteDatabase::getNextRow()
 {
+	if(!mMoreRows) return(false);
 
+	int rc=sqlite3_step(mCurrentStmt);
+	if(rc==SQLITE_ROW)
+	{
+		return(true);
+	}else if(rc==SQLITE_DONE)
+	{
+		return(false);
+	}else
+	{
+		cout << "SQL error:" << rc << endl;
+		return(false);
+	}
 }
 
 char* SqliteDatabase::getStr(int colIndex,std::string& retStr)
 {
-
+	retStr=(char*)sqlite3_column_text(mCurrentStmt, colIndex);
+	return((char*)retStr.c_str());
 }
+
 int32 SqliteDatabase::getInt(int colIndex)
 {
-
+	return(sqlite3_column_int(mCurrentStmt, colIndex));
 }
+
 float SqliteDatabase::getFloat(int colIndex)
 {
-
+	return(sqlite3_column_double(mCurrentStmt, colIndex));
 }
+
 bool SqliteDatabase::getBool(int colIndex)
 {
-
+	return(sqlite3_column_int(mCurrentStmt, colIndex));
 }
+
 bool SqliteDatabase::getBinary(int colIndex,unsigned char* buf,int maxSize)
 {
-
+	const void* blob=sqlite3_column_blob(mCurrentStmt, colIndex);
+	int size=sqlite3_column_bytes(mCurrentStmt, colIndex);
+	if(maxSize<size) size=maxSize;
+	memcpy(buf,blob,size);
+	return(true);
 }
 uint64 SqliteDatabase::getBigInt(int colIndex)
 {
-
+	return(sqlite3_column_int64(mCurrentStmt, colIndex));
 }