99 lines
2.7 KiB
C
99 lines
2.7 KiB
C
|
/**
|
|||
|
|
* @file mix_cc/sql/sqlapi_warpper.h
|
|||
|
|
* @brief 对SQL API++的包装，作为执行SQL语句的核心
|
|||
|
|
* @author Cat (null.null.null@qq.com)
|
|||
|
|
* @version 0.1
|
|||
|
|
* @date 2021-09-17
|
|||
|
|
*
|
|||
|
|
* Copyright: Baosight Co. Ltd.
|
|||
|
|
* DO NOT COPY/USE WITHOUT PERMISSION
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
#pragma once
|
|||
|
|
#include <SQLAPI.h>
|
|||
|
|
#include <db2_linux/sql.h>
|
|||
|
|
#include <db2_linux/sqlsystm.h>
|
|||
|
|
#include <boost/type_traits.hpp>
|
|||
|
|
#include <boost/type_index.hpp>
|
|||
|
|
#include <boost/algorithm/string.hpp>
|
|||
|
|
#include "mix_cc/exception.h"
|
|||
|
|
#include "mix_cc/type/mix_time.h"
|
|||
|
|
#include <memory>
|
|||
|
|
#include <string>
|
|||
|
|
|
|||
|
|
namespace mix_cc {
|
|||
|
|
namespace sql {
|
|||
|
|
struct sqlapi_warpper {
|
|||
|
|
public:
|
|||
|
|
struct cmd_t {
|
|||
|
|
protected:
|
|||
|
|
std::unique_ptr<SACommand> cmd_ptr;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
cmd_t() { cmd_ptr = std::make_unique<SACommand>(); }
|
|||
|
|
bool connect(SAConnection* conn_ptr) {
|
|||
|
|
cmd_ptr->setConnection(conn_ptr);
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
~cmd_t() {}
|
|||
|
|
|
|||
|
|
bool execute(std::string command) {
|
|||
|
|
try {
|
|||
|
|
cmd_ptr->setCommandText(_TSA(command.c_str()));
|
|||
|
|
cmd_ptr->Execute();
|
|||
|
|
} catch (const SAException& e) {
|
|||
|
|
throw Exception(-1, e.ErrText().GetMultiByteChars(),
|
|||
|
|
BOOST_CURRENT_LOCATION, {"sql", command});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
auto is_result_set() { return cmd_ptr->isResultSet(); }
|
|||
|
|
|
|||
|
|
auto rows_affacted() { return cmd_ptr->RowsAffected(); }
|
|||
|
|
|
|||
|
|
auto cancel() { return cmd_ptr->Cancel(); }
|
|||
|
|
|
|||
|
|
auto close() { return cmd_ptr->Close(); }
|
|||
|
|
|
|||
|
|
auto destroy() { return cmd_ptr->Destroy(); }
|
|||
|
|
|
|||
|
|
auto fetch_next() { return cmd_ptr->FetchNext(); }
|
|||
|
|
|
|||
|
|
auto command_text() { return cmd_ptr->CommandText().GetMultiByteChars(); }
|
|||
|
|
|
|||
|
|
template <typename Col>
|
|||
|
|
constexpr auto field(Col) {
|
|||
|
|
auto field_name =
|
|||
|
|
boost::to_upper_copy(std::string(Col::name_type::c_str()));
|
|||
|
|
if constexpr (boost::is_integral<typename Col::value_type>::value) {
|
|||
|
|
return cmd_ptr->Field(field_name.c_str()).asLong();
|
|||
|
|
} else if constexpr (boost::is_float<typename Col::value_type>::value) {
|
|||
|
|
return cmd_ptr->Field(field_name.c_str()).asDouble();
|
|||
|
|
} else if constexpr (std::is_same_v<typename Col::value_type,
|
|||
|
|
std::string>) {
|
|||
|
|
return cmd_ptr->Field(field_name.c_str()).asString();
|
|||
|
|
} else if constexpr (std::is_same_v<typename Col::value_type,
|
|||
|
|
mix_cc::mix_time_t>) {
|
|||
|
|
std::string tmp_time_str =
|
|||
|
|
cmd_ptr->Field(field_name.c_str()).asString().GetMultiByteChars();
|
|||
|
|
if (!tmp_time_str.empty()) {
|
|||
|
|
tmp_time_str.replace(tmp_time_str.find("T"), 1, " ");
|
|||
|
|
}
|
|||
|
|
return mix_cc::mix_time_t{tmp_time_str};
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
sqlapi_warpper() : cmd_() {}
|
|||
|
|
~sqlapi_warpper() {}
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
cmd_t cmd_;
|
|||
|
|
};
|
|||
|
|
} // namespace sql
|
|||
|
|
} // namespace mix_cc
|