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

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

生物技术

开发平台：

C/C++

lang_cmd.cpp：源码内容
  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: lang_cmd.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 19:20:51  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.13
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  10. /* $Id: lang_cmd.cpp,v 1000.1 2004/06/01 19:20: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:  TDS language command
  38.  *
  39.  */
  41. #include <ncbi_pch.hpp>
  42. #include <dbapi/driver/ftds/interfaces.hpp>
  45. BEGIN_NCBI_SCOPE
  48. /////////////////////////////////////////////////////////////////////////////
  49. //
  50. //  CTDS_LangCmd::
  51. //
  53. CTDS_LangCmd::CTDS_LangCmd(CTDS_Connection* conn, DBPROCESS* cmd,
  54.                            const string& lang_query,
  55.                            unsigned int nof_params) :
  56.     m_Connect(conn), m_Cmd(cmd), m_Query(lang_query), m_Params(nof_params)
  57. {
  59.     m_WasSent   =  false;
  60.     m_HasFailed =  false;
  61.     m_Res       =  0;
  62.     m_RowCount  = -1;
  63.     m_Status    =  0;
  64. }
  67. bool CTDS_LangCmd::More(const string& query_text)
  68. {
  69.     m_Query.append(query_text);
  70.     return true;
  71. }
  74. bool CTDS_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 CTDS_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 CTDS_LangCmd::Send()
  89. {
  90.     if (m_WasSent)
  91.         Cancel();
  93.     m_HasFailed = false;
  95.     if (!x_AssignParams()) {
  96.         dbfreebuf(m_Cmd);
  97.         m_HasFailed = true;
  98.         throw CDB_ClientEx(eDB_Error, 220003, "CTDS_LangCmd::Send",
  99.                            "cannot assign params");
  100.     }
  102.     if (dbcmd(m_Cmd, (char*)(m_Query.c_str())) != SUCCEED) {
  103.         dbfreebuf(m_Cmd);
  104.         m_HasFailed = true;
  105.         throw CDB_ClientEx(eDB_Fatal, 220001, "CTDS_LangCmd::Send",
  106.                            "dbcmd failed");
  107.     }
  109.     
  110.     m_Connect->TDS_SetTimeout();
  112.     if (dbsqlsend(m_Cmd) != SUCCEED) {
  113.         m_HasFailed = true;
  114.         throw CDB_ClientEx(eDB_Error, 220005, "CTDS_LangCmd::Send",
  115.                            "dbsqlsend failed");
  116.     }
  118.     m_WasSent = true;
  119.     m_Status = 0;
  120.     return true;
  121. }
  124. bool CTDS_LangCmd::WasSent() const
  125. {
  126.     return m_WasSent;
  127. }
  130. bool CTDS_LangCmd::Cancel()
  131. {
  132.     if (m_WasSent) {
  133.         if (m_Res) {
  134.             delete m_Res;
  135.             m_Res = 0;
  136.         }
  137.         m_WasSent = false;
  138.         return (dbcancel(m_Cmd) == SUCCEED);
  139.     }
  141.     dbfreebuf(m_Cmd);
  142.     m_Query.erase();
  143.     return true;
  144. }
  147. bool CTDS_LangCmd::WasCanceled() const
  148. {
  149.     return !m_WasSent;
  150. }
  153. CDB_Result* CTDS_LangCmd::Result()
  154. {
  155.     if (m_Res) {
  156.         if(m_RowCount < 0) {
  157.             m_RowCount= DBCOUNT(m_Cmd);
  158.         }
  159.         delete m_Res;
  160.         m_Res = 0;
  161.     }
  163.     if (!m_WasSent) {
  164.         throw CDB_ClientEx(eDB_Error, 220010, "CTDS_LangCmd::Result",
  165.                            "a command has to be sent first");
  166.     }
  168.     if (m_Status == 0) {
  169.         m_Status = 0x1;
  170.         if (dbsqlok(m_Cmd) != SUCCEED) {
  171.             m_WasSent = false;
  172.             m_HasFailed = true;
  173.             throw CDB_ClientEx(eDB_Error, 220011, "CTDS_LangCmd::Result",
  174.                                "dbsqlok failed");
  175.         }
  176.     }
  178.     if ((m_Status & 0x10) != 0) { // we do have a compute result
  179.         m_Res = new CTDS_ComputeResult(m_Cmd, &m_Status);
  180.         m_RowCount= 1;
  181.         return Create_Result(*m_Res);
  182.     }
  184.     while ((m_Status & 0x1) != 0) {
  185.         switch (dbresults(m_Cmd)) {
  186.         case SUCCEED:
  187.             if (DBCMDROW(m_Cmd) == SUCCEED) { // we may get rows in this result
  188.                 m_Res = new CTDS_RowResult(m_Cmd, &m_Status);
  189.                 m_RowCount = -1;
  190.                 return Create_Result(*m_Res);
  191.             } else {
  192.                 m_RowCount = DBCOUNT(m_Cmd);
  193.                 continue;
  194.             }
  195.         case NO_MORE_RESULTS:
  196.             m_Status = 2;
  197.             break;
  198.         default:
  199.             m_HasFailed = true;
  200.             throw CDB_ClientEx(eDB_Warning, 221016, "CTDS_RPCCmd::Result",
  201.                                "error encountered in command execution");
  202.         }
  203.         break;
  204.     }
  206.     // we've done with the row results at this point
  207.     // let's look at return parameters and ret status
  208.     if (m_Status == 2) {
  209.         m_Status = 4;
  210.         int n = dbnumrets(m_Cmd);
  211.         if (n > 0) {
  212.             m_Res = new CTDS_ParamResult(m_Cmd, n);
  213.             m_RowCount = 1;
  214.             return Create_Result(*m_Res);
  215.         }
  216.     }
  218.     if (m_Status == 4) {
  219.         m_Status = 6;
  220.         if (dbhasretstat(m_Cmd)) {
  221.             m_Res = new CTDS_StatusResult(m_Cmd);
  222.             m_RowCount = 1;
  223.             return Create_Result(*m_Res);
  224.         }
  225.     }
  227.     m_WasSent = false;
  228.     return 0;
  229. }
  231. bool CTDS_LangCmd::HasMoreResults() const
  232. {
  233.     return m_WasSent;
  234. }
  236. void CTDS_LangCmd::DumpResults()
  237. {
  238.     CDB_Result* dbres;
  239.     while(m_WasSent) {
  240.         dbres= Result();
  241.         if(dbres) {
  242.             if(m_Connect->m_ResProc) {
  243.                 m_Connect->m_ResProc->ProcessResult(*dbres);
  244.             }
  245.             else {
  246.                 while(dbres->Fetch());
  247.             }
  248.             delete dbres;
  249.         }
  250.     }
  251. }
  253. bool CTDS_LangCmd::HasFailed() const
  254. {
  255.     return m_HasFailed;
  256. }
  259. int CTDS_LangCmd::RowCount() const
  260. {
  261.     return (m_RowCount < 0)? DBCOUNT(m_Cmd) : m_RowCount;
  262. }
  265. void CTDS_LangCmd::Release()
  266. {
  267.     m_BR = 0;
  268.     if (m_WasSent) {
  269.         Cancel();
  270.         m_WasSent = false;
  271.     }
  272.     m_Connect->DropCmd(*this);
  273.     delete this;
  274. }
  277. CTDS_LangCmd::~CTDS_LangCmd()
  278. {
  279.     if (m_BR)
  280.         *m_BR = 0;
  281.     if (m_WasSent)
  282.         Cancel();
  283. }
  286. bool CTDS_LangCmd::x_AssignParams()
  287. {
  288.     static const char s_hexnum[] = "0123456789ABCDEF";
  290.     for (unsigned int n = 0; n < m_Params.NofParams(); n++) {
  291.         if(m_Params.GetParamStatus(n) == 0) continue;
  292.         const string& name  =  m_Params.GetParamName(n);
  293.         CDB_Object&   param = *m_Params.GetParam(n);
  294.         char          val_buffer[16*1024];
  295.         const char*   type;
  296.         string        cmd;
  298.         if (name.length() > kTDSMaxNameLen)
  299.             return false;
  301.         switch (param.GetType()) {
  302.         case eDB_Int:
  303.             type = "int";
  304.             break;
  305.         case eDB_SmallInt:
  306.             type = "smallint";
  307.             break;
  308.         case eDB_TinyInt:
  309.             type = "tinyint";
  310.             break;
  311.         case eDB_BigInt:
  312.             type = "numeric";
  313.             break;
  314.         case eDB_Char:
  315.         case eDB_VarChar:
  316.             type = "varchar(255)";
  317.             break;
  318. case eDB_LongChar: {
  319.     CDB_LongChar& lc = dynamic_cast<CDB_LongChar&> (param);
  320.     sprintf(val_buffer, "varchar(%d)", lc.Size());
  321.     type= val_buffer;
  322. break;
  323. }
  324.         case eDB_Binary:
  325.         case eDB_VarBinary:
  326.             type = "varbinary(255)";
  327.             break;
  328. case eDB_LongBinary: {
  329.     CDB_LongBinary& lb = dynamic_cast<CDB_LongBinary&> (param);
  330. if(lb.DataSize()*2 > (sizeof(val_buffer) - 4)) return false;
  331.     sprintf(val_buffer, "varbinary(%d)", lb.Size());
  332.     type= val_buffer;
  333. break;
  334. }
  335.         case eDB_Float:
  336.             type = "real";
  337.             break;
  338.         case eDB_Double:
  339.             type = "float";
  340.             break;
  341.         case eDB_SmallDateTime:
  342.             type = "smalldatetime";
  343.             break;
  344.         case eDB_DateTime:
  345.             type = "datetime";
  346.             break;            
  347.         default:
  348.             return false;
  349.         }
  351.         cmd += "declare " + name + ' ' + type + "nselect " + name + " = ";
  353.         if (!param.IsNULL()) {
  354.             switch (param.GetType()) {
  355.             case eDB_Int: {
  356.                 CDB_Int& val = dynamic_cast<CDB_Int&> (param);
  357.                 sprintf(val_buffer, "%dn", val.Value());
  358.                 break;
  359.             }
  360.             case eDB_SmallInt: {
  361.                 CDB_SmallInt& val = dynamic_cast<CDB_SmallInt&> (param);
  362.                 sprintf(val_buffer, "%dn", (int) val.Value());
  363.                 break;
  364.             }
  365.             case eDB_TinyInt: {
  366.                 CDB_TinyInt& val = dynamic_cast<CDB_TinyInt&> (param);
  367.                 sprintf(val_buffer, "%dn", (int) val.Value());
  368.                 break;
  369.             }
  370.             case eDB_BigInt: {
  371.                 CDB_BigInt& val = dynamic_cast<CDB_BigInt&> (param);
  372.                 sprintf(val_buffer, "%lldn", val.Value());
  373.                 break;
  374.             }
  375.             case eDB_Char: {
  376.                 CDB_Char& val = dynamic_cast<CDB_Char&> (param);
  377.                 const char* c = val.Value(); // No more than 255 chars
  378.                 size_t i = 0;
  379.                 val_buffer[i++] = ''';
  380.                 while (*c) {
  381.                     if (*c == ''')
  382.                         val_buffer[i++] = ''';
  383.                     val_buffer[i++] = *c++;
  384.                 }
  385.                 val_buffer[i++] = ''';
  386.                 val_buffer[i++] = 'n';
  387.                 val_buffer[i] = '';
  388.                 break;
  389.             }
  390.             case eDB_VarChar: {
  391.                 CDB_VarChar& val = dynamic_cast<CDB_VarChar&> (param);
  392.                 const char* c = val.Value();
  393.                 size_t i = 0;
  394.                 val_buffer[i++] = ''';
  395.                 while (*c) {
  396.                     if (*c == ''')
  397.                         val_buffer[i++] = ''';
  398.                     val_buffer[i++] = *c++;
  399.                 }
  400.                 val_buffer[i++] = ''';
  401.                 val_buffer[i++] = 'n';
  402.                 val_buffer[i] = '';
  403.                 break;
  404.             }
  405.             case eDB_LongChar: {
  406.                 CDB_LongChar& val = dynamic_cast<CDB_LongChar&> (param);
  407.                 const char* c = val.Value();
  408.                 size_t i = 0;
  409.                 val_buffer[i++] = ''';
  410.                 while (*c && (i < (sizeof(val_buffer)-3))) {
  411.                     if (*c == ''')
  412.                         val_buffer[i++] = ''';
  413.                     val_buffer[i++] = *c++;
  414.                 }
  415.                 val_buffer[i++] = ''';
  416.                 val_buffer[i++] = 'n';
  417.                 val_buffer[i] = '';
  418. if(*c != '') return false;
  419.                 break;
  420.             }
  421.             case eDB_Binary: {
  422.                 CDB_Binary& val = dynamic_cast<CDB_Binary&> (param);
  423.                 const unsigned char* c = (const unsigned char*) val.Value();
  424.                 size_t i = 0, size = val.Size();
  425.                 val_buffer[i++] = '0';
  426.                 val_buffer[i++] = 'x';
  427.                 for (size_t j = 0; j < size; j++) {
  428.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  429.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  430.                 }
  431.                 val_buffer[i++] = 'n';
  432.                 val_buffer[i++] = '';
  433.                 break;
  434.             }
  435.             case eDB_VarBinary: {
  436.                 CDB_VarBinary& val = dynamic_cast<CDB_VarBinary&> (param);
  437.                 const unsigned char* c = (const unsigned char*) val.Value();
  438.                 size_t i = 0, size = val.Size();
  439.                 val_buffer[i++] = '0';
  440.                 val_buffer[i++] = 'x';
  441.                 for (size_t j = 0; j < size; j++) {
  442.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  443.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  444.                 }
  445.                 val_buffer[i++] = 'n';
  446.                 val_buffer[i++] = '';
  447.                 break;
  448.             }
  449.             case eDB_LongBinary: {
  450.                 CDB_LongBinary& val = dynamic_cast<CDB_LongBinary&> (param);
  451.                 const unsigned char* c = (const unsigned char*) val.Value();
  452.                 size_t i = 0, size = val.DataSize();
  453.                 val_buffer[i++] = '0';
  454.                 val_buffer[i++] = 'x';
  455.                 for (size_t j = 0; j < size; j++) {
  456.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  457.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  458.                 }
  459.                 val_buffer[i++] = 'n';
  460.                 val_buffer[i++] = '';
  461.                 break;
  462.             }
  463.             case eDB_Float: {
  464.                 CDB_Float& val = dynamic_cast<CDB_Float&> (param);
  465.                 sprintf(val_buffer, "%En", (double) val.Value());
  466.                 break;
  467.             }
  468.             case eDB_Double: {
  469.                 CDB_Double& val = dynamic_cast<CDB_Double&> (param);
  470.                 sprintf(val_buffer, "%En", val.Value());
  471.                 break;
  472.             }
  473.             case eDB_SmallDateTime: {
  474.                 CDB_SmallDateTime& val =
  475.                     dynamic_cast<CDB_SmallDateTime&> (param);
  476.                 sprintf(val_buffer, "'%s'n",val.Value().AsString("M/D/Y h:m").c_str());
  477.                 break;
  478.             }
  479.             case eDB_DateTime: {
  480.                 CDB_DateTime& val =
  481.                     dynamic_cast<CDB_DateTime&> (param);
  482.                 sprintf(val_buffer,  "'%s:%.3d'n",   val.Value().AsString("M/D/Y h:m:s").c_str(),
  483. (int)(val.Value().NanoSecond()/1000000));
  484.                 break;
  485.             }
  486.             default:
  487.                 return false; // dummy for compiler
  488.             }
  489.             cmd += val_buffer;
  490.         } else
  491.             cmd += "NULLn";
  492.         if (dbcmd(m_Cmd, (char*) cmd.c_str()) != SUCCEED)
  493.             return false;
  494.     }
  495.     return true;
  496. }
  499. END_NCBI_SCOPE
  503. /*
  504.  * ===========================================================================
  505.  * $Log: lang_cmd.cpp,v $
  506.  * Revision 1000.1  2004/06/01 19:20:51  gouriano
  507.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.13
  508.  *
  509.  * Revision 1.13  2004/05/17 21:13:37  gorelenk
  510.  * Added include of PCH ncbi_pch.hpp
  511.  *
  512.  * Revision 1.12  2003/06/05 16:01:40  soussov
  513.  * adds code for DumpResults and for the dumped results processing
  514.  *
  515.  * Revision 1.11  2003/05/16 20:26:09  soussov
  516.  * adds code to skip parameter if it was not set
  517.  *
  518.  * Revision 1.10  2003/04/29 21:15:03  soussov
  519.  * new datatypes CDB_LongChar and CDB_LongBinary added
  520.  *
  521.  * Revision 1.9  2002/09/19 22:28:20  soussov
  522.  * changs order of result processing
  523.  *
  524.  * Revision 1.8  2002/07/22 20:11:07  soussov
  525.  * fixes the RowCount calculations
  526.  *
  527.  * Revision 1.7  2002/01/14 20:38:49  soussov
  528.  * timeout support for tds added
  529.  *
  530.  * Revision 1.6  2001/12/18 19:29:08  soussov
  531.  * adds conversion from nanosecs to milisecs for datetime args
  532.  *
  533.  * Revision 1.5  2001/12/18 17:56:39  soussov
  534.  * copy-paste bug in datetime processing fixed
  535.  *
  536.  * Revision 1.4  2001/12/18 16:42:44  soussov
  537.  * fixes bug in datetime argument processing
  538.  *
  539.  * Revision 1.3  2001/12/13 23:40:53  soussov
  540.  * changes double quotes to single quotes in SQL queries
  541.  *
  542.  * Revision 1.2  2001/11/06 18:00:02  lavr
  543.  * Formatted uniformly as the rest of the library
  544.  *
  545.  * Revision 1.1  2001/10/25 00:39:22  vakatov
  546.  * Initial revision
  547.  *
  548.  * ===========================================================================
  549.  */