server_common/wsproxy
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
aozhiwei 2023-04-24 12:03:30 +08:00
parent 52188b8ada
commit b424a98f9c
3 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/wsproxy/downstream.cc
View File

@ -49,9 +49,7 @@ void DownStream::OnClose()
ss::SS_WSP_SocketDisconnect msg; ss::SS_WSP_SocketDisconnect msg;
GetUpStream().lock()->SendMsg(socket_handle_, msg); GetUpStream().lock()->SendMsg(socket_handle_, msg);
} }
if (auto long_session = long_session_wp_.lock(); !long_session_wp_.expired()) { if (!long_session_wp_.expired()) {
LongSessionMgr::Instance()->DelSession(socket_handle_);
} else {
} }
} }

8
server/wsproxy/longsessionmgr.cc
View File

@ -88,3 +88,11 @@ void LongSessionMgr::ProcUdpPacket(a8::UdpPacket* pkt)
session->GetKcpSession()->OnRecvPacket(pkt); session->GetKcpSession()->OnRecvPacket(pkt);
} }
} }
void LongSessionMgr::DelSession(int socket_handle)
{
{
auto sesion = GetSession(socket_handle);
}
socket_handle_hash_.erase(socket_handle);
}

1
server/wsproxy/longsessionmgr.h
View File

@ -30,6 +30,7 @@ class LongSessionMgr : public a8::Singleton<LongSessionMgr>
void ProcUdpPacket(a8::UdpPacket* pkt); void ProcUdpPacket(a8::UdpPacket* pkt);
std::shared_ptr<LongSession> GetSession(int socket_handle); std::shared_ptr<LongSession> GetSession(int socket_handle);
std::shared_ptr<a8::UdpListener> GetUdpListener() { return udp_listener_; } std::shared_ptr<a8::UdpListener> GetUdpListener() { return udp_listener_; }
void DelSession(int socket_handle);
private: private:
std::shared_ptr<a8::UdpListener> udp_listener_; std::shared_ptr<a8::UdpListener> udp_listener_;