monero-project/monero
monero-project/monero
1
0
Fork 0
mirror of https://github.com/monero-project/monero.git synced 2024-11-18 00:37:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
monero/contrib/epee/include/ado_db_helper.h
fluffypony 6fc995fe5d License updated to BSD 3-clause
2014-07-23 15:03:52 +02:00

1098 lines
39 KiB
C++
Raw Blame History

// Copyright (c) 2014, The Monero Project
//
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without modification, are
// permitted provided that the following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this list of
// conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice, this list
// of conditions and the following disclaimer in the documentation and/or other
// materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its contributors may be
// used to endorse or promote products derived from this software without specific
// prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
// THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
// THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Parts of this file are originally copyright (c) 2006-2013, Andrey N. Sabelnikov
#ifndef _DB_ADO_HELPER_H_
#define _DB_ADO_HELPER_H_
#include <vector>
#include <comutil.h>
#include "string_coding.h"
#include "math_helper.h"
#include "file_io_utils.h"
#include "global_stream_operators.h"
#define BEGIN_TRY_SECTION() try {
#define CATCH_TRY_SECTION(ret_val) CATCH_TRY_SECTION_MESS(ret_val, "")
#define CATCH_TRY_SECTION_MESS(ret_val, mess_where) }\
catch(const std::exception&ex)\
{\
LOG_PRINT_J("DB_ERROR: " << ex.what(), LOG_LEVEL_0);\
return ret_val;\
}\
catch(const _com_error& comm_err)\
{\
const TCHAR* pstr = comm_err.Description();\
std::string descr = string_encoding::convert_to_ansii(pstr?pstr:TEXT(""));\
const TCHAR* pmessage = comm_err.ErrorMessage();\
pstr = comm_err.Source();\
std::string source = string_encoding::convert_to_ansii(pstr?pstr:TEXT(""));\
LOG_PRINT_J("COM_ERROR " << mess_where << ":\n\tDescriprion:" << descr << ", \n\t Message: " << string_encoding::convert_to_ansii(pmessage) << "\n\t Source: " << source, LOG_LEVEL_0);\
return ret_val;\
}\
catch(...)\
{\
LOG_PRINT_J("..._ERROR: Unknown error.", LOG_LEVEL_0);\
return ret_val;\
}\
namespace epee
{
namespace ado_db_helper
{
struct profile_entry
{
profile_entry():m_call_count(0), m_max_time(0), m_min_time(0)
{}
//std::string m_sql;
math_helper::average<DWORD, 10> m_avrg;
size_t m_call_count;
DWORD m_max_time;
DWORD m_min_time;
};
class profiler_manager
{
public:
typedef std::map<std::string, profile_entry> sqls_map;
profiler_manager(){}
static bool sort_by_timing(const sqls_map::iterator& a, const sqls_map::iterator& b)
{
return a->second.m_avrg.get_avg() > b->second.m_avrg.get_avg();
}
bool flush_log(const std::string& path)
{
CRITICAL_REGION_BEGIN(m_sqls_lock);
std::stringstream strm;
strm << "SQL PROFILE:\r\nStatements: " << m_sqls.size() << "\r\n";
std::list<sqls_map::iterator> m_sorted_by_time_sqls;
for(std::map<std::string, profile_entry>::iterator it = m_sqls.begin();it!=m_sqls.end();it++)
m_sorted_by_time_sqls.push_back(it);
m_sorted_by_time_sqls.sort(sort_by_timing);
for(std::list<sqls_map::iterator>::iterator it = m_sorted_by_time_sqls.begin();it!=m_sorted_by_time_sqls.end();it++)
{
strm << "---------------------------------------------------------------------------------------------------------\r\nSQL: " << (*it)->first << "\r\n";
strm << "\tavrg: " << (*it)->second.m_avrg.get_avg() << "\r\n\tmax: " << (*it)->second.m_max_time << "\r\n\tmin: " << (*it)->second.m_min_time << "\r\n\tcount: " << (*it)->second.m_call_count << "\r\n";
}
return file_io_utils::save_string_to_file(path.c_str(), strm.str());
CRITICAL_REGION_END();
}
bool push_entry(const std::string sql, DWORD time)
{
CRITICAL_REGION_BEGIN(m_sqls_lock);
profile_entry& entry_ref = m_sqls[sql];
entry_ref.m_avrg.push(time);
entry_ref.m_call_count++;
if(time > entry_ref.m_max_time) entry_ref.m_max_time = time;
if(time < entry_ref.m_min_time || entry_ref.m_min_time == 0) entry_ref.m_min_time = time;
CRITICAL_REGION_END();
return true;
}
bool get_entry_avarege(const std::string sql, DWORD& time)
{
CRITICAL_REGION_BEGIN(m_sqls_lock);
sqls_map::iterator it = m_sqls.find(sql);
if(it==m_sqls.end())
return false;
time = static_cast<DWORD>(it->second.m_avrg.get_avg());
CRITICAL_REGION_END();
return true;
}
private:
sqls_map m_sqls;
critical_section m_sqls_lock;
};
inline
profiler_manager* get_set_profiler(bool need_to_set = false, profiler_manager** pprofiler = NULL)
{
static profiler_manager* pmanager = NULL;
if(need_to_set)
pmanager = *pprofiler;
//else
// *pprofiler = pmanager;
return pmanager;
}
inline
bool init() // INIT and DEINIT are NOT THREAD SAFE operations, CALL it BEFOR u start using this wrapper.
{
profiler_manager* pmanager = new profiler_manager();
get_set_profiler(true, &pmanager);
return true;
}
inline
bool deinit()
{
profiler_manager* pmanager = get_set_profiler();
//get_set_profiler(false, &pmanager);
if(pmanager)
delete pmanager;
return true;
}
inline bool push_timing(const std::string sql, DWORD time)
{
profiler_manager* pmanager = get_set_profiler();
//get_set_profiler(false, &pmanager);
if(pmanager)
return pmanager->push_entry(sql, time);
return true;
}
inline bool flush_profiler(const std::string path)
{
profiler_manager* pmanager = get_set_profiler();
//get_set_profiler(false, &pmanager);
if(pmanager)
return pmanager->flush_log(path);
return true;
}
class timing_guard
{
DWORD m_start_time;
std::string m_sql;
public:
timing_guard(const std::string& sql)
{
m_start_time = ::GetTickCount();
m_sql = sql;
}
~timing_guard()
{
DWORD timing = ::GetTickCount() - m_start_time;
push_timing(m_sql, timing);
}
};
#define PROFILE_SQL(sql) timing_guard local_timing(sql)
typedef std::vector<std::vector<_variant_t> > table;
inline bool add_parametr(ADODB::_CommandPtr cmd, const std::string& parametr)
{
_variant_t param(parametr.c_str());
ADODB::ADO_LONGPTR size = sizeof(parametr);
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("", ADODB::adVarChar, ADODB::adParamInput, static_cast<long>(parametr.size()+1), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, const std::wstring& parametr)
{
_variant_t param(parametr.c_str());
ADODB::ADO_LONGPTR size = sizeof(parametr);
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("", ADODB::adVarWChar, ADODB::adParamInput, static_cast<long>(parametr.size()+2), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, const __int64 parametr)
{
_variant_t param(parametr);
ADODB::ADO_LONGPTR size = static_cast<long>(sizeof(parametr));
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adBigInt, ADODB::adParamInput, static_cast<long>(size), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, const unsigned __int64 parametr)
{
_variant_t param(parametr);
ADODB::ADO_LONGPTR size = static_cast<long>(sizeof(parametr));
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adUnsignedBigInt, ADODB::adParamInput, static_cast<long>(size), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, const int parametr)
{
_variant_t param(parametr);
ADODB::ADO_LONGPTR size = static_cast<long>(sizeof(parametr));
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adInteger, ADODB::adParamInput, static_cast<long>(size), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, const unsigned int parametr)
{
_variant_t param(parametr);
ADODB::ADO_LONGPTR size = static_cast<long>(sizeof(parametr));
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adUnsignedInt, ADODB::adParamInput, static_cast<long>(size), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, float parametr)
{
_variant_t param;
param.ChangeType(VT_R4);
param.fltVal = parametr;
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adSingle, ADODB::adParamInput, static_cast<long>(sizeof(float)), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, bool parametr)
{
_variant_t param;
param = parametr;
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adBoolean, ADODB::adParamInput, sizeof(parametr), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr(ADODB::_CommandPtr cmd, _variant_t parametr)
{
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adDBTimeStamp, ADODB::adParamInput, sizeof(parametr), parametr);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool add_parametr_as_double(ADODB::_CommandPtr cmd, const DATE parametr)
{
_variant_t param;
param.ChangeType(VT_R8);
param.dblVal = parametr;
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adDouble, ADODB::adParamInput, sizeof(float), param);
cmd->Parameters->Append(param_obj);
return true;
}
template<typename TParam>
inline bool add_parametr(ADODB::_CommandPtr cmd, const std::list<TParam> params)
{
for(std::list<TParam>::const_iterator it = params.begin(); it!=params.end(); it++)
if(!add_parametr(cmd, *it))
return false;
return true;
}
/*
inline bool add_parametr(ADODB::_CommandPtr cmd, const size_t parametr)
{
_variant_t param;
param.ChangeType(VT_I4);
param.intVal = parametr;
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adInteger, ADODB::adParamInput, sizeof(parametr), param);
cmd->Parameters->Append(param_obj);
return true;
}*/
inline bool add_parametr(ADODB::_CommandPtr cmd, const DATE parametr)
{
/*_variant_t param;
param.ChangeType(VT_R8);
param.dblVal = parametr;
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adDouble, ADODB::adParamInput, sizeof(float), param);
cmd->Parameters->Append(param_obj);*/
_variant_t param;
param.ChangeType(VT_DATE);
param.date = parametr;
ADODB::_ParameterPtr param_obj = cmd->CreateParameter("parametr", ADODB::adDBDate, ADODB::adParamInput, sizeof(parametr), param);
cmd->Parameters->Append(param_obj);
return true;
}
inline bool execute_helper(ADODB::_CommandPtr cmd, _variant_t* pcount_processed = NULL)
{
//BEGIN_TRY_SECTION();
cmd->Execute(pcount_processed, NULL, ADODB::adExecuteNoRecords);
//CATCH_TRY_SECTION(false);
return true;
}
inline bool select_helper(ADODB::_CommandPtr cmd, table& result_vector)
{
result_vector.clear();
//BEGIN_TRY_SECTION();
ADODB::_RecordsetPtr precordset = cmd->Execute(NULL, NULL, NULL);
if(!precordset)
{
LOG_ERROR("DB_ERROR: cmd->Execute returned NULL!!!");
return false;
}
//if(precordset->EndOfFile == EOF)
//{
// return true;
//}
/*try
{
if(precordset->MoveFirst()!= S_OK)
{
LOG_ERROR("DB_ERROR: Filed to move first!!!");
return false;
}
}
catch (...)
{
return true;
}*/
size_t current_record_index = 0;
while(precordset->EndOfFile != EOF)
{
result_vector.push_back(table::value_type());
size_t fields_count = precordset->Fields->Count;
result_vector[current_record_index].resize(fields_count);
for(size_t current_field_index = 0; current_field_index < fields_count; current_field_index++)
{
_variant_t var;
var.ChangeType(VT_I2);
var.intVal = static_cast<INT>(current_field_index);
result_vector[current_record_index][current_field_index] = precordset->Fields->GetItem(var)->Value;
}
precordset->MoveNext();
current_record_index++;
}
//CATCH_TRY_SECTION(false);
return true;
}
template<typename TParam1>
struct adapter_zero
{
};
template<typename TParam1>
struct adapter_single
{
TParam1 tparam1;
};
template<typename TParam1, typename TParam2>
struct adapter_double
{
TParam1 tparam1;
TParam2 tparam2;
};
template<typename TParam1, typename TParam2, typename TParam3>
struct adapter_triple
{
TParam1 tparam1;
TParam2 tparam2;
TParam3 tparam3;
};
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4>
struct adapter_quad
{
TParam1 tparam1;
TParam2 tparam2;
TParam3 tparam3;
TParam4 tparam4;
};
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5>
struct adapter_quanto
{
TParam1 tparam1;
TParam2 tparam2;
TParam3 tparam3;
TParam4 tparam4;
TParam5 tparam5;
};
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6>
struct adapter_sixto
{
TParam1 tparam1;
TParam2 tparam2;
TParam3 tparam3;
TParam4 tparam4;
TParam5 tparam5;
TParam6 tparam6;
};
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7>
struct adapter_sevento
{
TParam1 tparam1;
TParam2 tparam2;
TParam3 tparam3;
TParam4 tparam4;
TParam5 tparam5;
TParam6 tparam6;
TParam7 tparam7;
};
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7, typename TParam8, typename TParam9>
struct adapter_nine
{
TParam1 tparam1;
TParam2 tparam2;
TParam3 tparam3;
TParam4 tparam4;
TParam5 tparam5;
TParam6 tparam6;
TParam7 tparam7;
TParam8 tparam8;
TParam9 tparam9;
};
template<typename TParam1>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_zero<TParam1>& params)
{
return true;
}
template<typename TParam1>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_single<TParam1>& params)
{
return add_parametr(cmd, params.tparam1);
}
template<typename TParam1, typename TParam2>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_double<TParam1, TParam2>& params)
{
if(!add_parametr(cmd, params.tparam1)) return false;
return add_parametr(cmd, params.tparam2);
}
template<typename TParam1, typename TParam2, typename TParam3>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_triple<TParam1, TParam2, TParam3>& params)
{
if(!add_parametr(cmd, params.tparam1)) return false;
if(!add_parametr(cmd, params.tparam2)) return false;
return add_parametr(cmd, params.tparam3);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_quad<TParam1, TParam2, TParam3, TParam4>& params)
{
if(!add_parametr(cmd, params.tparam1)) return false;
if(!add_parametr(cmd, params.tparam2)) return false;
if(!add_parametr(cmd, params.tparam3)) return false;
return add_parametr(cmd, params.tparam4);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_quanto<TParam1, TParam2, TParam3, TParam4, TParam5>& params)
{
if(!add_parametr(cmd, params.tparam1)) return false;
if(!add_parametr(cmd, params.tparam2)) return false;
if(!add_parametr(cmd, params.tparam3)) return false;
if(!add_parametr(cmd, params.tparam4)) return false;
return add_parametr(cmd, params.tparam5);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_sixto<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6>& params)
{
if(!add_parametr(cmd, params.tparam1)) return false;
if(!add_parametr(cmd, params.tparam2)) return false;
if(!add_parametr(cmd, params.tparam3)) return false;
if(!add_parametr(cmd, params.tparam4)) return false;
if(!add_parametr(cmd, params.tparam5)) return false;
return add_parametr(cmd, params.tparam6);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_sevento<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6, TParam7>& params)
{
if(!add_parametr(cmd, params.tparam1)) return false;
if(!add_parametr(cmd, params.tparam2)) return false;
if(!add_parametr(cmd, params.tparam3)) return false;
if(!add_parametr(cmd, params.tparam4)) return false;
if(!add_parametr(cmd, params.tparam5)) return false;
if(!add_parametr(cmd, params.tparam6)) return false;
return add_parametr(cmd, params.tparam7);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7, typename TParam8, typename TParam9>
bool add_parametrs_multi(ADODB::_CommandPtr cmd, const adapter_nine<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6, TParam7, TParam8, TParam9>& params)
{
if(!add_parametr(cmd, params.tparam1)) return false;
if(!add_parametr(cmd, params.tparam2)) return false;
if(!add_parametr(cmd, params.tparam3)) return false;
if(!add_parametr(cmd, params.tparam4)) return false;
if(!add_parametr(cmd, params.tparam5)) return false;
if(!add_parametr(cmd, params.tparam6)) return false;
if(!add_parametr(cmd, params.tparam7)) return false;
if(!add_parametr(cmd, params.tparam8)) return false;
return add_parametr(cmd, params.tparam9);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7>
std::string print_parameters_multi(const adapter_sevento<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6, TParam7>& params)
{
std::stringstream strm;
strm << params.tparam1 << ", " << params.tparam2 << ", " << params.tparam3 << ", " << params.tparam4 << ", " << params.tparam5 << ", " << params.tparam6 << ", " << params.tparam7;
return strm.str();
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7, typename TParam8, typename TParam9>
std::string print_parameters_multi(const adapter_nine<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6, TParam7, TParam8, TParam9>& params)
{
std::stringstream strm;
strm << params.tparam1 << ", " << params.tparam2 << ", " << params.tparam3 << ", " << params.tparam4 << ", " << params.tparam5 << ", " << params.tparam6 << ", " << params.tparam7 << ", " << params.tparam8 << ", " << params.tparam9;
return strm.str();
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6>
std::string print_parameters_multi(const adapter_sixto<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6>& params)
{
std::stringstream strm;
strm << params.tparam1 << ", " << params.tparam2 << ", " << params.tparam3 << ", " << params.tparam4 << ", " << params.tparam5 << ", " << params.tparam6;
return strm.str();
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5>
std::string print_parameters_multi(const adapter_quanto<TParam1, TParam2, TParam3, TParam4, TParam5>& params)
{
std::stringstream strm;
strm << params.tparam1 << ", " << params.tparam2 << ", " << params.tparam3 << ", " << params.tparam4 << ", " << params.tparam5;
return strm.str();
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4>
std::string print_parameters_multi(const adapter_quad<TParam1, TParam2, TParam3, TParam4>& params)
{
std::stringstream strm;
strm << params.tparam1 << ", " << params.tparam2 << ", " << params.tparam3 << ", " << params.tparam4;
return strm.str();
}
template<typename TParam1, typename TParam2, typename TParam3>
std::string print_parameters_multi(const adapter_triple<TParam1, TParam2, TParam3>& params)
{
std::stringstream strm;
strm << params.tparam1 << ", " << params.tparam2 << ", " << params.tparam3;
return strm.str();
}
template<typename TParam>
std::string get_str_param(const TParam& prm)
{
std::stringstream strm;
strm << prm;
return strm.str();
}
template<typename TParam>
std::string get_str_param(const std::list<TParam>& prm_lst)
{
std::stringstream strm;
for(std::list<TParam>::const_iterator it = prm_lst.begin();it!=prm_lst.end();it++)
strm << get_str_param(*it) << ", ";
return strm.str();
}
template<typename TParam1, typename TParam2>
std::string print_parameters_multi(const adapter_double<TParam1, TParam2>& params)
{
std::stringstream strm;
strm << get_str_param(params.tparam1) << ", " << get_str_param(params.tparam2);
return strm.str();
}
template<typename TParam1>
std::string print_parameters_multi(const adapter_single<TParam1>& params)
{
std::stringstream strm;
strm << get_str_param(params.tparam1);
return strm.str();
}
template<typename TParam1>
std::string print_parameters_multi(const adapter_zero<TParam1>& params)
{
std::stringstream strm;
strm << "(no parametrs)";
return strm.str();
}
template<typename TParams>
bool execute_helper_multiparam(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParams& parametrs, _variant_t* pcount_processed = NULL)
{
PROFILE_SQL(sql_statment);
bool res = false;
BEGIN_TRY_SECTION();
ADODB::_CommandPtr cmd;
cmd.CreateInstance(__uuidof(ADODB::Command));
cmd->CommandText = _bstr_t(sql_statment.c_str());
if(!add_parametrs_multi(cmd, parametrs))
return false;
cmd->ActiveConnection = pconnection;
res = execute_helper(cmd, pcount_processed);
CATCH_TRY_SECTION_MESS(false, "while statment: " << sql_statment << " [params]: " << print_parameters_multi(parametrs));
return res;
}
template<typename TParams>
inline
bool select_helper_multiparam(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParams& parametrs, table& result_vector)
{
PROFILE_SQL(sql_statment);
bool res = false;
BEGIN_TRY_SECTION();
ADODB::_CommandPtr cmd;
cmd.CreateInstance(__uuidof(ADODB::Command));
cmd->CommandText = _bstr_t(sql_statment.c_str());
if(!add_parametrs_multi(cmd, parametrs))
return false;
cmd->ActiveConnection = pconnection;
res = select_helper(cmd, result_vector);
CATCH_TRY_SECTION_MESS(false, "while statment: " << sql_statment << " [params]: " << print_parameters_multi(parametrs));
return res;
}
template<typename TParams>
inline
bool select_helper_param_container(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParams& parametrs, table& result_vector)
{
PROFILE_SQL(sql_statment);
bool res = false;
BEGIN_TRY_SECTION();
ADODB::_CommandPtr cmd;
cmd.CreateInstance(__uuidof(ADODB::Command));
cmd->CommandText = _bstr_t(sql_statment.c_str());
for(TParams::const_iterator it = parametrs.begin(); it!=parametrs.end(); it++)
{
add_parametr(cmd, *it);
}
cmd->ActiveConnection = pconnection;
res = select_helper(cmd, result_vector);
CATCH_TRY_SECTION(false);
return res;
}
inline
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, _variant_t* pvt = NULL)
{
adapter_zero<int> params;
return execute_helper_multiparam(pconnection, sql_statment, params, pvt);
}
template<typename TParam>
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam& parametr)
{
adapter_single<TParam> params;
params.tparam1 = parametr;
return execute_helper_multiparam(pconnection, sql_statment, params);
}
template<typename TParam1, typename TParam2>
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1& parametr1, const TParam2& parametr2)
{
adapter_double<TParam1, TParam2> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
return execute_helper_multiparam(pconnection, sql_statment, params);
}
template<typename TParam1, typename TParam2, typename TParam3>
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1& parametr1, const TParam2& parametr2, const TParam3& parametr3)
{
adapter_triple<TParam1, TParam2, typename TParam3> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
return execute_helper_multiparam(pconnection, sql_statment, params);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4>
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1& parametr1, const TParam2& parametr2, const TParam3& parametr3, const TParam4& parametr4)
{
adapter_quad<TParam1, TParam2, TParam3, TParam4> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
return execute_helper_multiparam(pconnection, sql_statment, params);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5>
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1& parametr1, const TParam2& parametr2, const TParam3& parametr3, const TParam4& parametr4, const TParam5& parametr5)
{
adapter_quanto<TParam1, TParam2, TParam3, TParam4, TParam5> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
params.tparam5 = parametr5;
return execute_helper_multiparam(pconnection, sql_statment, params);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6>
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1& parametr1, const TParam2& parametr2, const TParam3& parametr3, const TParam4& parametr4, const TParam5& parametr5, const TParam6& parametr6)
{
adapter_sixto<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
params.tparam5 = parametr5;
params.tparam6 = parametr6;
return execute_helper_multiparam(pconnection, sql_statment, params);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7>
bool execute_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1& parametr1, const TParam2& parametr2, const TParam3& parametr3, const TParam4& parametr4, const TParam5& parametr5, const TParam6& parametr6, const TParam7& parametr7)
{
adapter_sevento<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6, TParam7> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
params.tparam5 = parametr5;
params.tparam6 = parametr6;
params.tparam7 = parametr7;
return execute_helper_multiparam(pconnection, sql_statment, params);
}
inline
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, table& result_vector)
{
adapter_zero<int> params;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam& parametr, table& result_vector)
{
adapter_single<TParam> params;
params.tparam1 = parametr;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam1, typename TParam2>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1 parametr1, const TParam2 parametr2, table& result_vector)
{
adapter_double<TParam1, TParam2> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam1, typename TParam2, typename TParam3>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1 parametr1, const TParam2 parametr2, const TParam3 parametr3, table& result_vector)
{
adapter_triple<TParam1, TParam2, typename TParam3> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1 parametr1, const TParam2 parametr2, const TParam3 parametr3, const TParam4 parametr4, table& result_vector)
{
adapter_quad<TParam1, TParam2, TParam3, TParam4> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1 parametr1, const TParam2 parametr2, const TParam3 parametr3, const TParam4 parametr4, const TParam5 parametr5, table& result_vector)
{
adapter_quanto<TParam1, TParam2, TParam3, TParam4, TParam5> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
params.tparam5 = parametr5;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1 parametr1, const TParam2 parametr2, const TParam3 parametr3, const TParam4 parametr4, const TParam5 parametr5, const TParam6 parametr6, table& result_vector)
{
adapter_sixto<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
params.tparam5 = parametr5;
params.tparam6 = parametr6;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1 parametr1, const TParam2 parametr2, const TParam3 parametr3, const TParam4 parametr4, const TParam5 parametr5, const TParam6 parametr6, const TParam7 parametr7, table& result_vector)
{
adapter_sevento<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6, TParam7> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
params.tparam5 = parametr5;
params.tparam6 = parametr6;
params.tparam7 = parametr7;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename TParam5, typename TParam6, typename TParam7, typename TParam8, typename TParam9>
bool select_helper(ADODB::_ConnectionPtr pconnection, const std::string& sql_statment, const TParam1 parametr1, const TParam2 parametr2, const TParam3 parametr3, const TParam4 parametr4, const TParam5 parametr5, const TParam6 parametr6, const TParam7 parametr7,const TParam8 parametr8,const TParam9 parametr9, table& result_vector)
{
adapter_nine<TParam1, TParam2, TParam3, TParam4, TParam5, TParam6, TParam7, TParam8, TParam9> params;
params.tparam1 = parametr1;
params.tparam2 = parametr2;
params.tparam3 = parametr3;
params.tparam4 = parametr4;
params.tparam5 = parametr5;
params.tparam6 = parametr6;
params.tparam7 = parametr7;
params.tparam8 = parametr8;
params.tparam9 = parametr9;
return select_helper_multiparam(pconnection, sql_statment, params, result_vector);
}
/************************************************************************/
/* */
/************************************************************************/
class per_thread_connection_pool
{
public:
bool init(const std::string& connection_string, const std::string& login, const std::string& pass)
{
m_connection_string = connection_string;
m_login = login;
m_password = pass;
if(!get_db_connection().GetInterfacePtr())
return false;
return true;
}
ADODB::_ConnectionPtr& get_db_connection()
{
//soci::session
m_db_connections_lock.lock();
boost::shared_ptr<ADODB::_ConnectionPtr>& conn_ptr = m_db_connections[::GetCurrentThreadId()];
m_db_connections_lock.unlock();
if(!conn_ptr.get())
{
conn_ptr.reset(new ADODB::_ConnectionPtr());
ADODB::_ConnectionPtr& conn = *conn_ptr.get();
//init new connection
BEGIN_TRY_SECTION();
//_bstr_t str = _bstr_t("Provider=SQLOLEDB;Data Source=SRV1;Integrated Security=SSPI;Initial Catalog=dispatcher;");
if(S_OK != conn.CreateInstance(__uuidof(ADODB::Connection)))
{
LOG_ERROR("Failed to Create, instance, was CoInitialize called ???!");
return conn;
}
HRESULT res = conn->Open(_bstr_t(m_connection_string.c_str()), _bstr_t(m_login.c_str()), _bstr_t(m_password.c_str()), NULL);
if(res != S_OK)
{
LOG_ERROR("Failed to connect do DB, connection str:" << m_connection_string);
return conn;
}
CATCH_TRY_SECTION_MESS(conn, "while creating another connection");
LOG_PRINT("New DB Connection added for threadid=" << ::GetCurrentThreadId(), LOG_LEVEL_0);
ado_db_helper::execute_helper(conn, "set enable_seqscan=false;");
return conn;
}
return *conn_ptr.get();
}
//----------------------------------------------------------------------------------------------
bool check_status()
{
ADODB::_ConnectionPtr& rconn = get_db_connection();
if(!ado_db_helper::execute_helper(rconn, "SET CLIENT_ENCODING TO 'SQL_ASCII'"))
{
try{
HRESULT res = rconn->Close();
}
catch(...)
{
};
BEGIN_TRY_SECTION();
HRESULT res = rconn->Open(_bstr_t(m_connection_string.c_str()), _bstr_t(m_login.c_str()), _bstr_t(m_password.c_str()), NULL);
if(res != S_OK)
{
LOG_PRINT("Failed to restore connection to local AI DB", LOG_LEVEL_1);
return false;
}
CATCH_TRY_SECTION(false);
}
return true;
}
protected:
private:
std::map<DWORD, boost::shared_ptr<ADODB::_ConnectionPtr> > m_db_connections;
critical_section m_db_connections_lock;
std::string m_connection_string;
std::string m_login;
std::string m_password;
};
template<typename TParam1, typename default_id_type, typename t_conn>
bool find_or_add_t(const std::string& sql_select_statment, const std::string& sql_insert_statment, OUT default_id_type& id, OUT bool& new_object_added, TParam1 parametr_1, t_conn& c)
{
ado_db_helper::adapter_single<TParam1> params;
params.tparam1 = parametr_1;
return find_or_add_t_multiparametred(sql_select_statment, sql_insert_statment, id, new_object_added, params, c);
}
template<typename TParam1, typename TParam2, typename default_id_type, typename t_conn>
bool find_or_add_t(const std::string& sql_select_statment, const std::string& sql_insert_statment, OUT default_id_type& id, OUT bool& new_object_added, TParam1 parametr_1, TParam2 parametr_2, t_conn& c)
{
ado_db_helper::adapter_double<TParam1, TParam2> params;
params.tparam1 = parametr_1;
params.tparam2 = parametr_2;
return find_or_add_t_multiparametred(sql_select_statment, sql_insert_statment, id, new_object_added, params, c);
}
template<typename TParam1, typename TParam2, typename TParam3, typename default_id_type, typename t_conn>
bool find_or_add_t(const std::string& sql_select_statment, const std::string& sql_insert_statment, OUT default_id_type& id, OUT bool& new_object_added, TParam1 parametr_1, TParam2 parametr_2, TParam3 parametr_3, t_conn& c)
{
ado_db_helper::adapter_triple<TParam1, TParam2, TParam3> params;
params.tparam1 = parametr_1;
params.tparam2 = parametr_2;
params.tparam3 = parametr_3;
return find_or_add_t_multiparametred(sql_select_statment, sql_insert_statment, id, new_object_added, params, c);
}
template<typename TParam1, typename TParam2, typename TParam3, typename TParam4, typename default_id_type, typename t_conn>
bool find_or_add_t(const std::string& sql_select_statment, const std::string& sql_insert_statment, OUT default_id_type& id, OUT bool& new_object_added, TParam1 parametr_1, TParam2 parametr_2, TParam3 parametr_3, TParam4 parametr_4, t_conn& c)
{
ado_db_helper::adapter_quad<TParam1, TParam2, TParam3, TParam4> params;
params.tparam1 = parametr_1;
params.tparam2 = parametr_2;
params.tparam3 = parametr_3;
params.tparam4 = parametr_4;
return find_or_add_t_multiparametred(sql_select_statment, sql_insert_statment, id, new_object_added, params, c);
}
template<typename TParams, typename default_id_type, typename t_conn>
bool find_or_add_t_multiparametred(const std::string& sql_select_statment, const std::string& sql_insert_statment, OUT default_id_type& id, OUT bool& new_object_added, TParams params, t_conn& c)
{
//CHECK_CONNECTION(false);
new_object_added = false;
ado_db_helper::table result_table;
bool res = select_helper_multiparam(c.get_db_connection(), sql_select_statment, params, result_table);
if(!result_table.size())
{
res = select_helper_multiparam(c.get_db_connection(), sql_insert_statment, params, result_table);
if(!res || !result_table.size())
{
//last time try to select
res = select_helper_multiparam(c.get_db_connection(), sql_select_statment, params, result_table);
CHECK_AND_ASSERT_MES(res, false, "Failed to execute statment: " << sql_select_statment);
CHECK_AND_ASSERT_MES(result_table.size(), false, "No records returned from statment: " << sql_select_statment);
}else
{
new_object_added = true;
}
}
BEGIN_TRY_SECTION()
id = result_table[0][0];
CATCH_TRY_SECTION_MESS(false, "while converting returned value [find_or_add_t_multiparametred()]");
return true;
}
}
}
#endif //!_DB_HELPER_H_
Reference in a new issue View git blame Copy permalink