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
|
#include "UnityPrefix.h"
#if ENABLE_CLUSTER_SYNC
#include "ClusterRendererModule.h"
#include "ClusterNode.h"
#include "ClusterNetwork.h"
#include "ClusterTransfer.h"
#include "Runtime/Interfaces/IClusterRenderer.h"
#include "Runtime/Utilities/Argv.h"
using namespace std;
bool ClusterRendererModule::IsInClusterTestMode;
ClusterRendererModule::ClusterRendererModule()
: m_Node(NULL)
, m_Context(NULL)
{
}
ClusterRendererModule::~ClusterRendererModule()
{
if(m_Node != NULL)
delete m_Node;
if(m_Context != NULL)
DestroyZMQContext(m_Context);
}
void ClusterRendererModule::ProcessServerArgs(vector<string> args)
{
m_Context = CreateZMQContext();
ClusterTransfer* sync = new ClusterTransfer();
MasterSocket* socket = new MasterSocket(m_Context, args[1].c_str(), args[2].c_str());
m_Node = new MasterNode(socket, sync, atoi(args[0].c_str()));
}
void ClusterRendererModule::ProcessClientArgs(vector<string> args)
{
m_Context = CreateZMQContext();
ClusterTransfer* sync = new ClusterTransfer();
SlaveSocket* socket = new SlaveSocket(m_Context, args[1].c_str(), args[2].c_str());
m_Node = new SlaveNode(socket, sync, atoi(args[0].c_str()));
}
// TODO: user memory manager NEW
void ClusterRendererModule::InitCluster()
{
AssertIf(m_Node != NULL);
vector<string> values;
if(HasARGV("server"))
{
ProcessServerArgs(GetValuesForARGV("server"));
}
else if(HasARGV("client"))
{
ProcessClientArgs(GetValuesForARGV("client"));
}
#ifdef DEBUG
IsInClusterTestMode = HasARGV("it");
// output some logs to indicate we are ready to go
// for IntegrationTest
LogString("ClusterReady");
#endif
}
void ClusterRendererModule::SynchronizeCluster()
{
// TODO check for NULL sockets
if(m_Node!= NULL)
m_Node->Sync();
}
bool ClusterRendererModule::IsMasterOfCluster()
{
return (m_Node != NULL) ? m_Node->IsMaster() : true;
}
void ClusterRendererModule::ShutdownCluster()
{
if(m_Node != NULL)
{
delete m_Node;
m_Node = NULL;
}
if(m_Context != NULL)
{
DestroyZMQContext(m_Context);
}
}
void InitializeClusterRendererModule ()
{
SetIClusterRenderer(UNITY_NEW_AS_ROOT(ClusterRendererModule, kMemClusterRenderer, "ClusterRendererInterface", ""));
}
void CleanupClusterRendererModule ()
{
ClusterRendererModule* module = reinterpret_cast<ClusterRendererModule*> (GetIClusterRenderer ());
UNITY_DELETE(module, kMemClusterRenderer);
SetIClusterRenderer (NULL);
}
#endif
|
