conn_impl.cpp

ncbi源码

/*
 * ===========================================================================
 * PRODUCTION $Log: conn_impl.cpp,v $
 * PRODUCTION Revision 1000.4  2004/06/01 19:18:16  gouriano
 * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.29
 * PRODUCTION
 * ===========================================================================
 */

/* $Id: conn_impl.cpp,v 1000.4 2004/06/01 19:18:16 gouriano Exp $
* ===========================================================================
*
*                            PUBLIC DOMAIN NOTICE
*               National Center for Biotechnology Information
*
*  This software/database is a "United States Government Work" under the
*  terms of the United States Copyright Act.  It was written as part of
*  the author's official duties as a United States Government employee and
*  thus cannot be copyrighted.  This software/database is freely available
*  to the public for use. The National Library of Medicine and the U.S.
*  Government have not placed any restriction on its use or reproduction.
*
*  Although all reasonable efforts have been taken to ensure the accuracy
*  and reliability of the software and data, the NLM and the U.S.
*  Government do not and cannot warrant the performance or results that
*  may be obtained by using this software or data. The NLM and the U.S.
*  Government disclaim all warranties, express or implied, including
*  warranties of performance, merchantability or fitness for any particular
*  purpose.
*
*  Please cite the author in any work or product based on this material.
*
* ===========================================================================
*
* File Name:  $Id: conn_impl.cpp,v 1000.4 2004/06/01 19:18:16 gouriano Exp $
*
* Author:  Michael Kholodov
*
* File Description:   Connection implementation
*
*
* $Log: conn_impl.cpp,v $
* Revision 1000.4  2004/06/01 19:18:16  gouriano
* PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.29
*
* Revision 1.29  2004/05/17 21:10:28  gorelenk
* Added include of PCH ncbi_pch.hpp
*
* Revision 1.28  2004/04/26 14:16:56  kholodov
* Modified: recreate the command objects each time the Get...() is called
*
* Revision 1.27  2004/04/12 14:25:33  kholodov
* Modified: resultset caching scheme, fixed single connection handling
*
* Revision 1.26  2004/04/08 15:56:58  kholodov
* Multiple bug fixes and optimizations
*
* Revision 1.25  2004/03/12 16:27:09  sponomar
* correct nested querys
*
* Revision 1.23  2004/03/08 22:15:19  kholodov
* Added: 3 new Get...() methods internally
*
* Revision 1.22  2004/02/27 14:37:33  kholodov
* Modified: set collection replaced by list for listeners
*
* Revision 1.21  2003/11/18 17:00:01  kholodov
* Added: CloneConnection() method
*
* Revision 1.20  2003/03/07 21:21:15  kholodov
* Added: IsAlive() method
*
* Revision 1.19  2003/01/09 16:10:17  sapojnik
* CConnection::Connect -- do not attempt SetDbName() if connection failed
*
* Revision 1.18  2002/11/27 17:19:49  kholodov
* Added: Error output redirection to CToMultiExHandler object.
*
* Revision 1.17  2002/10/03 18:50:00  kholodov
* Added: additional TRACE diagnostics about object deletion
* Fixed: setting parameters in IStatement object is fully supported
* Added: IStatement::ExecuteLast() to execute the last statement with
* different parameters if any
*
* Revision 1.16  2002/09/30 20:45:34  kholodov
* Added: ForceSingle() method to enforce single connection used
*
* Revision 1.15  2002/09/23 18:25:10  kholodov
* Added: GetDataSource() method.
*
* Revision 1.14  2002/09/18 18:49:27  kholodov
* Modified: class declaration and Action method to reflect
* direct inheritance of CActiveObject from IEventListener
*
* Revision 1.13  2002/09/16 19:34:40  kholodov
* Added: bulk insert support
*
* Revision 1.12  2002/09/09 20:48:56  kholodov
* Added: Additional trace output about object life cycle
* Added: CStatement::Failed() method to check command status
*
* Revision 1.11  2002/06/24 19:10:03  kholodov
* Added more trace diagnostics
*
* Revision 1.10  2002/06/24 18:06:49  kholodov
* Added more detailed diagnostics on connections
*
* Revision 1.9  2002/06/21 14:42:31  kholodov
* Added: reporting connection deletions in debug mode
*
* Revision 1.8  2002/05/16 22:11:11  kholodov
* Improved: using minimum connections possible
*
* Revision 1.7  2002/04/15 19:08:55  kholodov
* Changed GetContext() -> GetDriverContext
*
* Revision 1.6  2002/02/08 22:43:10  kholodov
* Set/GetDataBase() renamed to Set/GetDatabase() respectively
*
* Revision 1.5  2002/02/08 21:29:54  kholodov
* SetDataBase() restored, connection cloning algorithm changed
*
* Revision 1.4  2002/02/08 17:47:34  kholodov
* Removed SetDataBase() method
*
* Revision 1.3  2002/02/08 17:38:26  kholodov
* Moved listener registration to parent objects
*
* Revision 1.2  2002/02/06 22:20:09  kholodov
* Connections are cloned for CallableStatement and Cursor
*
* Revision 1.1  2002/01/30 14:51:21  kholodov
* User DBAPI implementation, first commit
*
*
*
*
*/

