mirror of
https://github.com/Xahau/xahaud.git
synced 2025-12-06 17:27:52 +00:00
87 lines
2.2 KiB
C++
87 lines
2.2 KiB
C++
#ifndef JOB_QUEUE__H
|
|
#define JOB_QUEUE__H
|
|
|
|
#include <map>
|
|
#include <set>
|
|
#include <vector>
|
|
|
|
#include <boost/thread/mutex.hpp>
|
|
#include <boost/thread/condition_variable.hpp>
|
|
#include <boost/function.hpp>
|
|
|
|
#include "types.h"
|
|
|
|
// Note that this queue should only be used for CPU-bound jobs
|
|
// It is primarily intended for signature checking
|
|
|
|
enum JobType
|
|
{ // must be in priority order, low to high
|
|
jtINVALID,
|
|
jtVALIDATION_ut, // A validation from an untrusted source
|
|
jtCLIENTOP_ut, // A client operation from a non-local/untrusted source
|
|
jtTRANSACTION, // A transaction received from the network
|
|
jtPROPOSAL_ut, // A proposal from an untrusted source
|
|
jtCLIENTOP_t, // A client operation from a trusted source
|
|
jtVALIDATION_t, // A validation from a trusted source
|
|
jtTRANSACTION_l, // A local transaction
|
|
jtPROPOSAL_t, // A proposal from a trusted source
|
|
jtADMIN, // An administrative operation
|
|
jtDEATH, // job of death, used internally
|
|
};
|
|
|
|
class Job
|
|
{
|
|
protected:
|
|
JobType mType;
|
|
uint64 mJobIndex;
|
|
boost::function<void(Job&)> mJob;
|
|
|
|
public:
|
|
Job() : mType(jtINVALID), mJobIndex(0) { ; }
|
|
Job(JobType type, uint64 index) : mType(type), mJobIndex(index) { ; }
|
|
|
|
Job(JobType type, uint64 index, const boost::function<void(Job&)>& job)
|
|
: mType(type), mJobIndex(index), mJob(job) { ; }
|
|
|
|
JobType getType() const { return mType; }
|
|
void doJob(void) { mJob(*this); }
|
|
|
|
bool operator<(const Job& j) const;
|
|
bool operator>(const Job& j) const;
|
|
bool operator<=(const Job& j) const;
|
|
bool operator>=(const Job& j) const;
|
|
|
|
static const char* toString(JobType);
|
|
};
|
|
|
|
class JobQueue
|
|
{
|
|
protected:
|
|
boost::mutex mJobLock;
|
|
boost::condition_variable mJobCond;
|
|
|
|
uint64 mLastJob;
|
|
std::set<Job> mJobSet;
|
|
std::map<JobType, int> mJobCounts;
|
|
int mThreadCount;
|
|
bool mShuttingDown;
|
|
|
|
|
|
void threadEntry(void);
|
|
|
|
public:
|
|
|
|
JobQueue() : mLastJob(0), mThreadCount(0), mShuttingDown(false) { ; }
|
|
|
|
void addJob(JobType type, const boost::function<void(Job&)>& job);
|
|
|
|
int getJobCount(JobType t); // Jobs at this priority
|
|
int getJobCountGE(JobType t); // All jobs at or greater than this priority
|
|
std::vector< std::pair<JobType, int> > getJobCounts();
|
|
|
|
void shutdown();
|
|
void setThreadCount(int c = 0);
|
|
};
|
|
|
|
#endif
|