transactd 1.0.1

data/source/bzs/db/protocol/tdap/client/client.h

@@ -0,0 +1,292 @@
+#ifndef BZS_DB_PROTOCOL_TDAP_CLIENT_CLIENT_H
+#define BZS_DB_PROTOCOL_TDAP_CLIENT_CLIENT_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 "sqlBuilder.h"
+#include "request.h"
+#include <bzs/db/blobBuffer.h>
+#include <boost/thread/mutex.hpp>
+#include <bzs/db/protocol/tdap/tdapSchema.h>
+#include <bzs/netsvc/client/tcpClient.h>
+#include <bzs/db/protocol/tdap/tdapcapi.h>
+#include <bzs/db/protocol/tdap/mysql/characterset.h>
+#include <bzs/env/compiler.h>
+#include <bzs/rtl/stl_uty.h>
+#include <vector>
+#ifdef _WIN32
+#include <mbstring.h>
+#endif
+extern bzs::netsvc::client::connections* m_cons;
+
+namespace bzs
+{
+namespace db
+{
+namespace protocol
+{
+namespace tdap
+{
+namespace client
+{
+#define TABELNAME_PREFIX "dbfile="
+
+
+
+class client
+{
+
+	mutex m_mutex;
+	request m_req;
+	posblk m_tmpPbk;
+	ushort_td m_op;
+	ushort_td m_preResult;
+	int m_charsetIndexServer;
+	std::string m_sql;
+	std::string m_serverCharData;
+	std::string m_serverCharData2;
+
+	uint_td m_tmplen;
+	bool m_logout;
+	blobBuffer m_blobBuffer;
+
+
+	std::vector<char> m_sendbuf;
+	inline bzs::netsvc::client::connection* con(){return m_req.cid->con;};
+	inline void setCon(bzs::netsvc::client::connection* con){m_req.cid->con=con;};
+	
+	inline void disconnect()
+	{
+		m_req.result = !m_cons->disconnect(con());
+		if (m_req.result==0)
+			setCon(NULL);
+	}
+	std::string getHostName(const char* uri)
+	{
+		std::vector<std::string> ss ;
+		split(ss, uri, "/"); /* btrv://serverName/dbName?dbfile=xxx */
+		if (ss.size()<3)
+			return "";
+		return ss[2];
+	}
+	std::string getTableName(const char* uri)
+	{
+		if (uri)
+		{
+			mbchar* en = (mbchar*)uri;
+			mbchar* st = _mbsstr(en, (mbchar*)TABELNAME_PREFIX);
+			if (st)
+			{
+				st +=strlen(TABELNAME_PREFIX);
+				mbchar* tmp = _mbsstr(st, (mbchar*)PSEPARATOR_A);
+				if (tmp)
+					st = tmp+1;
+				en = _mbsstr(st, (mbchar*)".");
+				if (en==NULL)
+					en = st + strlen((char*)st);
+				return std::string((const char*)st, en-st);
+			}
+		}
+		return "";
+	}
+	static void addSecondCharsetData(unsigned int destCodePage, std::string& src);
+public:
+	client():m_charsetIndexServer(-1)
+	{
+	}
+	request& req(){return m_req;}
+	inline void setParam(ushort_td op, posblk*   pbk,
+		   void_td* data, uint_td* datalen,
+		   void_td* keybuf, keylen_td keylen,
+		   char_td keyNum, clientID* cid)
+		   {
+			   m_op = op;
+			   m_req.op = op;
+			   m_req.pbk = pbk;
+			   m_req.data = data;
+			   m_req.datalen = datalen;
+			   m_req.keybuf = keybuf;
+			   m_req.keylen = keylen;
+			   m_req.keyNum = keyNum;
+			   m_req.cid = cid;
+			   m_req.paramMask = 0;
+			   m_preResult = 0;
+			   m_logout = false;
+			   if (m_req.pbk==NULL) m_req.pbk = &m_tmpPbk;
+
+			}
+	inline short_td result(){return m_preResult;}
+
+	inline bool stop_if()
+	{
+		if (m_op==TD_STOP_ENGINE)
+		{
+			m_req.result = !m_cons->disconnect(con());
+			if (m_req.result==0)
+				setCon(NULL);
+			return !m_req.result;
+		}
+		return false;
+	}
+	inline void connect()
+	{
+		if (!m_req.cid->con)
+		{
+			std::string host = getHostName((const char*)m_req.keybuf);
+			if (host=="")
+				m_req.result = ERROR_TD_HOSTNAME_NOT_FOUND;
+			setCon(m_cons->connect(host));  //if error throw exception
+
+		}
+
+	}
+	inline void create()
+	{
+		m_req.paramMask = P_MASK_ALL;
+		if ((m_req.datalen==NULL) || *m_req.datalen == 0)
+			m_req.paramMask = P_MASK_KEYONLY;
+		else if (m_req.keyNum > CR_SUBOP_SWAPNAME) //-126 swap -127 is rename. -128 is drop.
+		{
+			m_req.paramMask &= ~P_MASK_POSBLK;
+			std::string name = getTableName((const char*)m_req.keybuf);
+			if ((m_req.keyNum==1)|| (m_req.keyNum==2)) //make by tabledef
+			{
+				m_sql = sqlCreateTable(name.c_str(), (tabledef*)m_req.data, m_charsetIndexServer);
+				m_req.keyNum -= 2; // 1= exists check 2 = no exists check
+			}else
+				m_sql = sqlCreateTable(name.c_str(), (fileSpec*)m_req.data, m_charsetIndexServer);
+			m_req.data =(ushort_td*) m_sql.c_str();
+			m_tmplen = (uint_td)(m_sql.size()+1);
+			m_req.datalen = &m_tmplen;
+		}else if ((m_req.keyNum == CR_SUBOP_SWAPNAME)||(m_req.keyNum == CR_SUBOP_RENAME))
+		{
+			readServerCharsetIndex();
+			//m_sql = std::string((char*)m_req.data, *m_req.datalen);
+			m_sql = (char*)m_req.data;
+			addSecondCharsetData(mysql::codePage(m_charsetIndexServer), m_sql);
+			m_req.data =(ushort_td*) m_sql.c_str();
+			m_tmplen = (uint_td)(m_sql.size()+1);
+			m_req.datalen = &m_tmplen;
+		}
+	}
+	inline void cmdConnect()
+	{
+		if ((m_req.keyNum == LG_SUBOP_CONNECT)|| (m_req.keyNum == LG_SUBOP_NEWCONNECT))
+		{
+			if (con() && con()->isConnected())
+				m_preResult = 1;
+			else
+			{
+				std::string host = getHostName((const char*)m_req.keybuf);
+				if (host=="")
+					m_preResult = ERROR_TD_HOSTNAME_NOT_FOUND;
+				setCon(m_cons->connect(host, (m_req.keyNum == LG_SUBOP_NEWCONNECT))); //if error throw exception
+				buildDualChasetKeybuf();
+			}
+		}else if (m_req.keyNum == LG_SUBOP_DISCONNECT)
+		{
+			if (con())
+				m_logout = true;
+			else if (m_op==TD_CONNECT)
+				m_preResult = 1;
+		}
+		m_req.paramMask = P_MASK_KEYONLY;
+	}
+	inline ushort_td execute()
+	{
+		if (result()==0)
+		{
+			//mutex::scoped_lock lck(m_mutex);
+			if (!con())
+				m_preResult = ERROR_TD_NOT_CONNECTED;
+			else
+			{
+				char* p = con()->sendBuffer();
+				unsigned int size = m_req.serialize(p);
+				if ((m_req.paramMask & P_MASK_BLOBBODY) && m_blobBuffer.blobs())
+					size = m_req.serializeBlobBody(&m_blobBuffer
+						, p, con()->sendBufferSize(), con()->optionalBuffers());
+				p = con()->asyncWriteRead(size);
+				m_req.parse(p);
+				if (m_logout)
+					disconnect();
+				m_preResult = m_req.result;
+			}
+
+		}
+		return result();
+	}
+	inline ushort_td getBlobBuffer(const blobHeader** data)
+	{
+		*data = m_req.blobHeader;
+		return *data ? 0:1;;
+	}
+
+	inline ushort_td addBlob(const blob* data, bool endRow)
+	{
+		if (endRow)
+		{
+			if (m_blobBuffer.fieldCount()==0)
+				m_blobBuffer.setFieldCount((unsigned int)m_blobBuffer.blobs());
+			return 0;
+		}
+		if (data==NULL)
+			m_blobBuffer.clear();
+		else
+			m_blobBuffer.addBlob(*data);
+		return 0;
+	}
+	bool readServerCharsetIndex();
+
+	bool buildDualChasetKeybuf();
+	
+};
+
+#ifndef _MSC_VER
+extern __THREAD client* __THREAD_BCB g_client;
+#else
+extern DWORD g_tlsiID;
+#endif
+
+
+inline client* getClientThread()
+{
+#ifdef _MSC_VER
+	return (client*)TlsGetValue(g_tlsiID);
+#else
+	return g_client;
+#endif	
+}
+
+inline void setClientThread(client* v)
+{
+#ifdef _MSC_VER
+	TlsSetValue(g_tlsiID, v);
+#else
+	g_client = v;
+#endif
+}
+
+
+}//namespace client
+}//namespace tdap
+}//namespace protocol
+}//namespace db
+}//namespace bzs
+
+#endif//BZS_DB_PROTOCOL_TDAP_CLIENT_CLIENT_H

