int initializeSocketServer_RMSEG(void) { int res = FUNC_RETURN_OK; int netres = 0; char *allip = "0.0.0.0"; pgsocket RMListenSocket[HAWQRM_SERVER_PORT_COUNT]; for ( int i = 0 ; i < HAWQRM_SERVER_PORT_COUNT ; ++i ) { RMListenSocket[i] = PGINVALID_SOCKET; } /* Listen normal socket addresses. */ netres = StreamServerPort( AF_UNSPEC, allip, rm_segment_port, NULL, RMListenSocket, HAWQRM_SERVER_PORT_COUNT); /* If there are multiple segments in one host, which is common in old imp. * We can not make all segments work. So, if HAWQ RM SEG fails to start * socket server by listening the port, we accept this case and make it * silent. HAWQ RM will not recognize this segment and will not assign * tasks. */ if ( netres != STATUS_OK ) { res = REQUESTHANDLER_FAIL_START_SOCKET_SERVER; elog( LOG, "Can not create socket server. HostName=%s, Port=%d", allip, rm_segment_port); return res; } /* Initialize array for polling all file descriptors. */ initializeAsyncComm(); int validfdcount = 0; AsyncCommBuffer newbuffer = NULL; for ( int i = 0 ; i < HAWQRM_SERVER_PORT_COUNT ; ++i ) { if (RMListenSocket[i] != PGINVALID_SOCKET) { netres = registerFileDesc(RMListenSocket[i], NULL, ASYNCCOMM_READ, &AsyncCommBufferHandlersMsgServer, NULL, &newbuffer); if ( netres != FUNC_RETURN_OK ) { res = REQUESTHANDLER_FAIL_START_SOCKET_SERVER; elog(LOG, "Resource manager cannot track socket server."); break; } validfdcount++; InitHandler_Message(newbuffer); } } if ( res != FUNC_RETURN_OK ) { for ( int i = 0 ; i < HAWQRM_SERVER_PORT_COUNT ; ++i ) { if ( RMListenSocket[i] != PGINVALID_SOCKET ) close(RMListenSocket[i]); } return res; } elog(DEBUG5, "HAWQ RM SEG :: Starts accepting resource request. " "Listening normal socket port %s:%d. " "Total listened %d FDs.", allip, rm_segment_port, validfdcount); return res; }
void initializeSyncRPCComm(void) { initializeAsyncComm(); }