comprot_cli.cpp

ncbi源码

/*
 * ===========================================================================
 * PRODUCTION $Log: comprot_cli.cpp,v $
 * PRODUCTION Revision 1000.1  2004/06/01 19:21:03  gouriano
 * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.5
 * PRODUCTION
 * ===========================================================================
 */

/* $Id: comprot_cli.cpp,v 1000.1 2004/06/01 19:21:03 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.
 *
 * ===========================================================================
 *
 * Author:  Victor Sapojnikov
 *
 * File Description:
 *   Conveniency shortcut functions for CompactProtocol client,
 *   returning bool/int/void and accepting 0 or 1 parameters.
 *
 */

#include <ncbi_pch.hpp>
#include <dbapi/driver/gateway/comprot_cli.hpp>

BEGIN_NCBI_SCOPE

CSSSConnection* conn=NULL;

//USING_NCBI_SCOPE;

void comprot_errmsg()
{
  const char* pchAnswer;
  unsigned sz;
  IGate* pGate = conn->getProtocol();

  if( pGate->get_input_arg("Error", &pchAnswer, &sz) == IGate::eGood ) {
    conn->setErrMsg(pchAnswer);

    cerr << "comprot_errmsg=" << pchAnswer << "\n";
  }
  else {
    conn->setErrMsg("");
    cerr << "comprot_errmsg - no Error message from server\n";
  }

  CDB_Exception* ex = read_CDB_Exception(pGate);
  if( ex ) {
    // Re-throw the exception caught on a server
    throw ex;
  }
}


bool comprot_bool( const char *procName, int object )
{
  IGate* pGate = conn->getProtocol();
  pGate->set_RPC_call(procName);
  pGate->set_output_arg( "object", &object );

  pGate->send_data();
  pGate->send_done();

  int nOk;
  if (pGate->get_input_arg("result", &nOk) != IGate::eGood) {
    comprot_errmsg();
    return false;
  }

  return nOk;
}


int comprot_int( const char *procName, int object )
{
  IGate* pGate = conn->getProtocol();
  pGate->set_RPC_call(procName);
  pGate->set_output_arg( "object", &object );

  pGate->send_data();
  pGate->send_done();

  int res;
  if (pGate->get_input_arg("result", &res) != IGate::eGood) {
    comprot_errmsg();
    return 0;
  }

  return res;
}


void comprot_void( const char *procName, int object )
{
  IGate* pGate = conn->getProtocol();
  pGate->set_RPC_call(procName);
  pGate->set_output_arg( "object", &object );

  pGate->send_data();
  pGate->send_done();
}

char* comprot_chars( const char *procName, int object, char* buf, int len )
{
  IGate* pGate = conn->getProtocol();
  pGate->set_RPC_call(procName);
  pGate->set_output_arg( "object", &object );

  pGate->send_data();
  pGate->send_done();

  if (pGate->get_input_arg("result", buf, len) != IGate::eGood) {
    comprot_errmsg();
    return 0;
  }

  return buf;
}

CDB_Exception* read_CDB_Exception(IGate *pGate)
{
  int exceptionType;
  if( pGate->get_input_arg("exception", &exceptionType) != IGate::eGood )
    return NULL;

  int severity = (int)eDB_Unknown; // EDB_Severity::
  pGate->get_input_arg("severity", &severity);
  int code = 0;
  pGate->get_input_arg("code"    , &code    );
  const char* from="";
  pGate->get_input_arg("from"    , &from    );
  const char* msg="";
  pGate->get_input_arg("msg"     , &msg     );

  switch( (CDB_Exception::EType)exceptionType ) {
    case CDB_Exception::eRPC:
    {
      const char* proc="";
      pGate->get_input_arg("proc", &proc);
      int line = 0;
      pGate->get_input_arg("line", &line);

      return new CDB_RPCEx( (EDB_Severity)severity,code,from,msg,  proc,line );
    }
    case CDB_Exception::eSQL:
    {
      const char* state="";
      pGate->get_input_arg("state", &state);
      int line = 0;
      pGate->get_input_arg("line" , &line );

      return new CDB_SQLEx( (EDB_Severity)severity,code,from,msg,  state,line );
    }
    case CDB_Exception::eDS      : return new CDB_DSEx      ( (EDB_Severity)severity, code, from, msg );
    case CDB_Exception::eDeadlock: return new CDB_DeadlockEx( from, msg );
    case CDB_Exception::eTimeout : return new CDB_TimeoutEx ( code, from, msg );
    case CDB_Exception::eClient  : return new CDB_ClientEx  ( (EDB_Severity)severity, code, from, msg );
    case CDB_Exception::eMulti :
    {
      int count = 0;
      pGate->get_input_arg("count", &count);
      CDB_MultiEx* multi = new CDB_MultiEx(from); // , count
      for(int i=0;i<count;i++) {
        pGate->Fetch(0);
        CDB_Exception* ex = read_CDB_Exception(pGate);
        multi->Push(*ex);
        delete ex;
      }
      return multi;
    }
    default:
      cerr<< "Unrecognized exception in comprot_errmsg(): \n  type=" << exceptionType;
      cerr << " code=" << code << " severity=" << severity << "\n";
      cerr<< "  from=" << from << "\n";
      cerr<< "  msg=" << msg << "\n";
      //throw CDB_Exception( (CDB_Exception::EType)exceptionType,  (EDB_Severity)severity,code,from,msg );
      return NULL;
  }
}


END_NCBI_SCOPE