~drizzle-trunk/drizzle/development

/* Copyright (C) 2008 PrimeBase Technologies GmbH, Germany

 *

 * PrimeBase Media Stream for MySQL

 *

 * 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA

 *

 * Original author: Paul McCullagh

 * Continued development: Barry Leslie

 *

 * 2007-05-20

 *

 * H&G2JCtL

 *

 * Table handler.

 *

 */

#ifdef USE_PRAGMA_IMPLEMENTATION

#pragma implementation				// gcc: Class implementation

#endif

#ifdef DRIZZLED

#include <config.h>

#include <drizzled/current_session.h>

#include <drizzled/common.h>

#include <drizzled/plugin.h>

#include <drizzled/field.h>

#include <drizzled/session.h>

#include <drizzled/data_home.h>

#include <drizzled/error.h>

#include <drizzled/table.h>

#include <drizzled/plugin/transactional_storage_engine.h>

#include <drizzled/named_savepoint.h>

#define my_strdup(a,b) strdup(a)

using namespace drizzled;

using namespace drizzled::plugin;

#include "cslib/CSConfig.h"

#else

#include "cslib/CSConfig.h"

#include "mysql_priv.h"

#include <mysql/plugin.h>

#include <my_dir.h>

#endif 

#include <stdlib.h>

#include <time.h>

#include <inttypes.h>

#include "defs_ms.h"

#include "cslib/CSDefs.h"

#include "cslib/CSObject.h"

#include "cslib/CSGlobal.h"

#include "cslib/CSThread.h"

#include "cslib/CSStrUtil.h"

#include "cslib/CSLog.h"

#include "engine_ms.h"	

#include "ha_pbms.h"

#include "network_ms.h"

#include "connection_handler_ms.h"

#include "open_table_ms.h"

#include "database_ms.h"

#include "temp_log_ms.h"

#include "system_table_ms.h"

#include "mysql_ms.h"

#include "discover_ms.h"

#include "metadata_ms.h"

#include "transaction_ms.h"

#include "systab_httpheader_ms.h"

#include "system_table_ms.h"

#include "parameters_ms.h"

#include "pbmsdaemon_ms.h"

#include "version_ms.h"

/* Note: 'new' used here is NOT CSObject::new which is a DEBUG define*/

#ifdef new

#undef new

#endif

#ifdef DRIZZLED

static int pbms_done_func(void *);

class PBMSStorageEngine : public drizzled::plugin::TransactionalStorageEngine {

public:

	PBMSStorageEngine()

	: TransactionalStorageEngine(std::string("PBMS"), HTON_NO_FLAGS | HTON_HIDDEN) {}

	~PBMSStorageEngine()

	{

		pbms_done_func(NULL);

	}

	int close_connection(Session *);

	int doStartTransaction(Session *session, start_transaction_option_t options);

	int doCommit(Session *, bool);

	int doRollback(Session *, bool);

	Cursor *create(Table& table);

	bool doDropSchema(const drizzled::identifier::Schema&);

	/*

	* Indicates to a storage engine the start of a

	* new SQL statement.

	*/

	void doStartStatement(Session *session)

	{

		(void) session;

	}

	/*

	* Indicates to a storage engine the end of

	* the current SQL statement in the supplied

	* Session.

	*/

	void doEndStatement(Session *session)

	{

		(void) session;

	}

	int doCreateTable(Session&, Table&, const identifier::Table& ident, const drizzled::message::Table& );	

	int doDropTable(Session &, const identifier::Table& );

	int doRenameTable(Session&, const identifier::Table &from, const identifier::Table &to);

        void doGetTableIdentifiers(drizzled::CachedDirectory &dir,

                                   const drizzled::identifier::Schema &schema,

                                   drizzled::identifier::table::vector &set_of_identifiers) 

	{

		std::set<std::string> set_of_names;

		doGetTableNames(dir, schema, set_of_names);

		for (std::set<std::string>::iterator set_iter = set_of_names.begin(); set_iter != set_of_names.end(); ++set_iter)

		{

			set_of_identifiers.push_back(identifier::Table(schema, *set_iter));

		}

	}

