INT32 rtnCoordTransRollback::execute( MsgHeader *pMsg,
pmdEDUCB *cb,
INT64 &contextID,
rtnContextBuf *buf )
{
INT32 rc = SDB_OK;
pmdKRCB *pKrcb = pmdGetKRCB();
CoordCB *pCoordcb = pKrcb->getCoordCB();
netMultiRouteAgent *pRouteAgent = pCoordcb->getRouteAgent();
contextID = -1 ;
if ( !cb->isTransaction() )
{
rc = SDB_DPS_TRANS_NO_TRANS ;
goto error;
}
// add last op info
MON_SAVE_OP_DETAIL( cb->getMonAppCB(), MSG_BS_TRANS_ROLLBACK_REQ,
"TransactionID: 0x%016x(%llu)",
cb->getTransID(),
cb->getTransID() ) ;
rc = rtnCoordTransOperator::rollBack( cb, pRouteAgent ) ;
PD_RC_CHECK( rc, PDERROR, "Rollback transaction failed, rc: %d", rc ) ;
done:
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 ;
}
INT32 rtnCoordTransCommit::execute( MsgHeader *pMsg,
pmdEDUCB *cb,
INT64 &contextID,
rtnContextBuf *buf )
{
INT32 rc = SDB_OK ;
// add last op info
MON_SAVE_OP_DETAIL( cb->getMonAppCB(), MSG_BS_TRANS_COMMIT_REQ,
"TransactionID: 0x%016x(%llu)",
cb->getTransID(),
cb->getTransID() ) ;
rc = rtnCoord2PhaseCommit::execute( pMsg, cb, contextID, buf ) ;
PD_RC_CHECK( rc, PDERROR,
"Failed to commit the transaction(rc=%d)",
rc ) ;
// complete, delete transaction
cb->delTransaction() ;
done:
return rc;
error:
// rollback in session
goto done;
}
INT32 _rtnCoordSql::execute( MsgHeader *pMsg,
pmdEDUCB *cb,
INT64 &contextID,
rtnContextBuf *buf )
{
INT32 rc = SDB_OK ;
SQL_CB *sqlcb = pmdGetKRCB()->getSqlCB() ;
CHAR *sql = NULL ;
contextID = -1 ;
rc = msgExtractSql( (CHAR*)pMsg, &sql ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to extract sql" ) ;
rc = SDB_SYS ;
goto error ;
}
// add last op info
MON_SAVE_OP_DETAIL( cb->getMonAppCB(), pMsg->opCode,
"%s", sql ) ;
rc = sqlcb->exec( sql, cb, contextID ) ;
done:
return rc ;
error:
goto done ;
}
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::_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 ;
}
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 ;
}
//PD_TRACE_DECLARE_FUNCTION ( COORD_OPERATORDEL_EXE, "_coordDeleteOperator::execute" )
INT32 _coordDeleteOperator::execute( MsgHeader *pMsg,
pmdEDUCB *cb,
INT64 &contextID,
rtnContextBuf *buf )
{
INT32 rc = SDB_OK ;
INT32 rcTmp = SDB_OK ;
PD_TRACE_ENTRY ( COORD_OPERATORDEL_EXE ) ;
coordSendOptions sendOpt( TRUE ) ;
coordSendMsgIn inMsg( pMsg ) ;
coordProcessResult result ;
ROUTE_RC_MAP nokRC ;
result._pNokRC = &nokRC ;
coordCataSel cataSel ;
MsgRouteID errNodeID ;
BSONObj boDeletor ;
MsgOpDelete *pDelMsg = (MsgOpDelete *)pMsg ;
INT32 oldFlag = pDelMsg->flags ;
pDelMsg->flags |= FLG_DELETE_RETURNNUM ;
contextID = -1 ;
INT32 flag = 0;
CHAR *pCollectionName = NULL ;
CHAR *pDeletor = NULL ;
CHAR *pHint = NULL ;
rc = msgExtractDelete( (CHAR*)pMsg, &flag, &pCollectionName,
&pDeletor, &pHint ) ;
if( rc )
{
PD_LOG( PDERROR,"Failed to parse delete request, rc: %d", rc ) ;
pCollectionName = NULL ;
goto error ;
}
try
{
boDeletor = BSONObj( pDeletor ) ;
}
catch ( std::exception &e )
{
PD_RC_CHECK( SDB_INVALIDARG, PDERROR,
"Delete failed, received unexpected error: %s",
e.what() ) ;
}
MON_SAVE_OP_DETAIL( cb->getMonAppCB(), pMsg->opCode,
"Collection:%s, Deletor:%s, Hint:%s, "
"Flag:0x%08x(%u)",
pCollectionName,
boDeletor.toString().c_str(),
BSONObj(pHint).toString().c_str(),
oldFlag, oldFlag ) ;
rc = cataSel.bind( _pResource, pCollectionName, cb, FALSE, TRUE ) ;
if ( rc )
{
PD_LOG( PDERROR, "Get or update collection[%s]'s catalog info "
"failed, rc: %d", pCollectionName, rc ) ;
goto error ;
}
retry:
pDelMsg->version = cataSel.getCataPtr()->getVersion() ;
pDelMsg->w = 0 ;
rcTmp = doOpOnCL( cataSel, boDeletor, inMsg, sendOpt, cb, result ) ;
if ( SDB_OK == rcTmp && nokRC.empty() )
{
goto done ;
}
else if ( checkRetryForCLOpr( rcTmp, &nokRC, cataSel, inMsg.msg(),
cb, rc, &errNodeID, TRUE ) )
{
nokRC.clear() ;
_groupSession.getGroupCtrl()->incRetry() ;
goto retry ;
}
else
{
PD_LOG( PDERROR, "Delete failed on node[%s], rc: %d",
routeID2String( errNodeID ).c_str(), rc ) ;
goto error ;
}
done:
if ( oldFlag & FLG_DELETE_RETURNNUM )
{
contextID = _recvNum ;
}
if ( pCollectionName )
{
PD_AUDIT_OP( AUDIT_DML, MSG_BS_DELETE_REQ, AUDIT_OBJ_CL,
pCollectionName, rc,
"DeletedNum:%u, Deletor:%s, Hint:%s, Flag:0x%08x(%u)",
_recvNum, boDeletor.toString().c_str(),
BSONObj(pHint).toString().c_str(), oldFlag, oldFlag ) ;
}
PD_TRACE_EXITRC ( COORD_OPERATORDEL_EXE, rc ) ;
return rc ;
error:
if ( buf && nokRC.size() > 0 )
{
*buf = rtnContextBuf( coordBuildErrorObj( _pResource, rc,
cb, &nokRC ) ) ;
}
goto done ;
}
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 ;
}
//PD_TRACE_DECLARE_FUNCTION ( SDB_RTNCODEL_EXECUTE, "rtnCoordDelete::execute" )
INT32 rtnCoordDelete::execute( MsgHeader *pMsg,
pmdEDUCB *cb,
INT64 &contextID,
rtnContextBuf *buf )
{
INT32 rc = SDB_OK ;
INT32 rcTmp = SDB_OK ;
PD_TRACE_ENTRY ( SDB_RTNCODEL_EXECUTE ) ;
pmdKRCB *pKrcb = pmdGetKRCB() ;
CoordCB *pCoordcb = pKrcb->getCoordCB() ;
netMultiRouteAgent *pRouteAgent = pCoordcb->getRouteAgent() ;
// process define
rtnSendOptions sendOpt( TRUE ) ;
rtnSendMsgIn inMsg( pMsg ) ;
rtnProcessResult result ;
ROUTE_RC_MAP nokRC ;
result._pNokRC = &nokRC ;
CoordCataInfoPtr cataInfo ;
MsgRouteID errNodeID ;
UINT64 deleteNum = 0 ;
inMsg._pvtData = ( CHAR* )&deleteNum ;
inMsg._pvtType = PRIVATE_DATA_NUMBERLONG ;
BSONObj boDeletor ;
// fill default-reply(delete success)
MsgOpDelete *pDelMsg = (MsgOpDelete *)pMsg ;
INT32 oldFlag = pDelMsg->flags ;
pDelMsg->flags |= FLG_DELETE_RETURNNUM ;
contextID = -1 ;
INT32 flag = 0;
CHAR *pCollectionName = NULL ;
CHAR *pDeletor = NULL ;
CHAR *pHint = NULL ;
rc = msgExtractDelete( (CHAR*)pMsg, &flag, &pCollectionName,
&pDeletor, &pHint ) ;
if( rc )
{
PD_LOG( PDERROR,"Failed to parse delete request, rc: %d", rc ) ;
pCollectionName = NULL ;
goto error ;
}
try
{
boDeletor = BSONObj( pDeletor ) ;
}
catch ( std::exception &e )
{
PD_RC_CHECK( SDB_INVALIDARG, PDERROR,
"Delete failed, received unexpected error:%s",
e.what() ) ;
}
// add last op info
MON_SAVE_OP_DETAIL( cb->getMonAppCB(), pMsg->opCode,
"Collection:%s, Deletor:%s, Hint:%s, "
"Flag:0x%08x(%u)",
pCollectionName,
boDeletor.toString().c_str(),
BSONObj(pHint).toString().c_str(),
oldFlag, oldFlag ) ;
rc = rtnCoordGetCataInfo( cb, pCollectionName, FALSE, cataInfo ) ;
PD_RC_CHECK( rc, PDERROR, "Delete failed, failed to get the "
"catalogue info(collection name: %s), rc: %d",
pCollectionName, rc ) ;
retry:
do
{
pDelMsg->version = cataInfo->getVersion() ;
pDelMsg->w = 0 ;
if ( cataInfo->isMainCL() )
{
rcTmp = doOpOnMainCL( cataInfo, boDeletor, inMsg, sendOpt,
pRouteAgent, cb, result ) ;
}
else
{
rcTmp = doOpOnCL( cataInfo, boDeletor, inMsg, sendOpt,
pRouteAgent, cb, result ) ;
}
}while( FALSE ) ;
if ( SDB_OK == rcTmp && nokRC.empty() )
{
goto done ;
}
else if ( checkRetryForCLOpr( rcTmp, &nokRC, inMsg.msg(),
sendOpt._retryTimes,
cataInfo, cb, rc, &errNodeID, TRUE ) )
{
nokRC.clear() ;
++sendOpt._retryTimes ;
goto retry ;
}
else
{
PD_LOG( PDERROR, "Delete failed on node[%s], rc: %d",
routeID2String( errNodeID ).c_str(), rc ) ;
goto error ;
}
done:
if ( oldFlag & FLG_DELETE_RETURNNUM )
{
contextID = deleteNum ;
}
if ( pCollectionName )
{
/// AUDIT
PD_AUDIT_OP( AUDIT_DML, MSG_BS_DELETE_REQ, AUDIT_OBJ_CL,
pCollectionName, rc,
"DeletedNum:%u, Deletor:%s, Hint:%s, Flag:0x%08x(%u)",
deleteNum, boDeletor.toString().c_str(),
BSONObj(pHint).toString().c_str(), oldFlag, oldFlag ) ;
}
PD_TRACE_EXITRC ( SDB_RTNCODEL_EXECUTE, rc ) ;
return rc ;
error:
goto done ;
}