#include <ncbi_pch.hpp>
#include <dbapi/driver/public.hpp>
//#include <dbapi/driver/interfaces.hpp>
//#include <dbapi/driver/exception.hpp>

#include "conn_impl.hpp"
#include "ds_impl.hpp"
#include "stmt_impl.hpp"
#include "cstmt_impl.hpp"
#include "cursor_impl.hpp"
#include "bulkinsert.hpp"
#include "err_handler.hpp"

BEGIN_NCBI_SCOPE

// Implementation
CConnection::CConnection(CDataSource* ds)
    : m_ds(ds), m_connection(0), m_connCounter(1), m_connUsed(false),
      m_modeMask(0), m_forceSingle(false), m_multiExH(0),
      m_stmt(0), m_cstmt(0), m_cursor(0), m_bulkInsert(0)
{
    _TRACE("Default connection " << (void *)this << " created...");
    SetIdent("CConnection");
}

CConnection::CConnection(CDB_Connection *conn, CDataSource* ds)
    : m_ds(ds), m_connection(conn), m_connCounter(-1), m_connUsed(false),
      m_modeMask(0), m_forceSingle(false), m_multiExH(0),
      m_stmt(0), m_cstmt(0), m_cursor(0), m_bulkInsert(0)
{
    _TRACE("Auxiliary connection " << (void *)this << " created...");
    SetIdent("CConnection");
}


void CConnection::SetMode(EConnMode mode)
{
    m_modeMask |= mode;
}

void CConnection::ResetMode(EConnMode mode)
{
    m_modeMask &= ~mode;
}

IDataSource* CConnection::GetDataSource()
{
    return m_ds;
}

unsigned int CConnection::GetModeMask()
{
    return m_modeMask;
}

void CConnection::ForceSingle(bool enable)
{
    m_forceSingle = enable;
}

void CConnection::Connect(const string& user,
                          const string& password,
                          const string& server,
                          const string& database)
{

    m_connection = m_ds->
        GetDriverContext()->Connect(server,
                                    user,
                                    password,
                                    m_modeMask,
                                    m_ds->IsPoolUsed());
    if(m_connection) {
        SetDbName(database);
    }

}

CConnection::~CConnection()
{
    if( IsAux() ) {
        _TRACE("Auxiliary connection " << (void*)this << " is being deleted...");
    } else {
        _TRACE("Default connection " << (void*)this << " is being deleted...");
    }
    FreeResources();
    Notify(CDbapiDeletedEvent(this));
    _TRACE(GetIdent() << " " << (void*)this << " deleted.");
}


void CConnection::SetDatabase(const string& name)
{
  SetDbName(name);
}

string CConnection::GetDatabase()
{
    return m_database;
}

bool CConnection::IsAlive()
{
    if( m_connection != 0 )
        return m_connection->IsAlive();
    else
        return false;
}

void CConnection::SetDbName(const string& name,
                            CDB_Connection* conn)
{

    m_database = name;

    if( m_database.empty() )
        return;


    CDB_Connection* work = (conn == 0 ? m_connection : conn);
    string sql = "use " + m_database;
    CDB_LangCmd* cmd = work->LangCmd(sql.c_str());
    cmd->Send();
    while( cmd->HasMoreResults() ) {
        cmd->Result();
    }
    delete cmd;
}

CDB_Connection* CConnection::CloneCDB_Conn()
{
    CDB_Connection *temp = m_ds->
   GetDriverContext()->Connect(GetCDB_Connection()->ServerName(),
                               GetCDB_Connection()->UserName(),
                               GetCDB_Connection()->Password(),
                               m_modeMask,
                               true);
    _TRACE("CDB_Connection cloned");
    SetDbName(m_database, temp);
    return temp;
}

void CConnection::Close()
{
    FreeResources();