	void doGetTableNames(CachedDirectory&, 

					const identifier::Schema &schema, 

					std::set<std::string> &set_of_names) 

	{

		bool isPBMS = schema.compare("PBMS");

		if (isPBMS || PBMSParameters::isBLOBDatabase(schema.getSchemaName().c_str()))

			PBMSSystemTables::getSystemTableNames(isPBMS, set_of_names);

	}

	int doSetSavepoint(Session *thd, NamedSavepoint &savepoint);

	int doRollbackToSavepoint(Session *session, NamedSavepoint &savepoint);

	int doReleaseSavepoint(Session *session, NamedSavepoint &savepoint);

	const char **bas_ext() const;

  int doGetTableDefinition(Session&, const identifier::Table &identifier,

                                          drizzled::message::Table &table_proto)

  {

		int err;

		const char *tab_name = identifier.getTableName().c_str();

		// Set some required table proto info:

 		table_proto.set_schema(identifier.getSchemaName().c_str());

		table_proto.set_creation_timestamp(0);

		table_proto.set_update_timestamp(0);

		err = PBMSSystemTables::getSystemTableInfo(tab_name, table_proto);

		if (err)

			return err;

		return EEXIST;

  }

	bool doDoesTableExist(Session&, const identifier::Table &identifier)

	{

		const char *tab_name = identifier.getTableName().c_str();

		const char *db_name = identifier.getSchemaName().c_str();

		bool isPBMS = identifier.getSchemaName().compare("PBMS");

		if (isPBMS || PBMSParameters::isBLOBDatabase(db_name)) {

			return PBMSSystemTables::isSystemTable(isPBMS, tab_name);									 

		}

		return false;		

	}

};

PBMSStorageEngine	*pbms_hton;

#else

handlerton		*pbms_hton;

#endif

static const char *ha_pbms_exts[] = {

	NullS

};

/*

 * ---------------------------------------------------------------

 * UTILITIES

 */

#ifndef DRIZZLED

void pbms_take_part_in_transaction(void *thread)

{

	THD			*thd;

	if ((thd = (THD *) thread)) {

		trans_register_ha(thd, true, pbms_hton); 

	}

}

#endif

#ifdef DRIZZLED

const char **PBMSStorageEngine::bas_ext() const

#else

const char **ha_pbms::bas_ext() const

#endif

{

	return ha_pbms_exts;

}

#ifdef DRIZZLED

int PBMSStorageEngine::close_connection(Session *thd)

