Question

我使用std :: cout在控制台上打印日志。由于该程序是多线程的，因此如果在cout之后使用多个“ <<”运算符，则打印结果将会混乱。

例如，如果一个线程执行cout<< "A" << "B" << endl;，而另一个线程可能在A和B之间执行cout << "C";。结果将是“ ACB”。

因此，我将编写一个新类来继承ostream（实际上是basic_ostream<char, char_traits<char>>）并在初始化cout时添加一个锁，因此打印输出应遵循正确的顺序

Answer 1

一个选择是创建一个类，该类保存对流的引用，但在其整个生命周期内都保留一个锁。这是一个简单的示例：

#include <iostream>
#include <mutex>

struct LockedOstream {
    std::lock_guard<std::mutex> lg_;
    std::ostream& os_;
    LockedOstream(std::mutex& m, std::ostream& os)
      : lg_{m}
      , os_{os}
    { }
    std::ostream& stream() const { return os_; }
};

int main()
{
    std::mutex m;
    LockedOstream(m, std::cout).stream() << "foo " << "bar\n";
    //                        ^ locked now                   ^ unlocked now
}

只要构成单个输出“单位”的所有打印都在同一条语句中进行，此方法就起作用。

编辑：实际上，继承版本比我最初预期的要好很多。

#include <iostream>
#include <mutex>

class LockedOstream : public std::ostream {
    static std::mutex& getCoutMutex()
        // use a Meyers' singleton for the cout mutex to keep this header-only
    {
        static std::mutex m;
        return m;
    }

    std::lock_guard<std::mutex> lg_;

  public:
    // Generic constructor
    // You need to pass the right mutex to match the stream you want
    LockedOstream(std::mutex& m, std::ostream& os)
      : std::ostream(os.rdbuf())
      , lg_{m}
    { }

    // Cout specific constructor
    // Uses a mutex singleton specific for cout
    LockedOstream()
      : LockedOstream(getCoutMutex(), std::cout)
    { }
};

int main()
{
    LockedOstream() << "foo " << "bar\n";
    //            ^ locked now          ^ unlocked now
}

顺便说一句：

（尽管后者有时会引起争议，但至少了解并做出明智的选择是一个好主意）。

Answer 2

您可以定义自己的函数

template<typename... Ts>
void locked_print(std::ostream& stream, Ts&&... ts)
{
    static std::mutex mtx;
    std::lock_guard<std::mutex> guard(mtx);
    (stream << ... << std::forward<Ts>(ts));
}

如果要确保它是排他性的，可以将其命名为locked_print(std::cout, 1, 2, "bar");

Answer 3

由于outstream << x1 << x2 << ...是多个函数调用，因此除了锁定所有内容直到破坏同一流出之前，您无法在内部进行此操作。您可以在调用约束时强制使用约束：

{
  std::lock_guard<std::mutex> guard(global_mutex);
  // your print here
}

如何创建一个新类以继承ostream并将其用作cout但带锁

3 个答案: