用多路复用I/O模型实现支持多个客户端的通信服务
作者:HuangRG
引言
多路复用I/O模型是UNIX/LINUX用得的最多的一种I/O模型。这种I/O模型在技术上的实现是包括select()以及FD_XXX的几个宏及常量。在单个进程中支持的客户端数量由FD_SETSIZE决定。Solaris
10和Linux 9.0默认为1024个,Windows 2000是64个。本文用代码给出该I/O模型处理多Client的一种实现。
用select开发一个通信服务器
这个服务器包括接收模块、发送模块和套接口管理模块。我们只演示与select实现有关的部分,即接收模块和套接口管理模块。下面服务器类关系图:
tcp_listen是tcp实现监听功能,tcp_comm实现tcp的发送和接收,下在我们重点在于recv和sock_list这两个类。sock_list定义如下:
class sock_list
{
public:
// 功能:将一个值插入到list的尾部
// 参数:要插入的tcp_comm对象
void insert(tcp_comm* t);
// 功能:删除一个指定的元素
// 参数:指向要删除的tcp_comm对象的指针
void remove(tcp_comm* t);
// 功能:从队列取得所有的元素
// 参数:存放指向tcp_comm对象指针的向量
void get_all(std::vector<tcp_comm*>& vec);
private:
std::list<tcp_comm*> _sock_list;
};
sock_list实现tcp_comm对象的管理。下面我们来看recv类的定义:
class recv
{
public:
// 功能: 启动通信服务
bool start();
private:
// 功能: 接收核心函数,通过多路复用支持多个客户端
// 返回值: 正常, 0; 失败,-1
int do_select();
// 初始化套接口集合
void init_readfds(const vector<tcp*>& vec, fd_set* set);
private:
sock_list* _list;
tcp_listen* _listen; // 监听套接口
};
start()是给外部调用的:创建监听套接口对象;监听;执行do_select()。init_readfds将监听套接口对象和socket_list中的通信套接口对象加入到套接口读套接口集合中。do_select()比较重要,流程:
用init_readfds初始化读套接口集合;
执行select(),返回进行到下一步;
先检测监听套接口是否有连接到来,并将新的连接加入到sock_list中;
再检测sock_list中的套接口是否有数据可接收。
实现代码:
int recv::do_select()
{
fd_set readfds;
int listen = _listen->get_handle();
while (true)
{
vector<tcp_comm*> vec;
_list->get_all(vec); // 获得sock_list中的所有套接口句柄
init_readfds(vec, &readfds);
select(...);
// 检查监听套接字句柄,有新的连接到,建立新的连接
if (FD_ISSET(listen, &readfds))
{
tcp_comm* client = new tcp_comm;
_listen->accept(*client);
_list->insert(client);// 将已连接的套接字对象放到sock_list中
}
// 检查所有的客户端套接字句柄
vector<tcp_comm*>::iterator it;
for (it = vec.begin(); it != vec.end(); it++)
{
if (FD_ISSET((*it)->get_handle(), &readfds))
// 这里是*it作为参数进行数据接收
}
}
}
为减少代码量,我省去了出错处理、细节;本来类都在某个名称空间中的,也省去了。
多进程(线程)和select合作
上面是在一个进程中的情况,可以考虑多线程中每一个线程实现select。以solaris
10为例,开100个线程,一个线程支持1024个套接口,那就是100 *
1024,10万个。集群服务中的前端均衡器的实现可以考虑一下这个技术,不过效率将是一个极大的考验。
结论
本文重点讲述在单进程的环境使用select支持多个客户端。经常看见一些服务程序,用到了select也是来一个连接,fork一个子进程处理。在单进程中处理多个客户端的目的是使开发的程序更易理解、维护;多进程往往较单进程复杂些。