INT32 omHostNotifierJob::doit()
{
INT32 rc = SDB_OK ;
UINT64 count = 0 ;
_om->getRSManager()->registerEDU( eduCB() ) ;
while ( TRUE )
{
count++ ;
if ( eduCB()->isInterrupted() )
{
rc = SDB_APP_INTERRUPT ;
goto error ;
}
if ( count % 10 == 0 )
{
map< string, UINT32 > mapClusterVersion ;
_shareVersion->getVersionMap( mapClusterVersion );
_checkUpdateCluster( mapClusterVersion ) ;
_checkDeleteCluster( mapClusterVersion ) ;
}
ossSleep( OSS_ONE_SEC ) ;
}
done:
_om->getRSManager()->unregEUD( eduCB() );
return rc ;
error:
goto done ;
}
INT32 omTaskJob::doit()
{
INT32 rc = SDB_OK ;
UINT64 count = 0 ;
_om->getRSManager()->registerEDU( eduCB() ) ;
while ( TRUE )
{
count++ ;
if ( eduCB()->isInterrupted() )
{
rc = SDB_APP_INTERRUPT ;
goto error ;
}
if ( count % 10 == 0 )
{
_taskManager->run() ;
}
ossSleep( OSS_ONE_SEC ) ;
}
done:
_om->getRSManager()->unregEUD( eduCB() );
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onGetMoreReqMsg( MsgHeader * msg,
rtnContextBuf &buffObj,
INT64 &contextID )
{
INT32 rc = SDB_OK ;
INT32 numToRead = 0 ;
rc = msgExtractGetMore ( (CHAR*)msg, &numToRead, &contextID ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] extract get more msg failed, "
"rc: %d", getSession()->sessionName(), rc ) ;
MON_SAVE_OP_DETAIL( eduCB()->getMonAppCB(), msg->opCode,
"ContextID:%lld, NumToRead:%d",
contextID, numToRead ) ;
PD_LOG ( PDDEBUG, "GetMore: contextID:%lld\nnumToRead: %d", contextID,
numToRead ) ;
rc = rtnGetMore ( contextID, numToRead, buffObj, eduCB(), _pRTNCB ) ;
done:
return rc ;
error:
goto done ;
}
void _pmdCoordProcessor::_onDetach()
{
pmdDataProcessor::_onDetach() ;
_CoordCB *pCoordCB = pmdGetKRCB()->getCoordCB() ;
if ( eduCB() && pCoordCB )
{
netMultiRouteAgent *pRouteAgent = pCoordCB->getRouteAgent() ;
pRouteAgent->delSession( eduCB()->getTID() ) ;
}
}
INT32 _pmdDataProcessor::_onUpdateReqMsg( MsgHeader * msg, SDB_DPSCB *dpsCB )
{
INT32 rc = SDB_OK ;
INT32 flags = 0 ;
CHAR *pCollectionName = NULL ;
CHAR *pSelectorBuffer = NULL ;
CHAR *pUpdatorBuffer = NULL ;
CHAR *pHintBuffer = NULL ;
rc = msgExtractUpdate( (CHAR*)msg, &flags, &pCollectionName,
&pSelectorBuffer, &pUpdatorBuffer,
&pHintBuffer );
PD_RC_CHECK( rc, PDERROR, "Session[%s] extract update message failed, "
"rc: %d", getSession()->sessionName(), rc ) ;
try
{
BSONObj selector( pSelectorBuffer );
BSONObj updator( pUpdatorBuffer );
BSONObj hint( pHintBuffer );
MON_SAVE_OP_DETAIL( eduCB()->getMonAppCB(), msg->opCode,
"CL:%s, Match:%s, Updator:%s, Hint:%s",
pCollectionName,
selector.toString().c_str(),
updator.toString().c_str(),
hint.toString().c_str() ) ;
PD_LOG ( PDDEBUG, "Session[%s] Update: selctor: %s\nupdator: %s\n"
"hint: %s", getSession()->sessionName(),
selector.toString().c_str(),
updator.toString().c_str(), hint.toString().c_str() ) ;
rc = rtnUpdate( pCollectionName, selector, updator, hint,
flags, eduCB(), _pDMSCB, dpsCB ) ;
}
catch ( std::exception &e )
{
PD_LOG ( PDERROR, "Session[%s] Failed to create selector and updator "
"for update: %s", getSession()->sessionName(), e.what () ) ;
rc = SDB_INVALIDARG ;
goto error ;
}
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onAggrReqMsg( MsgHeader *msg, INT64 &contextID )
{
INT32 rc = SDB_OK ;
CHAR *pObjs = NULL ;
INT32 count = 0 ;
INT32 flags = 0 ;
CHAR *pCollectionName = NULL ;
rc = msgExtractAggrRequest( (CHAR*)msg, &pCollectionName,
&pObjs, count, &flags ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] extrace aggr msg failed, rc: %d",
getSession()->sessionName(), rc ) ;
try
{
BSONObj objs( pObjs ) ;
rc = rtnAggregate( pCollectionName, objs, count, flags, eduCB(),
_pDMSCB, contextID ) ;
}
catch( std::exception &e )
{
PD_LOG( PDERROR, "Session[%s] occurred exception in aggr: %s",
getSession()->sessionName(), e.what() ) ;
rc = SDB_INVALIDARG ;
goto error ;
}
done:
return rc ;
error:
goto done ;
}
void omHostNotifierJob::_checkUpdateCluster(
map<string, UINT32> &mapClusterVersion )
{
map<string, UINT32>::iterator iter = mapClusterVersion.begin() ;
while ( iter != mapClusterVersion.end() )
{
string clusterName = iter->first ;
UINT32 version = iter->second ;
_MAP_CLUSTER_ITER clusterIter = _mapClusters.find( clusterName ) ;
if ( clusterIter == _mapClusters.end() )
{
omClusterNotifier *notifier = NULL ;
notifier = SDB_OSS_NEW omClusterNotifier( eduCB(), _om,
clusterName ) ;
_mapClusters.insert( _MAP_CLUSTER_VALUE( clusterName,
notifier ) ) ;
clusterIter = _mapClusters.find( clusterName ) ;
}
omClusterNotifier *pNotifier = clusterIter->second ;
pNotifier->notify( version ) ;
iter++ ;
}
}
INT32 _pmdDataProcessor::_onWriteLobMsg( MsgHeader *msg )
{
INT32 rc = SDB_OK ;
UINT32 len = 0 ;
SINT64 offset = -1 ;
const CHAR *data = NULL ;
const MsgOpLob *header = NULL ;
rc = msgExtractWriteLobRequest( ( const CHAR * )msg, &header,
&len, &offset, &data ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to extract write msg:%d", rc ) ;
goto error ;
}
rc = rtnWriteLob( header->contextID, eduCB(), len, data ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to write lob:%d", rc ) ;
goto error ;
}
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onReadLobMsg( MsgHeader *msg,
rtnContextBuf &buffObj )
{
INT32 rc = SDB_OK ;
const MsgOpLob *header = NULL ;
SINT64 offset = -1 ;
UINT32 readLen = 0 ;
UINT32 length = 0 ;
const CHAR *data = NULL ;
rc = msgExtractReadLobRequest( ( const CHAR * )msg, &header,
&readLen, &offset ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to extract read msg:%d", rc ) ;
goto error ;
}
rc = rtnReadLob( header->contextID, eduCB(),
readLen, offset, &data, length ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to read lob:%d", rc ) ;
goto error ;
}
buffObj = rtnContextBuf( data, length, 0 ) ;
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onOpenLobMsg( MsgHeader *msg, SDB_DPSCB *dpsCB,
SINT64 &contextID,
rtnContextBuf &buffObj )
{
INT32 rc = SDB_OK ;
const MsgOpLob *header = NULL ;
BSONObj lob ;
BSONObj meta ;
rc = msgExtractOpenLobRequest( ( const CHAR * )msg, &header, lob ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to extract open msg:%d", rc ) ;
goto error ;
}
rc = rtnOpenLob( lob, header->flags, TRUE, eduCB(),
dpsCB, header->w, contextID, meta ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to open lob:%d", rc ) ;
goto error ;
}
buffObj = rtnContextBuf( meta.objdata(), meta.objsize(), 1 ) ;
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onDisconnectMsg()
{
PD_LOG( PDEVENT, "Session[%s, %lld] recv disconnect msg",
getSession()->sessionName(), eduCB()->getID() ) ;
getClient()->disconnect() ;
return SDB_OK ;
}
INT32 _pmdDataProcessor::_onKillContextsReqMsg( MsgHeader *msg )
{
PD_LOG ( PDDEBUG, "session[%s] _onKillContextsReqMsg",
getSession()->sessionName() ) ;
INT32 rc = SDB_OK ;
INT32 contextNum = 0 ;
INT64 *pContextIDs = NULL ;
rc = msgExtractKillContexts ( (CHAR*)msg, &contextNum, &pContextIDs ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] extract kill contexts msg failed, "
"rc: %d", getSession()->sessionName(), rc ) ;
if ( contextNum > 0 )
{
PD_LOG ( PDDEBUG, "KillContext: contextNum:%d\ncontextID: %lld",
contextNum, pContextIDs[0] ) ;
}
rc = rtnKillContexts ( contextNum, pContextIDs, eduCB(), _pRTNCB ) ;
done:
return rc ;
error:
goto done ;
}
// PD_TRACE_DECLARE_FUNCTION ( SDB__PMDSYNCJOB__DOUNIT, "_pmdSyncJob::_doUnit" )
void _pmdSyncJob::_doUnit( IDataSyncBase *pUnit )
{
PD_TRACE_ENTRY ( SDB__PMDSYNCJOB__DOUNIT ) ;
pmdEDUMgr *pEDUMgr = eduCB()->getEDUMgr() ;
BOOLEAN force = FALSE ;
pEDUMgr->activateEDU( eduCB() ) ;
if ( pUnit->canSync( force ) )
{
pUnit->sync( force, _pMgr->isSyncDeep(), eduCB() ) ;
eduCB()->incEventCount( 1 ) ;
}
pEDUMgr->waitEDU( eduCB() ) ;
PD_TRACE_EXIT( SDB__PMDSYNCJOB__DOUNIT ) ;
}
INT32 _pmdDataProcessor::_onInsertReqMsg( MsgHeader * msg )
{
INT32 rc = SDB_OK ;
INT32 flag = 0 ;
INT32 count = 0 ;
CHAR *pCollectionName = NULL ;
CHAR *pInsertor = NULL ;
rc = msgExtractInsert( (CHAR *)msg, &flag, &pCollectionName,
&pInsertor, count ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] extrace insert msg failed, rc: %d",
getSession()->sessionName(), rc ) ;
try
{
BSONObj insertor( pInsertor ) ;
MON_SAVE_OP_DETAIL( eduCB()->getMonAppCB(), msg->opCode,
"CL:%s, Insertors:%s, count: %d",
pCollectionName,
insertor.toString().c_str(),
count ) ;
PD_LOG ( PDDEBUG, "Session[%s] insert objs: %s\ncount: %d\n"
"collection: %s", getSession()->sessionName(),
insertor.toString().c_str(), count, pCollectionName ) ;
rc = rtnInsert( pCollectionName, insertor, count, flag, eduCB() ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] insert objs[%s, count:%d, "
"collection: %s] failed, rc: %d",
getSession()->sessionName(), insertor.toString().c_str(),
count, pCollectionName, rc ) ;
}
catch( std::exception &e )
{
PD_LOG( PDERROR, "Session[%s] insert objs occur exception: %s",
getSession()->sessionName(), e.what() ) ;
rc = SDB_INVALIDARG ;
goto error ;
}
done:
return rc ;
error:
goto done ;
}
void _pmdDataProcessor::_onDetach()
{
if ( DPS_INVALID_TRANS_ID != eduCB()->getTransID() )
{
INT32 rc = rtnTransRollback( eduCB(), getDPSCB() ) ;
if ( rc )
{
PD_LOG( PDERROR, "Session[%s] rollback trans info failed, rc: %d",
getSession()->sessionName(), rc ) ;
}
}
INT64 contextID = -1 ;
while ( -1 != ( contextID = eduCB()->contextPeek() ) )
{
_pRTNCB->contextDelete( contextID, NULL ) ;
}
}
INT32 _pmdDataProcessor::_onDelReqMsg( MsgHeader * msg, SDB_DPSCB *dpsCB )
{
INT32 rc = SDB_OK ;
INT32 flags = 0 ;
CHAR *pCollectionName = NULL ;
CHAR *pDeletorBuffer = NULL ;
CHAR *pHintBuffer = NULL ;
rc = msgExtractDelete ( (CHAR *)msg , &flags, &pCollectionName,
&pDeletorBuffer, &pHintBuffer ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] extract delete msg failed, rc: %d",
getSession()->sessionName(), rc ) ;
try
{
BSONObj deletor ( pDeletorBuffer ) ;
BSONObj hint ( pHintBuffer ) ;
MON_SAVE_OP_DETAIL( eduCB()->getMonAppCB(), msg->opCode,
"CL:%s, Deletor:%s, Hint:%s",
pCollectionName,
deletor.toString().c_str(),
hint.toString().c_str() ) ;
PD_LOG ( PDDEBUG, "Session[%s] Delete: deletor: %s\nhint: %s",
getSession()->sessionName(), deletor.toString().c_str(),
hint.toString().c_str() ) ;
rc = rtnDelete( pCollectionName, deletor, hint, flags, eduCB(),
_pDMSCB, dpsCB ) ;
}
catch ( std::exception &e )
{
PD_LOG ( PDERROR, "Session[%s] Failed to create deletor for "
"DELETE: %s", getSession()->sessionName(), e.what () ) ;
rc = SDB_INVALIDARG ;
goto error ;
}
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onInterruptMsg( MsgHeader *msg, SDB_DPSCB *dpsCB )
{
PD_LOG ( PDEVENT, "Session[%s, %lld] recieved interrupt msg",
getSession()->sessionName(), eduCB()->getID() ) ;
INT64 contextID = -1 ;
while ( -1 != ( contextID = eduCB()->contextPeek() ) )
{
_pRTNCB->contextDelete ( contextID, NULL ) ;
}
INT32 rcTmp = rtnTransRollback( eduCB(), dpsCB );
if ( rcTmp )
{
PD_LOG ( PDERROR, "Failed to rollback(rc=%d)", rcTmp );
}
eduCB()->clearTransInfo() ;
return SDB_OK ;
}
void _pmdCoordProcessor::_onAttach()
{
INT32 rc = SDB_OK ;
pmdDataProcessor::_onAttach() ;
_CoordCB *pCoordCB = pmdGetKRCB()->getCoordCB() ;
if ( NULL != pCoordCB )
{
netMultiRouteAgent *pRouteAgent = pCoordCB->getRouteAgent() ;
rc = pRouteAgent->addSession( eduCB() ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "Add coord session failed in session[%s], rc: %d",
getSession()->sessionName(), rc ) ;
}
}
}
INT32 _pmdDataProcessor::_onSQLMsg( MsgHeader *msg, INT64 &contextID )
{
CHAR *sql = NULL ;
INT32 rc = SDB_OK ;
SQL_CB *sqlcb = pmdGetKRCB()->getSqlCB() ;
rc = msgExtractSql( (CHAR*)msg, &sql ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] extract sql msg failed, rc: %d",
getSession()->sessionName(), rc ) ;
rc = sqlcb->exec( sql, eduCB(), contextID ) ;
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onTransRollbackMsg ( SDB_DPSCB *dpsCB )
{
INT32 rc = SDB_OK ;
if ( pmdGetDBRole() != SDB_ROLE_STANDALONE )
{
rc = SDB_PERM ;
PD_LOG( PDERROR, "In sharding mode, couldn't execute "
"transaction operation from local service" ) ;
goto error ;
}
else
{
rc = rtnTransRollback( eduCB(), dpsCB ) ;
}
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onCloseLobMsg( MsgHeader *msg )
{
INT32 rc = SDB_OK ;
const MsgOpLob *header = NULL ;
rc = msgExtractCloseLobRequest( ( const CHAR * )msg, &header ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to extract close msg:%d", rc ) ;
goto error ;
}
rc = rtnCloseLob( header->contextID, eduCB() ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to close lob:%d", rc ) ;
goto error ;
}
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onRemoveLobMsg( MsgHeader *msg, SDB_DPSCB *dpsCB )
{
INT32 rc = SDB_OK ;
BSONObj meta ;
const MsgOpLob *header = NULL ;
rc = msgExtractRemoveLobRequest( ( const CHAR * )msg, &header,
meta ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to extract remove msg:%d", rc ) ;
goto error ;
}
rc = rtnRemoveLob( meta, header->w, eduCB(), dpsCB ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to remove lob:%d", rc ) ;
goto error ;
}
done:
return rc ;
error:
goto done ;
}
INT32 _pmdDataProcessor::_onInterruptSelfMsg()
{
PD_LOG( PDEVENT, "Session[%s, %lld] recv interrupt self msg",
getSession()->sessionName(), eduCB()->getID() ) ;
return SDB_OK ;
}
// PD_TRACE_DECLARE_FUNCTION ( SDB__PMDSYNCJOB_DOIT, "_pmdSyncJob::doit" )
INT32 _pmdSyncJob::doit()
{
PD_TRACE_ENTRY ( SDB__PMDSYNCJOB_DOIT ) ;
pmdEDUMgr *pEDUMgr = eduCB()->getEDUMgr() ;
IDataSyncBase *pUnit = NULL ;
UINT32 timeout = 0 ;
BOOLEAN force = FALSE ;
pEDUMgr->activateEDU( eduCB() ) ;
while( !eduCB()->isForced() )
{
if ( isControlJob() )
{
_pMgr->checkLoad() ;
if ( _pMgr->getMainUnit() &&
_pMgr->getMainUnit()->canSync( force ) )
{
_pMgr->getMainUnit()->sync( force, _pMgr->isSyncDeep(),
eduCB() ) ;
eduCB()->incEventCount( 1 ) ;
}
else
{
_pMgr->getNtyEvent()->wait( PMD_SYNC_JOB_INTERVAL ) ;
}
}
else
{
pEDUMgr->waitEDU( eduCB() ) ;
pUnit = _pMgr->dispatchUnit() ;
if ( pUnit )
{
timeout = 0 ;
_doUnit( pUnit ) ;
_pMgr->pushBackUnit( pUnit ) ;
}
else
{
_pMgr->getEvent()->reset() ;
if ( SDB_TIMEOUT == _pMgr->getEvent()->wait( 5 * OSS_ONE_SEC ) )
{
timeout += ( 5 * OSS_ONE_SEC ) ;
}
else
{
timeout += 100 ;
}
if ( timeout >= (UINT32)_timeout )
{
break ;
}
}
}
}
_pMgr->exitJob( isControlJob() ) ;
PD_TRACE_EXIT( SDB__PMDSYNCJOB_DOIT ) ;
return SDB_OK ;
}
INT32 _pmdCoordProcessor::_processCoordMsg( MsgHeader *msg,
MsgOpReply &replyHeader,
rtnContextBuf &contextBuff )
{
INT32 rc = SDB_OK ;
if ( NULL != _pErrorObj )
{
SDB_OSS_DEL _pErrorObj ;
_pErrorObj = NULL ;
}
if ( NULL != _pResultBuff )
{
_pResultBuff = NULL ;
}
CoordCB *pCoordcb = _pKrcb->getCoordCB();
rtnCoordProcesserFactory *pProcesserFactory
= pCoordcb->getProcesserFactory();
if ( MSG_AUTH_VERIFY_REQ == msg->opCode )
{
rc = SDB_COORD_UNKNOWN_OP_REQ ;
goto done ;
}
else if ( MSG_BS_INTERRUPTE == msg->opCode ||
MSG_BS_INTERRUPTE_SELF == msg->opCode ||
MSG_BS_DISCONNECT == msg->opCode )
{
}
else if ( !getClient()->isAuthed() )
{
rc = getClient()->authenticate( "", "" ) ;
if ( rc )
{
goto done ;
}
}
switch ( msg->opCode )
{
case MSG_BS_GETMORE_REQ :
rc = SDB_COORD_UNKNOWN_OP_REQ ;
break ;
case MSG_BS_QUERY_REQ:
{
MsgOpQuery *pQueryMsg = ( MsgOpQuery * )msg ;
CHAR *pQueryName = pQueryMsg->name ;
SINT32 queryNameLen = pQueryMsg->nameLength ;
if ( queryNameLen > 0 && '$' == pQueryName[0] )
{
rtnCoordCommand *pCmdProcesser =
pProcesserFactory->getCommandProcesser( pQueryMsg ) ;
if ( NULL != pCmdProcesser )
{
rc = pCmdProcesser->execute( ( CHAR *)msg,
msg->messageLength,
eduCB(),
replyHeader,
&contextBuff ) ;
break ;
}
}
}
default:
{
rtnContextBase *pContext = NULL ;
rtnCoordOperator *pOperator =
pProcesserFactory->getOperator( msg->opCode ) ;
rc = pOperator->execute( ( CHAR* )msg,
msg->messageLength,
eduCB(),
replyHeader,
&contextBuff ) ;
if ( MSG_BS_QUERY_REQ == msg->opCode
&& ( ((MsgOpQuery*)msg)->flags & FLG_QUERY_WITH_RETURNDATA )
&& -1 != replyHeader.contextID
&& NULL != ( pContext = _pRTNCB->contextFind(
replyHeader.contextID ) ) )
{
rc = pContext->getMore( -1, contextBuff, eduCB() ) ;
if ( rc || pContext->eof() )
{
_pRTNCB->contextDelete( replyHeader.contextID, eduCB() ) ;
replyHeader.contextID = -1 ;
}
if ( SDB_DMS_EOC == rc )
{
rc = SDB_OK ;
}
else if ( rc )
{
PD_LOG( PDERROR, "Failed to query with return data, "
"rc: %d", rc ) ;
}
}
}
break;
}
if ( rc && contextBuff.size() == 0 )
{
if ( NULL != _pErrorObj )
{
contextBuff = rtnContextBuf( *_pErrorObj ) ;
}
else
{
BSONObj obj = utilGetErrorBson( rc, eduCB()->getInfo(
EDU_INFO_ERROR ) ) ;
contextBuff = rtnContextBuf( obj ) ;
}
}
done:
return rc ;
}
INT32 _pmdDataProcessor::_onQueryReqMsg( MsgHeader * msg,
SDB_DPSCB *dpsCB,
_rtnContextBuf &buffObj,
INT64 &contextID )
{
INT32 rc = SDB_OK ;
INT32 flags = 0 ;
CHAR *pCollectionName = NULL ;
CHAR *pQueryBuff = NULL ;
CHAR *pFieldSelector = NULL ;
CHAR *pOrderByBuffer = NULL ;
CHAR *pHintBuffer = NULL ;
INT64 numToSkip = -1 ;
INT64 numToReturn = -1 ;
_rtnCommand *pCommand = NULL ;
rc = msgExtractQuery ( (CHAR *)msg, &flags, &pCollectionName,
&numToSkip, &numToReturn, &pQueryBuff,
&pFieldSelector, &pOrderByBuffer, &pHintBuffer ) ;
PD_RC_CHECK( rc, PDERROR, "Session[%s] extract query msg failed, rc: %d",
getSession()->sessionName(), rc ) ;
if ( !rtnIsCommand ( pCollectionName ) )
{
rtnContextBase *pContext = NULL ;
try
{
BSONObj matcher ( pQueryBuff ) ;
BSONObj selector ( pFieldSelector ) ;
BSONObj orderBy ( pOrderByBuffer ) ;
BSONObj hint ( pHintBuffer ) ;
MON_SAVE_OP_DETAIL( eduCB()->getMonAppCB(), msg->opCode,
"CL:%s, Match:%s, Selector:%s, OrderBy:%s, "
"Hint:%s", pCollectionName,
matcher.toString().c_str(),
selector.toString().c_str(),
orderBy.toString().c_str(),
hint.toString().c_str() ) ;
PD_LOG ( PDDEBUG, "Session[%s] Query: matcher: %s\nselector: "
"%s\norderBy: %s\nhint:%s", getSession()->sessionName(),
matcher.toString().c_str(), selector.toString().c_str(),
orderBy.toString().c_str(), hint.toString().c_str() ) ;
rc = rtnQuery( pCollectionName, selector, matcher, orderBy,
hint, flags, eduCB(), numToSkip, numToReturn,
_pDMSCB, _pRTNCB, contextID, &pContext, TRUE ) ;
if ( rc )
{
goto error ;
}
if ( ( flags & FLG_QUERY_WITH_RETURNDATA ) && NULL != pContext )
{
rc = pContext->getMore( -1, buffObj, eduCB() ) ;
if ( rc || pContext->eof() )
{
_pRTNCB->contextDelete( contextID, eduCB() ) ;
contextID = -1 ;
}
if ( SDB_DMS_EOC == rc )
{
rc = SDB_OK ;
}
else if ( rc )
{
PD_LOG( PDERROR, "Session[%s] failed to query with return "
"data, rc: %d", getSession()->sessionName(), rc ) ;
goto error ;
}
}
}
catch ( std::exception &e )
{
PD_LOG ( PDERROR, "Session[%s] Failed to create matcher and "
"selector for QUERY: %s", getSession()->sessionName(),
e.what () ) ;
rc = SDB_INVALIDARG ;
goto error ;
}
}
else
{
rc = rtnParserCommand( pCollectionName, &pCommand ) ;
if ( SDB_OK != rc )
{
PD_LOG ( PDERROR, "Parse command[%s] failed[rc:%d]",
pCollectionName, rc ) ;
goto error ;
}
rc = rtnInitCommand( pCommand , flags, numToSkip, numToReturn,
pQueryBuff, pFieldSelector, pOrderByBuffer,
pHintBuffer ) ;
if ( SDB_OK != rc )
{
goto error ;
}
PD_LOG ( PDDEBUG, "Command: %s", pCommand->name () ) ;
rc = rtnRunCommand( pCommand, getSession()->getServiceType(),
eduCB(), _pDMSCB, _pRTNCB,
dpsCB, 1, &contextID ) ;
if ( rc )
{
goto error ;
}
}
done:
if ( pCommand )
{
rtnReleaseCommand( &pCommand ) ;
}
return rc ;
error:
goto done ;
}
