mirror of
https://gitee.com/TarsCloud/TarsCpp.git
synced 2025-01-05 17:42:24 +08:00
57 lines
1.5 KiB
C++
57 lines
1.5 KiB
C++
#include "PushImp.h"
|
||
#include "servant/Application.h"
|
||
#include "PushThread.h"
|
||
|
||
using namespace std;
|
||
|
||
//////////////////////////////////////////////////////
|
||
void PushImp::initialize()
|
||
{
|
||
//initialize servant here:
|
||
//...
|
||
}
|
||
|
||
//////////////////////////////////////////////////////
|
||
void PushImp::destroy()
|
||
{
|
||
//destroy servant here:
|
||
//...
|
||
}
|
||
|
||
|
||
int PushImp::doRequest(tars::CurrentPtr current, vector<char>& response)
|
||
{
|
||
// LOG_CONSOLE_DEBUG << endl;
|
||
//保存客户端的信息，以便对客户端进行push消息
|
||
(PushUser::mapMutex).lock();
|
||
map<string, CurrentPtr>::iterator it = PushUser::pushUser.find(current->getIp());
|
||
if(it == PushUser::pushUser.end())
|
||
{
|
||
PushUser::pushUser.insert(map<string, CurrentPtr>::value_type(current->getIp(), current));
|
||
LOG->debug() << "connect ip: " << current->getIp() << endl;
|
||
}
|
||
(PushUser::mapMutex).unlock();
|
||
//返回给客户端它自己请求的数据包，即原包返回(4字节长度+4字节requestid+buffer)
|
||
const vector<char>& request = current->getRequestBuffer();
|
||
response = request;
|
||
|
||
return 0;
|
||
}
|
||
//客户端关闭到服务端的连接，或者服务端发现客户端长时间未发送包过来，然后超过60s就关闭连接
|
||
//调用的方法
|
||
int PushImp::doClose(CurrentPtr current)
|
||
{
|
||
(PushUser::mapMutex).lock();
|
||
map<string, CurrentPtr>::iterator it = PushUser::pushUser.find(current->getIp());
|
||
if(it != PushUser::pushUser.end())
|
||
{
|
||
PushUser::pushUser.erase(it);
|
||
LOG->debug() << "close ip: " << current->getIp() << endl;
|
||
}
|
||
(PushUser::mapMutex).unlock();
|
||
|
||
return 0;
|
||
}
|
||
|
||
|