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

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

生物技术

开发平台：

C/C++

connection.cpp：源码内容
  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: connection.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 19:20:42  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.7
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  10. /* $Id: connection.cpp,v 1000.1 2004/06/01 19:20:42 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 connection
  38.  *
  39.  */
  41. #include <ncbi_pch.hpp>
  42. #include <dbapi/driver/ftds/interfaces.hpp>
  43. #include <string.h>
  46. BEGIN_NCBI_SCOPE
  49. CTDS_Connection::CTDS_Connection(CTDSContext* cntx, DBPROCESS* con,
  50.                                  bool reusable, const string& pool_name) :
  51.     m_Link(con), m_Context(cntx), m_Pool(pool_name), m_Reusable(reusable),
  52.     m_BCPAble(false), m_SecureLogin(false), m_ResProc(0)
  53. {
  54.     dbsetuserdata(m_Link, (BYTE*) this);
  55. }
  58. bool CTDS_Connection::IsAlive()
  59. {
  60.     return DBDEAD(m_Link) == FALSE;
  61. }
  64. CDB_LangCmd* CTDS_Connection::LangCmd(const string& lang_query,
  65.                                       unsigned int nof_parms)
  66. {
  67.     CTDS_LangCmd* lcmd = new CTDS_LangCmd(this, m_Link, lang_query, nof_parms);
  68.     m_CMDs.Add(lcmd);
  69.     return Create_LangCmd(*lcmd);
  70. }
  73. CDB_RPCCmd* CTDS_Connection::RPC(const string& rpc_name, unsigned int nof_args)
  74. {
  75.     CTDS_RPCCmd* rcmd = new CTDS_RPCCmd(this, m_Link, rpc_name, nof_args);
  76.     m_CMDs.Add(rcmd);
  77.     return Create_RPCCmd(*rcmd);
  78. }
  81. CDB_BCPInCmd* CTDS_Connection::BCPIn(const string& tab_name,
  82.                                      unsigned int nof_cols)
  83. {
  84.     if (!m_BCPAble) {
  85.         throw CDB_ClientEx(eDB_Error, 210003, "CTDS_Connection::BCPIn",
  86.                            "No bcp on this connection");
  87.     }
  88.     CTDS_BCPInCmd* bcmd = new CTDS_BCPInCmd(this, m_Link, tab_name, nof_cols);
  89.     m_CMDs.Add(bcmd);
  90.     return Create_BCPInCmd(*bcmd);
  91. }
  94. CDB_CursorCmd* CTDS_Connection::Cursor(const string& cursor_name,
  95.                                        const string& query,
  96.                                        unsigned int nof_params,
  97.                                        unsigned int)
  98. {
  99.     CTDS_CursorCmd* ccmd = new CTDS_CursorCmd(this, m_Link, cursor_name,
  100.                                               query, nof_params);
  101.     m_CMDs.Add(ccmd);
  102.     return Create_CursorCmd(*ccmd);
  103. }
  106. CDB_SendDataCmd* CTDS_Connection::SendDataCmd(I_ITDescriptor& descr_in,
  107.                                               size_t data_size, bool log_it)
  108. {
  109.     if (data_size < 1) {
  110.         throw CDB_ClientEx(eDB_Fatal, 210092,
  111.                            "CTDS_Connection::SendDataCmd",
  112.                            "wrong (zero) data size");
  113.     }
  115.     I_ITDescriptor* p_desc= 0;
  117.     // check what type of descriptor we've got
  118.     if(descr_in.DescriptorType() != CTDS_ITDESCRIPTOR_TYPE_MAGNUM) {
  119.         // this is not a native descriptor
  120.         p_desc= x_GetNativeITDescriptor(dynamic_cast<CDB_ITDescriptor&> (descr_in));
  121.         if(p_desc == 0) return false;
  122.     }
  123.     
  125.     C_ITDescriptorGuard d_guard(p_desc);
  127.     CTDS_ITDescriptor& desc = p_desc? dynamic_cast<CTDS_ITDescriptor&> (*p_desc) : 
  128.         dynamic_cast<CTDS_ITDescriptor&> (descr_in);
  130.     if (dbwritetext(m_Link,
  131.                     (char*) desc.m_ObjName.c_str(),
  132.                     desc.m_TxtPtr_is_NULL ? 0 : desc.m_TxtPtr,
  133.                     DBTXPLEN,
  134.                     desc.m_TimeStamp_is_NULL ? 0 : desc.m_TimeStamp,
  135.                     log_it ? TRUE : FALSE,
  136.                     data_size, 0) != SUCCEED ||
  137.         dbsqlok(m_Link) != SUCCEED ||
  138.         //        dbresults(m_Link) == FAIL) {
  139.         x_Results(m_Link) == FAIL) {
  140.         throw CDB_ClientEx(eDB_Error, 210093, "CTDS_Connection::SendDataCmd",
  141.                            "dbwritetext/dbsqlok/dbresults failed");
  142.     }
  144.     CTDS_SendDataCmd* sd_cmd = new CTDS_SendDataCmd(this, m_Link, data_size);
  145.     m_CMDs.Add(sd_cmd);
  146.     return Create_SendDataCmd(*sd_cmd);
  147. }
  150. bool CTDS_Connection::SendData(I_ITDescriptor& desc,
  151.                                CDB_Image& img, bool log_it){
  152.     return x_SendData(desc, dynamic_cast<CDB_Stream&> (img), log_it);
  153. }
  156. bool CTDS_Connection::SendData(I_ITDescriptor& desc,
  157.                                CDB_Text&  txt, bool log_it) {
  158.     return x_SendData(desc, dynamic_cast<CDB_Stream&> (txt), log_it);
  159. }
  162. bool CTDS_Connection::Refresh()
  163. {
  164.     // close all commands first
  165.     while (m_CMDs.NofItems()) {
  166.         CDB_BaseEnt* pCmd = static_cast<CDB_BaseEnt*> (m_CMDs.Get(0));
  167.         delete pCmd;
  168.         m_CMDs.Remove((int) 0);
  169.     }
  171.     // cancel all pending commands
  172.     if (dbcancel(m_Link) != CS_SUCCEED)
  173.         return false;
  175.     // check the connection status
  176.     return DBDEAD(m_Link) == FALSE;
  177. }
  180. const string& CTDS_Connection::ServerName() const
  181. {
  182.     return m_Server;
  183. }
  186. const string& CTDS_Connection::UserName() const
  187. {
  188.     return m_User;
  189. }
  192. const string& CTDS_Connection::Password() const
  193. {
  194.     return m_Passwd;
  195. }
  198. I_DriverContext::TConnectionMode CTDS_Connection::ConnectMode() const
  199. {
  200.     I_DriverContext::TConnectionMode mode = 0;
  201.     if ( m_BCPAble ) {
  202.         mode |= I_DriverContext::fBcpIn;
  203.     }
  204.     if ( m_SecureLogin ) {
  205.         mode |= I_DriverContext::fPasswordEncrypted;
  206.     }
  207.     return mode;
  208. }
  210. bool CTDS_Connection::IsReusable() const
  211. {
  212.     return m_Reusable;
  213. }
  216. const string& CTDS_Connection::PoolName() const
  217. {
  218.     return m_Pool;
  219. }
  222. I_DriverContext* CTDS_Connection::Context() const
  223. {
  224.     return const_cast<CTDSContext*> (m_Context);
  225. }
  228. void CTDS_Connection::PushMsgHandler(CDB_UserHandler* h)
  229. {
  230.     m_MsgHandlers.Push(h);
  231. }
  234. void CTDS_Connection::PopMsgHandler(CDB_UserHandler* h)
  235. {
  236.     m_MsgHandlers.Pop(h);
  237. }
  239. CDB_ResultProcessor* CTDS_Connection::SetResultProcessor(CDB_ResultProcessor* rp)
  240. {
  241.     CDB_ResultProcessor* r= m_ResProc;
  242.     m_ResProc= rp;
  243.     return r;
  244. }
  246. void CTDS_Connection::Release()
  247. {
  248.     m_BR = 0;
  249.     // close all commands first
  250.     while(m_CMDs.NofItems() > 0) {
  251.         CDB_BaseEnt* pCmd = static_cast<CDB_BaseEnt*> (m_CMDs.Get(0));
  252.         delete pCmd;
  253.         m_CMDs.Remove((int) 0);
  254.     }
  255. }
  258. CTDS_Connection::~CTDS_Connection()
  259. {
  260.     Refresh();
  261.     dbclose(m_Link);
  262. }
  265. void CTDS_Connection::DropCmd(CDB_BaseEnt& cmd)
  266. {
  267.     m_CMDs.Remove(static_cast<TPotItem> (&cmd));
  268. }
  271. bool CTDS_Connection::x_SendData(I_ITDescriptor& descr_in,
  272.                                  CDB_Stream& stream, bool log_it)
  273. {
  274.     size_t size = stream.Size();
  275.     if (size < 1)
  276.         return false;
  278.     I_ITDescriptor* p_desc= 0;
  280.     // check what type of descriptor we've got
  281.     if(descr_in.DescriptorType() != CTDS_ITDESCRIPTOR_TYPE_MAGNUM) {
  282.         // this is not a native descriptor
  283.         p_desc= x_GetNativeITDescriptor(dynamic_cast<CDB_ITDescriptor&> (descr_in));
  284.         if(p_desc == 0) return false;
  285.     }
  286.     
  288.     C_ITDescriptorGuard d_guard(p_desc);
  290.     CTDS_ITDescriptor& desc = p_desc? dynamic_cast<CTDS_ITDescriptor&> (*p_desc) : 
  291.         dynamic_cast<CTDS_ITDescriptor&> (descr_in);
  292.     // CTDS_ITDescriptor& desc = dynamic_cast<CTDS_ITDescriptor&> (descr_in);
  293.     char buff[1800]; // maximal page size
  295.     if (size <= sizeof(buff)) { // we could write a blob in one chunk
  296.         size_t s = stream.Read(buff, sizeof(buff));
  297.         if (dbwritetext(m_Link, (char*) desc.m_ObjName.c_str(),
  298.                         desc.m_TxtPtr_is_NULL ? 0 : desc.m_TxtPtr,
  299.                         DBTXPLEN,
  300.                         desc.m_TimeStamp_is_NULL ? 0 : desc.m_TimeStamp,
  301.                         log_it ? TRUE : FALSE, (DBINT) s, (BYTE*) buff)
  302.             != SUCCEED) {
  303.             throw CDB_ClientEx(eDB_Error, 210030, "CTDS_Connection::SendData",
  304.                                "dbwritetext failed");
  305.         }
  306.         return true;
  307.     }
  309.     // write it in chunks
  310.     if (dbwritetext(m_Link, (char*) desc.m_ObjName.c_str(),
  311.                     desc.m_TxtPtr_is_NULL ? 0 : desc.m_TxtPtr,
  312.                     DBTXPLEN,
  313.                     desc.m_TimeStamp_is_NULL ? 0 : desc.m_TimeStamp,
  314.                     log_it ? TRUE : FALSE, (DBINT) size, 0) != SUCCEED ||
  315.         dbsqlok(m_Link) != SUCCEED ||
  316.         //        dbresults(m_Link) == FAIL) {
  317.         x_Results(m_Link) == FAIL) {
  318.         throw CDB_ClientEx(eDB_Error, 210031, "CTDS_Connection::SendData",
  319.                            "dbwritetext/dbsqlok/dbresults failed");
  320.     }
  322.     while (size > 0) {
  323.         size_t s = stream.Read(buff, sizeof(buff));
  324.         if (s < 1) {
  325.             dbcancel(m_Link);
  326.             throw CDB_ClientEx(eDB_Fatal, 210032, "CTDS_Connection::SendData",
  327.                                "Text/Image data corrupted");
  328.         }
  329.         if (dbmoretext(m_Link, (DBINT) s, (BYTE*) buff) != SUCCEED) {
  330.             dbcancel(m_Link);
  331.             throw CDB_ClientEx(eDB_Error, 210033, "CTDS_Connection::SendData",
  332.                                "dbmoretext failed");
  333.         }
  334.         size -= s;
  335.     }
  337.     //    if (dbsqlok(m_Link) != SUCCEED || dbresults(m_Link) == FAIL) {
  338.     if (dbsqlok(m_Link) != SUCCEED || x_Results(m_Link) == FAIL) {
  339.         throw CDB_ClientEx(eDB_Error, 110034, "CTDS_Connection::SendData",
  340.                            "dbsqlok/dbresults failed");
  341.     }
  343.     return true;
  344. }
  346. I_ITDescriptor* CTDS_Connection::x_GetNativeITDescriptor(const CDB_ITDescriptor& descr_in)
  347. {
  348.     string q= "set rowcount 1nupdate ";
  349.     q+= descr_in.TableName();
  350.     q+= " set ";
  351.     q+= descr_in.ColumnName();
  352.     q+= "='0x0' where ";
  353.     q+= descr_in.SearchConditions();
  354.     q+= " nselect ";
  355.     q+= descr_in.ColumnName();
  356.     q+= " from ";
  357.     q+= descr_in.TableName();
  358.     q+= " where ";
  359.     q+= descr_in.SearchConditions();
  360.     q+= " nset rowcount 0";
  361.     
  362.     CDB_LangCmd* lcmd= LangCmd(q, 0);
  363.     if(!lcmd->Send()) {
  364.         throw CDB_ClientEx(eDB_Error, 210035, "CTDS_Connection::x_GetNativeITDescriptor",
  365.                            "can not send the language command");
  366.     }
  368.     CDB_Result* res;
  369.     I_ITDescriptor* descr= 0;
  370.     bool i;
  372.     while(lcmd->HasMoreResults()) {
  373.         res= lcmd->Result();
  374.         if(res == 0) continue;
  375.         if((res->ResultType() == eDB_RowResult) && (descr == 0)) {
  376.             EDB_Type tt= res->ItemDataType(0);
  377.             if(tt == eDB_Text || tt == eDB_Image) {
  378.                 while(res->Fetch()) {
  379.                     res->ReadItem(&i, 1);
  381.                     descr= new CTDS_ITDescriptor(m_Link, descr_in);
  382.                     // descr= res->GetImageOrTextDescriptor();
  383.                     if(descr) break;
  384.                 }
  385.             }
  386.         }
  387.         delete res;
  388.     }
  389.     delete lcmd;
  391.     return descr;
  392. }
  395. RETCODE CTDS_Connection::x_Results(DBPROCESS* pLink)
  396. {
  397.     unsigned int x_Status= 0x1;
  398.     CDB_Result* dbres;
  399.     I_Result* res= 0;
  401.     while ((x_Status & 0x1) != 0) {
  402.         switch (dbresults(pLink)) {
  403.         case SUCCEED:
  404.             if (DBCMDROW(pLink) == SUCCEED) { // we may get rows in this result
  405.                 if(!m_ResProc) {
  406.                     for(;;) {
  407.                         switch(dbnextrow(pLink)) {
  408.                         case NO_MORE_ROWS:
  409.                         case FAIL:
  410.                         case BUF_FULL: break;
  411.                         default: continue;
  412.                         }
  413.                         break;
  414.                     }
  415.                     continue;
  416.                 }
  417.                 res = new CTDS_RowResult(pLink, &x_Status);
  418.                 if(res) {
  419.                     dbres= Create_Result(*res);
  420.                     m_ResProc->ProcessResult(*dbres);
  421.                     delete dbres;
  422.                     delete res;
  423.                 }
  424.                 if ((x_Status & 0x10) != 0) { // we do have a compute result
  425.                     res = new CTDS_ComputeResult(pLink, &x_Status);
  426.                     if(res) {
  427.                         dbres= Create_Result(*res);
  428.                         m_ResProc->ProcessResult(*dbres);
  429.                         delete dbres;
  430.                         delete res;
  431.                     }
  432.                 }
  433.             } 
  434.             continue;
  435.             
  436.         case NO_MORE_RESULTS:
  437.             x_Status = 2;
  438.             break;
  439.         default:
  440.             return FAIL;
  441.         }
  442.         break;
  443.     }
  444.     
  445.     // we've done with the row results at this point
  446.     // let's look at return parameters and ret status
  447.     if (m_ResProc && x_Status == 2) {
  448.         x_Status = 4;
  449.         int n = dbnumrets(pLink);
  450.         if (n > 0) {
  451.             res = new CTDS_ParamResult(pLink, n);
  452.             if(res) {
  453.                 dbres= Create_Result(*res);
  454.                 m_ResProc->ProcessResult(*dbres);
  455.                 delete dbres;
  456.                 delete res;
  457.             }
  458.         }
  459.     }
  460.     
  461.     if (m_ResProc && x_Status == 4) {
  462.         if (dbhasretstat(pLink)) {
  463.             res = new CTDS_StatusResult(pLink);
  464.             if(res) {
  465.                 dbres= Create_Result(*res);
  466.                 m_ResProc->ProcessResult(*dbres);
  467.                 delete dbres;
  468.                 delete res;
  469.             }
  470.         }
  471.     }
  472.     return SUCCEED;
  473. }
  475. /////////////////////////////////////////////////////////////////////////////
  476. //
  477. //  CTDS_SendDataCmd::
  478. //
  481. CTDS_SendDataCmd::CTDS_SendDataCmd(CTDS_Connection* con, DBPROCESS* cmd,
  482.                                    size_t nof_bytes)
  483. {
  484.     m_Connect  = con;
  485.     m_Cmd      = cmd;
  486.     m_Bytes2go = nof_bytes;
  487. }
  490. size_t CTDS_SendDataCmd::SendChunk(const void* pChunk, size_t nof_bytes)
  491. {
  492.     if (!pChunk  ||  !nof_bytes) {
  493.         throw CDB_ClientEx(eDB_Fatal, 290000, "CTDS_SendDataCmd::SendChunk",
  494.                            "wrong (zero) arguments");
  495.     }
  496.     if (!m_Bytes2go)
  497.         return 0;
  499.     if (nof_bytes > m_Bytes2go)
  500.         nof_bytes = m_Bytes2go;
  502.     if (dbmoretext(m_Cmd, nof_bytes, (BYTE*) pChunk) != SUCCEED) {
  503.         dbcancel(m_Cmd);
  504.         throw CDB_ClientEx(eDB_Error, 290001, "CTDS_SendDataCmd::SendChunk",
  505.                            "dbmoretext failed");
  506.     }
  508.     m_Bytes2go -= nof_bytes;
  510.     if (m_Bytes2go <= 0) {
  511.         //        if (dbsqlok(m_Cmd) != SUCCEED || dbresults(m_Cmd) == FAIL) {
  512.         if (dbsqlok(m_Cmd) != SUCCEED || m_Connect->x_Results(m_Cmd) == FAIL) {
  513.             throw CDB_ClientEx(eDB_Error, 290002,
  514.                                "CTDS_SendDataCmd::SendChunk",
  515.                                "dbsqlok/results failed");
  516.         }
  517.     }
  519.     return nof_bytes;
  520. }
  523. void CTDS_SendDataCmd::Release()
  524. {
  525.     m_BR = 0;
  526.     if (m_Bytes2go > 0) {
  527.         dbcancel(m_Cmd);
  528.         m_Bytes2go = 0;
  529.     }
  530.     m_Connect->DropCmd(*this);
  531.     delete this;
  532. }
  535. CTDS_SendDataCmd::~CTDS_SendDataCmd()
  536. {
  537.     if (m_Bytes2go > 0)
  538.         dbcancel(m_Cmd);
  539.     if (m_BR)
  540.         *m_BR = 0;
  541. }
  544. END_NCBI_SCOPE
  548. /*
  549.  * ===========================================================================
  550.  * $Log: connection.cpp,v $
  551.  * Revision 1000.1  2004/06/01 19:20:42  gouriano
  552.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.7
  553.  *
  554.  * Revision 1.7  2004/05/17 21:13:37  gorelenk
  555.  * Added include of PCH ncbi_pch.hpp
  556.  *
  557.  * Revision 1.6  2003/06/05 16:01:40  soussov
  558.  * adds code for DumpResults and for the dumped results processing
  559.  *
  560.  * Revision 1.5  2002/12/03 19:21:24  soussov
  561.  * formatting
  562.  *
  563.  * Revision 1.4  2002/08/23 16:32:11  soussov
  564.  * fixes bug in ~CTDS_Connection()
  565.  *
  566.  * Revision 1.3  2002/03/26 15:35:10  soussov
  567.  * new image/text operations added
  568.  *
  569.  * Revision 1.2  2001/11/06 18:00:02  lavr
  570.  * Formatted uniformly as the rest of the library
  571.  *
  572.  * Revision 1.1  2001/10/25 00:39:22  vakatov
  573.  * Initial revision
  574.  *
  575.  * ===========================================================================
  576.  */