valitydev/thrift
14
0
Fork 0
mirror of https://github.com/valitydev/thrift.git synced 2024-11-07 02:45:22 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

THRIFT-3080: use select() instead poll() for early windows compatibility.

Browse Source
This commit is contained in:
Lei Feiwei 2015-04-04 22:12:07 +08:00 committed by Roger Meier
parent 38772c9c8d
commit b5ebcd199c
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
lib/cpp/src/thrift/server/TNonblockingServer.cpp
View File

@ -28,7 +28,10 @@
#include <thrift/transport/PlatformSocket.h>
#include <iostream>
#include <poll.h>
#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif
#ifdef HAVE_SYS_SOCKET_H
#include <sys/socket.h>
@ -1394,33 +1397,36 @@ bool TNonblockingIOThread::notify(TNonblockingServer::TConnection* conn) {
return false;
}
fd_set wfds, efds;
int ret = -1;
struct pollfd pfd = {fd, POLLOUT, 0};
int kSize = sizeof(conn);
const char * pos = (const char *)const_cast_sockopt(&conn);
while (kSize > 0) {
pfd.revents = 0;
ret = poll(&pfd, 1, -1);
FD_ZERO(&wfds);
FD_ZERO(&efds);
FD_SET(fd, &wfds);
FD_SET(fd, &efds);
ret = select(fd + 1, NULL, &wfds, &efds, NULL);
if (ret < 0) {
return false;
} else if (ret == 0) {
continue;
}
if (pfd.revents & POLLHUP || pfd.revents & POLLERR) {
::close(fd);
if (FD_ISSET(fd, &efds)) {
::THRIFT_CLOSESOCKET(fd);
return false;
}
if (pfd.revents & POLLOUT) {
if (FD_ISSET(fd, &wfds)) {
ret = send(fd, pos, kSize, 0);
if (ret < 0) {
if (errno == EAGAIN) {
continue;
}
::close(fd);
::THRIFT_CLOSESOCKET(fd);
return false;
}