// PD_TRACE_DECLARE_FUNCTION ( SDB_RTNCOINS_INSTOGROUP, "rtnCoordInsert::insertToAGroup" ) INT32 rtnCoordInsert::insertToAGroup( CHAR *pBuffer, UINT32 grpID, netMultiRouteAgent *pRouteAgent, pmdEDUCB *cb ) { INT32 rc = SDB_OK; PD_TRACE_ENTRY ( SDB_RTNCOINS_INSTOGROUP ) ; BOOLEAN isNeedRetry = FALSE; BOOLEAN hasRetry = FALSE; CoordGroupList groupLst; CoordGroupList successGroupLst; groupLst[grpID] = grpID; MsgHeader *pHead = (MsgHeader *)pBuffer ; if ( cb->isTransaction() ) { pHead->opCode = MSG_BS_TRANS_INSERT_REQ; } do { hasRetry = isNeedRetry; isNeedRetry = FALSE; REQUESTID_MAP sendNodes; rc = rtnCoordSendRequestToNodeGroups( pBuffer, groupLst, TRUE, pRouteAgent, cb, sendNodes ); if ( rc ) { rtnCoordClearRequest( cb, sendNodes ); } PD_RC_CHECK( rc, PDERROR, "Failed to insert on data-node, " "send request failed(rc=%d)", rc ) ; REPLY_QUE replyQue; rc = rtnCoordGetReply( cb, sendNodes, replyQue, MAKE_REPLY_TYPE( pHead->opCode ) ); PD_RC_CHECK( rc, PDWARNING, "Failed to insert on data-node, " "get reply failed(rc=%d)", rc ); rc = processReply( replyQue, successGroupLst, cb ); if ( SDB_CLS_NOT_PRIMARY == rc && !hasRetry ) { isNeedRetry = TRUE; rc = SDB_OK; } PD_RC_CHECK( rc, PDWARNING, "Failed to process the reply(rc=%d)", rc ); } while ( isNeedRetry ); done: PD_TRACE_EXITRC ( SDB_RTNCOINS_INSTOGROUP, rc ) ; return rc; error: goto done; }
// PD_TRACE_DECLARE_FUNCTION ( SDB_RTNCOINS_SENDTOGROUPS, "sendToGroups" ) INT32 rtnCoordInsert::sendToGroups( const GroupInsertMsgMap &groupMsgMap, MsgOpInsert *pSrcMsg, netMultiRouteAgent *pRouteAgent, pmdEDUCB *cb, REQUESTID_MAP &sendNodes ) { INT32 rc = SDB_OK; PD_TRACE_ENTRY ( SDB_RTNCOINS_SENDTOGROUPS ) ; SDB_ASSERT( pSrcMsg != NULL, "pSrcMsg can't be null!" ); GroupInsertMsgMap::const_iterator iterMap = groupMsgMap.begin(); INT32 headLen = ossRoundUpToMultipleX( offsetof(MsgOpInsert, name) + ossStrlen ( pSrcMsg->name ) + 1, 4 ) ; while( iterMap != groupMsgMap.end() ) { BSONObj boInsertor; CoordGroupList groupLst; groupLst[iterMap->first] = iterMap->first; if ( iterMap->second.dataList.size() == 0 ) { ++iterMap ; continue ; } pSrcMsg->header.messageLength = headLen + iterMap->second.dataLen ; pSrcMsg->header.routeID.value = 0; rc = rtnCoordSendRequestToNodeGroups( (MsgHeader *)pSrcMsg, groupLst, TRUE, pRouteAgent, cb, iterMap->second.dataList, sendNodes ) ; PD_RC_CHECK( rc, PDERROR, "Failed to send the insert request to " "group(goupID=%u)", iterMap->first ) ; ++iterMap ; } done: PD_TRACE_EXITRC ( SDB_RTNCOINS_SENDTOGROUPS, rc ) ; return rc ; error: goto done ; }
// PD_TRACE_DECLARE_FUNCTION ( SDB_RTNCOQUERY_QUERYTODNGROUP, "rtnCoordQuery::queryToDataNodeGroup" ) INT32 rtnCoordQuery::queryToDataNodeGroup( CHAR *pBuffer, CoordGroupList &groupLst, CoordGroupList &sendGroupLst, netMultiRouteAgent *pRouteAgent, pmdEDUCB *cb, rtnContextCoord *pContext, BOOLEAN sendToPrimary, std::set<INT32> *ignoreRCList ) { INT32 rc = SDB_OK; INT32 rcTmp = SDB_OK; PD_TRACE_ENTRY ( SDB_RTNCOQUERY_QUERYTODNGROUP ) ; BOOLEAN isNeedRetry = FALSE; BOOLEAN hasRetry = FALSE; MsgHeader *pHead = (MsgHeader *)pBuffer; SINT32 resCode = MAKE_REPLY_TYPE(pHead->opCode); BOOLEAN takeOver = FALSE ; do { hasRetry = isNeedRetry; isNeedRetry = FALSE; REQUESTID_MAP sendNodes; rcTmp= rtnCoordSendRequestToNodeGroups( pBuffer, groupLst, sendToPrimary, pRouteAgent, cb, sendNodes ); if ( rcTmp != SDB_OK ) { PD_LOG ( PDERROR, "Failed to query on data-node, send request " "failed(rc=%d)", rcTmp ) ; if ( SDB_OK == rc ) { rc = rcTmp; } } REPLY_QUE replyQue; rcTmp = rtnCoordGetReply( cb, sendNodes, replyQue, resCode ) ; if ( rcTmp != SDB_OK ) { PD_LOG ( PDWARNING, "Failed to query on data-node, get reply " "failed(rc=%d)", rcTmp ); if ( SDB_APP_INTERRUPT == rcTmp || SDB_OK == rc ) { rc = rcTmp; } } while ( !replyQue.empty() ) { MsgOpReply *pReply = NULL; takeOver = FALSE ; pReply = (MsgOpReply *)(replyQue.front()); rcTmp = pReply->flags; replyQue.pop(); UINT32 groupID = pReply->header.routeID.columns.groupID; if ( rcTmp != SDB_OK ) { if ( SDB_DMS_EOC == rcTmp || ( ignoreRCList != NULL && ignoreRCList->find( rcTmp ) != ignoreRCList->end() ) ) { sendGroupLst[ groupID ] = groupID; groupLst.erase( groupID ); rcTmp = SDB_OK; } else { cb->getCoordSession()->removeLastNode( groupID ); if ( SDB_CLS_FULL_SYNC == rcTmp ) { rtnCoordUpdateNodeStatByRC( pReply->header.routeID, rcTmp ); } if ( !hasRetry ) { if ( SDB_CLS_FULL_SYNC == rcTmp ) { isNeedRetry = TRUE; } else if ( SDB_CLS_NOT_PRIMARY == rcTmp && sendToPrimary ) { CoordGroupInfoPtr groupInfoTmp ; rcTmp = rtnCoordGetGroupInfo( cb, groupID, TRUE, groupInfoTmp ); if ( rcTmp ) { PD_LOG ( PDERROR, "Failed to update group info" "(groupID=%u, rc=%d)", pReply->header.routeID.columns.groupID, rcTmp ); if ( SDB_OK == rc ) { rc = rcTmp; } } isNeedRetry = TRUE ; } } } } else { rcTmp = pContext->addSubContext( pReply, takeOver ); if ( SDB_OK == rcTmp ) { sendGroupLst[ groupID ] = groupID; groupLst.erase( groupID ); } } if ( rcTmp != SDB_OK ) { if ( SDB_OK == rc || SDB_CLS_COORD_NODE_CAT_VER_OLD == rc ) { rc = rcTmp; } PD_LOG( PDERROR, "Failed to query on data node" "(groupID=%u, nodeID=%u, serviceID=%u, rc=%d)", pReply->header.routeID.columns.groupID, pReply->header.routeID.columns.nodeID, pReply->header.routeID.columns.serviceID, rcTmp ); } if ( !takeOver ) { SDB_OSS_FREE( pReply ) ; } } if ( rc != SDB_OK ) { PD_LOG ( PDERROR, "Failed to query on data-node(rc=%d)", rc ) ; break; } }while ( isNeedRetry ) ; PD_TRACE_EXITRC ( SDB_RTNCOQUERY_QUERYTODNGROUP, rc ) ; return rc; }