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);
}
}
void Worker::onMessage(Connection *client)
{
Buffer *pBuffer = client->getBuffer();
ClientData *pData = (ClientData*)client->getData();
LOG("client message, fd=%d, len=%d", client->getSocket().getFd(), pBuffer->size());
while(pBuffer->size())
{
bool proc = false;
Buffer *msg = new Buffer();
//自定义协议处理
if(pMaster->pScript && pMaster->pScript->hasInputProc()){
int ret = pMaster->pScript->procInput(client, pData->nrequest, pBuffer, msg);
if(ret < 0){
delete msg;
return ;
}else if(ret > 0){
proc = true;
}
}
//原生协议处理
if(!proc){
//check head
size_t hdrSize = 4;
if(pBuffer->size() < hdrSize){
LOG("message head not enough");
delete msg;
return ;
}
//check body
size_t msgSize = net::readNetInt32(pBuffer->data());
if(pBuffer->size() < hdrSize + msgSize){
LOG("message body size not enough, msgSize=%d, buffSize=%d", msgSize, pBuffer->size());
delete msg;
return ;
}
msg->append(pBuffer->data() + hdrSize, msgSize);
pBuffer->seek(hdrSize + msgSize);
}
pData->nrequest++;
LOG("process message, fd=%d, reqid=%d, proc=%d, buffer size=%d, msg len=%d",
client->getSocket().getFd(),
pData->nrequest,
proc,
pBuffer->size(),
msg->size());
//create request
if(!msg->empty()){
Backend *backend = new Backend(pEventLoop, client, msg);
backend->copyParam(pData->params);
backend->setCloseHandler(EV_CB(this, Worker::onBackendClose));
backend->setResponseHandler(EV_CB(this, Worker::onBackendResponse));
if(backend->run()){
pData->backends[backend] = 1;
}else{
delete backend;
LOG_ERR("not found backend server");
}
}
}
}
