我试图理解新C ++ 11标准中的基本多线程机制。我能想到的最基本的例子如下:
这个例子也被用在许多关于多线程的学校书籍中,关于通信过程的一切都很好。但是,在停止使用者线程时我遇到了问题。
我希望消费者能够运行直到它获得明确的停止信号(在大多数情况下,这意味着我等待生产者完成所以我可以在程序结束之前停止消费者)。不幸的是,C ++ 11线程缺少一种中断机制(例如我从Java中的多线程中知道)。因此,我必须使用像isRunning
这样的标志来表示我希望线程停止。
现在的主要问题是:在我停止生产者线程之后,队列为空并且消费者正在等待condition_variable
以在再次填充队列时获得信号。所以我需要在退出之前通过调用notify_all()
来唤醒线程。
我找到了一个有效的解决方案,但似乎有些混乱。 下面列出了示例代码(很抱歉,但不知何故,我无法为“最小”最小示例减少代码大小):
队列类:
class Queue{
public:
Queue() : m_isProgramStopped{ false } { }
void push(int i){
std::unique_lock<std::mutex> lock(m_mtx);
m_q.push(i);
m_cond.notify_one();
}
int pop(){
std::unique_lock<std::mutex> lock(m_mtx);
m_cond.wait(lock, [&](){ return !m_q.empty() || m_isProgramStopped; });
if (m_isProgramStopped){
throw std::exception("Program stopped!");
}
int x = m_q.front();
m_q.pop();
std::cout << "Thread " << std::this_thread::get_id() << " popped " << x << "." << std::endl;
return x;
}
void stop(){
m_isProgramStopped = true;
m_cond.notify_all();
}
private:
std::queue<int> m_q;
std::mutex m_mtx;
std::condition_variable m_cond;
bool m_isProgramStopped;
};
制片人:
class Producer{
public:
Producer(Queue & q) : m_q{ q }, m_counter{ 1 } { }
void produce(){
for (int i = 0; i < 5; i++){
m_q.push(m_counter++);
std::this_thread::sleep_for(std::chrono::milliseconds{ 500 });
}
}
void execute(){
m_t = std::thread(&Producer::produce, this);
}
void join(){
m_t.join();
}
private:
Queue & m_q;
std::thread m_t;
unsigned int m_counter;
};
消费者:
class Consumer{
public:
Consumer(Queue & q) : m_q{ q }, m_takeCounter{ 0 }, m_isRunning{ true }
{ }
~Consumer(){
std::cout << "KILL CONSUMER! - TOOK: " << m_takeCounter << "." << std::endl;
}
void consume(){
while (m_isRunning){
try{
m_q.pop();
m_takeCounter++;
}
catch (std::exception e){
std::cout << "Program was stopped while waiting." << std::endl;
}
}
}
void execute(){
m_t = std::thread(&Consumer::consume, this);
}
void join(){
m_t.join();
}
void stop(){
m_isRunning = false;
}
private:
Queue & m_q;
std::thread m_t;
unsigned int m_takeCounter;
bool m_isRunning;
};
最后是main()
:
int main(void){
Queue q;
Consumer cons{ q };
Producer prod{ q };
cons.execute();
prod.execute();
prod.join();
cons.stop();
q.stop();
cons.join();
std::cout << "END" << std::endl;
return EXIT_SUCCESS;
}
这是正确的方式来结束正在等待条件变量的线程还是有更好的方法?目前,队列需要知道程序是否已停止(我认为这会破坏组件的松散耦合),我需要明确调用队列上的stop()
,这似乎是正确的。
另外,如果队列为空,条件变量应该只用作信号,现在代表另一个条件 - 如果程序已经结束。如果我没有弄错,每次线程等待条件变量发生某些事件时,它还必须检查线程是否必须在继续执行之前停止(这似乎也是错误的)。
我是否有这些问题,因为我的整个设计都有问题,或者我错过了一些可以用来以干净的方式退出线程的机制?
答案 0 :(得分:6)
不,你的设计没有任何问题,这是解决这类问题的正常方法。
对于条件变量附加多个条件(例如,队列或程序停止的任何内容),这是完全有效的。关键是在wait
返回时检查条件中的位。
而不是在Queue
中有一个标志来指示程序正在停止,你应该将标志视为“我可以接受”。这是一个更好的整体范例,在多线程环境中效果更好。
此外,如果有人调用了pop
并且调用了stop
,那么您可以将bool try_pop(int &value)
替换为true
,而不是让false
抛出异常。返回了值,否则为bool is_stopped() const
。这样调用者可以检查失败以查看队列是否已停止(添加{{1}}方法)。虽然异常处理在这里起作用,但它有点沉重,并不是多线程程序中的特殊情况。
答案 1 :(得分:1)
wait
。控制权返回给线程,并且可以检查stop
。根据该值,可以wait
使用更多项目或完成执行。使用c ++对多线程的一个很好的介绍是C++11 Concurrency。