多线程生产者-消费者队列优雅停机实现合理性咨询

需求与伪代码 #

需求为给经典多线程生产者-消费者队列添加优雅停机机制，确保consumeItems线程处理完队列中所有元素后再停止。提供的伪代码如下：

生产者线程 #

thread addItems() {
while (!shutdown) {
    item = ItemFactory.produceItem();
    lock.acquire();
    try {
        while (self.queue.length() == MAX_QUEUE_SIZE) {
            conditionHasSpace.wait()
        }
        if (!shutdown) {
            self.queue.insert(order); // 注：此处疑似笔误，应为item
        }
        conditionHasItems.notify();
    }
    catch {
        handleException();
    }
    finally {
        lock.release()
    }
}
}

消费者线程 #

thread consumeItems() {
while (true) {
    lock.acquire();
    try { 
        while (self.queue.length() == 0) {
            if (shutdown) {
                exit;
            }
            conditionHasItems.wait();
        }
        self.queue.pop();
        conditionHasSpace.notify();
    }
    catch {
        handleException();
    }
    finally {
        lock.release();
    }
}
}

实现问题与风险分析 #

1. 不合理之处 #

• 生产者无效生产：生产者先调用ItemFactory.produceItem()生产元素，再获取锁判断shutdown状态。若生产完成后、获取锁前shutdown被置为true，该元素会直接丢弃，造成资源浪费。
• 消费者停机条件不严谨：消费者仅在队列空时检查shutdown并退出，但存在阻塞无法唤醒的场景，无法保证处理完所有元素后停机。

2. 僵死场景 #

最典型的僵死情况：当队列处于空状态时，消费者阻塞在conditionHasItems.wait()上。此时外部触发shutdown，生产者线程因shutdown为true已退出循环，不会再调用conditionHasItems.notify()唤醒消费者，导致消费者永久阻塞，无法退出。

3. 改进建议 #

调整生产者逻辑

将生产元素的操作移到锁内，避免无效生产，同时在等待空间时也检查shutdown状态：

thread addItems() {
while (!shutdown) {
    lock.acquire();
    try {
        // 队列满且未停机时等待
        while (self.queue.length() == MAX_QUEUE_SIZE && !shutdown) {
            conditionHasSpace.wait();
        }
        // 已停机则退出循环
        if (shutdown) {
            break;
        }
        item = ItemFactory.produceItem();
        self.queue.insert(item);
        conditionHasItems.notify();
    }
    catch {
        handleException();
    }
    finally {
        lock.release();
    }
}
}

修正消费者停机逻辑

外部触发shutdown时，必须主动唤醒所有等待的消费者；消费者在被唤醒后需重新检查状态，确保处理完队列剩余元素后再退出：

thread consumeItems() {
while (true) {
    lock.acquire();
    try { 
        while (self.queue.length() == 0) {
            if (shutdown) {
                // 队列空且已停机，退出线程
                return;
            }
            conditionHasItems.wait();
        }
        // 处理队列元素
        self.queue.pop();
        conditionHasSpace.notify();
    }
    catch {
        handleException();
    }
    finally {
        lock.release();
    }
}
}

外部触发shutdown的正确步骤：

1. 设置shutdown = true
2. 调用conditionHasItems.notifyAll()（唤醒所有等待的消费者）
3. 调用conditionHasSpace.notifyAll()（唤醒所有等待的生产者）

这样可确保生产者停止生产，消费者被唤醒后处理完队列中所有剩余元素，最终正常退出。

内容的提问来源于stack exchange，提问作者RutgerBlume