83
|
1 package christie.topology.manager;
|
|
2
|
|
3
|
|
4 import christie.annotation.Peek;
|
|
5 import christie.annotation.Take;
|
|
6 import christie.codegear.CodeGear;
|
|
7 import christie.codegear.CodeGearManager;
|
|
8 import christie.topology.HostMessage;
|
|
9
|
|
10 import java.util.HashMap;
|
|
11
|
|
12 public class CreateTreeTopology extends CodeGear{
|
|
13
|
|
14 @Take
|
|
15 HostMessage newHost;
|
|
16
|
|
17 @Take
|
|
18 int hostCount;
|
|
19
|
|
20 @Peek
|
|
21 HashMap<String, HostMessage> nameTable;
|
|
22
|
|
23 @Take
|
|
24 String MD5;
|
|
25
|
|
26 @Peek
|
|
27 HashMap<String, String> absCookieTable;
|
|
28
|
|
29 @Peek
|
|
30 ParentManager parentManager;
|
|
31
|
|
32 public CreateTreeTopology(){
|
|
33 }
|
|
34
|
|
35 @Override
|
|
36 protected void run(CodeGearManager cgm) {
|
95
|
37
|
83
|
38
|
|
39 String nodeName = "node" + hostCount;
|
94
|
40 String newHostName = newHost.getHostName();
|
|
41 int newHostPort = newHost.getPort();
|
|
42
|
95
|
43
|
94
|
44 cgm.createRemoteDGM(nodeName, newHostName, newHostPort);
|
83
|
45 getDGM(nodeName).put("nodeName", nodeName);
|
|
46 getDGM(nodeName).put("cookie", MD5);
|
|
47
|
|
48 absCookieTable.put(MD5, nodeName);
|
|
49 getLocalDGM().put("hostCount", hostCount + 1);
|
|
50
|
94
|
51 newHost.setAlive(true);
|
83
|
52 nameTable.put(nodeName, newHost);
|
|
53 parentManager.register(nodeName);
|
|
54
|
|
55 if (hostCount == 0) {
|
|
56 // どこにも繋がれるところがないので, ルートのとき.
|
|
57 // ルートなので, connectNodeNumもreverseCountも0でいい.
|
|
58 getDGM(nodeName).put("connectNodeNum", 0);
|
|
59 getDGM(nodeName).put("reverseCount", 0);
|
|
60
|
|
61 } else {
|
|
62 // 親のみにつながればいいので1
|
|
63 getDGM(nodeName).put("connectNodeNum", 1);
|
|
64 // put parent information own
|
|
65 String parentNodeName = parentManager.getMyParent();
|
94
|
66 HostMessage parentHost = nameTable.get(parentNodeName);
|
83
|
67
|
94
|
68 // 相手からhostNameとportはもらっているので, nodeの情報だけ与えれば良い.
|
114
|
69 parentHost.setNodeInfo(nodeName, "parent", parentNodeName);
|
94
|
70 getLocalDGM().put("nodeInfo", parentHost);
|
83
|
71 cgm.setup(new RecordTopology());
|
|
72
|
94
|
73 // newChildHost, newHostも同じ
|
114
|
74 newHost.setNodeInfo(parentNodeName, "child" + parentManager.getMyNumber(), nodeName);
|
94
|
75 getLocalDGM().put("nodeInfo", newHost);
|
83
|
76 cgm.setup(new RecordTopology());
|
|
77 }
|
85
|
78
|
|
79 getDGM(nodeName).put("start", "start");
|
100
|
80 getLocalDGM().put("startTime", System.currentTimeMillis());
|
85
|
81
|
92
|
82 cgm.setup(new CreateTreeTopology());
|
83
|
83 }
|
|
84 } |