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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
/****************************************************************************
* Copyright (C) 2009-2015 EPAM Systems
*
* This file is part of Indigo toolkit.
*
* This file may be distributed and/or modified under the terms of the
* GNU General Public License version 3 as published by the Free Software
* Foundation and appearing in the file LICENSE.GPL included in the
* packaging of this file.
*
* This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
* WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
***************************************************************************/
#ifndef __os_thread_wrapper_h__
#define __os_thread_wrapper_h__
//
// Thread wrapper based on command dispatcher:
// 1. osCommandDispatcher setups a osCommand
// 2. osCommand is executed (in the another thread) and
// result is stored in the osCommandResult object.
// 3. osCommandDispatcher handles osCommandResult result in
// the main thread.
//
// There is two options to handle results:
// OsCommandDispatcher::HANDLING_ORDER_ANY -
//
// Session ID for each thread, created by dispatcher, is
// released before thread exit automatically via TL_RELEASE_SESSION_ID.
//
// Note: OsCommand and OsCommandResult objects are reusable,
// so they shouldn't have specific parameters in
// constructors if the same dispatcher object is
// used many time.
//
#include "base_c/defs.h"
#include "base_cpp/array.h"
#include "base_cpp/ptr_array.h"
#include "base_cpp/os_sync_wrapper.h"
#include "base_cpp/cyclic_array.h"
namespace indigo {
class OsCommandResult
{
public:
virtual ~OsCommandResult () {};
virtual void clear () {};
};
class OsCommand
{
public:
virtual ~OsCommand () {};
virtual void clear () {};
virtual void execute (OsCommandResult &result) = 0;
int unique_id;
};
class Exception;
class OsCommandDispatcher
{
public:
enum { HANDLING_ORDER_ANY, HANDLING_ORDER_SERIAL };
OsCommandDispatcher (int handling_order, bool same_session_IDs);
virtual ~OsCommandDispatcher () {};
void run ();
void run (int nthreads);
void terminate ();
void markToTerminate ();
void _threadFunc (void);
protected:
void _run (int nthreads);
// For overloading
virtual OsCommand* _allocateCommand () = 0;
virtual OsCommandResult* _allocateResult ();
virtual bool _setupCommand (OsCommand &command) = 0;
virtual void _handleResult (OsCommandResult &result) {}
// Callback function to initialize thread-local variables
// Custom Session ID can be set in this callback function.
virtual void _prepareThread (void) {}
// Callback function to cleanup thread-local variables
virtual void _cleanupThread (void) {}
private:
// Methods
void _startStandalone ();
void _onMsgNeedTask ();
void _onMsgHandleResult ();
void _onMsgHandleException (Exception *exception);
void _recvCommandAndResult (OsCommandResult * &result, OsCommand * &command);
void _mainLoop ();
OsCommand* _getVacantCommand ();
OsCommandResult* _getVacantResult ();
void _handleException (Exception *exception);
void _handleResultWithCheck (OsCommandResult *result);
void _wakeSuspended ();
private:
// Variables
PtrArray<OsCommand> _availableCommands;
PtrArray<OsCommandResult> _availableResults;
CyclicArray<OsCommandResult*> _storedResults;
Array<OsSemaphore*> _syspendedThreads;
Exception *_exception_to_forward;
OsMessageSystem _baseMessageSystem;
OsMessageSystem _privateMessageSystem;
int _last_command_index;
int _expected_command_index;
int _handling_order;
int _left_thread_count;
bool _need_to_terminate;
qword _session_id;
int _last_unique_command_id;
bool _same_session_IDs;
qword _parent_session_ID;
};
}
int osGetProcessorsCount (void);
#endif // __cmd_thread_h__
|
