mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-18 21:44:02 +03:00
145 lines
3.3 KiB
C++
145 lines
3.3 KiB
C++
/* Copyright (C) 2014 InfiniDB, Inc.
|
|
|
|
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; version 2 of
|
|
the License.
|
|
|
|
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 Street, Fifth Floor, Boston,
|
|
MA 02110-1301, USA. */
|
|
|
|
#include <unistd.h>
|
|
//#define NDEBUG
|
|
#include <cassert>
|
|
#include <sstream>
|
|
#include <iomanip>
|
|
using namespace std;
|
|
|
|
#include "idberrorinfo.h"
|
|
using namespace logging;
|
|
|
|
#include "liboamcpp.h"
|
|
|
|
#include "libmysql_client.h"
|
|
|
|
namespace utils
|
|
{
|
|
|
|
LibMySQL::LibMySQL() : fCon(NULL), fRes(NULL)
|
|
{
|
|
}
|
|
|
|
|
|
LibMySQL::~LibMySQL()
|
|
{
|
|
if (fRes)
|
|
{
|
|
mysql_free_result(fRes);
|
|
}
|
|
|
|
fRes = NULL;
|
|
|
|
if (fCon)
|
|
{
|
|
mysql_close(fCon);
|
|
}
|
|
|
|
fCon = NULL;
|
|
}
|
|
|
|
|
|
int LibMySQL::init(const char* h, unsigned int p, const char* u, const char* w, const char* d)
|
|
{
|
|
int ret = 0;
|
|
|
|
fCon = mysql_init(NULL);
|
|
|
|
oam::Oam oam;
|
|
oam::oamModuleInfo_t moduleInfo;
|
|
moduleInfo = oam.getModuleInfo();
|
|
string moduleName = boost::get<0>(moduleInfo);
|
|
int serverTypeInstall = boost::get<5>(moduleInfo);
|
|
|
|
// This is single server installation so use um1 instead of pm1.
|
|
if ( serverTypeInstall == 2 )
|
|
moduleName.assign("um1");
|
|
|
|
oam::ModuleConfig moduleconfig;
|
|
oam.getSystemConfig(moduleName, moduleconfig);
|
|
|
|
if (!(moduleconfig.TLSCA.empty() || moduleconfig.TLSClientCert.empty() || moduleconfig.TLSClientKey.empty()))
|
|
{
|
|
mysql_ssl_set(fCon, moduleconfig.TLSClientKey.c_str(), moduleconfig.TLSClientCert.c_str(),
|
|
moduleconfig.TLSCA.c_str(), NULL, NULL);
|
|
}
|
|
|
|
if (fCon != NULL)
|
|
{
|
|
unsigned int tcp_option = MYSQL_PROTOCOL_TCP;
|
|
mysql_options(fCon, MYSQL_OPT_PROTOCOL, &tcp_option);
|
|
|
|
if (mysql_real_connect(fCon, h, u, w, d, p, NULL, 0) == NULL)
|
|
{
|
|
fErrStr = "fatal error running mysql_real_connect() in libmysql_client lib";
|
|
ret = mysql_errno(fCon);
|
|
}
|
|
else
|
|
{
|
|
mysql_set_character_set(fCon, "utf8");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fErrStr = "fatal error running mysql_init() in libmysql_client lib";
|
|
ret = -1;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
int LibMySQL::run(const char* query)
|
|
{
|
|
int ret = 0;
|
|
|
|
if (mysql_real_query(fCon, query, strlen(query)) != 0)
|
|
{
|
|
fErrStr = "fatal error runing mysql_real_query() in libmysql_client lib";
|
|
ret = -1;
|
|
}
|
|
|
|
fRes = mysql_use_result(fCon);
|
|
|
|
if (fRes == NULL)
|
|
{
|
|
fErrStr = "fatal error running mysql_use_result() or empty result set in libmysql_client lib";
|
|
ret = -1;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
void LibMySQL::handleMySqlError(const char* errStr, unsigned int errCode)
|
|
{
|
|
ostringstream oss;
|
|
oss << errStr << "(" << errCode << ")";
|
|
|
|
if (errCode == (unsigned int) - 1)
|
|
oss << "(null pointer)";
|
|
else
|
|
oss << "(" << errCode << ")";
|
|
|
|
throw IDBExcept(oss.str(), ERR_CROSS_ENGINE_CONNECT);
|
|
|
|
return;
|
|
}
|
|
|
|
} // namespace
|