-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathJobScheduler.h
103 lines (85 loc) · 1.94 KB
/
JobScheduler.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#ifndef JOB_SCHEDULER_H
#define JOB_SCHEDULER_H
#include <cstdlib>
#include <pthread.h>
#include <iostream>
#define THREADS_NUM 8
#define QUEUE_SIZE 1000
static pthread_mutex_t jobQueueMutex = PTHREAD_MUTEX_INITIALIZER; // mutex for JobQueue
extern pthread_mutex_t* jobsCounterMutexes; // mutex for jobsCounter
static pthread_cond_t jobQueueFullCond = PTHREAD_COND_INITIALIZER, jobQueueEmptyCond = PTHREAD_COND_INITIALIZER; // condition variables for JobQueue
extern int64_t* jobsCounter;
class Job
{
private:
int jobId;
public:
Job()
{
jobId = 0;
};
virtual ~Job() {}
// This method should be implemented by subclasses.
virtual void run() = 0;
virtual int setJobId(int jobId)
{
this->jobId = jobId;
return jobId;
};
int getJobId()
{
return jobId;
};
};
class JobListNode
{
public:
Job *job;
JobListNode *nextNode;
JobListNode(Job *job);
~JobListNode();
};
class JobQueue
{
private:
JobListNode *firstNode, *lastNode;
int maxSize, currentSize;
public:
JobQueue(int maxSize);
~JobQueue();
int getCurrentSize();
bool isEmpty();
bool isFull();
int insertJobAtEnd(Job *job);
JobListNode *getNodeFromStart();
};
// Class JobScheduler
class JobScheduler
{
private:
JobQueue *jobQueue;
pthread_t *threadIds;
int nextJobId, threadsNum;
public:
// Initializes the JobScheduler with the number of open threads.
JobScheduler(int threadsNum, int jobsMaxNum);
// Free all resources that are allocated by JobScheduler
~JobScheduler();
// Waits Until executed all jobs in the queue.void Barrier();
// Add a job in the queue and returns a job id
int schedule(Job *job, int queryIndex);
int getThreadsNum();
};
void *threadWork(void *arg);
class ExitJob : Job
{
public:
ExitJob() {}
~ExitJob() {}
void run() {}
int setJobId(int jobId)
{
return Job::setJobId(-1);
}
};
#endif