1212#include < brynet/net/SocketLibFunction.h>
1313
1414#include < brynet/net/EventLoop.h>
15- #include < brynet/net/DataSocket .h>
15+ #include < brynet/net/TcpConnection .h>
1616#include < brynet/net/TCPService.h>
1717#include < brynet/net/ListenThread.h>
1818
@@ -26,15 +26,15 @@ std::atomic_llong TotalRecvLen = ATOMIC_VAR_INIT(0);
2626std::atomic_llong SendPacketNum = ATOMIC_VAR_INIT (0 );
2727std::atomic_llong RecvPacketNum = ATOMIC_VAR_INIT (0 );
2828
29- std::vector<DataSocket:: PTR > clients;
30- TcpService::PTR service;
29+ std::vector<TcpConnection::Ptr > clients;
30+ TcpService::Ptr service;
3131
32- static void addClientID (const DataSocket:: PTR & session)
32+ static void addClientID (const TcpConnection::Ptr & session)
3333{
3434 clients.push_back (session);
3535}
3636
37- static void removeClientID (const DataSocket:: PTR & session)
37+ static void removeClientID (const TcpConnection::Ptr & session)
3838{
3939 for (auto it = clients.begin (); it != clients.end (); ++it)
4040 {
@@ -51,7 +51,7 @@ static size_t getClientNum()
5151 return clients.size ();
5252}
5353
54- static void broadCastPacket (const DataSocket:: PACKET_PTR & packet)
54+ static void broadCastPacket (const TcpConnection::PacketPtr & packet)
5555{
5656 auto packetLen = packet->size ();
5757 RecvPacketNum++;
@@ -81,18 +81,18 @@ int main(int argc, char** argv)
8181 auto mainLoop = std::make_shared<EventLoop>();
8282 auto listenThrean = ListenThread::Create ();
8383
84- listenThrean->startListen (false , " 0.0.0.0" , port, [mainLoop, listenThrean](TcpSocket::PTR socket) {
85- socket->SocketNodelay ();
86- socket->SetSendSize (32 * 1024 );
87- socket->SetRecvSize (32 * 1024 );
84+ listenThrean->startListen (false , " 0.0.0.0" , port, [mainLoop, listenThrean](TcpSocket::Ptr socket) {
85+ socket->setNodelay ();
86+ socket->setSendSize (32 * 1024 );
87+ socket->setRecvSize (32 * 1024 );
8888
89- auto enterCallback = [mainLoop](const DataSocket:: PTR & session) {
90- mainLoop->pushAsyncProc ([session]() {
89+ auto enterCallback = [mainLoop](const TcpConnection::Ptr & session) {
90+ mainLoop->pushAsyncFunctor ([session]() {
9191 addClientID (session);
9292 });
9393
94- session->setDisConnectCallback ([mainLoop](const DataSocket:: PTR & session) {
95- mainLoop->pushAsyncProc ([session]() {
94+ session->setDisConnectCallback ([mainLoop](const TcpConnection::Ptr & session) {
95+ mainLoop->pushAsyncFunctor ([session]() {
9696 removeClientID (session);
9797 });
9898 });
@@ -112,8 +112,8 @@ int main(int argc, char** argv)
112112 auto packet_len = rp.readUINT32 ();
113113 if (leftLen >= packet_len && packet_len >= HEAD_LEN )
114114 {
115- auto packet = DataSocket ::makePacket (parseStr, packet_len);
116- mainLoop->pushAsyncProc ([packet]() {
115+ auto packet = TcpConnection ::makePacket (parseStr, packet_len);
116+ mainLoop->pushAsyncFunctor ([packet]() {
117117 broadCastPacket (packet);
118118 });
119119
@@ -134,7 +134,7 @@ int main(int argc, char** argv)
134134 return totalProcLen;
135135 });
136136 };
137- service->addDataSocket (std::move (socket),
137+ service->addTcpConnection (std::move (socket),
138138 brynet::net::TcpService::AddSocketOption::WithEnterCallback (enterCallback),
139139 brynet::net::TcpService::AddSocketOption::WithMaxRecvBufferSize (1024 * 1024 ));
140140 });
0 commit comments