transactd 1.0.1

data/source/bzs/db/transactd/appModule.h

@@ -0,0 +1,80 @@
+#ifndef BZS_DB_TRANSACTD_APPMODULE_H
+#define BZS_DB_TRANSACTD_APPMODULE_H
+/*=================================================================
+   Copyright (C) 2012 2013 BizStation Corp All rights reserved.
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License
+   as published by the Free Software Foundation; either version 2
+   of the License, or (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software 
+   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  
+   02111-1307, USA.
+=================================================================*/
+
+#include <bzs/netsvc/server/IAppModule.h>
+#include <bzs/db/protocol/ICommandExecuter.h>
+
+
+class THD;
+
+namespace bzs
+{
+namespace db
+{
+namespace transactd
+{
+
+/** Feature
+ *  -Client authentication.
+ *  -Check tcp recieve of end of data.
+ *  -Tcp connection disconnect function.
+ *  -Execute client command.
+ *  -Anser of shutdown.
+ */
+class module :public netsvc::server::IAppModule,  private boost::noncopyable 
+{
+	friend class connManager;
+	mutable boost::mutex m_mutex;
+	boost::shared_ptr<protocol::ICommandExecuter> m_commandExecuter;
+	bool m_useThreadPool;
+	const boost::asio::ip::tcp::endpoint m_endpoint;
+	bzs::netsvc::server::iconnection* m_connection;
+	const char* m_readBuf;
+	size_t m_readSize;
+	bool perseLineEnd(const char* p, size_t size)const;
+	size_t onRead(const char* data, size_t size, bool& complete);
+	size_t onAccept(char* message, size_t bufsize);
+public:
+	module(const boost::asio::ip::tcp::endpoint& endpoint
+			, bzs::netsvc::server::iconnection* connection
+			, bool tpool,int type);
+	~module();
+	void reset();
+	int execute(char* result, size_t& size, netsvc::server::buffers*  optionalData);
+	void cleanup(){ m_commandExecuter->cleanup();};
+	bool isShutDown(){return m_commandExecuter->isShutDown();}
+	bool checkHost(const char* hostCheckname);
+	void disconnect();
+	boost::mutex& mutex()const{return m_mutex;};
+	
+};
+
+}//namespace transactd
+}//namespace db
+}//namespace bzs
+
+
+#endif //BZS_DB_TRANSACTD_APPMODULE_H
+
+
+
+
+

data/source/bzs/db/transactd/connManager.cpp

@@ -0,0 +1,201 @@
+/*=================================================================
+   Copyright (C) 2013 BizStation Corp All rights reserved.
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License
+   as published by the Free Software Foundation; either version 2
+   of the License, or (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software 
+   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  
+   02111-1307, USA.
+=================================================================*/
+
+#include "appModule.h"
+#include <bzs/db/protocol/tdap/mysql/tdapCommandExecuter.h>
+#include "connManager.h"
+
+namespace bzs
+{
+	using namespace netsvc::server;
+namespace db
+{
+	using namespace protocol::tdap::mysql;
+	using namespace engine::mysql;
+
+namespace transactd
+{
+
+connManager::~connManager()
+{
+
+
+}
+
+const module* getMod(unsigned __int64 conid)
+{
+	for (size_t i=0;i<modules.size();i++)
+	{
+		const module* mod = dynamic_cast<module*>(modules[i]);
+		if ((unsigned __int64)mod == conid)
+			return mod;
+	}
+	return NULL;
+}
+
+const database* connManager::getDatabase(const module* mod, int dbid)const
+{
+
+	igetDatabases* dbm = dynamic_cast<igetDatabases*>(mod->m_commandExecuter.get());
+	const databases& dbs = dbm->dbs();
+	if (dbid < (int)dbs.size())
+		return dbs[dbid].get();
+	return NULL;
+}
+
+void connManager::getConnectionList()const
+{
+	for (size_t i=0;i<modules.size();i++)
+	{
+		const module* mod = dynamic_cast<module*>(modules[i]);
+		if (mod && ((unsigned __int64)mod != m_me))
+		{
+			
+			
+			m_records.push_back(connection::record());
+			connection::record& rec = m_records[m_records.size()-1];
+			rec.conId = (unsigned __int64)mod;
+			boost::asio::ip::address adr = mod->m_endpoint.address();
+			strncpy_s(rec.name,64, adr.to_string().c_str(), 64);
+			
+		}
+	}
+}
+
+void connManager::getDatabaseList(const module* mod)const
+{
+	igetDatabases* dbm = dynamic_cast<igetDatabases*>(mod->m_commandExecuter.get());
+	const databases& dbs = dbm->dbs();
+	for (size_t j=0;j<dbs.size();j++)
+	{
+		
+		if (dbs[j])
+		{
+			m_records.push_back(connection::record());
+			connection::record& rec = m_records[m_records.size()-1];
+			rec.conId = (unsigned __int64)mod;
+			rec.cid = dbs[j]->clientID();
+			rec.dbid = (unsigned short)j;
+			strncpy_s(rec.name, 64, dbs[j]->name().c_str(),64);
+		}
+	}
+}
+
+const connManager::records& connManager::getRecords(unsigned __int64 conid, int dbid) const
+{
+	boost::mutex::scoped_lock lck(modulesMutex);
+	
+	if (conid==0)
+		getConnectionList();
+	else
+	{
+		const module* mod = getMod(conid);
+		if (mod)
+		{			
+			boost::try_mutex::scoped_try_lock m(mod->mutex(), boost::try_to_lock_t());
+			if (m.owns_lock())
+			{
+				if (dbid < 0)
+					getDatabaseList(mod);
+				else
+				{
+					const database* db = getDatabase(mod, dbid);
+					if (db)
+					{
+						const std::vector<boost::shared_ptr<table> >& tables = db->tables();
+						for (size_t k=0;k<tables.size();k++)
+						{
+							const table* tb = tables[k].get();
+							if (tb)
+							{
+								m_records.push_back(connection::record());
+								connection::record& rec = m_records[m_records.size()-1];
+								rec.conId = (unsigned __int64)mod;
+								rec.cid = db->clientID();
+					
+								strncpy_s(rec.name, 64, tb->name().c_str(),64);
+							}
+						}
+					}
+				}
+			}
+		}
+	}
+	return m_records;
+}
+
+void connManager::doDisconnect(unsigned __int64 conid)
+{
+	module* mod = const_cast<module*>(getMod(conid));
+	if (mod)
+	{
+		igetDatabases* dbm = dynamic_cast<igetDatabases*>(mod->m_commandExecuter.get());
+		const databases& dbs = dbm->dbs();
+		if (dbs.size())
+			mod->disconnect();
+	}
+}
+
+void connManager::disconnect(unsigned __int64 conid)
+{
+	boost::try_mutex::scoped_lock m(modulesMutex, boost::try_to_lock_t());
+	
+	if (m.owns_lock())
+		doDisconnect(conid);
+	else 
+	{
+		for (int i=0;i<20;i++)
+		{
+			if (m.try_lock())
+				return doDisconnect(conid);	
+			Sleep(100);
+		}
+	}
+}
+
+void connManager::doDisconnectAll()
+{
+	for (size_t i=0;i<modules.size();i++)
+	{
+		const module* mod = dynamic_cast<module*>(modules[i]);
+		doDisconnect((unsigned __int64)mod);
+	}	
+}
+
+void connManager::disconnectAll()
+{
+	boost::try_mutex::scoped_lock m(modulesMutex, boost::try_to_lock_t());
+	
+	if (m.owns_lock())
+		doDisconnectAll();
+	else 
+	{
+		for (int i=0;i<20;i++)
+		{
+			if (m.try_lock())
+				return doDisconnectAll();;	
+			Sleep(100);
+		}
+	}
+	
+}
+
+}//namespace transactd
+}//namespace db
+}//namespace bzs

