forked from metin2/server
160 lines
3.0 KiB
C++
160 lines
3.0 KiB
C++
#include "CStatement.h"
|
|
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
|
|
CStmt::CStmt()
|
|
{
|
|
m_pkStmt = NULL;
|
|
m_uiParamCount = 0;
|
|
m_uiResultCount = 0;
|
|
iRows = 0;
|
|
m_puiParamLen = NULL;
|
|
}
|
|
|
|
CStmt::~CStmt()
|
|
{
|
|
Destroy();
|
|
}
|
|
|
|
void CStmt::Destroy()
|
|
{
|
|
if (m_pkStmt)
|
|
{
|
|
mysql_stmt_close(m_pkStmt);
|
|
m_pkStmt = NULL;
|
|
}
|
|
|
|
if (m_puiParamLen)
|
|
{
|
|
free(m_puiParamLen);
|
|
m_puiParamLen = NULL;
|
|
}
|
|
}
|
|
|
|
void CStmt::Error(const char * c_pszMsg)
|
|
{
|
|
sys_log(0, "SYSERR: %s: [%d] %s", c_pszMsg, mysql_stmt_errno(m_pkStmt), mysql_stmt_error(m_pkStmt));
|
|
}
|
|
|
|
bool CStmt::Prepare(CAsyncSQL * sql, const char * c_pszQuery)
|
|
{
|
|
m_pkStmt = mysql_stmt_init(sql->GetSQLHandle());
|
|
m_stQuery = c_pszQuery;
|
|
|
|
if (mysql_stmt_prepare(m_pkStmt, m_stQuery.c_str(), m_stQuery.length()))
|
|
{
|
|
Error("mysql_stmt_prepare");
|
|
return false;
|
|
}
|
|
|
|
int iParamCount = 0;
|
|
|
|
for (unsigned int i = 0; i < m_stQuery.length(); ++i)
|
|
if (c_pszQuery[i] == '?')
|
|
++iParamCount;
|
|
|
|
if (iParamCount)
|
|
{
|
|
m_vec_param.resize(iParamCount);
|
|
memset(&m_vec_param[0], 0, sizeof(MYSQL_BIND) * iParamCount);
|
|
|
|
m_puiParamLen = (long unsigned int *) calloc(iParamCount, sizeof(long unsigned int));
|
|
}
|
|
|
|
m_vec_result.resize(48);
|
|
memset(&m_vec_result[0], 0, sizeof(MYSQL_BIND) * 48);
|
|
|
|
if (mysql_stmt_bind_result(m_pkStmt, &m_vec_result[0]))
|
|
{
|
|
Error("mysql_stmt_bind_result");
|
|
return 0;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CStmt::BindParam(enum_field_types type, void * p, int iMaxLen)
|
|
{
|
|
if (m_uiParamCount >= m_vec_param.size())
|
|
{
|
|
sys_err("too many parameter in query: %s", m_stQuery.c_str());
|
|
return false;
|
|
}
|
|
|
|
MYSQL_BIND * bind = &m_vec_param[m_uiParamCount];
|
|
|
|
bind->buffer_type = type;
|
|
bind->buffer = (void *) p;
|
|
bind->buffer_length = iMaxLen;
|
|
bind->length = m_puiParamLen + m_uiParamCount;
|
|
|
|
if (++m_uiParamCount == m_vec_param.size())
|
|
{
|
|
if (mysql_stmt_bind_param(m_pkStmt, &m_vec_param[0]))
|
|
{
|
|
Error("mysql_stmt_bind_param");
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CStmt::BindResult(enum_field_types type, void * p, int iMaxLen)
|
|
{
|
|
if (m_uiResultCount >= m_vec_result.size())
|
|
{
|
|
sys_err("too many result in query: %s", m_stQuery.c_str());
|
|
return false;
|
|
}
|
|
|
|
MYSQL_BIND * bind = &m_vec_result[m_uiResultCount++];
|
|
|
|
bind->buffer_type = type;
|
|
bind->buffer = (void *) p;
|
|
bind->buffer_length = iMaxLen;
|
|
return true;
|
|
}
|
|
|
|
int CStmt::Execute()
|
|
{
|
|
if (m_uiParamCount != m_vec_param.size())
|
|
{
|
|
sys_log(0, "Parameter not enough %d, expected %d query: %s", m_uiParamCount, m_vec_param.size(), m_stQuery.c_str());
|
|
return 0;
|
|
}
|
|
|
|
for (unsigned int i = 0; i < m_uiParamCount; ++i)
|
|
{
|
|
MYSQL_BIND * bind = &m_vec_param[i];
|
|
|
|
if (bind->buffer_type == MYSQL_TYPE_STRING)
|
|
{
|
|
*(m_puiParamLen + i) = strlen((const char *) bind->buffer);
|
|
sys_log(0, "param %d len %d buf %s", i, *m_puiParamLen, (const char *) bind->buffer);
|
|
}
|
|
}
|
|
|
|
if (mysql_stmt_execute(m_pkStmt))
|
|
{
|
|
Error("mysql_stmt_execute");
|
|
return 0;
|
|
}
|
|
|
|
if (mysql_stmt_store_result(m_pkStmt))
|
|
{
|
|
Error("mysql_store_result");
|
|
return 0;
|
|
}
|
|
|
|
iRows = mysql_stmt_num_rows(m_pkStmt);
|
|
return true;
|
|
}
|
|
|
|
bool CStmt::Fetch()
|
|
{
|
|
return !mysql_stmt_fetch(m_pkStmt);
|
|
}
|
|
|