move some useful code to universal_fiber.hpp

Author: microcai

example/unbufcpy/unbufcp4/unbufcp4.cpp

@@ -88,81 +88,6 @@ struct IOBuffer
 	}
 };
 
-template<typename T>
-class FiberChannel
-{
-	void wake_up_one_pusher()
-	{
-		if (m_push_awaiting.empty())
-			return;
-		auto top_waiter = m_push_awaiting.front();
-		// wake up .
-		m_push_awaiting.pop_front();
-		PostQueuedCompletionStatus(m_iocp, 0, 
-			(ULONG_PTR)(iocp::overlapped_proc_func) & process_stack_full_overlapped_event
-			, top_waiter);
-	}
-	void wake_up_one_poper()
-	{
-		if (m_pop_awaiting.empty())
-			return;
-		auto top_waiter = m_pop_awaiting.front();
-		// wake up .
-		m_pop_awaiting.pop_front();
-		PostQueuedCompletionStatus(m_iocp, 0,
-			(ULONG_PTR)(iocp::overlapped_proc_func)&process_stack_full_overlapped_event
-			, top_waiter);
-	}
-
-public:
-	T pop()
-	{
-		if (m_queue.empty())
-		{
-			FiberOVERLAPPED ov;
-			// yield
-			m_pop_awaiting.push_back(&ov);
-			get_overlapped_result(ov);
-		}
-		T r = m_queue.front();
-		m_queue.pop_front();
-
-		if (m_queue.size() < m_max_pending)
-		{
-			wake_up_one_pusher();
-		}
-
-		return r;
-	}
-
-	void push(T t)
-	{
-		m_queue.push_back(t);
-		wake_up_one_poper();
-		if (m_queue.size() > m_max_pending)
-		{
-			// sleep until wakeup.
-			FiberOVERLAPPED ov;
-			// yield
-			m_push_awaiting.push_back(&ov);
-			get_overlapped_result(ov);
-		}
-	}
-
-	FiberChannel(HANDLE iocp, long max_pending)
-		: m_max_pending(max_pending)
-		, m_iocp(iocp)
-	{
-	}
-
-	long m_max_pending = 1;
-	std::deque<T> m_queue;
-
-	HANDLE m_iocp;
-	std::deque<FiberOVERLAPPED*> m_pop_awaiting;
-	std::deque<FiberOVERLAPPED*> m_push_awaiting;
-};
-
 static void write_loop(FiberChannel<WSABUF>& channel, FiberChannel<uint64_t>& channel2, HANDLE destFile)
 {
 	DWORD written;

example/unbufcpy/unbufcp5/unbufcp5.cpp

@@ -1,4 +1,4 @@
-/*++
+/*++
 THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
 ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED
 TO THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
@@ -100,81 +100,6 @@ struct IOBuffer
 	}
 };
 
-template<typename T>
-class FiberChannel
-{
-	void wake_up_one_pusher()
-	{
-		if (m_push_awaiting.empty())
-			return;
-		auto top_waiter = m_push_awaiting.front();
-		// wake up .
-		m_push_awaiting.pop_front();
-		PostQueuedCompletionStatus(m_iocp, 0, 
-			(ULONG_PTR)(iocp::overlapped_proc_func) & process_stack_full_overlapped_event
-			, top_waiter);
-	}
-	void wake_up_one_poper()
-	{
-		if (m_pop_awaiting.empty())
-			return;
-		auto top_waiter = m_pop_awaiting.front();
-		// wake up .
-		m_pop_awaiting.pop_front();
-		PostQueuedCompletionStatus(m_iocp, 0,
-			(ULONG_PTR)(iocp::overlapped_proc_func)&process_stack_full_overlapped_event
-			, top_waiter);
-	}
-
-public:
-	T pop()
-	{
-		if (m_queue.empty())
-		{
-			FiberOVERLAPPED ov;
-			// yield
-			m_pop_awaiting.push_back(&ov);
-			get_overlapped_result(ov);
-		}
-		T r = m_queue.front();
-		m_queue.pop_front();
-
-		if (m_queue.size() < m_max_pending)
-		{
-			wake_up_one_pusher();
-		}
-
-		return r;
-	}
-
-	void push(T t)
-	{
-		m_queue.push_back(t);
-		wake_up_one_poper();
-		if (m_queue.size() > m_max_pending)
-		{
-			// sleep until wakeup.
-			FiberOVERLAPPED ov;
-			// yield
-			m_push_awaiting.push_back(&ov);
-			get_overlapped_result(ov);
-		}
-	}
-
-	FiberChannel(HANDLE iocp, long max_pending)
-		: m_max_pending(max_pending)
-		, m_iocp(iocp)
-	{
-	}
-
-	long m_max_pending = 1;
-	std::deque<T> m_queue;
-
-	HANDLE m_iocp;
-	std::deque<FiberOVERLAPPED*> m_pop_awaiting;
-	std::deque<FiberOVERLAPPED*> m_push_awaiting;
-};
-
 struct FileCopyer
 {
 	uint64_t cur_pos = 0;

uasync/include/universal_fiber.hpp

@@ -19,6 +19,7 @@ using iocp::init_winsock_api_pointer;
 #include <cstddef>
 #include <thread>
 #include <utility>
+#include <deque>
 
 #if defined (USE_BOOST_CONTEXT)
 #define USE_FCONTEXT
@@ -702,4 +703,87 @@ inline void create_detached_coroutine(void (Class::*mem_func_ptr)(Args...), Clas
 // <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 
 
+// >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
+// some useful utilities
+// >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
+
+// FiberChannel is a message channel for use with fibers.
+template<typename T>
+class FiberChannel
+{
+	void wake_up_one_pusher()
+	{
+		if (m_push_awaiting.empty())
+			return;
+		auto top_waiter = m_push_awaiting.front();
+		// wake up .
+		m_push_awaiting.pop_front();
+		PostQueuedCompletionStatus(m_iocp, 0,
+			(ULONG_PTR)(iocp::overlapped_proc_func)&process_stack_full_overlapped_event
+			, top_waiter);
+	}
+	void wake_up_one_poper()
+	{
+		if (m_pop_awaiting.empty())
+			return;
+		auto top_waiter = m_pop_awaiting.front();
+		// wake up .
+		m_pop_awaiting.pop_front();
+		PostQueuedCompletionStatus(m_iocp, 0,
+			(ULONG_PTR)(iocp::overlapped_proc_func)&process_stack_full_overlapped_event
+			, top_waiter);
+	}
+
+public:
+	FiberChannel(const FiberChannel&) = delete;
+	FiberChannel(FiberChannel&&) = delete;
+
+	FiberChannel(HANDLE iocp, long max_pending)
+		: m_max_pending(max_pending)
+		, m_iocp(iocp)
+	{
+	}
+
+	T pop()
+	{
+		if (m_queue.empty())
+		{
+			FiberOVERLAPPED ov;
+			// yield
+			m_pop_awaiting.push_back(&ov);
+			get_overlapped_result(ov);
+		}
+		T r = m_queue.front();
+		m_queue.pop_front();
+
+		if (m_queue.size() < m_max_pending)
+		{
+			wake_up_one_pusher();
+		}
+
+		return r;
+	}
+
+	void push(T t)
+	{
+		if (m_queue.size() > m_max_pending)
+		{
+			// sleep until wakeup.
+			FiberOVERLAPPED ov;
+			// yield
+			m_push_awaiting.push_back(&ov);
+			get_overlapped_result(ov);
+		}
+		m_queue.push_back(t);
+		wake_up_one_poper();
+	}
+
+	long m_max_pending = 1;
+	std::deque<T> m_queue;
+
+	HANDLE m_iocp;
+	std::deque<FiberOVERLAPPED*> m_pop_awaiting;
+	std::deque<FiberOVERLAPPED*> m_push_awaiting;
+};
+
 #endif // ___UNIVERSAL_FIBER__H___