data/source/bzs/db/transactd/connManager.h

@@ -0,0 +1,60 @@
+#ifndef BZS_DB_TRANSACTD_CONNMANAGER_H
+#define BZS_DB_TRANSACTD_CONNMANAGER_H
+/*=================================================================
+   Copyright (C) 2013 BizStation Corp All rights reserved.
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License
+   as published by the Free Software Foundation; either version 2
+   of the License, or (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software 
+   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  
+   02111-1307, USA.
+=================================================================*/
+#include "connectionRecord.h"
+#include <bzs/netsvc/server/IAppModule.h>
+
+
+namespace bzs
+{
+namespace db
+{
+	namespace engine{namespace mysql{class database;}}
+namespace transactd
+{
+
+class module;
+
+class connManager
+{
+public:
+
+	typedef std::vector<connection::record> records;
+private:
+	mutable records m_records;
+	__int64 m_me;
+	void getConnectionList()const;
+	void getDatabaseList(const module* mod) const;
+	const bzs::db::engine::mysql::database* getDatabase(const module* mod, int dbid)const;
+	void doDisconnect(unsigned __int64 conid);
+	void doDisconnectAll();
+public:
+	connManager(unsigned __int64 me):m_me(me){};
+	virtual ~connManager();
+	const records& getRecords(unsigned __int64 conid, int dbid) const;
+	void disconnect(unsigned __int64 conid);
+	void disconnectAll();
+};
+
+}//namespace transactd
+}//namespace db
+}//namespace bzs
+
+#endif//BZS_DB_TRANSACTD_CONNMANAGER_H

data/source/bzs/db/transactd/connectionRecord.h

@@ -0,0 +1,69 @@
+#ifndef BZS_DB_TRANSACTD_CONNECTIONRECORD_H
+#define BZS_DB_TRANSACTD_CONNECTIONRECORD_H
+/*=================================================================
+   Copyright (C) 2013 BizStation Corp All rights reserved.
+
+   This program is free software; you can redistribute it and/or
+   modify it under the terms of the GNU General Public License
+   as published by the Free Software Foundation; either version 2
+   of the License, or (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software 
+   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  
+   02111-1307, USA.
+=================================================================*/
+#include <bzs/env/compiler.h>
+#include <bzs/env/crosscompile.h>
+
+#pragma option -a-
+pragma_pack1
+
+namespace bzs
+{
+namespace db
+{
+namespace transactd
+{
+namespace connection
+{	
+	struct record
+	{
+		record():conId(0),dbid(0),cid(0)
+		{
+			name[0] = 0x00;
+			status = 0;
+		}
+		__int64 conId;
+		unsigned int	cid;
+		unsigned short  dbid;
+		char name[64];
+		union
+		{
+			char  status;
+			struct
+			{
+				char inTransaction:1;
+				char inSnapshot   :1;
+				char openNormal   :1;
+				char openReadOnly :1;
+				char openEx       :1;
+				char dummy        :3;
+			};
+		};
+	};
+
+}//connection
+}//transactd
+}//db
+}//bzs
+	
+#pragma option -a
+pragma_pop
+	
+#endif //BZS_DB_TRANSACTD_CONNECTIONRECORD_H