c++ - 安全线程终止

Question

工作线程处理数据库更新。当应用程序结束时，我想终止它但又不想让它离开数据库不一致。所以我在每个重要的数据库操作之前使用标志。过了一会儿，代码中充满了 if:

      void MyWorkerThread::RunMethod(){
      if (false == m_Manager->GetShutdownFlag()) {
         DoSomethingOnDB();
      }
      if (false == m_Manager->GetShutdownFlag()) {
         DoSomethingMoreOnDB();
      }
      ...

那么，有没有其他方法可以在不发送这么多 if 的情况下处理关机？

Answer 1

你最好通过一些同步对象(事件、锁、互斥锁、信号量、临界区等等......)等来同步线程，这样主线程就可以等待工作线程。

这里有一个竞争条件:如果在评估 if 条件之后和执行数据库操作之前立即引发关闭标志怎么办？

这是一个使用互斥量作为众所周知的同步原语的示例，但还有更好的方法。

主线程:

int main() {
    ... wait for signal to exit the app
    // the DB operations are running on another thread
    ...
    // assume that we start shutdown here
    // also assume that there is some global mutex g_mutex
    // following line blocks if mutex is locked in worker thread:
    std::lock_guard<std::mutex> lock(g_mutex); 
    Cleanup(); //  should also ensure that worker is stopped
}

工作线程:

void MyWorkerThread::RunMethod() 
{
  {
    std::lock_guard<std::mutex> lock(g_mutex); 
    DoSomethingOnDB();
  }
  // some other, non locked execution which doesn't prevent
  // main thread from exiting
  ...
  {
     std::lock_guard<std::mutex> lock(g_mutex);
     DoSomethingMoreOnDB();
  }
}

显然你不想重复所有的锁定，你应该包装它:

  void MyWorkerThread::RunMethod() 
  {
    Execute(DoSomethingOnDB);
    ...
    Execute(DoSomethingMoreOnDB);
  }

  void MyWorkerThread::Execute(DatabaseFn fn) 
  {
     std::lock_guard<std::mutex> lock(g_mutex); 
     fn();
  }