-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathWorker.h
More file actions
109 lines (107 loc) · 2.22 KB
/
Worker.h
File metadata and controls
109 lines (107 loc) · 2.22 KB
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
104
105
106
107
108
#pragma once
#include "stdafx.h"
class Worker
{
public:
Worker()
:enabled(true),isworking(false)
{
lasttime = GetTickCount64();
thread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)thread_fn, (LPVOID)this, 0, NULL);
InitializeCriticalSection(&cs);
awaker = ::CreateEvent(NULL,FALSE,//ñ àâòîñáðîñîì
FALSE,NULL);
finishtask = ::CreateEvent(NULL,FALSE,//ñ àâòîñáðîñîì
TRUE,NULL);
}
~Worker()
{
EnterCriticalSection(&cs);
enabled = false;
if (isworking)
printf("VAXVAXVAX\n");
isworking = false;
LeaveCriticalSection(&cs);
SetEvent(awaker);
::WaitForSingleObject(thread,INFINITE);
DeleteCriticalSection(&cs);
CloseHandle(awaker);
CloseHandle(finishtask);
}
DWORD GetLastFinishTime()
{
DWORD lstm;
EnterCriticalSection(&cs);
lstm = lasttime;
LeaveCriticalSection(&cs);
return lstm;
}
bool IsWorking()
{
bool ret;
EnterCriticalSection(&cs);
ret = isworking;
LeaveCriticalSection(&cs);
return ret;
}
int GiveTask(fn_type task)
{
if (!enabled)
return -1;
EnterCriticalSection(&cs);
if (!isworking)
{
task_field = task;
isworking = true;
SetEvent(awaker);
}
else
{
LeaveCriticalSection(&cs);
::WaitForSingleObject(finishtask,INFINITE);
EnterCriticalSection(&cs);
task_field = task;
isworking = true;
SetEvent(awaker);
}
LeaveCriticalSection(&cs);
return 0;
}
private:
fn_type task_field;
bool isworking;
DWORD lasttime;
CRITICAL_SECTION cs;
HANDLE awaker;//ñîáûòèå
HANDLE finishtask; //ñîáûòèå îêîí÷àíèÿ ðàáîòû
HANDLE thread;
bool enabled;
static void thread_fn(LPVOID param)
{
Worker* ths = (Worker*) param;
while (ths->enabled)
{
::WaitForSingleObject(ths->awaker,INFINITE);
EnterCriticalSection(&ths->cs);
if (!ths->enabled)
{
LeaveCriticalSection(&ths->cs);
return;
}
ths->isworking = true;
ths->lasttime = GetTickCount64();
ResetEvent(ths->finishtask);
LeaveCriticalSection(&ths->cs);
if (ths->task_field != NULL)
{
ths->task_field();
}
EnterCriticalSection(&ths->cs);
ths->lasttime = GetTickCount64();
ths->isworking = false;
SetEvent(ths->finishtask);
LeaveCriticalSection(&ths->cs);
}
}
};
typedef std::shared_ptr<Worker> worker_ptr;