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

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

生物技术

开发平台：

C/C++

bcp.cpp：源码内容
  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: bcp.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 19:20:00  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.10
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  10. /* $Id: bcp.cpp,v 1000.1 2004/06/01 19:20:00 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 bcp-in 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
  49. #include <dbapi/driver/util/numeric_convert.hpp>
  50. #include <string.h>
  53. BEGIN_NCBI_SCOPE
  56. /////////////////////////////////////////////////////////////////////////////
  57. //
  58. //  CDBL_BCPInCmd::
  59. //
  61. CDBL_BCPInCmd::CDBL_BCPInCmd(CDBL_Connection* con,
  62.                              DBPROCESS*       cmd,
  63.                              const string&    table_name,
  64.                              unsigned int     nof_columns) :
  65.     m_Connect(con), m_Cmd(cmd), m_Params(nof_columns),
  66.     m_WasSent(false), m_HasFailed(false),
  67.     m_HasTextImage(false), m_WasBound(false)
  68. {
  69.     if (bcp_init(cmd, (char*) table_name.c_str(), 0, 0, DB_IN) != SUCCEED) {
  70.         throw CDB_ClientEx(eDB_Fatal, 223001,
  71.                            "CDBL_BCPInCmd::CDBL_BCPInCmd", "bcp_init failed");
  72.     }
  73. }
  76. bool CDBL_BCPInCmd::Bind(unsigned int column_num, CDB_Object* param_ptr)
  77. {
  78.     return m_Params.BindParam(column_num,  kEmptyStr, param_ptr);
  79. }
  82. bool CDBL_BCPInCmd::x_AssignParams(void* pb)
  83. {
  84.     RETCODE r;
  86.     if (!m_WasBound) {
  87.         for (unsigned int i = 0; i < m_Params.NofParams(); i++) {
  88.             if (m_Params.GetParamStatus(i) == 0)
  89.                 continue;
  91.             CDB_Object& param = *m_Params.GetParam(i);
  93.             switch ( param.GetType() ) {
  94.             case eDB_Int: {
  95.                 CDB_Int& val = dynamic_cast<CDB_Int&> (param);
  96.                 // DBINT v = (DBINT) val.Value();
  97.                 r = bcp_bind(m_Cmd, (BYTE*) val.BindVal(), 0,
  98.                              val.IsNULL() ? 0 : -1, 0, 0, SYBINT4, i + 1);
  99.             }
  100.             break;
  101.             case eDB_SmallInt: {
  102.                 CDB_SmallInt& val = dynamic_cast<CDB_SmallInt&> (param);
  103.                 // DBSMALLINT v = (DBSMALLINT) val.Value();
  104.                 r = bcp_bind(m_Cmd, (BYTE*) val.BindVal(), 0,
  105.                              val.IsNULL() ? 0 : -1, 0, 0, SYBINT2, i + 1);
  106.             }
  107.             break;
  108.             case eDB_TinyInt: {
  109.                 CDB_TinyInt& val = dynamic_cast<CDB_TinyInt&> (param);
  110.                 // DBTINYINT v = (DBTINYINT) val.Value();
  111.                 r = bcp_bind(m_Cmd, (BYTE*) val.BindVal(), 0,
  112.                              val.IsNULL() ? 0 : -1, 0, 0, SYBINT1, i + 1);
  113.             }
  114.             break;
  115.             case eDB_BigInt: {
  116.                 CDB_BigInt& val = dynamic_cast<CDB_BigInt&> (param);
  117.                 DBNUMERIC* v = reinterpret_cast<DBNUMERIC*> (pb);
  118.                 Int8 v8 = val.Value();
  119.                 if (longlong_to_numeric(v8, 18, DBNUMERIC_val(v)) == 0)
  120.                     return false;
  121.                 r = bcp_bind(m_Cmd, (BYTE*) v, 0,
  122.                              val.IsNULL() ? 0 : -1, 0, 0, SYBNUMERIC, i + 1);
  123.                 pb = (void*) (v + 1);
  124.             }
  125.             break;
  126.             case eDB_Char: {
  127.                 CDB_Char& val = dynamic_cast<CDB_Char&> (param);
  128.                 r = bcp_bind(m_Cmd, (BYTE*) val.Value(), 0,
  129.                              val.IsNULL() ? 0 : -1,
  130.                              (BYTE*) "", 1, SYBCHAR, i + 1);
  131.             }
  132.             break;
  133.             case eDB_VarChar: {
  134.                 CDB_VarChar& val = dynamic_cast<CDB_VarChar&> (param);
  135.                 r = bcp_bind(m_Cmd, (BYTE*) val.Value(), 0,
  136.                              val.IsNULL() ? 0 : -1,
  137.                              (BYTE*) "", 1, SYBCHAR, i + 1);
  138.             }
  139.             break;
  140.             case eDB_Binary: {
  141.                 CDB_Binary& val = dynamic_cast<CDB_Binary&> (param);
  142.                 r = bcp_bind(m_Cmd, (BYTE*) val.Value(), 0,
  143.                              val.IsNULL() ? 0 : (DBINT) val.Size(),
  144.                              0, 0, SYBBINARY, i + 1);
  145.             }
  146.             break;
  147.             case eDB_VarBinary: {
  148.                 CDB_VarBinary& val = dynamic_cast<CDB_VarBinary&> (param);
  149.                 r = bcp_bind(m_Cmd, (BYTE*) val.Value(), 0,
  150.                              val.IsNULL() ? 0 : (DBINT) val.Size(),
  151.                              0, 0, SYBBINARY, i + 1);
  152.             }
  153.             break;
  154.             case eDB_Float: {
  155.                 CDB_Float& val = dynamic_cast<CDB_Float&> (param);
  156.                 // DBREAL v = (DBREAL) val.Value();
  157.                 r = bcp_bind(m_Cmd, (BYTE*) val.BindVal(), 0,
  158.                              val.IsNULL() ? 0 : -1, 0, 0, SYBREAL, i + 1);
  159.             }
  160.             break;
  161.             case eDB_Double: {
  162.                 CDB_Double& val = dynamic_cast<CDB_Double&> (param);
  163.                 // DBFLT8 v = (DBFLT8) val.Value();
  164.                 r = bcp_bind(m_Cmd, (BYTE*) val.BindVal(), 0,
  165.                              val.IsNULL() ? 0 : -1, 0, 0, SYBFLT8, i + 1);
  166.             }
  167.             break;
  168.             case eDB_SmallDateTime: {
  169.                 CDB_SmallDateTime& val =
  170.                     dynamic_cast<CDB_SmallDateTime&> (param);
  171.                 DBDATETIME4* dt = (DBDATETIME4*) pb;
  172.                 DBDATETIME4_days(dt)        = val.GetDays();
  173.                 DBDATETIME4_mins(dt)     = val.GetMinutes();
  174.                 r = bcp_bind(m_Cmd, (BYTE*) dt, 0, val.IsNULL() ? 0 : -1,
  175.                              0, 0, SYBDATETIME4,  i + 1);
  176.                 pb = (void*) (dt + 1);
  177.             }
  178.             break;
  179.             case eDB_DateTime: {
  180.                 CDB_DateTime& val = dynamic_cast<CDB_DateTime&> (param);
  181.                 DBDATETIME* dt = (DBDATETIME*) pb;
  182.                 dt->dtdays     = val.GetDays();
  183.                 dt->dttime     = val.Get300Secs();
  184.                 r = bcp_bind(m_Cmd, (BYTE*) dt, 0, val.IsNULL() ? 0 : -1,
  185.                              0, 0, SYBDATETIME, i + 1);
  186.                 pb = (void*) (dt + 1);
  187.             }
  188.             break;
  189.             case eDB_Text: {
  190.                 CDB_Text& val = dynamic_cast<CDB_Text&> (param);
  191.                 r = bcp_bind(m_Cmd, 0, 0,
  192.                              val.IsNULL() ? 0 : (DBINT) val.Size(),
  193.                              0, 0, SYBTEXT, i + 1);
  194.                 m_HasTextImage = true;
  195.             }
  196.             break;
  197.             case eDB_Image: {
  198.                 CDB_Image& val = dynamic_cast<CDB_Image&> (param);
  199.                 r = bcp_bind(m_Cmd, 0, 0,
  200.                              val.IsNULL() ? 0 : (DBINT) val.Size(),
  201.                              0, 0, SYBIMAGE, i + 1);
  202.                 m_HasTextImage = true;
  203.             }
  204.             break;
  205.             default:
  206.                 return false;
  207.             }
  208.             if (r != CS_SUCCEED)
  209.                 return false;
  210.         }
  211.         m_WasBound = true;
  212.     } else {
  213.         for (unsigned int i = 0; i < m_Params.NofParams(); i++) {
  214.             if (m_Params.GetParamStatus(i) == 0)
  215.                 continue;
  217.             CDB_Object& param = *m_Params.GetParam(i);
  219.             switch ( param.GetType() ) {
  220.             case eDB_Int: {
  221.                 CDB_Int& val = dynamic_cast<CDB_Int&> (param);
  222.                 // DBINT v = (DBINT) val.Value();
  223.                 r = bcp_colptr(m_Cmd, (BYTE*) val.BindVal(), i + 1)
  224.                     == SUCCEED &&
  225.                     bcp_collen(m_Cmd,  val.IsNULL() ? 0 : -1, i + 1)
  226.                     == SUCCEED ? SUCCEED : FAIL;
  227.             }
  228.             break;
  229.             case eDB_SmallInt: {
  230.                 CDB_SmallInt& val = dynamic_cast<CDB_SmallInt&> (param);
  231.                 // DBSMALLINT v = (DBSMALLINT) val.Value();
  232.                 r = bcp_colptr(m_Cmd, (BYTE*) val.BindVal(), i + 1)
  233.                     == SUCCEED &&
  234.                     bcp_collen(m_Cmd,  val.IsNULL() ? 0 : -1, i + 1)
  235.                     == SUCCEED ? SUCCEED : FAIL;
  236.             }
  237.             break;
  238.             case eDB_TinyInt: {
  239.                 CDB_TinyInt& val = dynamic_cast<CDB_TinyInt&> (param);
  240.                 // DBTINYINT v = (DBTINYINT) val.Value();
  241.                 r = bcp_colptr(m_Cmd, (BYTE*) val.BindVal(), i + 1)
  242.                     == SUCCEED &&
  243.                     bcp_collen(m_Cmd, val.IsNULL() ? 0 : -1, i + 1)
  244.                     == SUCCEED ? SUCCEED : FAIL;
  245.             }
  246.             break;
  247.             case eDB_BigInt: {
  248.                 CDB_BigInt& val = dynamic_cast<CDB_BigInt&> (param);
  249.                 DBNUMERIC* v = (DBNUMERIC*) pb;
  250.                 Int8 v8 = val.Value();
  251.                 if (longlong_to_numeric(v8, 18, DBNUMERIC_val(v)) == 0)
  252.                     return false;
  253.                 r = bcp_colptr(m_Cmd, (BYTE*) v, i + 1)
  254.                     == SUCCEED &&
  255.                     bcp_collen(m_Cmd,  val.IsNULL() ? 0 : -1, i + 1)
  256.                     == SUCCEED ? SUCCEED : FAIL;
  257.                 pb = (void*) (v + 1);
  258.             }
  259.             break;
  260.             case eDB_Char: {
  261.                 CDB_Char& val = dynamic_cast<CDB_Char&> (param);
  262.                 r = bcp_colptr(m_Cmd, (BYTE*) val.Value(), i + 1)
  263.                     == SUCCEED &&
  264.                     bcp_collen(m_Cmd, val.IsNULL() ? 0 : -1, i + 1)
  265.                     == SUCCEED ? SUCCEED : FAIL;
  266.             }
  267.             break;
  268.             case eDB_VarChar: {
  269.                 CDB_VarChar& val = dynamic_cast<CDB_VarChar&> (param);
  270.                 r = bcp_colptr(m_Cmd, (BYTE*) val.Value(), i + 1)
  271.                     == SUCCEED &&
  272.                     bcp_collen(m_Cmd, val.IsNULL() ? 0 : -1, i + 1)
  273.                     == SUCCEED ? SUCCEED : FAIL;
  274.             }
  275.             break;
  276.             case eDB_Binary: {
  277.                 CDB_Binary& val = dynamic_cast<CDB_Binary&> (param);
  278.                 r = bcp_colptr(m_Cmd, (BYTE*) val.Value(), i + 1)
  279.                     == SUCCEED &&
  280.                     bcp_collen(m_Cmd,
  281.                                val.IsNULL() ? 0 : (DBINT) val.Size(), i + 1)
  282.                     == SUCCEED ? SUCCEED : FAIL;
  283.             }
  284.             break;
  285.             case eDB_VarBinary: {
  286.                 CDB_VarBinary& val = dynamic_cast<CDB_VarBinary&> (param);
  287.                 r = bcp_colptr(m_Cmd, (BYTE*) val.Value(), i + 1)
  288.                     == SUCCEED &&
  289.                     bcp_collen(m_Cmd,
  290.                                val.IsNULL() ? 0 : (DBINT) val.Size(), i + 1)
  291.                     == SUCCEED ? SUCCEED : FAIL;
  292.             }
  293.             break;
  294.             case eDB_Float: {
  295.                 CDB_Float& val = dynamic_cast<CDB_Float&> (param);
  296.                 //DBREAL v = (DBREAL) val.Value();
  297.                 r = bcp_colptr(m_Cmd, (BYTE*) val.BindVal(), i + 1)
  298.                     == SUCCEED &&
  299.                     bcp_collen(m_Cmd,  val.IsNULL() ? 0 : -1, i + 1)
  300.                     == SUCCEED ? SUCCEED : FAIL;
  301.             }
  302.             break;
  303.             case eDB_Double: {
  304.                 CDB_Double& val = dynamic_cast<CDB_Double&> (param);
  305.                 //DBFLT8 v = (DBFLT8) val.Value();
  306.                 r = bcp_bind(m_Cmd, (BYTE*) val.BindVal(), 0,
  307.                              val.IsNULL() ? 0 : -1, 0, 0, SYBFLT8, i + 1);
  308.             }
  309.             break;
  310.             case eDB_SmallDateTime: {
  311.                 CDB_SmallDateTime& val =
  312.                     dynamic_cast<CDB_SmallDateTime&> (param);
  313.                 DBDATETIME4* dt = (DBDATETIME4*) pb;
  314.                 DBDATETIME4_days(dt)        = val.GetDays();
  315.                 DBDATETIME4_mins(dt)     = val.GetMinutes();
  316.                 r = bcp_colptr(m_Cmd, (BYTE*) dt, i + 1)
  317.                     == SUCCEED &&
  318.                     bcp_collen(m_Cmd, val.IsNULL() ? 0 : -1, i + 1)
  319.                     == SUCCEED ? SUCCEED : FAIL;
  320.                 pb = (void*) (dt + 1);
  321.             }
  322.             break;
  323.             case eDB_DateTime: {
  324.                 CDB_DateTime& val = dynamic_cast<CDB_DateTime&> (param);
  325.                 DBDATETIME* dt = (DBDATETIME*) pb;
  326.                 dt->dtdays     = val.GetDays();
  327.                 dt->dttime     = val.Get300Secs();
  328.                 r = bcp_colptr(m_Cmd, (BYTE*) dt, i + 1)
  329.                     == SUCCEED &&
  330.                     bcp_collen(m_Cmd, val.IsNULL() ? 0 : -1, i + 1)
  331.                     == SUCCEED ? SUCCEED : FAIL;
  332.                 pb = (void*) (dt + 1);
  333.             }
  334.             break;
  335.             case eDB_Text: {
  336.                 CDB_Text& val = dynamic_cast<CDB_Text&> (param);
  337.                 r = bcp_collen(m_Cmd, (DBINT) val.Size(), i + 1);
  338.             }
  339.             break;
  340.             case eDB_Image: {
  341.                 CDB_Image& val = dynamic_cast<CDB_Image&> (param);
  342.                 r = bcp_collen(m_Cmd, (DBINT) val.Size(), i + 1);
  343.             }
  344.             break;
  345.             default:
  346.                 return false;
  347.             }
  348.             if (r != CS_SUCCEED)
  349.                 return false;
  350.         }
  351.     }
  352.     return true;
  353. }
  356. bool CDBL_BCPInCmd::SendRow()
  357. {
  358.     char param_buff[2048]; // maximal row size, assured of buffer overruns
  360.     if (!x_AssignParams(param_buff)) {
  361.         m_HasFailed = true;
  362.         throw CDB_ClientEx(eDB_Error, 223004,
  363.                            "CDBL_BCPInCmd::SendRow", "cannot assign params");
  364.     }
  366.     if (bcp_sendrow(m_Cmd) != SUCCEED) {
  367.         m_HasFailed = true;
  368.         throw CDB_ClientEx(eDB_Error, 223005,
  369.                            "CDBL_BCPInCmd::SendRow", "bcp_sendrow failed");
  370.     }
  371.     m_WasSent = true;
  373.     if (m_HasTextImage) { // send text/image data
  374.         char buff[1800]; // text/image page size
  376.         for (unsigned int i = 0; i < m_Params.NofParams(); i++) {
  377.             if (m_Params.GetParamStatus(i) == 0)
  378.                 continue;
  380.             CDB_Object& param = *m_Params.GetParam(i);
  382.             if (param.GetType() != eDB_Text &&
  383.                 param.GetType() != eDB_Image)
  384.                 continue;
  386.             CDB_Stream& val = dynamic_cast<CDB_Stream&> (param);
  387.             size_t s = val.Size();
  388.             do {
  389.                 size_t l = val.Read(buff, sizeof(buff));
  390.                 if (l > s)
  391.                     l = s;
  392.                 if (bcp_moretext(m_Cmd, (DBINT) l, (BYTE*) buff) != SUCCEED) {
  393.                     m_HasFailed = true;
  394.                     throw CDB_ClientEx(eDB_Error, 223006,
  395.                                        "CDBL_BCPInCmd::SendRow",
  396.                                        param.GetType() == eDB_Text ?
  397.                                        "bcp_moretext for text failed" :
  398.                                        "bcp_moretext for image failed");
  399.                 }
  400.                 if (!l)
  401.                     break;
  402.                 s -= l;
  403.             } while (s);
  404.         }
  405.     }
  407.     return true;
  408. }
  411. bool CDBL_BCPInCmd::Cancel()
  412. {
  413.     if(m_WasSent) {
  414. DBINT outrow = bcp_done(m_Cmd);
  415. m_WasSent= false;
  416. return outrow == 0;
  417.     }
  418.     return true;
  419. }
  422. bool CDBL_BCPInCmd::CompleteBatch()
  423. {
  424.     if(m_WasSent) {
  425. CS_INT outrow = bcp_batch(m_Cmd);
  426. return outrow != -1;
  427.     }
  428.     return false;
  429. }
  432. bool CDBL_BCPInCmd::CompleteBCP()
  433. {
  434.     if(m_WasSent) {
  435. DBINT outrow = bcp_done(m_Cmd);
  436. m_WasSent= false;
  437. return outrow != -1;
  438.     }
  439.     return false;
  440. }
  443. void CDBL_BCPInCmd::Release()
  444. {
  445.     m_BR = 0;
  446.     if (m_WasSent) {
  447.         Cancel();
  448.         m_WasSent = false;
  449.     }
  450.     m_Connect->DropCmd(*this);
  451.     delete this;
  452. }
  455. CDBL_BCPInCmd::~CDBL_BCPInCmd()
  456. {
  457.     if (m_BR)
  458.         *m_BR = 0;
  459.     if (m_WasSent)
  460.         Cancel();
  461. }
  464. END_NCBI_SCOPE
  468. /*
  469.  * ===========================================================================
  470.  * $Log: bcp.cpp,v $
  471.  * Revision 1000.1  2004/06/01 19:20:00  gouriano
  472.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.10
  473.  *
  474.  * Revision 1.10  2004/05/18 18:30:36  gorelenk
  475.  * PCH <ncbi_pch.hpp> moved to correct place .
  476.  *
  477.  * Revision 1.9  2004/05/17 21:12:41  gorelenk
  478.  * Added include of PCH ncbi_pch.hpp
  479.  *
  480.  * Revision 1.8  2002/07/02 16:05:49  soussov
  481.  * splitting Sybase dblib and MS dblib
  482.  *
  483.  * Revision 1.7  2002/03/04 19:07:21  soussov
  484.  * fixed bug in m_WasSent flag setting
  485.  *
  486.  * Revision 1.6  2002/01/08 18:10:18  sapojnik
  487.  * Syabse to MSSQL name translations moved to interface_p.hpp
  488.  *
  489.  * Revision 1.5  2002/01/03 17:01:56  sapojnik
  490.  * fixing CR/LF mixup
  491.  *
  492.  * Revision 1.4  2002/01/03 15:46:23  sapojnik
  493.  * ported to MS SQL (about 12 'ifdef NCBI_OS_MSWIN' in 6 files)
  494.  *
  495.  * Revision 1.3  2001/10/24 16:38:53  lavr
  496.  * Explicit casts (where necessary) to eliminate 64->32 bit compiler warnings
  497.  *
  498.  * Revision 1.2  2001/10/22 18:38:49  soussov
  499.  * sending NULL instead of emty string fixed
  500.  *
  501.  * Revision 1.1  2001/10/22 15:19:55  lavr
  502.  * This is a major revamp (by Anton Lavrentiev, with help from Vladimir
  503.  * Soussov and Denis Vakatov) of the DBAPI "driver" libs originally
  504.  * written by Vladimir Soussov. The revamp follows the one of CTLib
  505.  * driver, and it involved massive code shuffling and grooming, numerous
  506.  * local API redesigns, adding comments and incorporating DBAPI to
  507.  * the C++ Toolkit.
  508.  *
  509.  * ===========================================================================
  510.  */