int main()
{
using namespace CpperoMQ;
// Prepare our context and sockets.
Context context;
RouterSocket frontend(context.createRouterSocket());
DealerSocket backend(context.createDealerSocket());
frontend.bind("tcp://*:5559");
backend.bind("tcp://*:5560");
auto frontendPollItem = isReceiveReady(frontend, [&frontend, &backend]()
{
bool more = true;
while (more)
{
// Process all parts of the message.
IncomingMessage inMsg;
inMsg.receive(frontend, more);
OutgoingMessage outMsg(inMsg.size(), inMsg.data());
outMsg.send(backend, more);
}
});
auto backendPollItem = isReceiveReady(backend, [&frontend, &backend]()
{
bool more = true;
while (more)
{
// Process all parts of the message.
IncomingMessage inMsg;
inMsg.receive(backend, more);
OutgoingMessage outMsg(inMsg.size(), inMsg.data());
outMsg.send(frontend, more);
}
});
Poller poller;
while (true)
{
poller.poll(frontendPollItem, backendPollItem);
}
return 0;
}
void Poller::onData(uv_poll_t* handle, int status, int events) {
Nan::HandleScope scope;
Poller* obj = static_cast<Poller*>(handle->data);
v8::Local<v8::Value> argv[2];
if (0 != status) {
// fprintf(stdout, "OnData Error status=%s events=%d\n", uv_strerror(status), events);
argv[0] = v8::Exception::Error(Nan::New<v8::String>(uv_strerror(status)).ToLocalChecked());
argv[1] = Nan::Undefined();
} else {
// fprintf(stdout, "OnData status=%d events=%d\n", status, events);
argv[0] = Nan::Null();
argv[1] = Nan::New<v8::Integer>(events);
}
// remove triggered events from the poll
int newEvents = obj->events & ~events;
obj->poll(newEvents);
obj->callback.Call(2, argv);
}
void SocketService::run(void)
{
timeout_t timer, expires;
SocketPort *port;
unsigned char buf;
#ifdef USE_POLL
Poller mfd;
pollfd * p_ufd;
int lastcount = 0;
// initialize ufd in all attached ports :
// probably don't need this but it can't hurt.
enterMutex();
port = first;
while(port)
{
port->ufd = 0;
port = port->next;
}
leaveMutex();
#else
struct timeval timeout, *tvp;
fd_set inp, out, err;
FD_ZERO(&inp);
FD_ZERO(&out);
FD_ZERO(&err);
int so;
#endif
setCancel(cancelDeferred);
for(;;)
{
timer = TIMEOUT_INF;
while(1 == ::read(iosync[0], (char *)&buf, 1))
{
if(buf)
{
onUpdate(buf);
continue;
}
setCancel(cancelImmediate);
sleep(TIMEOUT_INF);
exit();
}
#ifdef USE_POLL
bool reallocate = false;
MUTEX_START
onEvent();
port = first;
while(port)
{
onCallback(port);
if ( ( p_ufd = port->ufd ) ) {
if ( ( POLLHUP | POLLNVAL ) & p_ufd->revents ) {
// Avoid infinite loop from disconnected sockets
port->detect_disconnect = false;
p_ufd->events &= ~POLLHUP;
SocketPort* p = port;
port = port->next;
detach(p);
reallocate = true;
p->disconnect();
continue;
}
if ( ( POLLIN | POLLPRI ) & p_ufd->revents )
port->pending();
if ( POLLOUT & p_ufd->revents )
port->output();
} else {
reallocate = true;
}
retry:
expires = port->getTimer();
if(expires > 0)
if(expires < timer)
timer = expires;
if(!expires)
{
port->endTimer();
port->expired();
goto retry;
}
port = port->next;
}
//
// reallocate things if we saw a ServerPort without
// ufd set !
if ( reallocate || ( ( count + 1 ) != lastcount ) ) {
lastcount = count + 1;
p_ufd = mfd.getList( count + 1 );
// Set up iosync polling
p_ufd->fd = iosync[0];
p_ufd->events = POLLIN | POLLHUP;
p_ufd ++;
port = first;
while(port)
{
p_ufd->fd = port->so;
p_ufd->events =
( port->detect_disconnect ? POLLHUP : 0 )
| ( port->detect_output ? POLLOUT : 0 )
| ( port->detect_pending ? POLLIN : 0 )
;
port->ufd = p_ufd;
p_ufd ++;
port = port->next;
}
}
MUTEX_END
poll( mfd.getList(), lastcount, timer );
#else
MUTEX_START
onEvent();
port = first;
while(port)
{
onCallback(port);
so = port->so;
if(FD_ISSET(so, &err)) {
port->detect_disconnect = false;
SocketPort* p = port;
port = port->next;
p->disconnect();
continue;
}
if(FD_ISSET(so, &inp))
port->pending();
if(FD_ISSET(so, &out))
port->output();
retry:
expires = port->getTimer();
if(expires > 0)
if(expires < timer)
timer = expires;
// if we expire, get new scheduling now
if(!expires)
{
port->endTimer();
port->expired();
goto retry;
}
port = port->next;
}
FD_ZERO(&inp);
FD_ZERO(&out);
FD_ZERO(&err);
FD_SET(iosync[0],&inp);
port = first;
while(port)
{
so = port->so;
if(port->detect_pending)
FD_SET(so, &inp);
if(port->detect_output)
FD_SET(so, &out);
if(port->detect_disconnect)
FD_SET(so, &err);
port = port->next;
}
MUTEX_END
if(timer == TIMEOUT_INF)
tvp = NULL;
else
{
tvp = &timeout;
timeout.tv_sec = timer / 1000;
timeout.tv_usec = (timer % 1000) * 1000;
}
select(hiwater, &inp, &out, &err, tvp);
#endif
}
}
void* ServerLoop::auxiliaryfunc(void* a){
//fix select
Poller poll;
poll_event events[32];
ServerLoop* thisloop=(ServerLoop*)a;
std::map<int,int> acceptormap;
std::map<int,int> eventmap;
std::map<int,bool> needtaskindex;
int dispatchindex=0;
//fix
TcpAcceptor accs[MAXADDRESS];
assert(thisloop->acceptors_.size()<=MAXADDRESS);
for(int i=0;i<thisloop->acceptors_.size();i++){
address addr=thisloop->acceptors_[i];
accs[i].bindAddr(addr.ip,addr.port);
accs[i].listen();
acceptormap[accs[i].getfd()]=i;
poll.addFd(accs[i].getfd(), NULL);
}
for(int i=0;i<THREADSIZE+1;i++){
poll.addFd(thisloop->tasknoti[i].getRecvfd(),a);
eventmap[thisloop->tasknoti[i].getRecvfd()]=i;
needtaskindex[i]=false;
}
while(1){
int count=poll.wait(events, 32);
for(int i=0;i<count;i++){
if(events[i].fd==thisloop->sernoti.getRecvfd()){
thisloop->sernoti.disnotify();
continue;
}
if(events[i].ptr==NULL){
//dispatch average
int conn=accs[acceptormap[events[i].fd]].accept();
if(conn>0){
Task acceptask;
acceptask.fd_=conn;
acceptask.option_=RADD;
acceptask.home_=dispatchindex%THREADSIZE;
dispatchindex=(dispatchindex+1)%THREADSIZE;
thisloop->taskPut(acceptask);
//printf("accept%d\n",acceptask.home_);
}else{
printf("accept Error!\n");
}
}else{
int index=eventmap[events[i].fd];
;
needtaskindex[index]=true;
Notify::disnotifyFd(events[i].fd);
}
}
for(int i=0;i<THREADSIZE;i++){
if(needtaskindex[i]&&(!thisloop->childtaskqueue_[i].empty())){
thisloop->taskTake(i);
needtaskindex[i]=false;
}
}
thisloop->notifytask_=false;
}
return NULL;
}
void SocketService::run(void)
{
timeout_t timer, expires;
SocketPort *port;
unsigned char buf;
#ifndef WIN32
#ifdef USE_POLL
Poller mfd;
pollfd * p_ufd;
int lastcount = 0;
// initialize ufd in all attached ports :
// probably don't need this but it can't hurt.
enterMutex();
port = first;
while(port) {
port->ufd = 0;
port = port->next;
}
leaveMutex();
#else
struct timeval timeout, *tvp;
fd_set inp, out, err;
FD_ZERO(&inp);
FD_ZERO(&out);
FD_ZERO(&err);
int so;
#endif
#else // WIN32
int numHandle = 0;
HANDLE hv[MAXIMUM_WAIT_OBJECTS];
#endif
#ifdef WIN32
// FIXME: needed ?
ResetEvent(sync->GetSync());
#endif
setCancel(cancelDeferred);
for(;;) {
timer = TIMEOUT_INF;
#ifndef WIN32
while(1 == ::read(iosync[0], (char *)&buf, 1)) {
#else
for(;;) {
int f = sync->getFlag();
if (f < 0)
break;
buf = f;
#endif
if(buf) {
onUpdate(buf);
continue;
}
setCancel(cancelImmediate);
sleep(TIMEOUT_INF);
exit();
}
#ifndef WIN32
#ifdef USE_POLL
bool reallocate = false;
MUTEX_START
onEvent();
port = first;
while(port) {
onCallback(port);
if ( ( p_ufd = port->ufd ) ) {
if ( ( POLLHUP | POLLNVAL ) & p_ufd->revents ) {
// Avoid infinite loop from disconnected sockets
port->detect_disconnect = false;
p_ufd->events &= ~POLLHUP;
SocketPort* p = port;
port = port->next;
detach(p);
reallocate = true;
p->disconnect();
continue;
}
if ( ( POLLIN | POLLPRI ) & p_ufd->revents )
port->pending();
if ( POLLOUT & p_ufd->revents )
port->output();
} else {
reallocate = true;
}
retry:
expires = port->getTimer();
if(expires > 0)
if(expires < timer)
timer = expires;
if(!expires) {
port->endTimer();
port->expired();
goto retry;
}
port = port->next;
}
//
// reallocate things if we saw a ServerPort without
// ufd set !
if ( reallocate || ( ( count + 1 ) != lastcount ) ) {
lastcount = count + 1;
p_ufd = mfd.getList( count + 1 );
// Set up iosync polling
p_ufd->fd = iosync[0];
p_ufd->events = POLLIN | POLLHUP;
p_ufd ++;
port = first;
while(port) {
p_ufd->fd = port->so;
p_ufd->events =
( port->detect_disconnect ? POLLHUP : 0 )
| ( port->detect_output ? POLLOUT : 0 )
| ( port->detect_pending ? POLLIN : 0 )
;
port->ufd = p_ufd;
p_ufd ++;
port = port->next;
}
}
MUTEX_END
poll( mfd.getList(), lastcount, timer );
#else
MUTEX_START
onEvent();
port = first;
while(port) {
onCallback(port);
so = port->so;
if(FD_ISSET(so, &err)) {
port->detect_disconnect = false;
SocketPort* p = port;
port = port->next;
p->disconnect();
continue;
}
if(FD_ISSET(so, &inp))
port->pending();
if(FD_ISSET(so, &out))
port->output();
retry:
expires = port->getTimer();
if(expires > 0)
if(expires < timer)
timer = expires;
// if we expire, get new scheduling now
if(!expires) {
port->endTimer();
port->expired();
goto retry;
}
port = port->next;
}
FD_ZERO(&inp);
FD_ZERO(&out);
FD_ZERO(&err);
FD_SET(iosync[0],&inp);
port = first;
while(port) {
so = port->so;
if(port->detect_pending)
FD_SET(so, &inp);
if(port->detect_output)
FD_SET(so, &out);
if(port->detect_disconnect)
FD_SET(so, &err);
port = port->next;
}
MUTEX_END
if(timer == TIMEOUT_INF)
tvp = NULL;
else {
tvp = &timeout;
timeout.tv_sec = timer / 1000;
timeout.tv_usec = (timer % 1000) * 1000;
}
select(hiwater, &inp, &out, &err, tvp);
#endif
#else // WIN32
MUTEX_START
onEvent();
hv[0] = sync->GetSync();
numHandle = 1;
port = first;
while(port) {
onCallback(port);
long events = 0;
if(port->detect_pending)
events |= FD_READ;
if(port->detect_output)
events |= FD_WRITE;
if(port->detect_disconnect)
events |= FD_CLOSE;
// !!! ignore some socket on overflow !!!
if (events && numHandle < MAXIMUM_WAIT_OBJECTS) {
WSAEventSelect(port->so,port->event,events);
hv[numHandle++] = port->event;
}
retry:
expires = port->getTimer();
if(expires > 0)
if(expires < timer)
timer = expires;
// if we expire, get new scheduling now
if(!expires) {
port->endTimer();
port->expired();
goto retry;
}
port = port->next;
}
MUTEX_END
// FIXME: handle thread cancellation correctly
DWORD res = WaitForMultipleObjects(numHandle,hv,FALSE,timer);
switch (res) {
case WAIT_OBJECT_0:
break;
case WAIT_TIMEOUT:
break;
default:
// FIXME: handle failures (detach SocketPort)
if (res >= WAIT_OBJECT_0+1 && res <= WAIT_OBJECT_0+MAXIMUM_WAIT_OBJECTS) {
int curr = res - (WAIT_OBJECT_0);
WSANETWORKEVENTS events;
// search port
MUTEX_START
port = first;
while(port) {
if (port->event == hv[curr])
break;
port = port->next;
}
MUTEX_END
// if port not found ignore
if (!port || port->event != hv[curr])
break;
WSAEnumNetworkEvents(port->so,port->event,&events);
if(events.lNetworkEvents & FD_CLOSE) {
port->detect_disconnect = false;
port->disconnect();
continue;
}
if(events.lNetworkEvents & FD_READ)
port->pending();
if(events.lNetworkEvents & FD_WRITE)
port->output();
}
}
#endif
}
}