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

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

生物技术

开发平台：

C/C++

lang_cmd.cpp：源码内容
  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: lang_cmd.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 19:20:19  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.14
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  10. /* $Id: lang_cmd.cpp,v 1000.1 2004/06/01 19:20:19 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:  DBLib language command
  38.  *
  39.  */
  40. #include <ncbi_pch.hpp>
  42. #ifndef USE_MS_DBLIB
  43. #  include <dbapi/driver/dblib/interfaces.hpp>
  44. #  include <dbapi/driver/dblib/interfaces_p.hpp>
  45. #else
  46. #  include <dbapi/driver/msdblib/interfaces.hpp>
  47. #  include <dbapi/driver/msdblib/interfaces_p.hpp>
  48. #endif
  51. BEGIN_NCBI_SCOPE
  54. /////////////////////////////////////////////////////////////////////////////
  55. //
  56. //  CDBL_LangCmd::
  57. //
  59. CDBL_LangCmd::CDBL_LangCmd(CDBL_Connection* conn, DBPROCESS* cmd,
  60.                            const string& lang_query,
  61.                            unsigned int nof_params) :
  62.     m_Connect(conn), m_Cmd(cmd), m_Query(lang_query), m_Params(nof_params)
  63. {
  64.     m_WasSent   =  false;
  65.     m_HasFailed =  false;
  66.     m_Res       =  0;
  67.     m_RowCount  = -1;
  68.     m_Status    =  0;
  69. }
  72. bool CDBL_LangCmd::More(const string& query_text)
  73. {
  74.     m_Query.append(query_text);
  75.     return true;
  76. }
  79. bool CDBL_LangCmd::BindParam(const string& param_name, CDB_Object* param_ptr)
  80. {
  81.     return
  82.         m_Params.BindParam(CDB_Params::kNoParamNumber, param_name, param_ptr);
  83. }
  86. bool CDBL_LangCmd::SetParam(const string& param_name, CDB_Object* param_ptr)
  87. {
  88.     return
  89.         m_Params.SetParam(CDB_Params::kNoParamNumber, param_name, param_ptr);
  90. }
  93. bool CDBL_LangCmd::Send()
  94. {
  95.     if (m_WasSent)
  96.         Cancel();
  98.     m_HasFailed = false;
  100.     if (!x_AssignParams()) {
  101.         dbfreebuf(m_Cmd);
  102.         m_HasFailed = true;
  103.         throw CDB_ClientEx(eDB_Error, 220003, "CDBL_LangCmd::Send",
  104.                            "cannot assign params");
  105.     }
  107.     if (dbcmd(m_Cmd, (char*)(m_Query.c_str())) != SUCCEED) {
  108.         dbfreebuf(m_Cmd);
  109.         m_HasFailed = true;
  110.         throw CDB_ClientEx(eDB_Fatal, 220001, "CDBL_LangCmd::Send",
  111.                            "dbcmd failed");
  112.     }
  114.     if (dbsqlsend(m_Cmd) != SUCCEED) {
  115.         m_HasFailed = true;
  116.         throw CDB_ClientEx(eDB_Error, 220005, "CDBL_LangCmd::Send",
  117.                            "dbsqlsend failed");
  118.     }
  120.     m_WasSent = true;
  121.     m_Status = 0;
  122.     return true;
  123. }
  126. bool CDBL_LangCmd::WasSent() const
  127. {
  128.     return m_WasSent;
  129. }
  132. bool CDBL_LangCmd::Cancel()
  133. {
  134.     if (m_WasSent) {
  135.         if (m_Res) {
  136.             delete m_Res;
  137.             m_Res = 0;
  138.         }
  139.         m_WasSent = false;
  140.         return (dbcancel(m_Cmd) == SUCCEED);
  141.     }
  143.     dbfreebuf(m_Cmd);
  144.     m_Query.erase();
  145.     return true;
  146. }
  149. bool CDBL_LangCmd::WasCanceled() const
  150. {
  151.     return !m_WasSent;
  152. }
  155. CDB_Result* CDBL_LangCmd::Result()
  156. {
  157.     if (m_Res) {
  158.         if(m_RowCount < 0) {
  159.             m_RowCount = DBCOUNT(m_Cmd);
  160.         }
  161.         delete m_Res;
  162.         m_Res = 0;
  163.     }
  165.     if (!m_WasSent) {
  166.         throw CDB_ClientEx(eDB_Error, 220010, "CDBL_LangCmd::Result",
  167.                            "a command has to be sent first");
  168.     }
  170.     if (m_Status == 0) {
  171.         m_Status = 0x1;
  172.         if (dbsqlok(m_Cmd) != SUCCEED) {
  173.             m_WasSent = false;
  174.             m_HasFailed = true;
  175.             throw CDB_ClientEx(eDB_Error, 220011, "CDBL_LangCmd::Result",
  176.                                "dbsqlok failed");
  177.         }
  178.     }
  180.     if ((m_Status & 0x10) != 0) { // we do have a compute result
  181.         m_Res = new CDBL_ComputeResult(m_Cmd, &m_Status);
  182.         m_RowCount= 1;
  183.         return Create_Result(*m_Res);
  184.     }
  186.     while ((m_Status & 0x1) != 0) {
  187.         if ((m_Status & 0x20) != 0) { // check for return parameters from exec
  188.             m_Status ^= 0x20;
  189.             int n;
  190.             if ((n = dbnumrets(m_Cmd)) > 0) {
  191.                 m_Res = new CDBL_ParamResult(m_Cmd, n);
  192.                 m_RowCount= 1;
  193.                 return Create_Result(*m_Res);
  194.             }
  195.         }
  197.         if ((m_Status & 0x40) != 0) { // check for ret status
  198.             m_Status ^= 0x40;
  199.             if (dbhasretstat(m_Cmd)) {
  200.                 m_Res = new CDBL_StatusResult(m_Cmd);
  201.                 m_RowCount= 1;
  202.                 return Create_Result(*m_Res);
  203.             }
  204.         }
  205.         switch (dbresults(m_Cmd)) {
  206.         case SUCCEED:
  207.             m_Status |= 0x60;
  208.             if (DBCMDROW(m_Cmd) == SUCCEED) { // we could get rows in result
  210. // This optimization is currently unavailable for MS dblib...
  211. #ifndef MS_DBLIB_IN_USE /*Text,Image*/
  212.                 if (dbnumcols(m_Cmd) == 1) {
  213.                     int ct = dbcoltype(m_Cmd, 1);
  214.                     if ((ct == SYBTEXT) || (ct == SYBIMAGE)) {
  215.                         m_Res = new CDBL_BlobResult(m_Cmd);
  216.                     }
  217.                 }
  218. #endif
  219.                 if (!m_Res)
  220.                     m_Res = new CDBL_RowResult(m_Cmd, &m_Status);
  221.                 m_RowCount= -1;
  222.                 return Create_Result(*m_Res);
  223.             } else {
  224.                 m_RowCount = DBCOUNT(m_Cmd);
  225.                 continue;
  226.             }
  227.         case NO_MORE_RESULTS:
  228.             m_Status = 2;
  229.             break;
  230.         default:
  231.             m_HasFailed = true;
  232.             throw CDB_ClientEx(eDB_Warning, 220016, "CDBL_LangCmd::Result",
  233.                                "an error was encountered by server");
  234.         }
  235.         break;
  236.     }
  238.     m_WasSent = false;
  239.     return 0;
  240. }
  243. bool CDBL_LangCmd::HasMoreResults() const
  244. {
  245.     return m_WasSent;
  246. }
  248. void CDBL_LangCmd::DumpResults()
  249. {
  250.     CDB_Result* dbres;
  251.     while(m_WasSent) {
  252.         dbres= Result();
  253.         if(dbres) {
  254.             if(m_Connect->m_ResProc) {
  255.                 m_Connect->m_ResProc->ProcessResult(*dbres);
  256.             }
  257.             else {
  258.                 while(dbres->Fetch());
  259.             }
  260.             delete dbres;
  261.         }
  262.     }
  263. }
  265. bool CDBL_LangCmd::HasFailed() const
  266. {
  267.     return m_HasFailed;
  268. }
  271. int CDBL_LangCmd::RowCount() const
  272. {
  273.     return (m_RowCount < 0)? DBCOUNT(m_Cmd) : m_RowCount;
  274. }
  277. void CDBL_LangCmd::Release()
  278. {
  279.     m_BR = 0;
  280.     if (m_WasSent) {
  281.         Cancel();
  282.         m_WasSent = false;
  283.     }
  284.     m_Connect->DropCmd(*this);
  285.     delete this;
  286. }
  289. CDBL_LangCmd::~CDBL_LangCmd()
  290. {
  291.     if (m_BR)
  292.         *m_BR = 0;
  293.     if (m_WasSent)
  294.         Cancel();
  295. }
  298. bool CDBL_LangCmd::x_AssignParams()
  299. {
  300.     static const char s_hexnum[] = "0123456789ABCDEF";
  302.     for (unsigned int n = 0; n < m_Params.NofParams(); n++) {
  303.         if(m_Params.GetParamStatus(n) == 0) continue;
  304.         const string& name  =  m_Params.GetParamName(n);
  305.         CDB_Object&   param = *m_Params.GetParam(n);
  306.         char          val_buffer[1024];
  307.         const char*   type;
  308.         string        cmd;
  310.         if (name.length() > kDBLibMaxNameLen)
  311.             return false;
  313.         switch (param.GetType()) {
  314.         case eDB_Int:
  315.             type = "int";
  316.             break;
  317.         case eDB_SmallInt:
  318.             type = "smallint";
  319.             break;
  320.         case eDB_TinyInt:
  321.             type = "tinyint";
  322.             break;
  323.         case eDB_BigInt:
  324.             type = "numeric";
  325.             break;
  326.         case eDB_Char:
  327.         case eDB_VarChar:
  328.             type = "varchar(255)";
  329.             break;
  330.         case eDB_Binary:
  331.         case eDB_VarBinary:
  332.             type = "varbinary(255)";
  333.             break;
  334.         case eDB_Float:
  335.             type = "real";
  336.             break;
  337.         case eDB_Double:
  338.             type = "float";
  339.             break;
  340.         case eDB_SmallDateTime:
  341.             type = "smalldatetime";
  342.             break;
  343.         case eDB_DateTime:
  344.             type = "datetime";
  345.             break;
  346.         default:
  347.             return false;
  348.         }
  350.         cmd += "declare " + name + ' ' + type + "nselect " + name + " = ";
  352.         if (!param.IsNULL()) {
  353.             switch (param.GetType()) {
  354.             case eDB_Int: {
  355.                 CDB_Int& val = dynamic_cast<CDB_Int&> (param);
  356.                 sprintf(val_buffer, "%dn", val.Value());
  357.                 break;
  358.             }
  359.             case eDB_SmallInt: {
  360.                 CDB_SmallInt& val = dynamic_cast<CDB_SmallInt&> (param);
  361.                 sprintf(val_buffer, "%dn", (int) val.Value());
  362.                 break;
  363.             }
  364.             case eDB_TinyInt: {
  365.                 CDB_TinyInt& val = dynamic_cast<CDB_TinyInt&> (param);
  366.                 sprintf(val_buffer, "%dn", (int) val.Value());
  367.                 break;
  368.             }
  369.             case eDB_BigInt: {
  370.                 CDB_BigInt& val = dynamic_cast<CDB_BigInt&> (param);
  371.                 sprintf(val_buffer, "%lldn", val.Value());
  372.                 break;
  373.             }
  374.             case eDB_Char: {
  375.                 CDB_Char& val = dynamic_cast<CDB_Char&> (param);
  376.                 const char* c = val.Value(); // No more than 255 chars
  377.                 size_t i = 0;
  378.                 val_buffer[i++] = '"';
  379.                 while (*c) {
  380.                     if (*c == '"')
  381.                         val_buffer[i++] = '"';
  382.                     val_buffer[i++] = *c++;
  383.                 }
  384.                 val_buffer[i++] = '"';
  385.                 val_buffer[i++] = 'n';
  386.                 val_buffer[i] = '';
  387.                 break;
  388.             }
  389.             case eDB_VarChar: {
  390.                 CDB_VarChar& val = dynamic_cast<CDB_VarChar&> (param);
  391.                 const char* c = val.Value();
  392.                 size_t i = 0;
  393.                 val_buffer[i++] = '"';
  394.                 while (*c) {
  395.                     if (*c == '"')
  396.                         val_buffer[i++] = '"';
  397.                     val_buffer[i++] = *c++;
  398.                 }
  399.                 val_buffer[i++] = '"';
  400.                 val_buffer[i++] = 'n';
  401.                 val_buffer[i] = '';
  402.                 break;
  403.             }
  404.             case eDB_Binary: {
  405.                 CDB_Binary& val = dynamic_cast<CDB_Binary&> (param);
  406.                 const unsigned char* c = (const unsigned char*) val.Value();
  407.                 size_t i = 0, size = val.Size();
  408.                 val_buffer[i++] = '0';
  409.                 val_buffer[i++] = 'x';
  410.                 for (size_t j = 0; j < size; j++) {
  411.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  412.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  413.                 }
  414.                 val_buffer[i++] = 'n';
  415.                 val_buffer[i++] = '';
  416.                 break;
  417.             }
  418.             case eDB_VarBinary: {
  419.                 CDB_VarBinary& val = dynamic_cast<CDB_VarBinary&> (param);
  420.                 const unsigned char* c = (const unsigned char*) val.Value();
  421.                 size_t i = 0, size = val.Size();
  422.                 val_buffer[i++] = '0';
  423.                 val_buffer[i++] = 'x';
  424.                 for (size_t j = 0; j < size; j++) {
  425.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  426.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  427.                 }
  428.                 val_buffer[i++] = 'n';
  429.                 val_buffer[i++] = '';
  430.                 break;
  431.             }
  432.             case eDB_Float: {
  433.                 CDB_Float& val = dynamic_cast<CDB_Float&> (param);
  434.                 sprintf(val_buffer, "%En", (double) val.Value());
  435.                 break;
  436.             }
  437.             case eDB_Double: {
  438.                 CDB_Double& val = dynamic_cast<CDB_Double&> (param);
  439.                 sprintf(val_buffer, "%En", val.Value());
  440.                 break;
  441.             }
  442.             case eDB_SmallDateTime: {
  443.                 CDB_SmallDateTime& val =
  444.                     dynamic_cast<CDB_SmallDateTime&> (param);
  445.                 sprintf(val_buffer, "'%s'n",
  446. val.Value().AsString("M/D/Y h:m").c_str());
  447.                 break;
  448.             }
  449.             case eDB_DateTime: {
  450.                 CDB_DateTime& val =
  451.                     dynamic_cast<CDB_DateTime&> (param);
  452.                 sprintf(val_buffer, "'%s:%.3d'n",
  453. val.Value().AsString("M/D/Y h:m:s").c_str(),
  454. (int)(val.Value().NanoSecond()/1000000));
  455.                 break;
  456.             }
  457.             default:
  458.                 return false; // dummy for compiler
  459.             }
  460.             cmd += val_buffer;
  461.         } else
  462.             cmd += "NULLn";
  463.         if (dbcmd(m_Cmd, (char*) cmd.c_str()) != SUCCEED)
  464.             return false;
  465.     }
  466.     return true;
  467. }
  470. END_NCBI_SCOPE
  474. /*
  475.  * ===========================================================================
  476.  * $Log: lang_cmd.cpp,v $
  477.  * Revision 1000.1  2004/06/01 19:20:19  gouriano
  478.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.14
  479.  *
  480.  * Revision 1.14  2004/05/18 18:30:36  gorelenk
  481.  * PCH <ncbi_pch.hpp> moved to correct place .
  482.  *
  483.  * Revision 1.13  2003/06/05 16:01:13  soussov
  484.  * adds code for DumpResults and for the dumped results processing
  485.  *
  486.  * Revision 1.12  2003/05/16 20:25:20  soussov
  487.  * adds code to skip parameter if it was not set
  488.  *
  489.  * Revision 1.11  2002/07/22 20:02:59  soussov
  490.  * fixes the RowCount calculations
  491.  *
  492.  * Revision 1.10  2002/07/02 16:05:49  soussov
  493.  * splitting Sybase dblib and MS dblib
  494.  *
  495.  * Revision 1.9  2002/01/11 20:11:43  vakatov
  496.  * Fixed CVS logs from prev. revision that messed up the compilation
  497.  *
  498.  * Revision 1.8  2002/01/10 22:05:52  sapojnik
  499.  * MS-specific workarounds needed to use blobs via I_ITDescriptor
  500.  * (see Text,Image)
  501.  *
  502.  * Revision 1.7  2002/01/08 18:10:18  sapojnik
  503.  * Syabse to MSSQL name translations moved to interface_p.hpp
  504.  *
  505.  * Revision 1.6  2001/12/18 19:29:22  soussov
  506.  * adds conversion from nanosecs to milisecs for datetime args
  507.  *
  508.  * Revision 1.5  2001/12/18 17:56:53  soussov
  509.  * copy-paste bug in datetime processing fixed
  510.  *
  511.  * Revision 1.4  2001/12/18 16:47:11  soussov
  512.  * fixes bug in datetime argument processing
  513.  *
  514.  * Revision 1.3  2001/12/13 23:48:44  soussov
  515.  * inserts space after declare
  516.  *
  517.  * Revision 1.2  2001/10/24 16:41:13  lavr
  518.  * File description changed to be of the same style as in other files
  519.  *
  520.  * Revision 1.1  2001/10/22 15:19:55  lavr
  521.  * This is a major revamp (by Anton Lavrentiev, with help from Vladimir
  522.  * Soussov and Denis Vakatov) of the DBAPI "driver" libs originally
  523.  * written by Vladimir Soussov. The revamp follows the one of CTLib
  524.  * driver, and it involved massive code shuffling and grooming, numerous
  525.  * local API redesigns, adding comments and incorporating DBAPI to
  526.  * the C++ Toolkit.
  527.  *
  528.  * ===========================================================================
  529.  */