blob: 617730ad013c6dcbe922ead082d0503597bbfb27 (
plain)
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
|
/*
* Copyright (C) 2008-2016 TrinityCore <http://www.trinitycore.org/>
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2 of the License, or (at your
* option) any later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License along
* with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "ZmqWorker.h"
#include "ZmqContext.h"
ZmqWorker::ZmqWorker(std::string const& taskUri, std::string const& resUri) :
_taskUri(taskUri), _resultsUri(resUri)
{
}
ZmqWorker::~ZmqWorker()
{
delete _taskQueue;
delete _results;
}
void ZmqWorker::HandleOpen()
{
_taskQueue = sIpcContext->CreateNewSocket(zmqpp::socket_type::pull);
_results = sIpcContext->CreateNewSocket(zmqpp::socket_type::push);
_taskQueue->connect(_taskUri);
_results->connect(_resultsUri);
_poller->add(*_taskQueue);
}
void ZmqWorker::HandleClose()
{
_taskQueue->close();
_results->close();
}
void ZmqWorker::Run()
{
while (!ProcessExit())
{
_poller->poll();
if (_poller->events(*_taskQueue) & zmqpp::poller::poll_in)
PerformWork();
}
}
void ZmqWorker::PerformWork()
{
int32 op1;
do
{
zmqpp::message msg;
_taskQueue->receive(msg);
Dispatch(msg);
_taskQueue->get(zmqpp::socket_option::events, op1);
} while (op1 & zmqpp::poller::poll_in);
}
|
