X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=include%2Fthreadengine.h;h=c11f2d817734d58a7de096b2836a8cf20b843a63;hb=abe0f102d7899780200b3d86557316ef71b57dab;hp=7d27b867ac8b911c34131bd0d066e132eb57c4e6;hpb=9dbd7d10ac81c6445c58f4abb4f0eff240a8f9d2;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/include/threadengine.h b/include/threadengine.h index 7d27b867a..c11f2d817 100644 --- a/include/threadengine.h +++ b/include/threadengine.h @@ -2,8 +2,8 @@ * | Inspire Internet Relay Chat Daemon | * +------------------------------------+ * - * InspIRCd: (C) 2002-2008 InspIRCd Development Team - * See: http://www.inspircd.org/wiki/index.php/Credits + * InspIRCd: (C) 2002-2009 InspIRCd Development Team + * See: http://wiki.inspircd.org/Credits * * This program is free but copyrighted software; see * the file COPYING for details. @@ -20,34 +20,153 @@ #include "inspircd_config.h" #include "base.h" -class InspIRCd; +#ifdef WINDOWS +#include "threadengines/threadengine_win32.h" +#endif + +class ThreadData; +/** Derive from this class to implement your own threaded sections of + * code. Be sure to keep your code thread-safe and not prone to deadlocks + * and race conditions if you MUST use threading! + */ class CoreExport Thread : public Extensible { + private: + /** Set to true when the thread is to exit + */ + bool ExitFlag; + protected: + /** Get thread's current exit status. + * (are we being asked to exit?) + */ + bool GetExitFlag() + { + return ExitFlag; + } public: - Thread() { }; - virtual ~Thread() { }; - virtual void Run() = 0; -}; + /** Opaque thread state managed by threading engine + */ + ThreadData* state; -class CoreExport ThreadEngine : public Extensible -{ - protected: + /** Set Creator to NULL at this point + */ + Thread() : ExitFlag(false), state(NULL) + { + } - InspIRCd* ServerInstance; - Thread* NewThread; + virtual ~Thread(); - public: + /** Override this method to put your actual + * threaded code here. + */ + virtual void Run() = 0; - ThreadEngine(InspIRCd* Instance); + /** Signal the thread to exit gracefully. + */ + virtual void SetExitFlag() + { + ExitFlag = true; + } +}; - virtual ~ThreadEngine(); - virtual bool Mutex(bool enable) = 0; +class CoreExport QueuedThread : public Thread +{ + ThreadQueueData queue; + protected: + /** Waits for an enqueue operation to complete + * You MUST hold the queue lock when you call this. + * It will be unlocked while you wait, and will be relocked + * before the function returns + */ + void WaitForQueue() + { + queue.Wait(); + } + public: + /** Lock queue. + */ + void LockQueue() + { + queue.Lock(); + } + /** Unlock queue. + */ + void UnlockQueue() + { + queue.Unlock(); + } + /** Unlock queue and wake up worker + */ + void UnlockQueueWakeup() + { + queue.Wakeup(); + queue.Unlock(); + } + virtual void SetExitFlag() + { + queue.Lock(); + Thread::SetExitFlag(); + queue.Wakeup(); + queue.Unlock(); + } +}; - virtual void Run() = 0; +class CoreExport SocketThread : public Thread +{ + ThreadQueueData queue; + ThreadSignalData signal; + protected: + /** Waits for an enqueue operation to complete + * You MUST hold the queue lock when you call this. + * It will be unlocked while you wait, and will be relocked + * before the function returns + */ + void WaitForQueue() + { + queue.Wait(); + } + /** Notifies parent by making the SignalFD ready to read + * No requirements on locking + */ + void NotifyParent(); + public: + SocketThread(InspIRCd* SI); + virtual ~SocketThread(); + /** Lock queue. + */ + void LockQueue() + { + queue.Lock(); + } + /** Unlock queue. + */ + void UnlockQueue() + { + queue.Unlock(); + } + /** Unlock queue and send wakeup to worker + */ + void UnlockQueueWakeup() + { + queue.Wakeup(); + queue.Unlock(); + } + virtual void SetExitFlag() + { + queue.Lock(); + Thread::SetExitFlag(); + queue.Wakeup(); + queue.Unlock(); + } - virtual void Create(Thread* thread_to_init) = 0; + /** + * Called in the context of the parent thread after a notification + * has passed through the socket + */ + virtual void OnNotify() = 0; }; #endif +