moduo之服务器TcpServer

简介TcpServer实现了多线程主从Reactor服务器结构TcpServerAcceptorTcpConnectionEventLoopThreadPoolTcpServer相关的回调连接建立回调ConnectionCallback connectionCallback_连接上数据到来时的回调MessageCallback messageCallback_连接上写完成回调WriteCompleteCallback writeCompleteCallback_typedefstd::functionvoid(constTcpConnectionPtr)ConnectionCallback;typedefstd::functionvoid(constTcpConnectionPtr)WriteCompleteCallback;typedefstd::functionvoid(constTcpConnectionPtr,Buffer*,Timestamp)MessageCallback;Acceptor新连接回调voidnewConnection(intsockfd,constInetAddresspeerAddr);TcpConnection的四个回调voidsetConnectionCallback(constConnectionCallbackcb){connectionCallback_cb;}/// Set message callback./// Not thread safe.voidsetMessageCallback(constMessageCallbackcb){messageCallback_cb;}/// Set write complete callback./// Not thread safe.voidsetWriteCompleteCallback(constWriteCompleteCallbackcb){writeCompleteCallback_cb;}voidremoveConnection(constTcpConnectionPtrconn);事件循环线程池线程启动初始回调voidsetThreadInitCallback(constThreadInitCallbackcb){threadInitCallback_cb;}Acceptor连接回调从事件循环池中轮循方式选取一个创建TcpConnection将TcpServer设置的connectionCallback_,messageCallback_和writeCompleteCallback_传递到TcpConnection同时设置关闭连接的回调removeConnection创建连接建立的任务connectEstablished放入连接事件循环的队列中异步执行voidTcpServer::newConnection(intsockfd,constInetAddresspeerAddr){loop_-assertInLoopThread();EventLoop*ioLoopthreadPool_-getNextLoop();charbuf[64];snprintf(buf,sizeofbuf,-%s#%d,ipPort_.c_str(),nextConnId_);nextConnId_;string connNamename_buf;LOG_INFOTcpServer::newConnection [name_] - new connection [connName] from peerAddr.toIpPort();InetAddresslocalAddr(sockets::getLocalAddr(sockfd));// FIXME poll with zero timeout to double confirm the new connection// FIXME use make_shared if necessaryTcpConnectionPtrconn(newTcpConnection(ioLoop,connName,sockfd,localAddr,peerAddr));connections_[connName]conn;conn-setConnectionCallback(connectionCallback_);conn-setMessageCallback(messageCallback_);conn-setWriteCompleteCallback(writeCompleteCallback_);conn-setCloseCallback(std::bind(TcpServer::removeConnection,this,_1));// FIXME: unsafeioLoop-runInLoop(std::bind(TcpConnection::connectEstablished,conn));}连接建立回调connectEstablished设置连接状态为kConnected事件通道绑定连接对象开启读事件执行事件回调connectionCallback_voidTcpConnection::connectEstablished(){loop_-assertInLoopThread();assert(state_kConnecting);setState(kConnected);channel_-tie(shared_from_this());channel_-enableReading();connectionCallback_(shared_from_this());}关闭连接回调添加任务removeConnectionInLoop到acceptor所在的事件循环队列中异步执行连接容器中删除对应连接添加任务connectDestroyed到连接所在的事件循环队列中异步执行voidTcpServer::removeConnection(constTcpConnectionPtrconn){// FIXME: unsafeloop_-runInLoop(std::bind(TcpServer::removeConnectionInLoop,this,conn));}voidTcpServer::removeConnectionInLoop(constTcpConnectionPtrconn){loop_-assertInLoopThread();LOG_INFOTcpServer::removeConnectionInLoop [name_] - connection conn-name();size_t nconnections_.erase(conn-name());(void)n;assert(n1);EventLoop*ioLoopconn-getLoop();ioLoop-queueInLoop(std::bind(TcpConnection::connectDestroyed,conn));}