TcpConnection用来管理连接,包括连接的状态、消息的处理(发、收),该类是muduo中唯一一个实用shared_ptr来进行管理的类,也是唯一继承enable_shared_from_this的类,这源于TcpConnection类的模糊的生命期。在本章的结尾我们会进行分析。
我们首先来分析一下私有成员:
enum StateE { kConnecting, kConnected, kDisconnected}; //连接的状态
void setState(StateE s) { state_ = s; } //设置状态
//处理读、写、关闭、错误事件
void handleRead();
void handleWrite();
void handleClose();
void handleError();
EventLoop *loop_; //记录所属的EventLoop
std::string name_; //记录连接的名称
StateE state_; //记录连接的状态
std::unique_ptr<Socket> socket_; //不明所以,暂且认为记录文件描述符fd
std::unique_ptr<Channel> channel_; //管理fd,设置各种类型的回调,并加入poller队列
InetAddress localAddr_; //记录本地地址
InetAddress peerAddr_; //记录对端地址
ConnectionCallback connectionCallback_; //连接回调函数
MessageCallback messageCallback_; //消息到来回调函数
CloseCallback closeCallback_; //关闭回调
构造函数如下:
TcpConnection::TcpConnection(EventLoop *loop, const std::string name,
int sockfd, const InetAddress &localAddr, const InetAddress &peerAddr)
:loop_(loop),
name_(name),
socket_(new Socket(sockfd)),
localAddr_(localAddr),
peerAddr_(peerAddr),
channel_(new Channel(loop_, socket_->fd())),
state_(kConnecting)
{
std::cout << "TcpConnection::ctor[" << name_ << "] at " << this
<< " fd=" << sockfd<<"\n";
channel_->setReadCallback(
std::bind(&TcpConnection::handleRead, this)
);
channel_->setCloseCallback(
std::bind(&TcpConnection::handleClose, this));
/*
channel_->setErrorCallback(
std::bind(&TcpConnection::handleError, this));
*/
}
除了对成员进行初始化外,在Channel中设置了读、写(当前代码没有加上)、关闭、错误回调函数。
另外还有两个重要的public成员函数:
void connectEstablished();
void connectDestroyed();
一个是连接建立,一个是连接销毁,定义如下:
void TcpConnection::connectEstablished()
{
loop_->assertInLoopThread();
setState(kConnected);
channel_->enableReading();
connectionCallback_(shared_from_this());
}
void TcpConnection::connectDestroyed()
{
loop_->assertInLoopThread();
setState(kDisconnected);
channel_->disableAll();
loop_->removeChannel(channel_.get());
}
连接建立,首先设置状态为kConnected,接着调用channel的enableReading,将connfd加入poller队列,最后调用connectionCallback_回调,注意,这里用的是shared_from_this()。
连接销毁,同样,首先设置状态为kDisconnected,然后调用channel的disableAll,将connfd从polle队列中删除,最后调用loop的removeChannel将channel删除
这两个函数供上层调用,即TcpServer调用。
下面我们来分析一下TcpConnection的生命期。首先TcpConnectionPtr是一个智能指针类型——shared_ptr,TcpConnection首次创建是在TcpServer::newConnection函数中
TcpConnectionPtr conn(
new TcpConnection(loop_, connName, sockfd, localAddr, peerAddr)
);
此时,引用计数为1,而后为了管理,conn被放入connections_(是一个map)中,引用计数为2。又由于onConnection和onMessage都是传入的shared_from_this(),所以引用计数都会增加。TcpServer::newConnection结束后,引用计数会减1。就是说,如果没有调用回调函数,那么conn的引用计数最终为1,即存在于connections_中。那么什么时候引用计数为0,conn被析构呢?上层TcpServer调用removeConnection,该函数会接受一个conn的引用,会增加引用计数,接着调用connections_删除map中的conn,引用计数减1,接着调用conn的connectionDestroy,处理与conn相关的channel,当次函数结束的时候,引用计数减为0(如果此时没有调用messageCallback或者其它回调的话),conn会被析构。
我们开始的时候提到,用shared_ptr来管理TcpConnection,是因为TcpConnection的生命期比较模糊。在完整的muduo中,主线程只负责接受连接,然后在TcpConnection::newConnection中启动一个线程,把TcpConnection放入该线程处理。conn的各种事件都在新的线程中处理。比如当conn正在处理一个request,此时线程是“阻塞”在这个处理上的,客户端这时候断开连接,线程的EventLoop会收到close事件,但是此时线程“阻塞”在处理request中,并不能马上处理close事件,只有当request结束后,才会处理close事件。所以,对服务器端来说,只有当conn不在被使用且客户端断开连接的时候,conn才会被析构。书上说使用shared_ptr管理连接,可以防止串话。
关于TcpConnection生命期分析的不知道对不对,如果有不正确的地方,见谅。