From 250e30d73f09e9da2b5a81d0fbae63744ae12a73 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Tue, 2 Apr 2019 08:47:15 +0800
Subject: *misc

---
 .../libs/asura-lib-utils/threading/semaphore.cpp   | 88 ++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 source/libs/asura-lib-utils/threading/semaphore.cpp

(limited to 'source/libs/asura-lib-utils/threading/semaphore.cpp')

diff --git a/source/libs/asura-lib-utils/threading/semaphore.cpp b/source/libs/asura-lib-utils/threading/semaphore.cpp
new file mode 100644
index 0000000..d59ec78
--- /dev/null
+++ b/source/libs/asura-lib-utils/threading/semaphore.cpp
@@ -0,0 +1,88 @@
+#include "../exceptions/exception.h"
+#include "../type.h"
+
+#include "semaphore.h"
+
+namespace AsuraEngine
+{
+	namespace Threading
+	{
+
+#define try_create_semaphore(impl) \
+		if (!mImpl)												\
+		{																	\
+			try															\
+			{																\
+				mImpl = new impl(init_count);	\
+			}																\
+			catch (Exception& e)						\
+			{																\
+				mImpl = nullptr;							\
+			}																\
+		}
+
+		Semaphore::Semaphore(unsigned int init_count)
+			: mImpl(nullptr)
+		{
+#ifdef ASURA_THREAD_WIN32 
+			try_create_semaphore(SemaphoreWin32);
+#endif
+			ASSERT(mImpl);
+		}
+
+		Semaphore::~Semaphore()
+		{
+			if (mImpl) delete mImpl;
+		}
+
+		void Semaphore::Signal()
+		{
+			ASSERT(mImpl);
+			mImpl->Signal();
+		}
+
+		void Semaphore::Wait(int timeout)
+		{
+			ASSERT(mImpl);
+			mImpl->Wait(timeout);
+		}
+
+#if ASURA_THREAD_WIN32 
+
+		SemaphoreWin32::SemaphoreWin32(unsigned int init_value)
+			: SemaphoreImpl(init_value)
+		{
+			mSem = CreateSemaphore(NULL, init_value, UINT_MAX, NULL);
+			if (!mSem)
+				throw Exception("Cant use win32 semaphore.");
+		}
+
+		SemaphoreWin32::~SemaphoreWin32()
+		{
+			CloseHandle(mSem);
+		}
+
+		void SemaphoreWin32::Signal()
+		{
+			InterlockedIncrement(&mCount);
+			if (ReleaseSemaphore(mSem, 1, NULL) == FALSE)
+				InterlockedDecrement(&mCount);
+		}
+
+		bool SemaphoreWin32::Wait(int timeout)
+		{
+			int result;
+			result = WaitForSingleObject(mSem, timeout < 0 ? INFINITE : timeout);
+			if (result == WAIT_OBJECT_0)
+			{
+				InterlockedDecrement(&mCount);
+				return true;
+			}
+			else
+				return false;
+		}
+
+#endif // ASURA_THREAD_WIN32 
+
+	}
+}
\ No newline at end of file
-- 
cgit v1.1-26-g67d0