前面几节我们讨论了非阻塞IO的基本概念、Buffer的设计以及非阻塞connect的实现，现在我们使用它们来完成客户端的编写。
我们在http://www.cnblogs.com/inevermore/p/4049165.html中提出过，客户端需要监听stdin、stdout和sockfd。
这里需要注意的是
所以每次执行poll之前，需要重新装填poll的events数组。
完整的代码如下：
#include "sysutil.h"
#include "buffer.h" int main(int argc, char const *argv[])
{
//创建client套接字
int sockfd = tcp_client(8934);
//调用非阻塞connect函数
int ret = nonblocking_connect(sockfd, "192.168.44.136", 9981, 5000);
if(ret == -1)
{
fprintf(stderr, "Timeout .\n");
exit(EXIT_FAILURE);
} //将三个fd设置为Non-Blocking
activate_nonblock(sockfd);
activate_nonblock(STDIN_FILENO);
activate_nonblock(STDOUT_FILENO); buffer_t recvbuf; //sockfd -> Buffer -> stdout
buffer_t sendbuf; //stdin -> Buffer -> sockfd //初始化缓冲区
buffer_init(&recvbuf);
buffer_init(&sendbuf); struct pollfd pfd[10]; while(1)
{
//初始化
int ix;
for(ix = 0; ix != 3; ++ix)
{
pfd[ix].fd = -1;
pfd[ix].events = 0;
} //重新装填events数组
if(buffer_is_readable(&sendbuf))
{
pfd[0].fd = sockfd;
pfd[0].events |= kWriteEvent;
}
if(buffer_is_writeable(&sendbuf))
{
pfd[1].fd = STDIN_FILENO;
pfd[1].events |= kReadEvent;
}
if(buffer_is_readable(&recvbuf))
{
pfd[2].fd = STDOUT_FILENO;
pfd[2].events |= kWriteEvent;
}
if(buffer_is_writeable(&recvbuf))
{
pfd[0].fd = sockfd;
pfd[0].events |= kReadEvent;
} //监听fd数组
int nready = poll(pfd, 3, 5000);
if(nready == -1)
ERR_EXIT("poll");
else if(nready == 0)
{
printf("timeout\n");
continue;
}
else
{
int i;
for(i = 0; i < 3; ++i)
{
int fd = pfd[i].fd;
if(fd == sockfd && pfd[i].revents & kReadEvent)
{
//从sockfd接收数据到recvbuf
if(buffer_read(&recvbuf, fd) == 0)
{
fprintf(stderr, "server close.\n");
exit(EXIT_SUCCESS);
}
} if(fd == sockfd && pfd[i].revents & kWriteEvent)
buffer_write(&sendbuf, fd); //将sendbuf中的数据写入sockfd if(fd == STDIN_FILENO && pfd[i].revents & kReadEvent)
{
//从stdin接收数据写入sendbuf
if(buffer_read(&sendbuf, fd) == 0)
{
fprintf(stderr, "exit.\n");
exit(EXIT_SUCCESS);
}
} if(fd == STDOUT_FILENO && pfd[i].revents & kWriteEvent)
buffer_write(&recvbuf, fd); //将recvbuf中的数据输出至stdout
}
}
} }
从以上的代码可以看出，大部分操作被封装进了buffer的实现中。
测试服务器，我暂时使用muduo库编写一个，代码如下：
#include <muduo/net/TcpServer.h>
#include <muduo/net/InetAddress.h>
#include <muduo/net/TcpConnection.h>
#include <muduo/base/Timestamp.h>
#include <muduo/net/EventLoop.h>
#include <muduo/base/Logging.h>
using namespace muduo;
using namespace muduo::net; void onMessage(const TcpConnectionPtr &conn, Buffer *buf, Timestamp t)
{
string s(buf->retrieveAllAsString());
LOG_INFO << "recv msg : " << s.size() << " at: " << t.toFormattedString();
conn->send(s);
} int main(int argc, char const *argv[])
{
EventLoop loop;
InetAddress addr("192.168.44.136", 9981);
TcpServer server(&loop, addr, "EchoServer");
server.setMessageCallback(&onMessage);
server.start(); loop.loop(); return 0;
}
读者如果使用上述的代码需要安装muduo网络库。
采用以下命令编译：
g++ server.cpp -lmuduo_net -lmuduo_base -lpthread -o server
下文用poll实现非阻塞的服务器端。