{

#else

static int pbms_close_connection(handlerton *hton, THD* thd)

{

	(void)hton;

#endif

	MSEngine::closeConnection(thd);

	return 0;

}

/*

 * ---------------------------------------------------------------

 * HANDLER INTERFACE

 */

#ifdef DRIZZLED

Cursor *PBMSStorageEngine::create(Table& table)

{

	PBMSStorageEngine * const hton = this;

	return new ha_pbms(hton, table);

}

#else

static handler *pbms_create_handler(handlerton *hton, TABLE_SHARE *table, MEM_ROOT *mem_root)

{

	return new (mem_root) ha_pbms(hton, table);

}

#endif

#ifdef DRIZZLED

int PBMSStorageEngine::doStartTransaction(Session *thd, start_transaction_option_t options)

{

	(void)thd;

	(void)options;

	return 0;

}

int PBMSStorageEngine::doCommit(Session *thd, bool all)

{

#else

static int pbms_commit(handlerton *, THD *thd, bool all)

{

#endif

	int			err = 0;

	CSThread	*self;

	PBMSResultRec result;

	// I am not interesed in single statement transactions.

	if (all == false)

		return 0;

	if (MSEngine::enterConnection(thd, &self, &result, false))

		return 0;

	inner_();

	try_(a) {

		MSTransactionManager::commit();

	}

	catch_(a) {

		err = MSEngine::exceptionToResult(&self->myException, &result);

	}

	cont_(a);

	self->myIsAutoCommit = true;

	return_(err);

}

#ifdef DRIZZLED

int PBMSStorageEngine::doRollback(THD *thd, bool all)

{

#else

static int pbms_rollback(handlerton *, THD *thd, bool all)

{

#endif

	int			err = 0;

	CSThread	*self;

	PBMSResultRec result;

	UNUSED(all);

	if (MSEngine::enterConnection(thd, &self, &result, false))

		return 0;

	inner_();

	try_(a) {

		MSTransactionManager::rollback();

	}

	catch_(a) {

		err = MSEngine::exceptionToResult(&self->myException, &result);

	}

	cont_(a);

	self->myIsAutoCommit = true;

	return_(err);

}

#ifdef DRIZZLED

int PBMSStorageEngine::doSetSavepoint(Session *thd, NamedSavepoint &savepoint)

{

	int			err = 0;

	CSThread	*self;

	PBMSResultRec result;

	if (MSEngine::enterConnection(thd, &self, &result, false))

		return 0;

	inner_();

	try_(a) {

		MSTransactionManager::setSavepoint(savepoint.getName().c_str());

	}

	catch_(a) {

		err = MSEngine::exceptionToResult(&self->myException, &result);

	}

	cont_(a);

	return_(err);

}

int PBMSStorageEngine::doRollbackToSavepoint(Session *session, NamedSavepoint &savepoint)

{

	int			err = 0;

	CSThread	*self;

	PBMSResultRec result;

	if (MSEngine::enterConnection(session, &self, &result, false))

		return 0;

	inner_();

	try_(a) {

		MSTransactionManager::rollbackTo(savepoint.getName().c_str());

	}

	catch_(a) {

		err = MSEngine::exceptionToResult(&self->myException, &result);

	}

	cont_(a);

	return_(err);

}

int PBMSStorageEngine::doReleaseSavepoint(Session *session, NamedSavepoint &savepoint)

{

	int			err = 0;

	CSThread	*self;

	PBMSResultRec result;

	if (MSEngine::enterConnection(session, &self, &result, false))

		return 0;

	inner_();

	try_(a) {

		MSTransactionManager::releaseSavepoint(savepoint.getName().c_str());

	}

	catch_(a) {

		err = MSEngine::exceptionToResult(&self->myException, &result);

	}

	cont_(a);

	return_(err);

}

#else

static int pbms_savepoint_set(handlerton *hton, THD *thd, void *sv)

{

	int			err = 0;

	CSThread	*self;

	PBMSResultRec result;

	if (MSEngine::enterConnection(thd, &self, &result, false))

		return 0;

	*((uint32_t*)sv) = self->myStmtCount;

	return 0;	

}

static int pbms_savepoint_rollback(handlerton *hton, THD *thd, void *sv)

{

	int			err = 0;

	CSThread	*self;

	PBMSResultRec result;

	if (MSEngine::enterConnection(thd, &self, &result, false))

		return 0;

	inner_();

	try_(a) {

		MSTransactionManager::rollbackToPosition(*((uint32_t*)sv));

	}

	catch_(a) {

		err = MSEngine::exceptionToResult(&self->myException, &result);

	}

	cont_(a);

	return_(err);

}

static int pbms_savepoint_release(handlerton *hton, THD *thd, void *sv)

{

	return 0;

}

#endif

#ifdef DRIZZLED

bool  PBMSStorageEngine::doDropSchema(const drizzled::identifier::Schema &schema)

{

	CSThread *self;

	PBMSResultRec result;

	if (MSEngine::enterConnectionNoThd(&self, &result))

		return false;

	inner_();

	try_(a) {

		MSDatabase::dropDatabase(schema.getSchemaName().c_str());

	}

	catch_(a);

	self->logException();

	cont_(a);

	return_(false);

}

#else

static void pbms_drop_database(handlerton *, char *path)

{

	CSThread *self;

	char db_name[PATH_MAX];

	PBMSResultRec result;

	if (MSEngine::enterConnectionNoThd(&self, &result))

		return;

	inner_();

	cs_strcpy(PATH_MAX, db_name, cs_last_directory_of_path(path));

	cs_remove_dir_char(db_name);

	try_(a) {

		MSDatabase::dropDatabase(db_name);

	}

	catch_(a);

	self->logException();

	cont_(a);

	exit_();

}

#endif

static bool pbms_started = false;

#ifdef DRIZZLED

int pbms_init_func(module::Context &registry);

int pbms_init_func(module::Context &registry)

#else

int pbms_init_func(void *p);

int pbms_discover_system_tables(handlerton *hton, THD* thd, const char *db, const char *name, uchar **frmblob, size_t *frmlen);

int pbms_init_func(void *p)

#endif

{

	PBMSResultRec		result;

	int					err;

	int					my_res = 0;

	CSThread			*thread;

	ASSERT(!pbms_started);

	pbms_started = false;

	PBMSDaemon::setDaemonState(PBMSDaemon::DaemonStartUp);

	{

		char info[120];

		snprintf(info, 120, "PrimeBase Media Stream (PBMS) Daemon %s loaded...", PBMSVersion::getCString());

		CSL.logLine(NULL, CSLog::Protocol, info);

	}

	CSL.logLine(NULL, CSLog::Protocol, "Barry Leslie, PrimeBase Technologies GmbH, http://www.primebase.org");

	if ((err = MSEngine::startUp(&result))) {

		CSL.logLine(NULL, CSLog::Error, result.mr_message);

		PBMSDaemon::setDaemonState(PBMSDaemon::DaemonError);

		return(1);

	}

#ifdef DRIZZLED

		pbms_hton= new PBMSStorageEngine();

		registry.add(pbms_hton);

#else

	pbms_hton = (handlerton *) p;

	pbms_hton->state = SHOW_OPTION_YES;

	pbms_hton->close_connection = pbms_close_connection; /* close_connection, cleanup thread related data. */

	pbms_hton->create = pbms_create_handler;

	pbms_hton->flags = HTON_CAN_RECREATE | HTON_HIDDEN;

	pbms_hton->drop_database = pbms_drop_database; /* Drop a database */

	pbms_hton->discover = pbms_discover_system_tables;

	pbms_hton->commit = pbms_commit; /* commit */

	pbms_hton->rollback = pbms_rollback; /* rollback */

	pbms_hton->savepoint_offset = 4;

	pbms_hton->savepoint_set = pbms_savepoint_set;

	pbms_hton->savepoint_rollback = pbms_savepoint_rollback; 

	pbms_hton->savepoint_release = pbms_savepoint_release; 

#endif

	/* Startup the Media Stream network: */

	cs_init_memory();

	CSThread::startUp();

	if (!(thread = CSThread::newCSThread())) {

		CSException::logOSError(CS_CONTEXT, ENOMEM);

		PBMSDaemon::setDaemonState(PBMSDaemon::DaemonError);

		return(1);

	}

	if (!CSThread::attach(thread)) {

		PBMSDaemon::setDaemonState(PBMSDaemon::DaemonError);

		thread->myException.log(NULL);

		CSThread::shutDown();

		cs_exit_memory();

		MSEngine::shutDown();

		return(1);

	}

	enter_();

	try_(a) {

		thread->threadName = CSString::newString("startup");

		MSDatabase::startUp(PBMSParameters::getDefaultMetaDataHeaders());

		MSTableList::startUp();

		MSSystemTableShare::startUp();

		MSNetwork::startUp(PBMSParameters::getPortNumber());

		MSTransactionManager::startUp();

		MSNetwork::startNetwork();

	}

	catch_(a) {

		self->logException();

		my_res = 1;

	}

	cont_(a);

	if (my_res) {

		try_(b) {

			MSNetwork::shutDown();

			MSTransactionManager::shutDown();

			MSSystemTableShare::shutDown();

			MSDatabase::stopThreads();

			MSTableList::shutDown();

			MSDatabase::shutDown();

			CSThread::shutDown();

		}

		catch_(b) {

			self->logException();

		}

		cont_(b);

	}

	outer_();

	CSThread::detach(thread);

	if (my_res) {

		cs_exit_memory();

		MSEngine::shutDown();

	}

	else {

		srandom(time(NULL));

		pbms_started = true;

	}

	if (pbms_started)

		PBMSDaemon::setDaemonState(PBMSDaemon::DaemonRunning);

	else

		PBMSDaemon::setDaemonState(PBMSDaemon::DaemonError);

	return(my_res);

}

#ifdef DRIZZLED

static int pbms_done_func(void *)

#else

int pbms_done_func(void *)

#endif

{

	CSThread	*thread;

	if (!pbms_started)

		return 0;

	PBMSDaemon::setDaemonState(PBMSDaemon::DaemonShuttingDown);

	CSL.logLine(NULL, CSLog::Protocol, "PrimeBase Media Stream (PBMS) Daemon shutdown...");

	/* Shutdown the Media Stream network. */

	if (!(thread = CSThread::newCSThread()))

		CSException::logOSError(CS_CONTEXT, ENOMEM);

	else if (!CSThread::attach(thread))

		thread->myException.log(NULL);

	else {

		enter_();

		try_(a) {

			thread->threadName = CSString::newString("shutdown");

			MSNetwork::shutDown();

			MSSystemTableShare::shutDown();

			/* Ensure that the database threads are stopped before

			 * freeing the tables.

			 */

			MSDatabase::stopThreads();

			MSTableList::shutDown();

			/* Databases must be shutdown after table because tables

			 * have references to repositories.

			 */

			MSDatabase::shutDown();

			/* Shutdown the transaction manager after the databases

			 * incase they want to commit or rollback a transaction.

			 */

			MSTransactionManager::shutDown();

		}

		catch_(a) {

			self->logException();

		}

		cont_(a);

		outer_();

		CSThread::shutDown();

		CSThread::detach(thread);

	}

	MSEngine::shutDown();

	cs_exit_memory();

	CSL.logLine(NULL, CSLog::Protocol, "PrimeBase Media Stream (PBMS) Daemon shutdown completed");

	pbms_started = false;

	return(0);

}

#ifdef DRIZZLED

ha_pbms::ha_pbms(handlerton *hton, Table& table_arg) : handler(*hton, table_arg),

#else

ha_pbms::ha_pbms(handlerton *hton, TABLE_SHARE *table_arg) : handler(hton, table_arg),

#endif

ha_open_tab(NULL),

ha_error(0)

{

	memset(&ha_result, 0, sizeof(PBMSResultRec));

}

#ifndef DRIZZLED

MX_TABLE_TYPES_T ha_pbms::table_flags() const

{

	return (

		/* We need this flag because records are not packed

		 * into a table which means #ROWID != offset

		 */

		HA_REC_NOT_IN_SEQ |

		HA_CAN_SQL_HANDLER |

#if MYSQL_VERSION_ID > 50119

		/* We can do row logging, but not statement, because

		 * MVCC is not serializable!

		 */

		HA_BINLOG_ROW_CAPABLE |

#endif

		/*

		 * Auto-increment is allowed on a partial key.

		 */

		0);

}

#endif

int ha_pbms::open(const char *table_path, int , uint )

{

	CSThread *self;

	if ((ha_error = MSEngine::enterConnection(current_thd, &self, &ha_result, true)))

		return 1;

	inner_();

	try_(a) {

#ifdef DRIZZLED

		ha_open_tab = MSSystemTableShare::openSystemTable(table_path, getTable());

		ha_lock.init(&ha_open_tab->myShare->myThrLock);

#else

		ha_open_tab = MSSystemTableShare::openSystemTable(table_path, table);

		thr_lock_data_init(&ha_open_tab->myShare->myThrLock, &ha_lock, NULL);

#endif

		ref_length = ha_open_tab->getRefLen();

	}

	catch_(a) {