ssdfasd
34c75e7129
- Containers: String, Array, HashMap 容器实现及测试 - Memory: Allocator, LinearAllocator, PoolAllocator, ProxyAllocator, MemoryManager 实现及测试 - Threading: Mutex, SpinLock, ReadWriteLock, Thread, Task, TaskSystem 实现及测试 - 修复Windows平台兼容性: _aligned_malloc, std::hash特化 - 修复构建错误和测试用例问题
62 lines
1.3 KiB
C++
62 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include <atomic>
|
|
#include <functional>
|
|
#include <cstdint>
|
|
|
|
namespace XCEngine {
|
|
namespace Threading {
|
|
|
|
enum class TaskPriority : uint8_t {
|
|
Critical = 0,
|
|
High = 1,
|
|
Normal = 2,
|
|
Low = 3,
|
|
Idle = 4
|
|
};
|
|
|
|
enum class TaskStatus : uint8_t {
|
|
Pending,
|
|
Scheduled,
|
|
Running,
|
|
Completed,
|
|
Failed,
|
|
Canceled
|
|
};
|
|
|
|
class ITask {
|
|
public:
|
|
virtual ~ITask() = default;
|
|
|
|
virtual void Execute() = 0;
|
|
virtual void OnComplete() {}
|
|
virtual void OnCancel() {}
|
|
|
|
TaskPriority GetPriority() const { return m_priority; }
|
|
TaskStatus GetStatus() const { return m_status.load(); }
|
|
uint64_t GetId() const { return m_id; }
|
|
void SetId(uint64_t id) { m_id = id; }
|
|
|
|
void SetPriority(TaskPriority priority) { m_priority = priority; }
|
|
void SetStatus(TaskStatus status) { m_status = status; }
|
|
|
|
void AddRef() { m_refCount.fetch_add(1); }
|
|
void Release() {
|
|
if (m_refCount.fetch_sub(1) == 1) {
|
|
delete this;
|
|
}
|
|
}
|
|
|
|
protected:
|
|
ITask() = default;
|
|
explicit ITask(TaskPriority priority) : m_priority(priority) {}
|
|
|
|
TaskPriority m_priority = TaskPriority::Normal;
|
|
std::atomic<TaskStatus> m_status{TaskStatus::Pending};
|
|
uint64_t m_id = 0;
|
|
std::atomic<uint32_t> m_refCount{1};
|
|
};
|
|
|
|
} // namespace Threading
|
|
} // namespace XCEngine
|