add database class

2 files changed, 136 insertions, 0 deletions

diff --git a/src/server/database.cpp b/src/server/database.cpp
new file mode 100644
index 0000000..4438cc2
--- /dev/null
+++ b/src/server/database.cpp
@@ -0,0 +1,71 @@
+#include "database.hpp"
+
+DB_ROW::~DB_ROW()
+{
+	row = 0;
+	/* mariadb documentation:
+	Memory associated to MYSQL_ROW will be freed when calling
+	mysql_free_result() function. */
+}
+
+char *DB_ROW::operator[](const int index)
+{
+ 	return row[index];
+}
+//////////////////////////////////////////////////////////////
+
+AnswerDB::~AnswerDB()
+{
+	mysql_free_result(result);
+}
+
+DB_ROW *AnswerDB::GetNextRow()
+{
+	if(current_row)
+		delete current_row;
+
+	MYSQL_ROW row = mysql_fetch_row(result);
+	if(!row)
+		return 0;
+
+	current_row = new DB_ROW(&row, fields);
+	return current_row;
+}
+
+//////////////////////////////////////////////////////////////
+
+DatabaseManager::~DatabaseManager()
+{
+	mysql_close(connection);
+}
+
+DatabaseManager *DatabaseManager::Connect(const char *host, const char *user,
+	const char *pass, const char *db_name, unsigned int port,
+    const char *unix_socket, unsigned long flags)
+{
+	MYSQL *mysql = 0;
+	mysql = mysql_init(mysql);
+
+	mysql = mysql_real_connect(mysql, host, user, pass,
+		db_name, port, unix_socket, flags);
+	if(!mysql)
+		return 0;
+
+	return new DatabaseManager(mysql);
+}
+
+AnswerDB *DatabaseManager::QuerySelect(const char *sql)
+{
+	int state = mysql_query(connection, sql);
+	if(state != 0)
+		return 0;
+		// #TODO: error msg
+
+	MYSQL_RES *result = mysql_store_result(connection);
+	if(!result) 
+		return 0;
+
+	unsigned int fields = mysql_num_fields(result);
+	unsigned long rows = mysql_num_rows(result);
+	return new AnswerDB(result, fields, rows);
+}
\ No newline at end of file
diff --git a/src/server/database.hpp b/src/server/database.hpp
new file mode 100644
index 0000000..8be6f06
--- /dev/null
+++ b/src/server/database.hpp
@@ -0,0 +1,65 @@
+#ifndef WC_DATABASE_H
+#define WC_DATABASE_H
+
+#include <mysql.h>
+
+class DatabaseManager;
+
+class DB_ROW {
+	 MYSQL_ROW row;
+	 const unsigned int fields;
+public:
+	 DB_ROW(MYSQL_ROW *r, const unsigned int flds) :
+	 	row(*r), fields(flds) {}
+	 ~DB_ROW();
+	 char *operator[](const int index);
+
+	 const unsigned int GetFieldCount() const { return fields; }
+};
+
+class AnswerDB {
+	friend class DatabaseManager; // ?
+
+	MYSQL_RES *result;
+	DB_ROW *current_row;
+	const unsigned int fields;
+	unsigned long rows;
+
+	AnswerDB(MYSQL_RES *res, unsigned int flds, unsigned long rws) :
+		result(res), current_row(0), fields(flds), rows(rws) {}
+public:
+	~AnswerDB();
+
+	DB_ROW *GetNextRow();
+	const unsigned int GetFieldCount() const { return fields; }
+	unsigned long GetRowCount() const { return rows; }
+};
+
+class DatabaseManager {
+	MYSQL *connection;
+
+	DatabaseManager(MYSQL *db) : connection(db) {}
+public:
+	~DatabaseManager();
+
+	static DatabaseManager *Connect(const char *host, const char *user,
+		const char *pass, const char *db_name, unsigned int port,
+    	const char *unix_socket = 0, unsigned long flags = 0);
+
+	AnswerDB *QuerySelect(const char *sql);
+};
+
+#endif
+
+/*
+
+надо как-то вызывать ~AnswerDB!!
+
+Answer DB *ans = server->Query("SELECT password FROM users WHERE name = '%s' ONLY ONE", strlen(..))
+DB_ROW *row = ans->GetNextRow();
+if(row)
+	name = row[0];
+	password = row[1];
+
+
+*/
\ No newline at end of file