tyler.durden
/
icinga2
mirror of https://github.com/Icinga/icinga2.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use multiple threads for socket IO 

refs #11014
This commit is contained in:
Gunnar Beutner 2016-02-02 11:17:33 +01:00
parent 3f95e86c73
commit 532655d482
3 changed files with 73 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/base/process.cpp
View File

@ -45,7 +45,7 @@ extern char **environ;
using namespace icinga; using namespace icinga;
#define IOTHREADS 2 #define IOTHREADS 4
static boost::mutex l_ProcessMutex[IOTHREADS]; static boost::mutex l_ProcessMutex[IOTHREADS];
static std::map<Process::ProcessHandle, Process::Ptr> l_Processes[IOTHREADS]; static std::map<Process::ProcessHandle, Process::Ptr> l_Processes[IOTHREADS];

111
lib/base/socketevents.cpp
View File

@ -44,36 +44,44 @@ struct EventDescription
Object::Ptr LifesupportReference; Object::Ptr LifesupportReference;
}; };
static boost::thread l_SocketIOThread; #define IOTHREADS 8
static boost::once_flag l_SocketIOOnceFlag = BOOST_ONCE_INIT; static boost::once_flag l_SocketIOOnceFlag = BOOST_ONCE_INIT;
static SOCKET l_SocketIOEventFDs[2]; static boost::thread l_SocketIOThreads[IOTHREADS];
static boost::mutex l_SocketIOMutex; static SOCKET l_SocketIOEventFDs[IOTHREADS][2];
static boost::condition_variable l_SocketIOCV; static boost::mutex l_SocketIOMutex[IOTHREADS];
static bool l_SocketIOFDChanged; static boost::condition_variable l_SocketIOCV[IOTHREADS];
static std::map<SOCKET, SocketEventDescriptor> l_SocketIOSockets; static bool l_SocketIOFDChanged[IOTHREADS];
static std::map<SOCKET, int> l_SocketIOEventChanges; static std::map<SOCKET, SocketEventDescriptor> l_SocketIOSockets[IOTHREADS];
static std::map<SOCKET, int> l_SocketIOEventChanges[IOTHREADS];
int SocketEvents::m_NextID = 0;
void SocketEvents::InitializeThread(void) void SocketEvents::InitializeThread(void)
{ {
Socket::SocketPair(l_SocketIOEventFDs);
Utility::SetNonBlockingSocket(l_SocketIOEventFDs[0]); for (int i = 0; i < IOTHREADS; i++) {
Utility::SetNonBlockingSocket(l_SocketIOEventFDs[1]); Socket::SocketPair(l_SocketIOEventFDs[i]);
Utility::SetNonBlockingSocket(l_SocketIOEventFDs[i][0]);
Utility::SetNonBlockingSocket(l_SocketIOEventFDs[i][1]);
#ifndef _WIN32 #ifndef _WIN32
Utility::SetCloExec(l_SocketIOEventFDs[0]); Utility::SetCloExec(l_SocketIOEventFDs[i][0]);
Utility::SetCloExec(l_SocketIOEventFDs[1]); Utility::SetCloExec(l_SocketIOEventFDs[i][1]);
#endif /* _WIN32 */ #endif /* _WIN32 */
SocketEventDescriptor sed; SocketEventDescriptor sed;
sed.Events = POLLIN; sed.Events = POLLIN;
l_SocketIOSockets[l_SocketIOEventFDs[0]] = sed; l_SocketIOSockets[i][l_SocketIOEventFDs[i][0]] = sed;
l_SocketIOFDChanged[i] = true;
l_SocketIOThread = boost::thread(&SocketEvents::ThreadProc); l_SocketIOThreads[i] = boost::thread(&SocketEvents::ThreadProc, i);
}
} }
void SocketEvents::ThreadProc(void) void SocketEvents::ThreadProc(int tid)
{ {
Utility::SetThreadName("SocketIO"); Utility::SetThreadName("SocketIO");
@ -82,17 +90,17 @@ void SocketEvents::ThreadProc(void)
for (;;) { for (;;) {
{ {
boost::mutex::scoped_lock lock(l_SocketIOMutex); boost::mutex::scoped_lock lock(l_SocketIOMutex[tid]);
if (l_SocketIOFDChanged) { if (l_SocketIOFDChanged[tid]) {
pfds.resize(l_SocketIOSockets.size()); pfds.resize(l_SocketIOSockets[tid].size());
descriptors.resize(l_SocketIOSockets.size()); descriptors.resize(l_SocketIOSockets[tid].size());
int i = 0; int i = 0;
typedef std::map<SOCKET, SocketEventDescriptor>::value_type kv_pair; typedef std::map<SOCKET, SocketEventDescriptor>::value_type kv_pair;
BOOST_FOREACH(const kv_pair& desc, l_SocketIOSockets) { BOOST_FOREACH(const kv_pair& desc, l_SocketIOSockets[tid]) {
if (desc.second.EventInterface) if (desc.second.EventInterface)
desc.second.EventInterface->m_PFD = &pfds[i]; desc.second.EventInterface->m_PFD = &pfds[i];
@ -103,8 +111,8 @@ void SocketEvents::ThreadProc(void)
i++; i++;
} }
l_SocketIOFDChanged = false; l_SocketIOFDChanged[tid] = false;
l_SocketIOCV.notify_all(); l_SocketIOCV[tid].notify_all();
} }
} }
@ -119,18 +127,18 @@ void SocketEvents::ThreadProc(void)
std::vector<EventDescription> events; std::vector<EventDescription> events;
{ {
boost::mutex::scoped_lock lock(l_SocketIOMutex); boost::mutex::scoped_lock lock(l_SocketIOMutex[tid]);
if (l_SocketIOFDChanged) if (l_SocketIOFDChanged[tid])
continue; continue;
for (int i = 0; i < pfds.size(); i++) { for (int i = 0; i < pfds.size(); i++) {
if ((pfds[i].revents & (POLLIN | POLLOUT | POLLHUP | POLLERR)) == 0) if ((pfds[i].revents & (POLLIN | POLLOUT | POLLHUP | POLLERR)) == 0)
continue; continue;
if (pfds[i].fd == l_SocketIOEventFDs[0]) { if (pfds[i].fd == l_SocketIOEventFDs[tid][0]) {
char buffer[512]; char buffer[512];
if (recv(l_SocketIOEventFDs[0], buffer, sizeof(buffer), 0) < 0) if (recv(l_SocketIOEventFDs[tid][0], buffer, sizeof(buffer), 0) < 0)
Log(LogCritical, "SocketEvents", "Read from event FD failed."); Log(LogCritical, "SocketEvents", "Read from event FD failed.");
continue; continue;
@ -162,22 +170,24 @@ void SocketEvents::ThreadProc(void)
void SocketEvents::WakeUpThread(bool wait) void SocketEvents::WakeUpThread(bool wait)
{ {
if (boost::this_thread::get_id() == l_SocketIOThread.get_id()) int tid = m_ID % IOTHREADS;
if (boost::this_thread::get_id() == l_SocketIOThreads[tid].get_id())
return; return;
if (wait) { if (wait) {
boost::mutex::scoped_lock lock(l_SocketIOMutex); boost::mutex::scoped_lock lock(l_SocketIOMutex[tid]);
l_SocketIOFDChanged = true; l_SocketIOFDChanged[tid] = true;
while (l_SocketIOFDChanged) { while (l_SocketIOFDChanged[tid]) {
(void) send(l_SocketIOEventFDs[1], "T", 1, 0); (void) send(l_SocketIOEventFDs[tid][1], "T", 1, 0);
boost::system_time const timeout = boost::get_system_time() + boost::posix_time::milliseconds(50); boost::system_time const timeout = boost::get_system_time() + boost::posix_time::milliseconds(50);
l_SocketIOCV.timed_wait(lock, timeout); l_SocketIOCV[tid].timed_wait(lock, timeout);
} }
} else { } else {
(void) send(l_SocketIOEventFDs[1], "T", 1, 0); (void) send(l_SocketIOEventFDs[tid][1], "T", 1, 0);
} }
} }
@ -185,7 +195,7 @@ void SocketEvents::WakeUpThread(bool wait)
* Constructor for the SocketEvents class. * Constructor for the SocketEvents class.
*/ */
SocketEvents::SocketEvents(const Socket::Ptr& socket, Object *lifesupportObject) SocketEvents::SocketEvents(const Socket::Ptr& socket, Object *lifesupportObject)
: m_FD(socket->GetFD()), m_PFD(NULL) : m_ID(m_NextID++), m_FD(socket->GetFD()), m_PFD(NULL)
{ {
boost::call_once(l_SocketIOOnceFlag, &SocketEvents::InitializeThread); boost::call_once(l_SocketIOOnceFlag, &SocketEvents::InitializeThread);
@ -199,8 +209,10 @@ SocketEvents::~SocketEvents(void)
void SocketEvents::Register(Object *lifesupportObject) void SocketEvents::Register(Object *lifesupportObject)
{ {
int tid = m_ID % IOTHREADS;
{ {
boost::mutex::scoped_lock lock(l_SocketIOMutex); boost::mutex::scoped_lock lock(l_SocketIOMutex[tid]);
VERIFY(m_FD != INVALID_SOCKET); VERIFY(m_FD != INVALID_SOCKET);
@ -209,10 +221,10 @@ void SocketEvents::Register(Object *lifesupportObject)
desc.EventInterface = this; desc.EventInterface = this;
desc.LifesupportObject = lifesupportObject; desc.LifesupportObject = lifesupportObject;
VERIFY(l_SocketIOSockets.find(m_FD) == l_SocketIOSockets.end()); VERIFY(l_SocketIOSockets[tid].find(m_FD) == l_SocketIOSockets[tid].end());
l_SocketIOSockets[m_FD] = desc; l_SocketIOSockets[tid][m_FD] = desc;
l_SocketIOFDChanged = true; l_SocketIOFDChanged[tid] = true;
m_Events = true; m_Events = true;
} }
@ -222,14 +234,16 @@ void SocketEvents::Register(Object *lifesupportObject)
void SocketEvents::Unregister(void) void SocketEvents::Unregister(void)
{ {
int tid = m_ID % IOTHREADS;
{ {
boost::mutex::scoped_lock lock(l_SocketIOMutex); boost::mutex::scoped_lock lock(l_SocketIOMutex[tid]);
if (m_FD == INVALID_SOCKET) if (m_FD == INVALID_SOCKET)
return; return;
l_SocketIOSockets.erase(m_FD); l_SocketIOSockets[tid].erase(m_FD);
l_SocketIOFDChanged = true; l_SocketIOFDChanged[tid] = true;
m_FD = INVALID_SOCKET; m_FD = INVALID_SOCKET;
m_Events = false; m_Events = false;
@ -241,15 +255,17 @@ void SocketEvents::Unregister(void)
void SocketEvents::ChangeEvents(int events) void SocketEvents::ChangeEvents(int events)
{ {
int tid = m_ID % IOTHREADS;
{ {
boost::mutex::scoped_lock lock(l_SocketIOMutex); boost::mutex::scoped_lock lock(l_SocketIOMutex[tid]);
if (m_FD == INVALID_SOCKET) if (m_FD == INVALID_SOCKET)
BOOST_THROW_EXCEPTION(std::runtime_error("Tried to read/write from a closed socket.")); BOOST_THROW_EXCEPTION(std::runtime_error("Tried to read/write from a closed socket."));
std::map<SOCKET, SocketEventDescriptor>::iterator it = l_SocketIOSockets.find(m_FD); std::map<SOCKET, SocketEventDescriptor>::iterator it = l_SocketIOSockets[tid].find(m_FD);
if (it == l_SocketIOSockets.end()) if (it == l_SocketIOSockets[tid].end())
return; return;
if (it->second.Events == events) if (it->second.Events == events)
@ -257,10 +273,10 @@ void SocketEvents::ChangeEvents(int events)
it->second.Events = events; it->second.Events = events;
if (m_PFD && boost::this_thread::get_id() == l_SocketIOThread.get_id()) if (m_PFD && boost::this_thread::get_id() == l_SocketIOThreads[tid].get_id())
m_PFD->events = events; m_PFD->events = events;
else else
l_SocketIOFDChanged = true; l_SocketIOFDChanged[tid] = true;
} }
WakeUpThread(); WakeUpThread();
@ -268,7 +284,8 @@ void SocketEvents::ChangeEvents(int events)
bool SocketEvents::IsHandlingEvents(void) const bool SocketEvents::IsHandlingEvents(void) const
{ {
boost::mutex::scoped_lock lock(l_SocketIOMutex); int tid = m_ID % IOTHREADS;
boost::mutex::scoped_lock lock(l_SocketIOMutex[tid]);
return m_Events; return m_Events;
} }

9
lib/base/socketevents.hpp
View File

@ -52,14 +52,17 @@ protected:
SocketEvents(const Socket::Ptr& socket, Object *lifesupportObject); SocketEvents(const Socket::Ptr& socket, Object *lifesupportObject);
private: private:
int m_ID;
SOCKET m_FD; SOCKET m_FD;
bool m_Events; bool m_Events;
pollfd *m_PFD; pollfd *m_PFD;
static void InitializeThread(void); static int m_NextID;
static void ThreadProc(void);
static void WakeUpThread(bool wait = false); static void InitializeThread(void);
static void ThreadProc(int tid);
void WakeUpThread(bool wait = false);
int GetPollEvents(void) const; int GetPollEvents(void) const;