first commit

inc/hgl/thread/DataPost.h

@@ -0,0 +1,156 @@
+#ifndef HGL_THREAD_DATA_POST_INCLUDE
+#define HGL_THREAD_DATA_POST_INCLUDE
+
+#include<hgl/thread/ThreadMutex.h>
+#include<hgl/thread/Semaphore.h>
+#include<hgl/type/Stack.h>
+
+namespace hgl
+{
+    /**
+     * 多线程数据投递模板<br>
+     * 需要注意每个任务都会重新由空闲线程来进行获取，所以请将连续的任务合并为一个任务，而不是一次大量投递。
+     */
+    template<typename T> class DataPost
+    {
+    protected:
+
+        List<T *> data_list[2];
+
+        int post_index,recv_index;
+        int recv_offset;
+
+        ThreadMutex post_lock,recv_lock;
+
+    protected:
+
+        void _Swap()
+        {
+            if(recv_index){recv_index=0;post_index=1;}
+                      else{recv_index=1;post_index=0;}
+
+            recv_offset=0;
+        }
+
+    public:
+
+        DataPost()
+        {
+            post_index=0;
+            recv_index=1;
+
+            recv_offset=0;
+        }
+
+        virtual ~DataPost()=default;
+
+        /**
+         * 投递一个数据
+         */
+        void Post(T *obj)
+        {
+            if(!obj)return;
+
+            post_lock.Lock();
+                data_list[post_index].Add(obj);
+            post_lock.Unlock();
+        }
+
+        /**
+         * 投递一批数据
+         */
+        void Post(T **obj,int count)
+        {
+            if(!obj)return;
+
+            post_lock.Lock();
+                data_list[post_index].Add(obj,count);
+            post_lock.Unlock();
+        }
+
+    public:
+
+        /**
+         * 获取一个数据
+         */
+        T *Receive()
+        {
+            T *obj=nullptr;
+
+            recv_lock.Lock();
+                int count=data_list[recv_index].GetCount();
+
+                if(recv_offset<count)
+                {
+                    obj=*(data_list[recv_index].GetData()+recv_offset);
+
+                    ++recv_offset;
+                }
+                else
+                {
+                    data_list[recv_index].ClearData();      //清空接收区的数据
+
+                    post_lock.Lock();
+                    _Swap();
+                    post_lock.Unlock();
+
+                    count=data_list[recv_index].GetCount();
+                    if(count>0)                             //如果换出来的区有数据
+                    {
+                        obj=*(data_list[recv_index].GetData()+recv_offset);
+
+                        ++recv_offset;
+                    }
+                }
+            recv_lock.Unlock();
+
+            return obj;
+        }
+    };//template<typename T> class DataPost
+
+    /**
+    * 信号自动交换数据访问模板
+    */
+    template<typename T> class SemDataPost:public DataPost<T>
+    {
+        Semaphore sem;
+
+    public:
+
+        using DataPost<T>::DataPost;
+        ~SemDataPost()=default;
+
+        /**
+        * 释放接收信号
+        * @param count 信号个数
+        */
+        void PostSem(int count=1)
+        {
+            sem.Post(count);
+        }
+
+        /**
+        * 等待获取一个信号并获取数据
+        * @param time_out 等待时长
+        */
+        T *WaitSemReceive(const double time_out=5)
+        {
+            if(!sem.Acquire(time_out))
+                return(nullptr);
+
+            return this->Receive();
+        }
+
+        /**
+        * 尝试获取一个信号并获取数据
+        */
+        T *TrySemReceive()
+        {
+            if(!sem.TryAcquire())
+                return(nullptr);
+
+            return this->Receive();
+        }
+    };//template<typename T> class SemDataPost:public DataPost<T>
+}//namespace hgl
+#endif//HGL_THREAD_DATA_POST_INCLUDE