一个多线程的隐患

    今天查看项目代码的时候发现下面这段代码：

 

internal MoveTask Dequeue(string host)
        {
            if (host == null)
                throw new ArgumentNullException("host");

            lock (m_QueueLock)
            {
                if (!m_MoveTaskQueueDictionary.ContainsKey(host) || (m_MoveTaskQueueDictionary[host].Count == 0))
                    return null;
            }

            lock (m_QueueLock)
            {
                return m_MoveTaskQueueDictionary[host].Dequeue();
            }
        }

 

     当时这段代码实现时分了两个Lock为的就是对锁的晚占用，早释放。没想到却带来了多线程环境下可能发生的隐患：某个时刻某个Host的队列长度刚好等于1，这时线程A拥有了锁，那么在第一个Lock检查时该Host队列长度大于0，退出锁。此时另外一个线程B在A退出锁之后也进入了锁，也在第一个Lock检查时发现同一个Host的队列长度大于0，然后退出锁，准备进入第二个Lock。这时A线程由于B线程退出了锁，从而得以进入第二个Lock，对队列执行Dequeue方法后安全退出了锁。然后B线程进入锁执行Dequeue方法，这时队列为空，从而引发InvalidOperationException。

     解决办法很简单，合并这两个Lock为一个即可：

internal MoveTask Dequeue(string host)
        {
            if (host == null)
                throw new ArgumentNullException("host");

            lock (m_QueueLock)
            {
                if (!m_MoveTaskQueueDictionary.ContainsKey(host) || (m_MoveTaskQueueDictionary[host].Count == 0))
                    return null;

                return m_MoveTaskQueueDictionary[host].Dequeue();
            }
        }