tagging ychat-0.6ychat-0.6

1 files changed, 190 insertions, 0 deletions

diff --git a/mman.cpp b/mman.cpp
new file mode 100644
index 0000000..5576723
--- /dev/null
+++ b/mman.cpp
@@ -0,0 +1,190 @@
+#ifndef MMAN_CXX
+#define MMAN_CXX
+
+#include "mman.h"
+#include "mcon.h"
+#include "mcon.h"
+#include "s_mutx.h"
+#include "s_ncur.h"
+#include "s_tool.h"
+
+using namespace std;
+
+mman::mman(int i_initial, int i_max)
+{
+    pthread_mutex_init( &mut_i_used_con  , NULL);
+
+    this->i_used_connections=0;
+
+    if( i_max > MAXMSQL )
+    {
+#ifdef VERBOSE
+    cerr << MYLIMIT << MAXMSQL << endl;
+#endif
+
+    i_max = MAXMSQL;
+
+    }
+
+    if( i_initial > i_max )
+        i_initial = i_max;
+
+    this->i_initial_connections = i_initial;
+    this->i_max_connections = i_max;
+
+#ifdef VERBOSE
+    cout << MYINITC << i_initial << endl
+         << MYINITM << i_max << endl;
+#endif
+}
+void mman::init( string host, string user, string passwd, string db, unsigned int port)
+{
+    this->s_host=host;
+    this->s_user=user;
+    this->s_pass=passwd;
+    this->s_db=db;
+    this->i_port=port;
+
+    for(int i=0; i<this->i_initial_connections;i++)
+        this->mysql.push_back(new_connection( ));
+
+}
+
+#ifdef NCURSES
+void
+mman::print_init_ncurses()
+{
+        string s_tmp( MYINITC );
+        s_tmp.append( s_tool::int2string( i_initial_connections ) );
+        s_ncur::get
+            ().print( s_tmp );
+
+        string s_tmp2( MYINITM );
+        s_tmp2.append( s_tool::int2string( i_max_connections ) );
+        s_ncur::get
+            ().print( s_tmp2 );
+        print_used_connections(1);
+}
+void
+mman::print_used_connections( bool b_refresh )
+{
+    pthread_mutex_lock  ( &s_mutx::get().mut_stdout );
+    mvprintw( NCUR_MYSQL_X,NCUR_MYSQL_Y, "MySQL: %d ", i_used_connections);
+    if ( b_refresh )
+     refresh();
+    pthread_mutex_unlock( &s_mutx::get().mut_stdout );
+}
+#endif
+
+mman::~mman()
+{
+
+    for(int i = 0; i < this->mysql.size();i++)
+    {
+        if(this->mysql[i] != NULL)
+            mysql_close(this->mysql[i]);
+    }
+
+    pthread_mutex_destroy( &mut_i_used_con );
+}
+MYSQL *mman::get_connection()
+{
+    if(!this->mysql.empty())
+    {
+        MYSQL *x=this->mysql[this->mysql.size()-1];
+        this->mysql.pop_back();
+        if(mysql_ping(x)==0)
+        {
+            pthread_mutex_lock  ( &mut_i_used_con );
+            this->i_used_connections--;
+#ifdef NCURSES
+            print_used_connections(1);
+#endif
+            pthread_mutex_unlock( &mut_i_used_con );
+            mysql_close(x);
+            return new_connection();
+        }
+        return x;
+    }
+
+#ifdef NCURSES
+    {
+        s_ncur::get
+            ().print( MYERROR );
+    }
+#endif
+#ifdef SERVMSG
+    cerr << MYERROR << endl;
+#endif
+
+    return NULL;
+}
+MYSQL *mman::new_connection( )
+{
+
+    pthread_mutex_lock  ( &mut_i_used_con );
+    if(this->i_used_connections>this->i_max_connections)
+    {
+#ifdef NCURSES
+	s_ncur::get().shutdown();
+    	cerr << MYERRO2 << endl;
+#endif
+#ifdef SERVMSG
+    	cerr << MYERRO2 << endl;
+#endif
+        return NULL;
+    }
+    pthread_mutex_unlock( &mut_i_used_con );
+
+    MYSQL *ms = mysql_init(NULL);
+    if(!ms)
+    {
+#ifdef NCURSES
+	s_ncur::get().shutdown();
+    	cerr << MYERRO1 << endl;
+#endif
+#ifdef SERVMSG
+    	cerr << MYERRO1 << endl;
+#endif
+        exit(1);
+    }
+    if(mysql_real_connect( ms, this->s_host.c_str(), this->s_user.c_str(), this->s_pass.c_str(), this->s_db.c_str()
+                           , this->i_port, NULL, 0 )==NULL)
+    {
+#ifdef NCURSES
+	s_ncur::get().shutdown();
+        cerr << MYMANAG << mysql_error(ms) << endl;
+#endif
+#ifdef SERVMSG
+        cerr << MYMANAG << mysql_error(ms) << endl;
+#endif
+        exit(1);
+    }
+
+    pthread_mutex_lock  ( &mut_i_used_con );
+    this->i_used_connections++;
+#ifdef NCURSES
+    print_used_connections(0);
+#endif
+    pthread_mutex_unlock( &mut_i_used_con );
+    return ms;
+}
+void mman::free_connection( MYSQL *msql)
+{
+    pthread_mutex_lock  ( &mut_i_used_con );
+    this->i_used_connections--;
+#ifdef NCURSES
+    print_used_connections(1);
+#endif
+    pthread_mutex_unlock( &mut_i_used_con );
+
+    if(mysql_ping( msql )==0)
+    {
+        mysql_close( msql );
+        this->mysql.push_back( msql );
+    }
+    else
+        this->mysql.push_back( new_connection() );
+}
+
+#endif