muduo网络库源码解析 五

26 篇文章 1 订阅

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生命期分析的不知道对不对,如果有不正确的地方,见谅。




  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值