Initial commit

2026-07-09 11:27:59 +08:00
commit 3f1edda843
126 changed files with 34446 additions and 0 deletions
@@ -0,0 +1,245 @@
+#include "Databases.h"
+#include "thread.h"
+#include "debug.h"
+
+#ifdef _WIN32
+	#pragma comment(lib,"libmysql.lib")
+#else
+	#include <string.h>
+#endif
+
+CSema * g_pSema = NULL;
+
+Databases::Databases()
+{
+	m_nfieldNum = 0;
+	m_record = NULL;
+	m_result = NULL;
+	InitMysql();
+
+    g_pSema = new CSema(NULL, 1);
+}
+Databases::~Databases()
+{
+    if( g_pSema ){
+        delete g_pSema;
+        g_pSema = NULL;
+    }
+
+	mysql_free_result(m_result);
+	mysql_close(&m_mysql);
+}
+
+int Databases::ConnectDatabases(char *IP, char *user, char *pass ,char *db)
+{
+    m_srvIP = IP;
+    m_userName=user;
+    m_password = pass;
+    m_dbName=db;
+
+    if (!mysql_real_connect(&m_mysql,
+       						m_srvIP.c_str(), m_userName.c_str(), m_password.c_str(), m_dbName.c_str(),
+                            0, 0, CLIENT_MULTI_STATEMENTS)){
+
+        DebugPrint( "SQL CONNECT error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+        return -1;
+    }
+
+    return 0;
+}
+
+int Databases::InitMysql()
+{
+    if(!mysql_init(&m_mysql)){
+        DebugPrint( "SQL INIT error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+        return -1;
+    }
+
+    char value = 1;
+    mysql_options(&m_mysql, MYSQL_OPT_RECONNECT, (char *)&value);
+	mysql_options(&m_mysql,MYSQL_SET_CHARSET_NAME, "utf8");
+
+    //禁用SSL
+	mysql_ssl_mode sslmode = SSL_MODE_DISABLED;
+	mysql_options(&m_mysql, MYSQL_OPT_SSL_MODE, (void*)&sslmode);
+
+    return 0;
+}
+int Databases::FreeResult()
+{
+    //mysql_free_result(m_result);
+    return 0;
+}
+
+void Databases::CloseMysql()
+{
+    //mysql_free_result(m_result);
+    mysql_close(&m_mysql);
+    m_nfieldNum = 0;
+    m_record = NULL;
+    m_result = NULL;
+}
+
+int	Databases::Query( char *query )
+{
+	FreeResult();
+
+	if(mysql_ping(&m_mysql))
+	{
+		DebugPrint( "SQL PING error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+		printf("尝试重新连接……!\n");
+
+		CloseMysql();		
+
+		if( InitMysql() == -1)
+			return -1;
+
+
+		if (!mysql_real_connect(&m_mysql, m_srvIP.c_str(), m_userName.c_str(), m_password.c_str(),
+        						m_dbName.c_str(), 0, 0, CLIENT_MULTI_STATEMENTS)){
+			DebugPrint( "SQL CONNECT error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+			return -1;
+		}
+	}
+
+	if(mysql_query(&m_mysql, query))
+	{
+		DebugPrint( "SQL QUERY error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+		printf("sql: %s\n ",query);
+		return -1;
+	}	
+
+	m_result = mysql_store_result(&m_mysql);
+	
+	if( m_result ){
+		m_record = mysql_fetch_row(m_result);
+		m_nfieldNum = mysql_num_fields(m_result);
+	}
+
+	return 0;
+}
+
+int Databases::CallPro(const char *call, char *select, MYSQL_ROW* row)
+{
+    if( g_pSema->ActP(1000) ){
+        return -1;
+    }
+
+	int ret = 0;
+
+	FreeResult();
+
+	if(mysql_ping(&m_mysql))
+	{
+		DebugPrint( "SQL PING error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+		printf("尝试重新连接……!\n");
+
+		CloseMysql();		
+
+		if( InitMysql() == -1){
+			ret = -1;
+		}
+		else if (!mysql_real_connect(&m_mysql, m_srvIP.c_str(), m_userName.c_str(), m_password.c_str(),
+        						m_dbName.c_str(), 0, 0, CLIENT_MULTI_STATEMENTS)){
+			DebugPrint( "SQL CONNECT error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+		}
+	}
+
+	if( !ret && mysql_real_query(&m_mysql,call,(unsigned int)strlen(call)))
+	{			
+		DebugPrint( "SQL QUERY error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+		printf("proc: %s\n ",call);
+
+		printf("尝试重新连接……!\n");	
+		CloseMysql();
+		if( InitMysql()
+        		|| !mysql_real_connect(&m_mysql, m_srvIP.c_str(), m_userName.c_str(), m_password.c_str(),
+                			m_dbName.c_str(), 0, 0, CLIENT_MULTI_STATEMENTS)
+                || mysql_real_query(&m_mysql,call,(unsigned int)strlen(call)) ){
+			printf( "recover failed.\n" );
+			ret = -1;
+		}
+		else{
+			printf( "recovered.\n" );
+		}
+	}		
+	
+	if( !ret && select && strlen(select)>0 ){
+		if(mysql_query(&m_mysql, select))
+		{	
+			DebugPrint( "SQL QUERY error %d : %s\n", mysql_errno(&m_mysql), mysql_error(&m_mysql) );
+			printf("sql: %s\n ",select);
+			ret = -1;
+		}	
+		else{
+			int i, maxLen = 0;			
+			
+			MYSQL_RES* result = mysql_store_result(&m_mysql);
+			if( result ){
+				int nfieldNum = mysql_num_fields(result);
+				if( nfieldNum>0 && row ){
+					MYSQL_ROW record = mysql_fetch_row(result);
+					if( record && record[0] ){
+						*row = new char*[nfieldNum];
+						unsigned long* lengths = mysql_fetch_lengths(result);
+
+						for(i = 0; i < nfieldNum; i++){
+							if( lengths[i] ){
+								(*row)[i] = new char[lengths[i]+1];
+								memcpy( (*row)[i], record[i], lengths[i] );
+								(*row)[i][lengths[i]] = '\0';
+							}
+							else{
+								(*row)[i] = NULL;
+							}
+						}
+						
+						ret = nfieldNum;
+					}
+				}
+				
+				mysql_free_result(result);
+			}
+			else{
+				ret = -1;
+			}
+/*
+			m_nfieldNum = mysql_num_fields(m_result);	
+			if( m_nfieldNum>0 ){
+				unsigned long* length = mysql_fetch_lengths(m_result);	
+				if( length[0]>0 ){
+					m_record = mysql_fetch_row(m_result);
+					
+					*record = (char**)new char[length[0]];
+					memcpy( *record, m_record, length[0] );
+				}
+			}
+*/
+		}
+	}
+	
+    g_pSema->ActV();
+
+	return ret;
+}
+
+int Databases::GetFieldData(int i, char *chData)
+{
+
+	if( i < m_nfieldNum )		
+		strcpy(chData,m_record[i]);	
+	else
+		return -1;
+		
+	return 0;
+}
+
+MYSQL_ROW Databases::GetRecord()
+{
+		return m_record;
+}
+
+my_ulonglong Databases::GetInsertID()
+{
+	return mysql_insert_id(&m_mysql);
+}