// returns { ok : true } if target was created successfully
// returns { ok : false } and { errmsg : <msg> } if target creation unsuccessful
// (if not of master of a network, we can't connect to it, invalid custom collection prefix)
BSONObj PingMonitorThreadManager::createTarget( HostAndPort& hp , bool on=true , int interval=15 , string customCollectionPrefix="" ){
BSONObjBuilder toReturn;
BSONObj connInfo = getConnInfo( hp );
// check for connection errors
if( connInfo["errmsg"].trueValue() ){
toReturn.append( "ok" , false );
toReturn.append( ERRMSG , connInfo["errmsg"].valuestrsafe() );
return toReturn.obj();
}
// if custom collectionPrefix setting is empty, use default
if( customCollectionPrefix.empty() ){
customCollectionPrefix = connInfo["collectionPrefix"].valuestrsafe();
}
// if custom collectionPrefix setting is set, check that collectionPrefix is not already in use
// and make sure custom collection prefix does not contain invalid character $
else{
for(map< HostAndPort , PingMonitor* >::iterator i = targets.begin(); i!=targets.end(); i++){
if( i->second->getCollectionPrefix().compare( customCollectionPrefix ) == 0){
toReturn.append( "ok" , false );
toReturn.append( ERRMSG , ALREADY_USING_COLLECTION );
return toReturn.obj();
}
}
if( customCollectionPrefix.find("$") != string::npos ){
toReturn.append( "ok" , false );
toReturn.append( ERRMSG , INVALID_COLLECTION_CHARACTER );
return toReturn.obj();
}
}
// create a PingMonitor target with this host
PingMonitor *pmt = new PingMonitor( hp , on , interval , customCollectionPrefix , connInfo["networkType"].valuestrsafe() );
pmt->go();
targets[ hp ] = pmt;
toReturn.append("ok" , true);
return toReturn.obj();
}
///handle close
int CwxAppHandler4UnixConn::close(CWX_HANDLE )
{
CWX_UINT16 ucLocState = m_conn.getState();
char const* szState;
int iCloseReturn=0;
if (getApp()->isStopped())
{
if (-1 != this->index()) reactor()->cancelTimer(this);
delete this;
return 0;
}
switch(ucLocState)
{
case CwxAppConnInfo::IDLE:
szState = "IDLE"; ///一定没有在reactor注册
break;
case CwxAppConnInfo::CONNECTING:
szState = "CONNECTING";///可能在重连的时候再reactor注册
CWX_ERROR(("Failure to connect to unix-file:%s, errno=%d",
m_strConnectPathFile.c_str(),
getConnErrNo()));
if (reactor()) reactor()->removeHandler(this);
break;
case CwxAppConnInfo::TIMEOUT:
szState = "TIMEOUT"; ///可能注册了timeout
if (-1 != this->index()) reactor()->cancelTimer(this);
break;
case CwxAppConnInfo::ESTABLISHING:
szState = "ESTABLISHING";///可能注册了消息收发
if (CWX_INVALID_HANDLE != getHandle())
if (reactor()) reactor()->removeHandler(this, false);
break;
case CwxAppConnInfo::ESTABLISHED:
szState = "ESTABLISHED";///一定注册了消息收发
if (CWX_INVALID_HANDLE != getHandle())
if (reactor()) reactor()->removeHandler(this, false);
break;
case CwxAppConnInfo::FAILED:
szState = "FAILED";///一定没有注册消息收发
break;
default:
szState = "Unknown";
CWX_ASSERT(0);
break;
}
if (CWX_INVALID_HANDLE != getHandle())
{
CWX_DEBUG(("Connection closed. State = %s, unix file=%s, Active-Close=%s",
szState,
m_strConnectPathFile.c_str(),
m_conn.isActiveClose()?"yes":"no"));
}
m_conn.setState(CwxAppConnInfo::FAILED);
//reconnection or close
if (CwxAppConnInfo::ESTABLISHED == ucLocState)
{
//re-dispatch all msg
while(this->getNextMsg() == 1)
{
if (this->m_curSndingMsg && m_curSndingMsg->send_ctrl().isFailNotice())
{
this->getApp()->onFailSendMsg(m_curSndingMsg);
}
if (m_curSndingMsg) CwxMsgBlockAlloc::free(m_curSndingMsg);
this->m_curSndingMsg = NULL;
}
iCloseReturn = this->getApp()->connClosed(*this);
m_conn.setFailConnNum(1);
//remove recieved data.
if (this->m_recvMsgData) CwxMsgBlockAlloc::free(this->m_recvMsgData);
this->m_recvMsgData = NULL;
}
else if (m_conn.isActiveConn())
{
m_conn.setFailConnNum(m_conn.getFailConnNum() + 1);
iCloseReturn = this->getApp()->onFailConnect(*this);
}
if (getHandle () != CWX_INVALID_HANDLE)
{
::close(getHandle());
setHandle(CWX_INVALID_HANDLE);
}
if (m_conn.isNeedReconnect() && (iCloseReturn>=0))
{
CWX_UINT32 uiInternal = 0;
if (m_conn.isReconn())
{
uiInternal = m_conn.getReconnDelay();
m_conn.setReconn(false);
}
else
{
if (iCloseReturn > 0)
{
uiInternal = iCloseReturn;
}
else
{
uiInternal = 2 * m_conn.getMinRetryInternal() * getConnInfo().getFailConnNum();
if (uiInternal > getConnInfo().getMaxRetryInternal()) uiInternal = getConnInfo().getMaxRetryInternal();
uiInternal *= 1000;
}
}
CWX_DEBUG(("Reconnect to address %s after %d mili-second.", m_strConnectPathFile.c_str(), uiInternal));
if (uiInternal)
{
m_conn.setState(CwxAppConnInfo::TIMEOUT);
if (this->reactor()->scheduleTimer(this, CwxTimeValue(uiInternal/1000, (uiInternal%1000) * 1000)) == -1)
{
CWX_ERROR(("Failure schedule_timer to noticeReconnect for conn id[%u]", m_conn.getConnId()));
if (reactor()) reactor()->removeHandlerByConnId(m_conn.getConnId());
delete this;
return 0;
}
}
else
{
if (-1 == getApp()->getUnixConnector()->connect(this,
m_strConnectPathFile.c_str()))
{
CWX_ERROR(("Failure to reconnect unix-file=%s, errno=%d",
m_strConnectPathFile.c_str(),
errno));
this->close();
}
}
}
else
{
reactor()->removeFromConnMap(m_conn.getConnId());
getApp()->getHandlerCache()->cacheUnixHandle(this);
}
return 0;
}
