void Worker::closeClient(Connection *client)
{
ClientData *pData = (ClientData*)client->getData();
std::string sid = pData->session.getId();
LOG("close client, fd=%d, sid=%s", client->getSocket().getFd(), sid.c_str());
//router通知
sendToRouter(ROUTER_MSG_CLOSE, SID_LENGTH, pData->session.getId(), 0, NULL);
//后端通知
Config *pConfig = Config::getInstance();
if(pConfig->bEventClose){
Backend *backend = new Backend(pEventLoop, NULL, NULL);
backend->copyParam(pData->params);
backend->addParam("EVENT", sizeof("EVENT") - 1, "2", 1);
backend->setCloseHandler(EV_CB(this, Worker::onBackendClose));
backend->setResponseHandler(EV_CB(this, Worker::onBackendResponse));
if(!backend->run()){
delete backend;
}
}
//释放后端请求
for(BackendList::const_iterator it = pData->backends.begin(); it != pData->backends.end(); ++it){
Backend *backend = it->first;
backend->setClient(NULL);
backend->shutdown();
}
//取消频道订阅
for(ChannelSet::const_iterator it2 = pData->channels.begin(); it2 != pData->channels.end(); ++it2){
const std::string &chname = it2->first;
ChannelList::iterator cit = arrChannels.find(chname);
if(cit != arrChannels.end()){
cit->second.erase(client);
if(cit->second.empty()){
sendToRouter(ROUTER_MSG_CH_UNSUB, chname.size(), chname.c_str(), 0, NULL);
arrChannels.erase(cit);
}
}
}
//计数更新
pMaster->delClient(nId);
//客户端列表更新
arrClients.erase(sid);
//lua更新
if(pMaster->pScript && pMaster->pScript->hasCloseProc()){
pMaster->pScript->procClose(client);
}
delete pData;
delete client;
}
void Worker::createClient(int fd, const char *host, int port)
{
//创建session对像
Session sess(nPid, fd);
LOG("new client, fd=%d, host=%s, port=%d, sid=%s", fd, host, port, sess.getId());
//环镜变量
ClientData *pData = new ClientData();
char szPort[8];
int nPort = sprintf(szPort, "%d", port);
pData->session = sess;
pData->nrequest = 0;
Backend::makeParam(pData->params, "REMOTE_ADDR", sizeof("REMOTE_ADDR") - 1, host, strlen(host));
Backend::makeParam(pData->params, "REMOTE_PORT", sizeof("REMOTE_PORT") - 1, szPort, nPort);
Backend::makeParam(pData->params, "SESSIONID", sizeof("SESSIONID") - 1, sess.getId(), SID_LENGTH);
//创建连接对像
Connection *client = new Connection(pEventLoop, fd);
client->setHostAndPort(host, port);
client->setData(pData);
client->setMessageHandler(EV_CB(this, Worker::onMessage));
client->setCloseHandler(EV_CB(this, Worker::onClose));
//client->setWriteCompleteHandler(EL_CB(this, Worker::onWriteComplete));
arrClients[sess.getId()] = client;
//计数更新
pMaster->addClient(nId);
//Router通知
sendToRouter(ROUTER_MSG_CONN, SID_LENGTH, sess.getId(), 0, NULL);
//backend通知
Config *pConfig = Config::getInstance();
if(pConfig->bEventConnect){
Backend *backend = new Backend(pEventLoop, client, NULL);
backend->copyParam(pData->params);
backend->addParam("EVENT", sizeof("EVENT") - 1, "1", 1);
backend->setCloseHandler(EV_CB(this, Worker::onBackendClose));
backend->setResponseHandler(EV_CB(this, Worker::onBackendResponse));
if(backend->run()){
pData->backends[backend] = 1;
}else{
delete backend;
}
}
//lua通知
if(pMaster->pScript && pMaster->pScript->hasConnectProc()){
pMaster->pScript->procConnect(client);
}
}
