mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-03 20:37:50 +08:00
76 lines
1.6 KiB
C++
76 lines
1.6 KiB
C++
/*
|
|
* StealRunnerQueue.h
|
|
*
|
|
* Copyright (c) hikyuu.org
|
|
*
|
|
* Created on: 2020-4-26
|
|
* Author: fasiondog
|
|
*/
|
|
|
|
#pragma once
|
|
#ifndef HIKYUU_UTILITIES_TASK_STEAL_RUNNER_QUEUE_H
|
|
#define HIKYUU_UTILITIES_TASK_STEAL_RUNNER_QUEUE_H
|
|
|
|
#include <deque>
|
|
#include <mutex>
|
|
#include "StealTaskBase.h"
|
|
|
|
namespace hku {
|
|
|
|
/**
|
|
* 工作线程任务队列
|
|
* @ingroup TaskGroup
|
|
*/
|
|
class StealRunnerQueue {
|
|
public:
|
|
/** 构造函数 */
|
|
StealRunnerQueue() = default;
|
|
~StealRunnerQueue() = default;
|
|
|
|
// 禁用赋值构造和赋值重载
|
|
StealRunnerQueue(const StealRunnerQueue& other) = delete;
|
|
StealRunnerQueue& operator=(const StealRunnerQueue& other) = delete;
|
|
|
|
/**
|
|
* 将任务插入队列头部
|
|
* @param task 任务
|
|
*/
|
|
void push_front(const StealTaskPtr& task);
|
|
|
|
/**
|
|
* 将任务插入队列尾部
|
|
* @param task 任务
|
|
*/
|
|
void push_back(const StealTaskPtr& task);
|
|
|
|
/** 队列是否为空 */
|
|
bool empty() const;
|
|
|
|
/** 队列当前大小 */
|
|
size_t size() const {
|
|
return m_queue.size();
|
|
}
|
|
|
|
/**
|
|
* 尝试从队列头部弹出一条数数据，失败返回空指针
|
|
* @param res 存储弹出的数据
|
|
* @return 如果原本队列为空指针
|
|
*/
|
|
StealTaskPtr try_pop();
|
|
|
|
/**
|
|
* 尝试从队列尾部偷取一条数据，失败返回空指针
|
|
* @param res 存储偷取的数据
|
|
* @return 返回任务指针或空指针
|
|
*/
|
|
StealTaskPtr try_steal();
|
|
|
|
private:
|
|
std::deque<StealTaskPtr> m_queue;
|
|
mutable std::mutex m_mutex;
|
|
};
|
|
|
|
} /* namespace hku */
|
|
|
|
#endif /* HIKYUU_UTILITIES_TASK_STEAL_RUNNER_QUEUE_H */
|