1 files changed, 83 insertions, 0 deletions
diff --git a/libmysqlpp/embeddedmy-server.cpp b/libmysqlpp/embeddedmy-server.cpp
new file mode 100644
index 0000000..a0fc3e3
--- /dev/null
+++ b/libmysqlpp/embeddedmy-server.cpp
@@ -0,0 +1,83 @@
+#include "embeddedmy-server.h"
+#include "embeddedmy-connection.h"
+#include <stdexcept>
+#include <buffer.h>
+#include <mysql.h>
+#include <boost/filesystem/convenience.hpp>
+#include <fstream>
+
+namespace MySQL {
+	namespace Embedded {
+		ServerPtr
+		Server::get(const boost::filesystem::path & path, const std::vector<std::string> & extraOpts)
+		{
+			if (instance) {
+				if (path == instance->path) {
+					return instance;
+				}
+				throw std::runtime_error(stringbf("Only one embedded server per process. [old=%s, new=%s]", instance->path, path));
+			}
+			instance = new Server(path, extraOpts);
+			return instance;
+		}
+
+		ServerPtr
+		Server::getMock(const boost::filesystem::path & path)
+		{
+			boost::filesystem::create_directories(path / "mysql");
+			if (instance) {
+				if (path == instance->path && dynamic_cast<MockServer *>(instance)) {
+					return instance;
+				}
+				throw std::runtime_error(stringbf("Only one embedded server per process. [old=%s, new=%s]", instance->path, path));
+			}
+			auto i = new MockServer(path);
+			i->initialize();
+			return (instance = i);
+		}
+
+		Server::Server(const boost::filesystem::path & p, const std::vector<std::string> & extraOpts) :
+			path(p)
+		{
+			const auto datadir = stringbf("--datadir=%s", path.string());
+			std::vector<const char *> opts;
+			opts.push_back(typeid(this).name());
+			opts.push_back(datadir.c_str());
+			for (auto & opt : extraOpts) {
+				opts.push_back(opt.c_str());
+			}
+			opts.push_back(nullptr);
+			static const char * groups[] = { NULL };
+			mysql_library_init(opts.size() - 1, (char**)&opts.front(), (char**)groups);
+		}
+
+		Server::~Server()
+		{
+			mysql_library_end();
+			instance = nullptr;
+		}
+
+		MockServer::MockServer(const boost::filesystem::path & p) :
+			Server(p, { "--bootstrap" })
+		{
+		}
+
+		void
+		MockServer::initialize()
+		{
+			Connection initialize(this, "mysql");
+			std::ifstream sql1("/usr/share/mysql/mysql_system_tables.sql");
+			initialize.executeScript(sql1, path);
+			std::ifstream sql2("/usr/share/mysql/mysql_system_tables_data.sql");
+			initialize.executeScript(sql2, path);
+		}
+
+		MockServer::~MockServer()
+		{
+			boost::filesystem::remove_all(path);
+		}
+
+		Server * Server::instance = nullptr;
+	}
+}
+