From 3a160a80c788d81e48acf19a2cf68f29cf125dae Mon Sep 17 00:00:00 2001 From: Karen Arutyunov Date: Thu, 25 Jan 2024 18:52:59 +0300 Subject: Turn libodb-mysql repository into package for muti-package repository --- libodb-mysql/odb/mysql/connection.hxx | 231 ++++++++++++++++++++++++++++++++++ 1 file changed, 231 insertions(+) create mode 100644 libodb-mysql/odb/mysql/connection.hxx (limited to 'libodb-mysql/odb/mysql/connection.hxx') diff --git a/libodb-mysql/odb/mysql/connection.hxx b/libodb-mysql/odb/mysql/connection.hxx new file mode 100644 index 0000000..31b24ed --- /dev/null +++ b/libodb-mysql/odb/mysql/connection.hxx @@ -0,0 +1,231 @@ +// file : odb/mysql/connection.hxx +// license : GNU GPL v2; see accompanying LICENSE file + +#ifndef ODB_MYSQL_CONNECTION_HXX +#define ODB_MYSQL_CONNECTION_HXX + +#include + +#include + +#include + +#include +#include +#include +#include +#include +#include +#include + +#include +#include + +#include + +namespace odb +{ + namespace mysql + { + class statement_cache; + class connection_factory; + + class connection; + typedef details::shared_ptr connection_ptr; + + class LIBODB_MYSQL_EXPORT connection: public odb::connection + { + public: + typedef mysql::statement_cache statement_cache_type; + typedef mysql::database database_type; + + virtual + ~connection (); + + connection (connection_factory&); + connection (connection_factory&, MYSQL* handle); + + database_type& + database (); + + public: + virtual transaction_impl* + begin (); + + public: + using odb::connection::execute; + + virtual unsigned long long + execute (const char* statement, std::size_t length); + + // Query preparation. + // + public: + template + prepared_query + prepare_query (const char* name, const char*); + + template + prepared_query + prepare_query (const char* name, const std::string&); + + template + prepared_query + prepare_query (const char* name, const mysql::query_base&); + + template + prepared_query + prepare_query (const char* name, const odb::query_base&); + + // SQL statement tracing. + // + public: + typedef mysql::tracer tracer_type; + + void + tracer (tracer_type& t) + { + odb::connection::tracer (t); + } + + void + tracer (tracer_type* t) + { + odb::connection::tracer (t); + } + + using odb::connection::tracer; + + public: + bool + failed () const + { + return failed_; + } + + void + mark_failed () + { + failed_ = true; + } + + // Ping the server to make sure the connection is still alive. Return + // true if successful, mark the connection as failed and return false + // otherwise. This function can also throw database_exception. + // + bool + ping (); + + public: + MYSQL* + handle () + { + return handle_; + } + + statement_cache_type& + statement_cache () + { + return *statement_cache_; + } + + public: + statement* + active () + { + return active_; + } + + void + active (statement* s) + { + active_ = s; + + if (s == 0 && stmt_handles_.size () > 0) + free_stmt_handles (); + } + + // Cancel and clear the active statement, if any. + // + void + clear () + { + if (active_ != 0) + clear_ (); + } + + public: + MYSQL_STMT* + alloc_stmt_handle (); + + void + free_stmt_handle (auto_handle&); + + private: + connection (const connection&); + connection& operator= (const connection&); + + private: + void + free_stmt_handles (); + + void + clear_ (); + + private: + friend class transaction_impl; // invalidate_results() + + private: + bool failed_; + + MYSQL mysql_; + auto_handle handle_; + + statement* active_; + + // Keep statement_cache_ after handle_ so that it is destroyed before + // the connection is closed. + // + details::unique_ptr statement_cache_; + + // List of "delayed" statement handles to be freed next time there + // is no active statement. + // + typedef std::vector stmt_handles; + stmt_handles stmt_handles_; + }; + + class LIBODB_MYSQL_EXPORT connection_factory: + public odb::connection_factory + { + public: + typedef mysql::database database_type; + + virtual void + database (database_type&); + + database_type& + database () {return *db_;} + + virtual connection_ptr + connect () = 0; + + virtual + ~connection_factory (); + + connection_factory (): db_ (0) {} + + // Needed to break the circular connection_factory-database dependency + // (odb::connection_factory has the odb::database member). + // + protected: + database_type* db_; + }; + } +} + +#include + +#include + +#endif // ODB_MYSQL_CONNECTION_HXX -- cgit v1.1