C# 阻塞队列(Block Queue)实现

杀马特。学长 韩版系。学妹 提交于 2020-11-30 02:38:57

C#实现的阻塞队列,功能类似GO里的channel。

public class CBlockQueue<T>
    {
        private readonly Queue<T> queue = new Queue<T>();
        private readonly int maxSize;
        bool closing;

        /// <summary>
        /// CBlockQueue init
        /// </summary>
        /// <param >maxSize</param>
        public CBlockQueue(int maxSize)
        {
            this.maxSize = maxSize;
        }

        /// <summary>
        /// Enqueue
        /// </summary>
        /// <param >item</param>
        public void Enqueue(T item)
        {
            lock (queue)
            {
                while (queue.Count >= maxSize)
                {
                    Monitor.Wait(queue);
                }
                queue.Enqueue(item);
                if (queue.Count == 1)
                {
                    // wake up any blocked dequeue
                    Monitor.PulseAll(queue);
                }
            }
        }

        /// <summary>
        /// TryDequeue 
        /// </summary>
        /// <param >value</param>
        public bool TryDequeue(out T value)
        {
            lock (queue)
            {
                while (queue.Count == 0)
                {
                    if (closing)
                    {
                        value = default(T);
                        return false;
                    }
                    Monitor.Wait(queue);
                }
                value = queue.Dequeue();
                if (queue.Count == maxSize - 1)
                {
                    // wake up any blocked enqueue
                    Monitor.PulseAll(queue);
                }
                return true;
            }
        }

        /// <summary>
        /// Get Queue Count 
        /// </summary>
        /// <param ></param>
        public int Count()
        {
            lock (queue)
            {
                return queue.Count;
            }
        }

        /// <summary>
        /// Close Queue 
        /// </summary>
        /// <param ></param>
        public void Close()
        {
            lock (queue)
            {
                closing = true;
                Monitor.PulseAll(queue);
            }
        }

    }

  

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!