我必须创建一个线程安全的工作队列,它可以在不同的线程中弹出工作,并在工作线程上处理它.这项工作可能非常通用,所以我在考虑使用带捕获的lambdas作为允许这种情况的好方法.我有以下代码作为启动器:
#include <iostream>
#include <vector>
#include <functional>
typedef std::function<void()> Task;
typedef std::vector<Task> TaskQueue;
class Queue
{
public:
void flush() {
for (auto it : m_queue) {
it();
}
}
// will add thread safe locks later.
void queue(Task task) {
m_queue.push_back(task);
}
private:
TaskQueue m_queue;
};
Queue q;
class WorkMaker
{
public:
WorkMaker(int inA) : m_a(inA) {}
void MakeWork() {
q.queue([&]{
std::cout << this->m_a << std::endl;
});
}
private:
int m_a;
};
int main()
{
WorkMaker w1(1);
WorkMaker w2(2);
w1.MakeWork();
w2.MakeWork();
q.flush();
return 0;
}
Run Code Online (Sandbox Code Playgroud)
这段代码是否存在固有的无法解决的问题,还是编译器会对其进行优化?还是将lambda传递给std::function参数,通过值复制lambda或只是指向它的指针?
编辑:
我想我可以通过使用shared_ptr并将它们传递给lambda来解决内存所有权问题.请考虑以下修改:
typedef std::function<void()> Task;
typedef std::deque<Task> TaskQueue;
class Queue
{
public:
void flush() {
while (!m_queue.empty()) {
auto it = m_queue.front();
m_queue.pop_front();
it();
}
}
// will add thread safe locks later.
void queue(Task task) {
m_queue.push_back(task);
}
private:
TaskQueue m_queue;
};
Queue q;
class WorkMaker : public std::enable_shared_from_this<WorkMaker>
{
public:
WorkMaker(int inA) : m_a(inA) {}
~WorkMaker() { std::cout << "Destroy " << m_a << std::endl; }
void MakeWork() {
std::shared_ptr<WorkMaker> self = shared_from_this();
q.queue([self]{
std::cout << self->m_a << std::endl;
});
}
int m_a;
};
int main()
{
{
auto w1 = std::make_shared<WorkMaker>(1);
auto w2 = std::make_shared<WorkMaker>(2);
w1->MakeWork();
w2->MakeWork();
}
q.flush();
return 0;
}
Run Code Online (Sandbox Code Playgroud)
我得到了所需的输出:
1
Destroy 1
2
Destory 2
Run Code Online (Sandbox Code Playgroud)
这std::function将创建函数指针、lambda 或它所引用的任何内容的私有副本。通常,从对象引用此副本std::function,以便以后避免进一步复制。
以这种方式使用对象并没有什么特别慢的地方std::function。但是,您可能应该考虑将 替换std::vector为std::deque.