我必须创建一个线程安全的工作队列,它可以在不同的线程中弹出工作,并在工作线程上处理它。这项工作可能非常通用,所以我在考虑使用带捕获的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;
}
这段代码是否存在本质上无法解决的问题,还是编译器会对其进行优化?还是将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;
}
我得到了所需的输出:
1
Destroy 1
2
Destory 2
答案 0 :(得分:1)
std::function
将创建函数指针,lambda或其引用的任何内容的私有副本。通常,此副本从std::function
对象引用,以便稍后进一步复制。
以这种方式使用std::function
对象没有什么特别慢的。但是,您应该考虑将std::vector
替换为std::deque
。