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

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

生物技术

开发平台：

C/C++

cursor.cpp：源码内容
  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: cursor.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 19:20:49  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.11
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  10. /* $Id: cursor.cpp,v 1000.1 2004/06/01 19:20:49 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 cursor command
  38.  *
  39.  */
  41. #include <ncbi_pch.hpp>
  42. #include <dbapi/driver/ftds/interfaces.hpp>
  45. BEGIN_NCBI_SCOPE
  48. /////////////////////////////////////////////////////////////////////////////
  49. //
  50. //  CTDS_CursorCmd::
  51. //
  53. CTDS_CursorCmd::CTDS_CursorCmd(CTDS_Connection* con, DBPROCESS* cmd,
  54.                                const string& cursor_name, const string& query,
  55.                                unsigned int nof_params) :
  56.     m_Connect(con), m_Cmd(cmd), m_Name(cursor_name), m_LCmd(0), m_Query(query),
  57.     m_Params(nof_params), m_IsOpen(false), m_HasFailed(false),
  58.     m_IsDeclared(false), m_Res(0), m_RowCount(-1)
  59. {
  60. }
  63. bool CTDS_CursorCmd::BindParam(const string& param_name, CDB_Object* param_ptr)
  64. {
  65.     return
  66.         m_Params.BindParam(CDB_Params::kNoParamNumber, param_name, param_ptr);
  67. }
  69. static bool for_update_of(const string& q)
  70. {
  71.     if((q.find("update") == string::npos) && 
  72.        (q.find("UPDATE") == string::npos))
  73.         return false;
  75.     if((q.find("for update") != string::npos) || 
  76.        (q.find("FOR UPDATE") != string::npos)) 
  77.         return true;
  79.     // TODO: add more logic here to find "for update" clause
  80.     return false;
  81. }
  83. CDB_Result* CTDS_CursorCmd::Open()
  84. {
  85.     if (m_IsOpen) { // need to close it first
  86.         Close();
  87.     }
  89.     m_HasFailed = false;
  91.     // declare the cursor
  92.     if (!x_AssignParams()) {
  93.         m_HasFailed = true;
  94.         throw CDB_ClientEx(eDB_Error, 222003, "CTDS_CursorCmd::Open",
  95.                            "cannot assign params");
  96.     }
  99.     m_LCmd = 0;
  100.     string cur_feat;
  101.     if(for_update_of(m_Query)) {
  102.         cur_feat= " cursor FORWARD_ONLY SCROLL_LOCKS for ";
  103.     }
  104.     else {
  105.         cur_feat= " cursor FORWARD_ONLY for ";
  106.     }
  107.         
  108.     string buff = "declare " + m_Name + cur_feat + m_Query;
  110.     try {
  111.         m_LCmd = m_Connect->LangCmd(buff);
  112.         m_LCmd->Send();
  113.         m_LCmd->DumpResults();
  114. #if 0
  115.         while (m_LCmd->HasMoreResults()) {
  116.             CDB_Result* r = m_LCmd->Result();
  117.             if (r) {
  118.                 while (r->Fetch())
  119.                     ;
  120.                 delete r;
  121.             }
  122.         }
  123. #endif
  124.         delete m_LCmd;
  125.     } catch (CDB_Exception& e) {
  126.         if (m_LCmd) {
  127.             delete m_LCmd;
  128.             m_LCmd = 0;
  129.         }
  130.         throw CDB_ClientEx(eDB_Error, 222001, "CTDS_CursorCmd::Open",
  131.                            "failed to declare cursor");
  132.     }
  133.     m_IsDeclared = true;
  135.     // open the cursor
  136.     m_LCmd = 0;
  137.     buff = "open " + m_Name;
  139.     try {
  140.         m_LCmd = m_Connect->LangCmd(buff);
  141.         m_LCmd->Send();
  142.         m_LCmd->DumpResults();
  143. #if 0
  144.         while (m_LCmd->HasMoreResults()) {
  145.             CDB_Result* r = m_LCmd->Result();
  146.             if (r) {
  147.                 while (r->Fetch())
  148.                     ;
  149.                 delete r;
  150.             }
  151.         }
  152. #endif
  153.         delete m_LCmd;
  154.     } catch (CDB_Exception& e) {
  155.         if (m_LCmd) {
  156.             delete m_LCmd;
  157.             m_LCmd = 0;
  158.         }
  159.         throw CDB_ClientEx(eDB_Error, 222002, "CTDS_CursorCmd::Open",
  160.                            "failed to open cursor");
  161.     }
  162.     m_IsOpen = true;
  164.     m_LCmd = 0;
  165.     buff = "fetch " + m_Name;
  167.     m_LCmd = m_Connect->LangCmd(buff);
  168.     m_Res = new CTDS_CursorResult(m_LCmd);
  169.     return Create_Result(*m_Res);
  170. }
  173. bool CTDS_CursorCmd::Update(const string&, const string& upd_query)
  174. {
  175.     if (!m_IsOpen)
  176.         return false;
  178.     CDB_LangCmd* cmd = 0;
  180.     try {
  181. while(m_LCmd->HasMoreResults()) {
  182.     CDB_Result* r= m_LCmd->Result();
  183.     if(r) delete r;
  184. }
  186.         string buff = upd_query + " where current of " + m_Name;
  187.         cmd = m_Connect->LangCmd(buff);
  188.         cmd->Send();
  189.         cmd->DumpResults();
  190. #if 0
  191.         while (cmd->HasMoreResults()) {
  192.             CDB_Result* r = cmd->Result();
  193.             if (r) {
  194.                 while (r->Fetch())
  195.                     ;
  196.                 delete r;
  197.             }
  198.         }
  199. #endif
  200.         delete cmd;
  201.     } catch (CDB_Exception& e) {
  202.         if (cmd)
  203.             delete cmd;
  204.         throw CDB_ClientEx(eDB_Error, 222004, "CTDS_CursorCmd::Update",
  205.                            "update failed");
  206.     }
  208.     return true;
  209. }
  211. I_ITDescriptor* CTDS_CursorCmd::x_GetITDescriptor(unsigned int item_num)
  212. {
  213.     if(!m_IsOpen || (m_Res == 0)) {
  214.         return 0;
  215.     }
  216.     while(m_Res->CurrentItemNo() < item_num) {
  217.         if(!m_Res->SkipItem()) return 0;
  218.     }
  219.     
  220.     I_ITDescriptor* desc= new CTDS_ITDescriptor(m_Cmd, item_num+1);
  221.     return desc;
  222. }
  224. bool CTDS_CursorCmd::UpdateTextImage(unsigned int item_num, CDB_Stream& data, 
  225.     bool log_it)
  226. {
  227.     I_ITDescriptor* desc= x_GetITDescriptor(item_num);
  228.     C_ITDescriptorGuard d_guard(desc);
  230.     if(desc) {
  231.         while(m_LCmd->HasMoreResults()) {
  232.             CDB_Result* r= m_LCmd->Result();
  233.             if(r) delete r;
  234.         }
  235.         
  236.         return m_Connect->x_SendData(*desc, data, log_it);
  237.     }
  238.     return false;
  239. }
  241. CDB_SendDataCmd* CTDS_CursorCmd::SendDataCmd(unsigned int item_num, size_t size, 
  242.     bool log_it)
  243. {
  244.     I_ITDescriptor* desc= x_GetITDescriptor(item_num);
  245.     C_ITDescriptorGuard d_guard(desc);
  247.     if(desc) {
  248.         m_LCmd->DumpResults();
  249. #if 0
  250.         while(m_LCmd->HasMoreResults()) {
  251.             CDB_Result* r= m_LCmd->Result();
  252.             if(r) delete r;
  253.         }
  254. #endif
  255.         
  256.         return m_Connect->SendDataCmd(*desc, size, log_it);
  257.     }
  258.     return 0;
  259. }     
  261. bool CTDS_CursorCmd::Delete(const string& table_name)
  262. {
  263.     if (!m_IsOpen)
  264.         return false;
  266.     CDB_LangCmd* cmd = 0;
  268.     try {
  269. while(m_LCmd->HasMoreResults()) {
  270.     CDB_Result* r= m_LCmd->Result();
  271.     if(r) delete r;
  272. }
  274.         string buff = "delete " + table_name + " where current of " + m_Name;
  275.         cmd = m_Connect->LangCmd(buff);
  276.         cmd->Send();
  277.         cmd->DumpResults();
  278. #if 0
  279.         while (cmd->HasMoreResults()) {
  280.             CDB_Result* r = cmd->Result();
  281.             if (r) {
  282.                 while (r->Fetch())
  283.                     ;
  284.                 delete r;
  285.             }
  286.         }
  287. #endif
  288.         delete cmd;
  289.     } catch (CDB_Exception& e) {
  290.         if (cmd)
  291.             delete cmd;
  292.         throw CDB_ClientEx(eDB_Error, 222004, "CTDS_CursorCmd::Update",
  293.                            "update failed");
  294.     }
  296.     return true;
  297. }
  300. int CTDS_CursorCmd::RowCount() const
  301. {
  302.     return m_RowCount;
  303. }
  306. bool CTDS_CursorCmd::Close()
  307. {
  308.     if (!m_IsOpen)
  309.         return false;
  311.     if (m_Res) {
  312.         delete m_Res;
  313.         m_Res = 0;
  314.     }
  316.     if (m_LCmd)
  317.         delete m_LCmd;
  319.     if (m_IsOpen) {
  320.         string buff = "close " + m_Name;
  321.         m_LCmd = 0;
  322.         try {
  323.             m_LCmd = m_Connect->LangCmd(buff);
  324.             m_LCmd->Send();
  325.             m_LCmd->DumpResults();
  326. #if 0
  327.             while (m_LCmd->HasMoreResults()) {
  328.                 CDB_Result* r = m_LCmd->Result();
  329.                 if (r) {
  330.                     while (r->Fetch())
  331.                         ;
  332.                     delete r;
  333.                 }
  334.             }
  335. #endif
  336.             delete m_LCmd;
  337.         } catch (CDB_Exception& e) {
  338.             if (m_LCmd)
  339.                 delete m_LCmd;
  340.             m_LCmd = 0;
  341.             throw CDB_ClientEx(eDB_Error, 222003, "CTDS_CursorCmd::Close",
  342.                                "failed to close cursor");
  343.         }
  345.         m_IsOpen = false;
  346.         m_LCmd = 0;
  347.     }
  349.     if (m_IsDeclared) {
  350.         string buff = "deallocate " + m_Name;
  351.         m_LCmd = 0;
  352.         try {
  353.             m_LCmd = m_Connect->LangCmd(buff);
  354.             m_LCmd->Send();
  355.             m_LCmd->DumpResults();
  356. #if 0
  357.             while (m_LCmd->HasMoreResults()) {
  358.                 CDB_Result* r = m_LCmd->Result();
  359.                 if (r) {
  360.                     while (r->Fetch())
  361.                         ;
  362.                     delete r;
  363.                 }
  364.             }
  365. #endif
  366.             delete m_LCmd;
  367.         } catch (CDB_Exception& e) {
  368.             if (m_LCmd)
  369.                 delete m_LCmd;
  370.             m_LCmd = 0;
  371.             throw CDB_ClientEx(eDB_Error, 222003, "CTDS_CursorCmd::Close",
  372.                                "failed to deallocate cursor");
  373.         }
  375.         m_IsDeclared = false;
  376.         m_LCmd = 0;
  377.     }
  379.     return true;
  380. }
  383. void CTDS_CursorCmd::Release()
  384. {
  385.     m_BR = 0;
  386.     if (m_IsOpen) {
  387.         Close();
  388.         m_IsOpen = false;
  389.     }
  390.     m_Connect->DropCmd(*this);
  391.     delete this;
  392. }
  395. CTDS_CursorCmd::~CTDS_CursorCmd()
  396. {
  397.     if (m_BR)
  398.         *m_BR = 0;
  399.     if (m_IsOpen)
  400.         Close();
  401. }
  404. bool CTDS_CursorCmd::x_AssignParams()
  405. {
  406.     static const char s_hexnum[] = "0123456789ABCDEF";
  408.     for (unsigned int n = 0; n < m_Params.NofParams(); n++) {
  409.         const string& name = m_Params.GetParamName(n);
  410.         if (name.empty())
  411.             continue;
  412.         CDB_Object& param = *m_Params.GetParam(n);
  413.         char val_buffer[16*1024];
  415.         if (!param.IsNULL()) {
  416.             switch (param.GetType()) {
  417.             case eDB_Int: {
  418.                 CDB_Int& val = dynamic_cast<CDB_Int&> (param);
  419.                 sprintf(val_buffer, "%d", val.Value());
  420.                 break;
  421.             }
  422.             case eDB_SmallInt: {
  423.                 CDB_SmallInt& val = dynamic_cast<CDB_SmallInt&> (param);
  424.                 sprintf(val_buffer, "%d", (int) val.Value());
  425.                 break;
  426.             }
  427.             case eDB_TinyInt: {
  428.                 CDB_TinyInt& val = dynamic_cast<CDB_TinyInt&> (param);
  429.                 sprintf(val_buffer, "%d", (int) val.Value());
  430.                 break;
  431.             }
  432.             case eDB_BigInt: {
  433.                 CDB_BigInt& val = dynamic_cast<CDB_BigInt&> (param);
  434.                 Int8 v8 = val.Value();
  435.                 sprintf(val_buffer, "%lld", v8);
  436.                 break;
  437.             }
  438.             case eDB_Char: {
  439.                 CDB_Char& val = dynamic_cast<CDB_Char&> (param);
  440.                 const char* c = val.Value(); // NB: 255 bytes at most
  441.                 size_t i = 0;
  442.                 val_buffer[i++] = ''';
  443.                 while (*c) {
  444.                     if (*c == ''')
  445.                         val_buffer[i++] = ''';
  446.                     val_buffer[i++] = *c++;
  447.                 }
  448.                 val_buffer[i++] = ''';
  449.                 val_buffer[i] = '';
  450.                 break;
  451.             }
  452.             case eDB_VarChar: {
  453.                 CDB_VarChar& val = dynamic_cast<CDB_VarChar&> (param);
  454.                 const char* c = val.Value(); // NB: 255 bytes at most
  455.                 size_t i = 0;
  456.                 val_buffer[i++] = ''';
  457.                 while (*c) {
  458.                     if (*c == ''')
  459.                         val_buffer[i++] = ''';
  460.                     val_buffer[i++] = *c++;
  461.                 }
  462.                 val_buffer[i++] = ''';
  463.                 val_buffer[i] = '';
  464.                 break;
  465.             }
  466.             case eDB_LongChar: {
  467.                 CDB_LongChar& val = dynamic_cast<CDB_LongChar&> (param);
  468.                 const char* c = val.Value(); // NB: 255 bytes at most
  469.                 size_t i = 0;
  470.                 val_buffer[i++] = ''';
  471.                 while (*c && (i < sizeof(val_buffer) - 2)) {
  472.                     if (*c == ''')
  473.                         val_buffer[i++] = ''';
  474.                     val_buffer[i++] = *c++;
  475.                 }
  476. if(*c != '') return false; 
  477.                 val_buffer[i++] = ''';
  478.                 val_buffer[i] = '';
  479.                 break;
  480.             }
  481.             case eDB_Binary: {
  482.                 CDB_Binary& val = dynamic_cast<CDB_Binary&> (param);
  483.                 const unsigned char* c = (const unsigned char*) val.Value();
  484.                 size_t i = 0, size = val.Size();
  485.                 val_buffer[i++] = '0';
  486.                 val_buffer[i++] = 'x';
  487.                 for (size_t j = 0; j < size; j++) {
  488.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  489.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  490.                 }
  491.                 val_buffer[i++] = '';
  492.                 break;
  493.             }
  494.             case eDB_VarBinary: {
  495.                 CDB_VarBinary& val = dynamic_cast<CDB_VarBinary&> (param);
  496.                 const unsigned char* c = (const unsigned char*) val.Value();
  497.                 size_t i = 0, size = val.Size();
  498.                 val_buffer[i++] = '0';
  499.                 val_buffer[i++] = 'x';
  500.                 for (size_t j = 0; j < size; j++) {
  501.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  502.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  503.                 }
  504.                 val_buffer[i++] = '';
  505.                 break;
  506.             }
  507.             case eDB_LongBinary: {
  508.                 CDB_LongBinary& val = dynamic_cast<CDB_LongBinary&> (param);
  509.                 const unsigned char* c = (const unsigned char*) val.Value();
  510.                 size_t i = 0, size = val.DataSize();
  511. if(size*2 > sizeof(val_buffer) - 4) return false;
  512.                 val_buffer[i++] = '0';
  513.                 val_buffer[i++] = 'x';
  514.                 for (size_t j = 0; j < size; j++) {
  515.                     val_buffer[i++] = s_hexnum[c[j] >> 4];
  516.                     val_buffer[i++] = s_hexnum[c[j] & 0x0F];
  517.                 }
  518.                 val_buffer[i++] = '';
  519.                 break;
  520.             }
  521.             case eDB_Float: {
  522.                 CDB_Float& val = dynamic_cast<CDB_Float&> (param);
  523.                 sprintf(val_buffer, "%E", (double) val.Value());
  524.                 break;
  525.             }
  526.             case eDB_Double: {
  527.                 CDB_Double& val = dynamic_cast<CDB_Double&> (param);
  528.                 sprintf(val_buffer, "%E", val.Value());
  529.                 break;
  530.             }
  531.             case eDB_SmallDateTime: {
  532.                 CDB_SmallDateTime& val =
  533.                     dynamic_cast<CDB_SmallDateTime&> (param);
  534.                 string t = val.Value().AsString("M/D/Y h:m");
  535.                 sprintf(val_buffer, "'%s'", t.c_str());
  536.                 break;
  537.             }
  538.             case eDB_DateTime: {
  539.                 CDB_DateTime& val =
  540.                     dynamic_cast<CDB_DateTime&> (param);
  541.                 string t = val.Value().AsString("M/D/Y h:m:s");
  542.                 sprintf(val_buffer, "'%s:%.3d'", t.c_str(),
  543. (int)(val.Value().NanoSecond()/1000000));
  544.                 break;
  545.             }
  546.             default:
  547.                 return false;
  548.             }
  549.         } else
  550.             strcpy(val_buffer, "NULL");
  552.         // substitute the param
  553.         g_SubstituteParam(m_Query, name, val_buffer);
  554.     }
  556.     return true;
  557. }
  560. END_NCBI_SCOPE
  564. /*
  565.  * ===========================================================================
  566.  * $Log: cursor.cpp,v $
  567.  * Revision 1000.1  2004/06/01 19:20:49  gouriano
  568.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.11
  569.  *
  570.  * Revision 1.11  2004/05/17 21:13:37  gorelenk
  571.  * Added include of PCH ncbi_pch.hpp
  572.  *
  573.  * Revision 1.10  2003/06/05 16:01:40  soussov
  574.  * adds code for DumpResults and for the dumped results processing
  575.  *
  576.  * Revision 1.9  2003/04/29 21:15:03  soussov
  577.  * new datatypes CDB_LongChar and CDB_LongBinary added
  578.  *
  579.  * Revision 1.8  2003/02/28 23:27:24  soussov
  580.  * fixes double quote bug in char/varchar parameters substitute
  581.  *
  582.  * Revision 1.7  2002/12/03 19:18:58  soussov
  583.  * adopting the TDS8 cursors
  584.  *
  585.  * Revision 1.6  2002/05/16 21:36:45  soussov
  586.  * fixes the memory leak in text/image processing
  587.  *
  588.  * Revision 1.5  2002/03/26 15:35:10  soussov
  589.  * new image/text operations added
  590.  *
  591.  * Revision 1.4  2001/12/18 19:29:08  soussov
  592.  * adds conversion from nanosecs to milisecs for datetime args
  593.  *
  594.  * Revision 1.3  2001/12/18 17:56:38  soussov
  595.  * copy-paste bug in datetime processing fixed
  596.  *
  597.  * Revision 1.2  2001/11/06 18:00:02  lavr
  598.  * Formatted uniformly as the rest of the library
  599.  *
  600.  * Revision 1.1  2001/10/25 00:39:22  vakatov
  601.  * Initial revision
  602.  *
  603.  * ===========================================================================
  604.  */