data/source/bzs/db/protocol/tdap/client/connMgr.cpp

@@ -0,0 +1,144 @@
+/*=================================================================
+   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/tstring.h>
+#pragma hdrstop
+
+#include "connMgr.h"
+#include "database.h"
+
+
+#pragma package(smart_init)
+
+namespace bzs
+{
+namespace db
+{
+namespace protocol
+{
+namespace tdap
+{
+namespace client
+{
+
+
+connMgr::connMgr(database* db):nstable(db)
+{
+    m_db = db;
+}
+
+connMgr::~connMgr()
+{
+
+}
+database*  connMgr::db()const{return m_db;}
+
+void connMgr::connect(const _TCHAR* uri)
+{
+    m_db->connect(uri, true);
+    m_stat = m_db->stat();
+    if (m_stat == 0)
+        m_uri = uri;
+}
+
+void connMgr::disconnect()
+{
+    if (m_uri != _T(""))
+    {
+        m_db->disconnect(m_uri.c_str());
+        m_stat = m_db->stat();
+        if (m_stat == 0)
+            m_uri = _T("");
+    }
+}
+
+void  connMgr::allocBuffer()
+{
+    m_records.resize(64000/sizeof(connMgr::record));
+
+    m_datalen = m_buflen = (uint_td)(m_records.size()*sizeof(connMgr::record));
+    m_pdata = (void*)&m_records[0];
+    memset(m_pdata, 0, m_datalen);
+    setIsOpen(true);
+    m_keybuf = &m_params[0];
+    m_params[0] = 0;
+    m_params[1] = 0;
+    m_keylen = sizeof(m_params);
+}
+
+const connMgr::records& connMgr::connections()
+{
+    allocBuffer();
+    m_keynum = TD_STSTCS_READ;
+    tdap(TD_STASTISTICS);
+    if (m_stat == 0)
+        m_records.resize(m_datalen/sizeof(connMgr::record));
+    return m_records;
+}
+
+const connMgr::records& connMgr::databases(__int64 connid)
+{
+    allocBuffer();
+    m_keynum = TD_STSTCS_READ;
+    m_params[0] = connid;
+    m_params[1] = -1;
+    tdap(TD_STASTISTICS);
+    if (m_stat == 0)
+        m_records.resize(m_datalen/sizeof(connMgr::record));
+    return m_records;
+}
+
+const connMgr::records& connMgr::tables(__int64 connid, int dbid)
+{
+    allocBuffer();
+    m_keynum = TD_STSTCS_READ;
+    m_params[0] = connid;
+    m_params[1] = dbid;
+    tdap(TD_STASTISTICS);
+    if (m_stat == 0)
+        m_records.resize(m_datalen/sizeof(connMgr::record));
+    return m_records;
+}
+
+void connMgr::disconnectOne(__int64 connid)
+{
+    m_keynum = TD_STSTCS_DISCONNECT_ONE;
+    record& rec = m_records[0];
+    rec.conId = connid;
+    tdap(TD_STASTISTICS);
+}
+
+void connMgr::disconnectAll()
+{
+    m_keynum = TD_STSTCS_DISCONNECT_ALL;
+    tdap(TD_STASTISTICS);
+
+}
+
+short_td connMgr::stat()
+{
+    return m_stat;
+}
+
+
+
+}//namespace client
+}//namespace tdap
+}//namespace protocol
+}//namespace db
+}//namespace bzs

data/source/bzs/db/protocol/tdap/client/connMgr.h

@@ -0,0 +1,82 @@
+#ifndef BZS_DB_PROTOCOL_TDAP_CLIENT_CTDCONNMGR_H
+#define BZS_DB_PROTOCOL_TDAP_CLIENT_CTDCONNMGR_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 "nsTable.h"
+#include <bzs/db/transactd/connectionRecord.h>
+#include <vector>
+
+
+namespace bzs
+{
+namespace db
+{
+namespace protocol
+{
+namespace tdap
+{
+namespace client
+{
+
+class database;
+#pragma warning(disable:4251)
+
+class AGRPACK connMgr : private nstable
+{
+public:
+    typedef  bzs::db::transactd::connection::record  record;
+    typedef  std::vector<record> records;
+protected:
+
+
+private:
+    std::vector<record> m_records;
+	__int64 m_params[2];
+    database* m_db;
+    std::_tstring m_uri;
+    void allocBuffer();
+    void writeRecordData(){};
+    void onReadAfter(){};
+
+     ~connMgr();
+public:
+	explicit connMgr(database* db);
+
+    void connect(const _TCHAR* uri);
+    void disconnect();
+    const records& connections();
+    const records& databases(__int64 connid);
+    const records& tables(__int64 connid, int dbid);
+    void disconnectOne(__int64 connid);
+    void disconnectAll();
+    short_td stat();
+
+    database* db()const;
+    using nstable::tdapErr;
+};
+
+#pragma warning(default:4251)
+
+
+}//namespace client
+}//namespace tdap
+}//namespace protocol
+}//namespace db
+}//namespace bzs
+#endif //BZS_DB_PROTOCOL_TDAP_CLIENT_CTDCONNMGR_H