开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数值算法/人工智能 > 生物技术 > 查看源码
lang_cmd.cpp
资源名称：ncbi_cxx_win--Jul_08_2004.zip　[点击查看]
上传用户：yhdzpy8989
上传日期：2007-06-13
资源大小：13604k
文件大小：17k
源码类别：

生物技术

开发平台：

C/C++

lang_cmd.cpp：源码内容
  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: lang_cmd.cpp,v $
  4.  * PRODUCTION Revision 1000.2  2004/06/01 19:21:51  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.8
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  10. /* $Id: lang_cmd.cpp,v 1000.2 2004/06/01 19:21:51 gouriano Exp $
  11.  * ===========================================================================
  12.  *
  13.  *                            PUBLIC DOMAIN NOTICE
  14.  *               National Center for Biotechnology Information
  15.  *
  16.  *  This software/database is a "United States Government Work" under the
  17.  *  terms of the United States Copyright Act.  It was written as part of
  18.  *  the author's official duties as a United States Government employee and
  19.  *  thus cannot be copyrighted.  This software/database is freely available
  20.  *  to the public for use. The National Library of Medicine and the U.S.
  21.  *  Government have not placed any restriction on its use or reproduction.
  22.  *
  23.  *  Although all reasonable efforts have been taken to ensure the accuracy
  24.  *  and reliability of the software and data, the NLM and the U.S.
  25.  *  Government do not and cannot warrant the performance or results that
  26.  *  may be obtained by using this software or data. The NLM and the U.S.
  27.  *  Government disclaim all warranties, express or implied, including
  28.  *  warranties of performance, merchantability or fitness for any particular
  29.  *  purpose.
  30.  *
  31.  *  Please cite the author in any work or product based on this material.
  32.  *
  33.  * ===========================================================================
  34.  *
  35.  * Author:  Vladimir Soussov
  36.  *
  37.  * File Description:  ODBC language command
  38.  *
  39.  */
  41. #include <ncbi_pch.hpp>
  42. #include <stdio.h>
  43. #include <dbapi/driver/odbc/interfaces.hpp>
  46. BEGIN_NCBI_SCOPE
  49. /////////////////////////////////////////////////////////////////////////////
  50. //
  51. //  CODBC_LangCmd::
  52. //
  54. CODBC_LangCmd::CODBC_LangCmd(CODBC_Connection* conn, SQLHSTMT cmd,
  55.                            const string& lang_query,
  56.                            unsigned int nof_params) :
  57.     m_Connect(conn), m_Cmd(cmd), m_Query(lang_query), m_Params(nof_params), 
  58.     m_Reporter(&conn->m_MsgHandlers, SQL_HANDLE_STMT, cmd)
  59. {
  60.     m_WasSent   =  false;
  61.     m_HasFailed =  false;
  62.     m_Res       =  0;
  63.     m_RowCount  = -1;
  64. }
  67. bool CODBC_LangCmd::More(const string& query_text)
  68. {
  69.     m_Query.append(query_text);
  70.     return true;
  71. }
  74. bool CODBC_LangCmd::BindParam(const string& param_name, CDB_Object* param_ptr)
  75. {
  76.     return
  77.         m_Params.BindParam(CDB_Params::kNoParamNumber, param_name, param_ptr);
  78. }
  81. bool CODBC_LangCmd::SetParam(const string& param_name, CDB_Object* param_ptr)
  82. {
  83.     return
  84.         m_Params.SetParam(CDB_Params::kNoParamNumber, param_name, param_ptr);
  85. }
  88. bool CODBC_LangCmd::Send()
  89. {
  90.     if (m_WasSent)
  91.         Cancel();
  93.     m_HasFailed = false;
  95.     CMemPot bindGuard;
  96.     string q_str;
  98. if(m_Params.NofParams() > 0) {
  99. SQLINTEGER* indicator= (SQLINTEGER*)
  100. bindGuard.Alloc(m_Params.NofParams()*sizeof(SQLINTEGER));
  102. if (!x_AssignParams(q_str, bindGuard, indicator)) {
  103. SQLFreeStmt(m_Cmd, SQL_RESET_PARAMS);
  104. m_HasFailed = true;
  105. throw CDB_ClientEx(eDB_Error, 420003, "CODBC_LangCmd::Send",
  106.                            "cannot assign params");
  107. }
  108. }
  110.     string* real_query;
  111.     if(!q_str.empty()) {
  112.         q_str.append(m_Query);
  113.         real_query= &q_str;
  114.     }
  115.     else {
  116.         real_query= &m_Query;
  117.     }
  119.     switch(SQLExecDirect(m_Cmd, (SQLCHAR*)real_query->c_str(), SQL_NTS)) {
  120.     case SQL_SUCCESS:
  121.         m_hasResults= true;
  122.         break;
  124.     case SQL_NO_DATA:
  125.         m_hasResults= false;
  126.         m_RowCount= 0;
  127.         break;
  129.     case SQL_ERROR:
  130.         m_Reporter.ReportErrors();
  131.         SQLFreeStmt(m_Cmd, SQL_RESET_PARAMS);
  132.         m_HasFailed = true;
  133.         throw CDB_ClientEx(eDB_Fatal, 420001, "CODBC_LangCmd::Send",
  134.                            "SQLExecDirect failed");
  136.     case SQL_SUCCESS_WITH_INFO:
  137.         m_Reporter.ReportErrors();
  138.         m_hasResults= true;
  139.         break;
  141.     case SQL_STILL_EXECUTING:
  142.         m_Reporter.ReportErrors();
  143.         SQLFreeStmt(m_Cmd, SQL_RESET_PARAMS);
  144.         m_HasFailed = true;
  145.         throw CDB_ClientEx(eDB_Fatal, 420002, "CODBC_LangCmd::Send",
  146.                            "Some other query is executing on this connection");
  147.         
  148.     case SQL_INVALID_HANDLE:
  149.         m_HasFailed= true;
  150.         throw CDB_ClientEx(eDB_Fatal, 420004, "CODBC_LangCmd::Send",
  151.                            "The statement handler is invalid (memory corruption suspected)");
  152.         
  153.     default:
  154.         m_Reporter.ReportErrors();
  155.         SQLFreeStmt(m_Cmd, SQL_RESET_PARAMS);
  156.         m_HasFailed = true;
  157.         throw CDB_ClientEx(eDB_Fatal, 420005, "CODBC_LangCmd::Send",
  158.                            "Unexpected error");
  159.         
  160.     }
  162.     m_WasSent = true;
  163.     return true;
  164. }
  167. bool CODBC_LangCmd::WasSent() const
  168. {
  169.     return m_WasSent;
  170. }
  173. bool CODBC_LangCmd::Cancel()
  174. {
  175.     if (m_WasSent) {
  176.         if (m_Res) {
  177.             delete m_Res;
  178.             m_Res = 0;
  179.         }
  180.         m_WasSent = false;
  181.         switch(SQLFreeStmt(m_Cmd, SQL_CLOSE)) {
  182.         case SQL_SUCCESS_WITH_INFO: m_Reporter.ReportErrors();
  183.         case SQL_SUCCESS:           break;
  184.         case SQL_ERROR:             m_Reporter.ReportErrors();
  185.         default:                    return false;
  186.         }
  187.     }
  189.     SQLFreeStmt(m_Cmd, SQL_RESET_PARAMS);
  190.     // m_Query.erase();
  191.     return true;
  192. }
  195. bool CODBC_LangCmd::WasCanceled() const
  196. {
  197.     return !m_WasSent;
  198. }
  201. CDB_Result* CODBC_LangCmd::Result()
  202. {
  203.     if (m_Res) {
  204.         delete m_Res;
  205.         m_Res = 0;
  206.         m_hasResults= xCheck4MoreResults();
  207.     }
  209.     if (!m_WasSent) {
  210.         throw CDB_ClientEx(eDB_Error, 420010, "CODBC_LangCmd::Result",
  211.                            "a command has to be sent first");
  212.     }
  214.     if(!m_hasResults) {
  215. m_WasSent= false;
  216.         return 0;
  217.     }
  219.     SQLSMALLINT nof_cols= 0;
  221.     while(m_hasResults) {
  222.         switch(SQLNumResultCols(m_Cmd, &nof_cols)) {
  223.         case SQL_SUCCESS_WITH_INFO:
  224.             m_Reporter.ReportErrors();
  226.         case SQL_SUCCESS:
  227.             break;
  229.         case SQL_ERROR:
  230.             m_Reporter.ReportErrors();
  231.             throw CDB_ClientEx(eDB_Error, 420011, "CODBC_LangCmd::Result",
  232.                                "SQLNumResultCols failed");
  233.         default:
  234.             throw CDB_ClientEx(eDB_Error, 420012, "CODBC_LangCmd::Result",
  235.                                "SQLNumResultCols failed (memory corruption suspected)");
  236.         }
  238.         if(nof_cols < 1) { // no data in this result set
  239. SQLINTEGER rc;
  240. switch(SQLRowCount(m_Cmd, &rc)) {
  241. case SQL_SUCCESS_WITH_INFO:
  242. m_Reporter.ReportErrors(); 
  243. case SQL_SUCCESS: break;
  244. case SQL_ERROR:
  245. m_Reporter.ReportErrors();
  246. throw CDB_ClientEx(eDB_Error, 420013, "CODBC_LangCmd::Result",
  247.                                "SQLRowCount failed");
  248. default:
  249. throw CDB_ClientEx(eDB_Error, 420014, "CODBC_LangCmd::Result",
  250. "SQLRowCount failed (memory corruption suspected)");
  251. }
  253.             m_RowCount = rc;
  254.             m_hasResults= xCheck4MoreResults();
  255.             continue;
  256.         }
  258.         m_Res = new CODBC_RowResult(nof_cols, m_Cmd, m_Reporter);
  259.         return Create_Result(*m_Res);
  260.     }
  262.     m_WasSent = false;
  263.     return 0;
  264. }
  267. bool CODBC_LangCmd::HasMoreResults() const
  268. {
  269.     return m_hasResults;
  270. }
  272. void CODBC_LangCmd::DumpResults()
  273. {
  274.     CDB_Result* dbres;
  275.     while(m_WasSent) {
  276.         dbres= Result();
  277.         if(dbres) {
  278.             if(m_Connect->m_ResProc) {
  279.                 m_Connect->m_ResProc->ProcessResult(*dbres);
  280.             }
  281.             else {
  282.                 while(dbres->Fetch());
  283.             }
  284.             delete dbres;
  285.         }
  286.     }
  287. }
  289. bool CODBC_LangCmd::HasFailed() const
  290. {
  291.     return m_HasFailed;
  292. }
  295. int CODBC_LangCmd::RowCount() const
  296. {
  297.     return m_RowCount;
  298. }
  301. void CODBC_LangCmd::Release()
  302. {
  303.     m_BR = 0;
  304.     if (m_WasSent) {
  305.         Cancel();
  306.         m_WasSent = false;
  307.     }
  308.     m_Connect->DropCmd(*this);
  309.     delete this;
  310. }
  313. CODBC_LangCmd::~CODBC_LangCmd()
  314. {
  315.     if (m_BR)
  316.         *m_BR = 0;
  317.     if (m_WasSent)
  318.         Cancel();
  319.     SQLFreeHandle(SQL_HANDLE_STMT, m_Cmd);
  320. }
  323. bool CODBC_LangCmd::x_AssignParams(string& cmd, CMemPot& bind_guard, SQLINTEGER* indicator)
  324. {
  325.     for (unsigned int n = 0; n < m_Params.NofParams(); n++) {
  326.         if(m_Params.GetParamStatus(n) == 0) continue;
  327.         const string& name  =  m_Params.GetParamName(n);
  328.         CDB_Object&   param = *m_Params.GetParam(n);
  329.         const char*   type;
  330. char tbuf[64];
  332. SQLRETURN rc_from_bind;
  334.         switch (param.GetType()) {
  335.         case eDB_Int: {
  336.             CDB_Int& val = dynamic_cast<CDB_Int&> (param);
  337.             type = "int";
  338.             indicator[n]= 4;
  339.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_SLONG, 
  340.                              SQL_INTEGER, 4, 0, val.BindVal(), 4, indicator+n);
  341.             break;
  342.         }
  343.         case eDB_SmallInt: {
  344.             CDB_SmallInt& val = dynamic_cast<CDB_SmallInt&> (param);
  345.             type = "smallint";
  346.             indicator[n]= 2;
  347.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_SSHORT, 
  348.                              SQL_SMALLINT, 2, 0, val.BindVal(), 2, indicator+n);
  349.             break;
  350.         }
  351.         case eDB_TinyInt: {
  352.             CDB_TinyInt& val = dynamic_cast<CDB_TinyInt&> (param);
  353.             type = "tinyint";
  354.             indicator[n]= 1;
  355.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_UTINYINT, 
  356.                              SQL_TINYINT, 1, 0, val.BindVal(), 1, indicator+n);
  357.             break;
  358.         }
  359.         case eDB_BigInt: {
  360.             CDB_BigInt& val = dynamic_cast<CDB_BigInt&> (param);
  361.             type = "numeric";
  362.             indicator[n]= 8;
  363.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_SBIGINT, 
  364.                              SQL_NUMERIC, 18, 0, val.BindVal(), 18, indicator+n);
  365.             
  366.             break;
  367.         }
  368.         case eDB_Char: {
  369.             CDB_Char& val = dynamic_cast<CDB_Char&> (param);
  370.             type= "varchar(255)";
  371.             indicator[n]= SQL_NTS;
  372.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_CHAR, 
  373.                              SQL_VARCHAR, 255, 0, (void*)val.Value(), 256, indicator+n);
  374.             break;
  375.         }
  376.         case eDB_VarChar: {
  377.             CDB_VarChar& val = dynamic_cast<CDB_VarChar&> (param);
  378.             type = "varchar(255)";
  379.             indicator[n]= SQL_NTS;
  380.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_CHAR, 
  381.                              SQL_VARCHAR, 255, 0, (void*)val.Value(), 256, indicator+n);
  382.             break;
  383.         }
  384.         case eDB_LongChar: {
  385.             CDB_LongChar& val = dynamic_cast<CDB_LongChar&> (param);
  386. sprintf(tbuf,"varchar(%d)", val.Size());
  387.             type= tbuf;
  388.             indicator[n]= SQL_NTS;
  389.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_CHAR, 
  390.                              SQL_VARCHAR, val.Size(), 0, (void*)val.Value(), val.Size(), indicator+n);
  391.             break;
  392.         }
  393.         case eDB_Binary: {
  394.             CDB_Binary& val = dynamic_cast<CDB_Binary&> (param);
  395.             type = "varbinary(255)";
  396.             indicator[n]= val.Size();
  397.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_BINARY, 
  398.                              SQL_VARBINARY, 255, 0, (void*)val.Value(), 255, indicator+n);
  399.             break;
  400.         }
  401.         case eDB_VarBinary: {
  402.             CDB_VarBinary& val = dynamic_cast<CDB_VarBinary&> (param);
  403.             type = "varbinary(255)";
  404.             indicator[n]= val.Size();
  405.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_BINARY, 
  406.                              SQL_VARBINARY, 255, 0, (void*)val.Value(), 255, indicator+n);
  407.             break;
  408.         }
  409.         case eDB_LongBinary: {
  410.             CDB_LongBinary& val = dynamic_cast<CDB_LongBinary&> (param);
  411. sprintf(tbuf,"varbinary(%d)", val.Size());
  412.             type= tbuf;
  413.             type = "varbinary(255)";
  414.             indicator[n]= val.DataSize();
  415.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_BINARY, 
  416.                              SQL_VARBINARY, val.Size(), 0, (void*)val.Value(), val.Size(), indicator+n);
  417.             break;
  418.         }
  419.         case eDB_Float: {
  420.             CDB_Float& val = dynamic_cast<CDB_Float&> (param);
  421.             type = "real";
  422.             indicator[n]= 4;
  423.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_FLOAT, 
  424.                              SQL_REAL, 4, 0, val.BindVal(), 4, indicator+n);
  425.             break;
  426.         }
  427.         case eDB_Double: {
  428.             CDB_Double& val = dynamic_cast<CDB_Double&> (param);
  429.             type = "float";
  430.             indicator[n]= 8;
  431.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_DOUBLE, 
  432.                              SQL_FLOAT, 8, 0, val.BindVal(), 8, indicator+n);
  433.             break;
  434.         }
  435.         case eDB_SmallDateTime: {
  436.             CDB_SmallDateTime& val = dynamic_cast<CDB_SmallDateTime&> (param);
  437.             type = "smalldatetime";
  438.             SQL_TIMESTAMP_STRUCT* ts= 0;
  439.             if(!val.IsNULL()) {
  440.                 ts= (SQL_TIMESTAMP_STRUCT*)bind_guard.Alloc(sizeof(SQL_TIMESTAMP_STRUCT));
  441.                 const CTime& t= val.Value();
  442.                 ts->year= t.Year();
  443.                 ts->month= t.Month();
  444.                 ts->day= t.Day();
  445.                 ts->hour= t.Hour();
  446.                 ts->minute= t.Minute();
  447.                 ts->second= 0;
  448.                 ts->fraction= 0;
  449.                 indicator[n]= sizeof(SQL_TIMESTAMP_STRUCT);
  450.             }
  451.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_TYPE_TIMESTAMP, 
  452.                              SQL_TYPE_TIMESTAMP, 16, 0, (void*)ts, sizeof(SQL_TIMESTAMP_STRUCT), 
  453.                              indicator+n);
  454.             break;
  455.         }
  456.         case eDB_DateTime: {
  457.             CDB_DateTime& val = dynamic_cast<CDB_DateTime&> (param);
  458.             type = "datetime";
  459.             SQL_TIMESTAMP_STRUCT* ts= 0;
  460.             if(!val.IsNULL()) {
  461.                 ts= (SQL_TIMESTAMP_STRUCT*)bind_guard.Alloc(sizeof(SQL_TIMESTAMP_STRUCT));
  462.                 const CTime& t= val.Value();
  463.                 ts->year= t.Year();
  464.                 ts->month= t.Month();
  465.                 ts->day= t.Day();
  466.                 ts->hour= t.Hour();
  467.                 ts->minute= t.Minute();
  468.                 ts->second= t.Second();
  469.                 ts->fraction= t.NanoSecond()/1000000;
  470. ts->fraction*= 1000000; /* MSSQL has a bug - it can not handle fraction of msecs */
  471.                 indicator[n]= sizeof(SQL_TIMESTAMP_STRUCT);
  472.             }
  473.             rc_from_bind= SQLBindParameter(m_Cmd, n+1, SQL_PARAM_INPUT, SQL_C_TYPE_TIMESTAMP, 
  474.                              SQL_TYPE_TIMESTAMP, 23, 3, ts, sizeof(SQL_TIMESTAMP_STRUCT), 
  475.                              indicator+n);
  476.             break;
  477.         }
  478.         default:
  479.             return false;
  480.         }
  482. switch(rc_from_bind) {
  483.         case SQL_SUCCESS_WITH_INFO:
  484.         m_Reporter.ReportErrors();
  486. case SQL_SUCCESS:
  487. break;
  489. case SQL_ERROR:
  490. m_Reporter.ReportErrors();
  491. throw CDB_ClientEx(eDB_Error, 420066, "CODBC_LangCmd::x_AssignParams",
  492.                                "SQLBindParameter failed");
  493. default:
  494. throw CDB_ClientEx(eDB_Error, 420067, "CODBC_LangCmd::x_AssignParams",
  495.                                "SQLBindParameter failed (memory corruption suspected)");
  496. }
  497.         cmd += "declare " + name + ' ' + type + ";select " + name + " = ?;";
  499.         if(param.IsNULL()) {
  500.             indicator[n]= SQL_NULL_DATA;
  501.         }
  502.     }
  503.     return true;
  504. }
  507. bool CODBC_LangCmd::xCheck4MoreResults()
  508. {
  509.     switch(SQLMoreResults(m_Cmd)) {
  510.     case SQL_SUCCESS_WITH_INFO: m_Reporter.ReportErrors();
  511.     case SQL_SUCCESS:           return true;
  512.     case SQL_NO_DATA:           return false;
  513.     case SQL_ERROR:             
  514.         m_Reporter.ReportErrors();
  515.         throw CDB_ClientEx(eDB_Error, 420014, "CODBC_LangCmd::xCheck4MoreResults",
  516.                            "SQLMoreResults failed");
  517.     default:
  518.         throw CDB_ClientEx(eDB_Error, 420015, "CODBC_LangCmd::xCheck4MoreResults",
  519.                            "SQLMoreResults failed (memory corruption suspected)");
  520.     }
  521. }
  523. END_NCBI_SCOPE
  527. /*
  528.  * ===========================================================================
  529.  * $Log: lang_cmd.cpp,v $
  530.  * Revision 1000.2  2004/06/01 19:21:51  gouriano
  531.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.8
  532.  *
  533.  * Revision 1.8  2004/05/17 21:16:06  gorelenk
  534.  * Added include of PCH ncbi_pch.hpp
  535.  *
  536.  * Revision 1.7  2003/11/07 17:14:20  soussov
  537.  * work around the odbc bug. It can not handle properly the fractions of msecs
  538.  *
  539.  * Revision 1.6  2003/11/06 20:33:32  soussov
  540.  * fixed bug in DateTime bindings
  541.  *
  542.  * Revision 1.5  2003/06/05 16:02:04  soussov
  543.  * adds code for DumpResults and for the dumped results processing
  544.  *
  545.  * Revision 1.4  2003/05/16 20:26:44  soussov
  546.  * adds code to skip parameter if it was not set
  547.  *
  548.  * Revision 1.3  2003/05/08 20:40:08  soussov
  549.  * adds stdio.h for sprintf
  550.  *
  551.  * Revision 1.2  2003/05/08 20:30:24  soussov
  552.  * CDB_LongChar CDB_LongBinary added
  553.  *
  554.  * Revision 1.1  2002/06/18 22:06:24  soussov
  555.  * initial commit
  556.  *
  557.  *
  558.  * ===========================================================================
  559.  */