base for mastership term implementation
Change-Id: Ie354096ad8835536839e3c1504e19a2cbff866c1
Showing
7 changed files
with
106 additions
and
2 deletions
... | @@ -56,6 +56,13 @@ public interface MastershipService { | ... | @@ -56,6 +56,13 @@ public interface MastershipService { |
56 | Set<DeviceId> getDevicesOf(NodeId nodeId); | 56 | Set<DeviceId> getDevicesOf(NodeId nodeId); |
57 | 57 | ||
58 | /** | 58 | /** |
59 | + * Returns the mastership term service for getting term information. | ||
60 | + * | ||
61 | + * @return the MastershipTermService for this mastership manager | ||
62 | + */ | ||
63 | + MastershipTermService requestTermService(); | ||
64 | + | ||
65 | + /** | ||
59 | * Adds the specified mastership change listener. | 66 | * Adds the specified mastership change listener. |
60 | * | 67 | * |
61 | * @param listener the mastership listener | 68 | * @param listener the mastership listener | ... | ... |
... | @@ -55,4 +55,13 @@ public interface MastershipStore extends Store<MastershipEvent, MastershipStoreD | ... | @@ -55,4 +55,13 @@ public interface MastershipStore extends Store<MastershipEvent, MastershipStoreD |
55 | * @return a mastership event | 55 | * @return a mastership event |
56 | */ | 56 | */ |
57 | MastershipEvent setMaster(NodeId nodeId, DeviceId deviceId); | 57 | MastershipEvent setMaster(NodeId nodeId, DeviceId deviceId); |
58 | + | ||
59 | + /** | ||
60 | + * Returns the current master and number of past mastership hand-offs | ||
61 | + * (terms) for a device. | ||
62 | + * | ||
63 | + * @param deviceId the device identifier | ||
64 | + * @return the current master's ID and the term value for device, or null | ||
65 | + */ | ||
66 | + MastershipTerm getTermFor(DeviceId deviceId); | ||
58 | } | 67 | } | ... | ... |
1 | package org.onlab.onos.cluster; | 1 | package org.onlab.onos.cluster; |
2 | 2 | ||
3 | -public class MastershipTerm { | 3 | +import java.util.Objects; |
4 | - private final NodeId master = null; | 4 | + |
5 | +public final class MastershipTerm { | ||
6 | + | ||
7 | + private final NodeId master; | ||
5 | private int termNumber; | 8 | private int termNumber; |
9 | + | ||
10 | + private MastershipTerm(NodeId master, int term) { | ||
11 | + this.master = master; | ||
12 | + this.termNumber = term; | ||
13 | + } | ||
14 | + | ||
15 | + public static MastershipTerm of(NodeId master, int term) { | ||
16 | + return new MastershipTerm(master, term); | ||
17 | + } | ||
18 | + | ||
19 | + public NodeId master() { | ||
20 | + return master; | ||
21 | + } | ||
22 | + | ||
23 | + public int termNumber() { | ||
24 | + return termNumber; | ||
25 | + } | ||
26 | + | ||
27 | + @Override | ||
28 | + public int hashCode() { | ||
29 | + return Objects.hash(master, termNumber); | ||
30 | + } | ||
31 | + | ||
32 | + @Override | ||
33 | + public boolean equals(Object other) { | ||
34 | + if (other instanceof MastershipTerm) { | ||
35 | + MastershipTerm that = (MastershipTerm) other; | ||
36 | + if (!this.master.equals(that.master)) { | ||
37 | + return false; | ||
38 | + } | ||
39 | + if (this.termNumber != that.termNumber) { | ||
40 | + return false; | ||
41 | + } | ||
42 | + return true; | ||
43 | + } | ||
44 | + return false; | ||
45 | + } | ||
6 | } | 46 | } | ... | ... |
... | @@ -40,4 +40,9 @@ public class MastershipServiceAdapter implements MastershipService { | ... | @@ -40,4 +40,9 @@ public class MastershipServiceAdapter implements MastershipService { |
40 | @Override | 40 | @Override |
41 | public void removeListener(MastershipListener listener) { | 41 | public void removeListener(MastershipListener listener) { |
42 | } | 42 | } |
43 | + | ||
44 | + @Override | ||
45 | + public MastershipTermService requestTermService() { | ||
46 | + return null; | ||
47 | + } | ||
43 | } | 48 | } | ... | ... |
... | @@ -12,6 +12,8 @@ import org.onlab.onos.cluster.MastershipEvent; | ... | @@ -12,6 +12,8 @@ import org.onlab.onos.cluster.MastershipEvent; |
12 | import org.onlab.onos.cluster.MastershipListener; | 12 | import org.onlab.onos.cluster.MastershipListener; |
13 | import org.onlab.onos.cluster.MastershipService; | 13 | import org.onlab.onos.cluster.MastershipService; |
14 | import org.onlab.onos.cluster.MastershipStore; | 14 | import org.onlab.onos.cluster.MastershipStore; |
15 | +import org.onlab.onos.cluster.MastershipTerm; | ||
16 | +import org.onlab.onos.cluster.MastershipTermService; | ||
15 | import org.onlab.onos.cluster.NodeId; | 17 | import org.onlab.onos.cluster.NodeId; |
16 | import org.onlab.onos.event.AbstractListenerRegistry; | 18 | import org.onlab.onos.event.AbstractListenerRegistry; |
17 | import org.onlab.onos.event.EventDeliveryService; | 19 | import org.onlab.onos.event.EventDeliveryService; |
... | @@ -103,6 +105,12 @@ public class MastershipManager | ... | @@ -103,6 +105,12 @@ public class MastershipManager |
103 | return store.getDevices(nodeId); | 105 | return store.getDevices(nodeId); |
104 | } | 106 | } |
105 | 107 | ||
108 | + | ||
109 | + @Override | ||
110 | + public MastershipTermService requestTermService() { | ||
111 | + return new InternalMastershipTermService(); | ||
112 | + } | ||
113 | + | ||
106 | @Override | 114 | @Override |
107 | public void addListener(MastershipListener listener) { | 115 | public void addListener(MastershipListener listener) { |
108 | checkNotNull(listener); | 116 | checkNotNull(listener); |
... | @@ -124,4 +132,13 @@ public class MastershipManager | ... | @@ -124,4 +132,13 @@ public class MastershipManager |
124 | } | 132 | } |
125 | } | 133 | } |
126 | 134 | ||
135 | + private class InternalMastershipTermService implements MastershipTermService { | ||
136 | + | ||
137 | + @Override | ||
138 | + public MastershipTerm getMastershipTerm(DeviceId deviceId) { | ||
139 | + return store.getTermFor(deviceId); | ||
140 | + } | ||
141 | + | ||
142 | + } | ||
143 | + | ||
127 | } | 144 | } | ... | ... |
... | @@ -15,6 +15,7 @@ import org.onlab.onos.cluster.ClusterService; | ... | @@ -15,6 +15,7 @@ import org.onlab.onos.cluster.ClusterService; |
15 | import org.onlab.onos.cluster.MastershipEvent; | 15 | import org.onlab.onos.cluster.MastershipEvent; |
16 | import org.onlab.onos.cluster.MastershipStore; | 16 | import org.onlab.onos.cluster.MastershipStore; |
17 | import org.onlab.onos.cluster.MastershipStoreDelegate; | 17 | import org.onlab.onos.cluster.MastershipStoreDelegate; |
18 | +import org.onlab.onos.cluster.MastershipTerm; | ||
18 | import org.onlab.onos.cluster.NodeId; | 19 | import org.onlab.onos.cluster.NodeId; |
19 | import org.onlab.onos.net.DeviceId; | 20 | import org.onlab.onos.net.DeviceId; |
20 | import org.onlab.onos.net.MastershipRole; | 21 | import org.onlab.onos.net.MastershipRole; |
... | @@ -115,4 +116,10 @@ public class DistributedMastershipStore | ... | @@ -115,4 +116,10 @@ public class DistributedMastershipStore |
115 | return nodeId.equals(master) ? MastershipRole.MASTER : MastershipRole.STANDBY; | 116 | return nodeId.equals(master) ? MastershipRole.MASTER : MastershipRole.STANDBY; |
116 | } | 117 | } |
117 | 118 | ||
119 | + @Override | ||
120 | + public MastershipTerm getTermFor(DeviceId deviceId) { | ||
121 | + // TODO Auto-generated method stub | ||
122 | + return null; | ||
123 | + } | ||
124 | + | ||
118 | } | 125 | } | ... | ... |
... | @@ -3,11 +3,13 @@ package org.onlab.onos.net.trivial.impl; | ... | @@ -3,11 +3,13 @@ package org.onlab.onos.net.trivial.impl; |
3 | import static org.slf4j.LoggerFactory.getLogger; | 3 | import static org.slf4j.LoggerFactory.getLogger; |
4 | 4 | ||
5 | import java.util.Collections; | 5 | import java.util.Collections; |
6 | +import java.util.HashMap; | ||
6 | import java.util.HashSet; | 7 | import java.util.HashSet; |
7 | import java.util.Map; | 8 | import java.util.Map; |
8 | import java.util.Set; | 9 | import java.util.Set; |
9 | import java.util.concurrent.ConcurrentHashMap; | 10 | import java.util.concurrent.ConcurrentHashMap; |
10 | import java.util.concurrent.ConcurrentMap; | 11 | import java.util.concurrent.ConcurrentMap; |
12 | +import java.util.concurrent.atomic.AtomicInteger; | ||
11 | 13 | ||
12 | import org.apache.felix.scr.annotations.Activate; | 14 | import org.apache.felix.scr.annotations.Activate; |
13 | import org.apache.felix.scr.annotations.Component; | 15 | import org.apache.felix.scr.annotations.Component; |
... | @@ -18,6 +20,7 @@ import org.onlab.onos.cluster.DefaultControllerNode; | ... | @@ -18,6 +20,7 @@ import org.onlab.onos.cluster.DefaultControllerNode; |
18 | import org.onlab.onos.cluster.MastershipEvent; | 20 | import org.onlab.onos.cluster.MastershipEvent; |
19 | import org.onlab.onos.cluster.MastershipStore; | 21 | import org.onlab.onos.cluster.MastershipStore; |
20 | import org.onlab.onos.cluster.MastershipStoreDelegate; | 22 | import org.onlab.onos.cluster.MastershipStoreDelegate; |
23 | +import org.onlab.onos.cluster.MastershipTerm; | ||
21 | import org.onlab.onos.cluster.NodeId; | 24 | import org.onlab.onos.cluster.NodeId; |
22 | import org.onlab.onos.net.DeviceId; | 25 | import org.onlab.onos.net.DeviceId; |
23 | import org.onlab.onos.net.MastershipRole; | 26 | import org.onlab.onos.net.MastershipRole; |
... | @@ -47,6 +50,7 @@ public class SimpleMastershipStore | ... | @@ -47,6 +50,7 @@ public class SimpleMastershipStore |
47 | //devices mapped to their masters, to emulate multiple nodes | 50 | //devices mapped to their masters, to emulate multiple nodes |
48 | protected final ConcurrentMap<DeviceId, NodeId> masterMap = | 51 | protected final ConcurrentMap<DeviceId, NodeId> masterMap = |
49 | new ConcurrentHashMap<>(); | 52 | new ConcurrentHashMap<>(); |
53 | + protected final Map<DeviceId, AtomicInteger> termMap = new HashMap<>(); | ||
50 | 54 | ||
51 | @Activate | 55 | @Activate |
52 | public void activate() { | 56 | public void activate() { |
... | @@ -63,17 +67,23 @@ public class SimpleMastershipStore | ... | @@ -63,17 +67,23 @@ public class SimpleMastershipStore |
63 | 67 | ||
64 | NodeId node = masterMap.get(deviceId); | 68 | NodeId node = masterMap.get(deviceId); |
65 | if (node == null) { | 69 | if (node == null) { |
70 | + synchronized (this) { | ||
66 | masterMap.put(deviceId, nodeId); | 71 | masterMap.put(deviceId, nodeId); |
72 | + termMap.put(deviceId, new AtomicInteger()); | ||
73 | + } | ||
67 | return new MastershipEvent(MASTER_CHANGED, deviceId, nodeId); | 74 | return new MastershipEvent(MASTER_CHANGED, deviceId, nodeId); |
68 | } | 75 | } |
69 | 76 | ||
70 | if (node.equals(nodeId)) { | 77 | if (node.equals(nodeId)) { |
71 | return null; | 78 | return null; |
72 | } else { | 79 | } else { |
80 | + synchronized (this) { | ||
73 | masterMap.put(deviceId, nodeId); | 81 | masterMap.put(deviceId, nodeId); |
82 | + termMap.get(deviceId).incrementAndGet(); | ||
74 | return new MastershipEvent(MASTER_CHANGED, deviceId, nodeId); | 83 | return new MastershipEvent(MASTER_CHANGED, deviceId, nodeId); |
75 | } | 84 | } |
76 | } | 85 | } |
86 | + } | ||
77 | 87 | ||
78 | @Override | 88 | @Override |
79 | public NodeId getMaster(DeviceId deviceId) { | 89 | public NodeId getMaster(DeviceId deviceId) { |
... | @@ -114,4 +124,13 @@ public class SimpleMastershipStore | ... | @@ -114,4 +124,13 @@ public class SimpleMastershipStore |
114 | return role; | 124 | return role; |
115 | } | 125 | } |
116 | 126 | ||
127 | + @Override | ||
128 | + public MastershipTerm getTermFor(DeviceId deviceId) { | ||
129 | + if (masterMap.get(deviceId) == null) { | ||
130 | + return null; | ||
131 | + } | ||
132 | + return MastershipTerm.of( | ||
133 | + masterMap.get(deviceId), termMap.get(deviceId).get()); | ||
134 | + } | ||
135 | + | ||
117 | } | 136 | } | ... | ... |
-
Please register or login to post a comment