bool CODBC_RPCCmd::Send()
{
Cancel();
SetHasFailed(false);
m_HasStatus = false;
// make a language command
string main_exec_query("declare @STpROCrETURNsTATUS int;\nexec @STpROCrETURNsTATUS=");
main_exec_query += GetQuery();
string param_result_query;
CMemPot bindGuard;
string q_str;
if(GetBindParamsImpl().NofParams() > 0) {
SQLLEN* indicator = (SQLLEN*)
bindGuard.Alloc(GetBindParamsImpl().NofParams() * sizeof(SQLLEN));
if (!x_AssignParams(q_str, main_exec_query, param_result_query,
bindGuard, indicator)) {
ResetParams();
SetHasFailed();
string err_message = "Cannot assign params." + GetDbgInfo();
DATABASE_DRIVER_ERROR( err_message, 420003 );
}
}
if(NeedToRecompile()) main_exec_query += " with recompile";
q_str += main_exec_query + ";\nselect STpROCrETURNsTATUS=@STpROCrETURNsTATUS";
if(!param_result_query.empty()) {
q_str += ";\nselect " + param_result_query;
}
switch(SQLExecDirect(GetHandle(), CODBCString(q_str, GetClientEncoding()), SQL_NTS)) {
case SQL_SUCCESS:
m_HasMoreResults = true;
break;
case SQL_NO_DATA:
m_HasMoreResults = true; /* this is a bug in SQLExecDirect it returns SQL_NO_DATA if
status result is the only result of RPC */
m_RowCount = 0;
break;
case SQL_ERROR:
ReportErrors();
ResetParams();
SetHasFailed();
{
string err_message = "SQLExecDirect failed." + GetDbgInfo();
DATABASE_DRIVER_ERROR( err_message, 420001 );
}
case SQL_SUCCESS_WITH_INFO:
ReportErrors();
m_HasMoreResults = true;
break;
case SQL_STILL_EXECUTING:
ReportErrors();
ResetParams();
SetHasFailed();
{
string err_message = "Some other query is executing on this connection." +
GetDbgInfo();
DATABASE_DRIVER_ERROR( err_message, 420002 );
}
case SQL_INVALID_HANDLE:
SetHasFailed();
{
string err_message = "The statement handler is invalid (memory corruption suspected)." +
GetDbgInfo();
DATABASE_DRIVER_ERROR( err_message, 420004 );
}
default:
ReportErrors();
ResetParams();
SetHasFailed();
{
string err_message = "Unexpected error." + GetDbgInfo();
DATABASE_DRIVER_ERROR( err_message, 420005 );
}
}
SetWasSent();
return true;
}
bool CTL_BCPInCmd::Send(void)
{
unsigned int i;
CS_INT datalen = 0;
size_t len = 0;
char buff[2048];
CheckIsDead();
if ( !WasSent() ) {
// we need to init the bcp
CheckSFB(blk_init(x_GetSybaseCmd(), CS_BLK_IN,
(CS_CHAR*) GetQuery().data(), GetQuery().size()),
"blk_init failed", 123001);
SetWasSent();
// check what needs to be default
CS_DATAFMT fmt;
for (i = 0; i < GetBindParamsImpl().NofParams(); i++) {
if (GetBindParamsImpl().GetParamStatus(i) != 0) {
continue;
}
SetHasFailed((Check(blk_describe(x_GetSybaseCmd(),
i + 1,
&fmt)) != CS_SUCCEED));
CHECK_DRIVER_ERROR(
HasFailed(),
"blk_describe failed (check the number of "
"columns in a table)." + GetDbgInfo(),
123002 );
}
}
SetHasFailed(!x_AssignParams());
CHECK_DRIVER_ERROR( HasFailed(), "Cannot assign the params." + GetDbgInfo(), 123004 );
switch ( Check(blk_rowxfer(x_GetSybaseCmd())) ) {
case CS_BLK_HAS_TEXT:
for (i = 0; i < GetBindParamsImpl().NofParams(); i++) {
if (GetBindParamsImpl().GetParamStatus(i) == 0)
continue;
CDB_Object& param = *GetBindParamsImpl().GetParam(i);
if (param.IsNULL()) {
continue;
}
else if (param.GetType() == eDB_Text || param.GetType() == eDB_Image) {
CDB_Stream& par = dynamic_cast<CDB_Stream&> (param);
for (datalen = (CS_INT) par.Size(); datalen > 0;
datalen -= (CS_INT) len)
{
len = par.Read(buff, sizeof(buff));
SetHasFailed((Check(blk_textxfer(x_GetSybaseCmd(),
(CS_BYTE*) buff,
(CS_INT) len,
0)
) == CS_FAIL));
CHECK_DRIVER_ERROR(
HasFailed(),
"blk_textxfer failed for the text/image field." + GetDbgInfo(), 123005
);
}
}
}
case CS_SUCCEED:
++m_RowCount;
return true;
default:
SetHasFailed();
CHECK_DRIVER_ERROR( HasFailed(), "blk_rowxfer failed." + GetDbgInfo(), 123007 );
}
return false;
}
bool CODBC_BCPInCmd::Send(void)
{
char param_buff[2048]; // maximal row size, assured of buffer overruns
if (!x_AssignParams(param_buff)) {
SetHasFailed();
string err_message = "Cannot assign params." + GetDbgInfo();
DATABASE_DRIVER_ERROR( err_message, 423004 );
}
SetWasSent();
if (bcp_sendrow(GetHandle()) != SUCCEED) {
SetHasFailed();
ReportErrors();
string err_message = "bcp_sendrow failed." + GetDbgInfo();
DATABASE_DRIVER_ERROR( err_message, 423005 );
}
if (m_HasTextImage) { // send text/image data
char buff[1800]; // text/image page size
for (unsigned int i = 0; i < GetBindParamsImpl().NofParams(); ++i) {
if (GetBindParamsImpl().GetParamStatus(i) == 0)
continue;
CDB_Object& param = *GetBindParamsImpl().GetParam(i);
if (param.GetType() != eDB_Image &&
(param.GetType() != eDB_Text || param.IsNULL()))
continue;
CDB_Stream& val = dynamic_cast<CDB_Stream&> (param);
size_t left_bytes = val.Size();
size_t len = 0;
size_t valid_len = 0;
size_t invalid_len = 0;
do {
invalid_len = len - valid_len;
if (valid_len < len) {
memmove(buff, buff + valid_len, invalid_len);
}
len = val.Read(buff + invalid_len, sizeof(buff) - invalid_len);
if (len > left_bytes) {
len = left_bytes;
}
valid_len = CUtf8::GetValidBytesCount( CTempString(buff, len));
CODBCString odbc_str(buff, len);
// !!! TODO: Decode to UCS2 if needed !!!!
if (bcp_moretext(GetHandle(),
(DBINT) valid_len,
(LPCBYTE)static_cast<const char*>(odbc_str)
) != SUCCEED) {
SetHasFailed();
ReportErrors();
string err_text;
if (param.GetType() == eDB_Text) {
err_text = "bcp_moretext for text failed.";
} else {
err_text = "bcp_moretext for image failed.";
}
err_text += GetDbgInfo();
DATABASE_DRIVER_ERROR( err_text, 423006 );
}
if (!valid_len) {
break;
}
left_bytes -= valid_len;
} while (left_bytes);
}
}
return true;
}