Madan Jampani

Use partitionId instead of clusterName in CopycatTransport

Change-Id: I8e7ab3863a36944ac9e48e187037fb43695ebde3
...@@ -17,6 +17,7 @@ package org.onosproject.store.primitives.impl; ...@@ -17,6 +17,7 @@ package org.onosproject.store.primitives.impl;
17 17
18 import static com.google.common.base.Preconditions.checkNotNull; 18 import static com.google.common.base.Preconditions.checkNotNull;
19 19
20 +import org.onosproject.cluster.PartitionId;
20 import org.onosproject.store.cluster.messaging.MessagingService; 21 import org.onosproject.store.cluster.messaging.MessagingService;
21 22
22 import io.atomix.catalyst.transport.Client; 23 import io.atomix.catalyst.transport.Client;
...@@ -48,25 +49,25 @@ public class CopycatTransport implements Transport { ...@@ -48,25 +49,25 @@ public class CopycatTransport implements Transport {
48 } 49 }
49 50
50 private final Mode mode; 51 private final Mode mode;
51 - private final String clusterName; 52 + private final PartitionId partitionId;
52 private final MessagingService messagingService; 53 private final MessagingService messagingService;
53 54
54 - public CopycatTransport(Mode mode, String clusterName, MessagingService messagingService) { 55 + public CopycatTransport(Mode mode, PartitionId partitionId, MessagingService messagingService) {
55 this.mode = checkNotNull(mode); 56 this.mode = checkNotNull(mode);
56 - this.clusterName = checkNotNull(clusterName); 57 + this.partitionId = checkNotNull(partitionId);
57 this.messagingService = checkNotNull(messagingService); 58 this.messagingService = checkNotNull(messagingService);
58 } 59 }
59 60
60 @Override 61 @Override
61 public Client client() { 62 public Client client() {
62 - return new CopycatTransportClient(clusterName, 63 + return new CopycatTransportClient(partitionId,
63 messagingService, 64 messagingService,
64 mode); 65 mode);
65 } 66 }
66 67
67 @Override 68 @Override
68 public Server server() { 69 public Server server() {
69 - return new CopycatTransportServer(clusterName, 70 + return new CopycatTransportServer(partitionId,
70 messagingService); 71 messagingService);
71 } 72 }
72 } 73 }
......
...@@ -19,7 +19,9 @@ import static com.google.common.base.Preconditions.checkNotNull; ...@@ -19,7 +19,9 @@ import static com.google.common.base.Preconditions.checkNotNull;
19 19
20 import java.util.Set; 20 import java.util.Set;
21 import java.util.concurrent.CompletableFuture; 21 import java.util.concurrent.CompletableFuture;
22 +
22 import org.apache.commons.lang.math.RandomUtils; 23 import org.apache.commons.lang.math.RandomUtils;
24 +import org.onosproject.cluster.PartitionId;
23 import org.onosproject.store.cluster.messaging.MessagingService; 25 import org.onosproject.store.cluster.messaging.MessagingService;
24 26
25 import com.google.common.collect.Sets; 27 import com.google.common.collect.Sets;
...@@ -34,13 +36,13 @@ import io.atomix.catalyst.util.concurrent.ThreadContext; ...@@ -34,13 +36,13 @@ import io.atomix.catalyst.util.concurrent.ThreadContext;
34 */ 36 */
35 public class CopycatTransportClient implements Client { 37 public class CopycatTransportClient implements Client {
36 38
37 - private final String clusterName; 39 + private final PartitionId partitionId;
38 private final MessagingService messagingService; 40 private final MessagingService messagingService;
39 private final CopycatTransport.Mode mode; 41 private final CopycatTransport.Mode mode;
40 private final Set<CopycatTransportConnection> connections = Sets.newConcurrentHashSet(); 42 private final Set<CopycatTransportConnection> connections = Sets.newConcurrentHashSet();
41 43
42 - CopycatTransportClient(String clusterName, MessagingService messagingService, CopycatTransport.Mode mode) { 44 + CopycatTransportClient(PartitionId partitionId, MessagingService messagingService, CopycatTransport.Mode mode) {
43 - this.clusterName = checkNotNull(clusterName); 45 + this.partitionId = checkNotNull(partitionId);
44 this.messagingService = checkNotNull(messagingService); 46 this.messagingService = checkNotNull(messagingService);
45 this.mode = checkNotNull(mode); 47 this.mode = checkNotNull(mode);
46 } 48 }
...@@ -51,7 +53,7 @@ public class CopycatTransportClient implements Client { ...@@ -51,7 +53,7 @@ public class CopycatTransportClient implements Client {
51 CopycatTransportConnection connection = new CopycatTransportConnection( 53 CopycatTransportConnection connection = new CopycatTransportConnection(
52 nextConnectionId(), 54 nextConnectionId(),
53 CopycatTransport.Mode.CLIENT, 55 CopycatTransport.Mode.CLIENT,
54 - clusterName, 56 + partitionId,
55 remoteAddress, 57 remoteAddress,
56 messagingService, 58 messagingService,
57 context); 59 context);
......
...@@ -32,6 +32,7 @@ import java.util.function.Consumer; ...@@ -32,6 +32,7 @@ import java.util.function.Consumer;
32 import org.apache.commons.io.IOUtils; 32 import org.apache.commons.io.IOUtils;
33 import org.onlab.packet.IpAddress; 33 import org.onlab.packet.IpAddress;
34 import org.onlab.util.Tools; 34 import org.onlab.util.Tools;
35 +import org.onosproject.cluster.PartitionId;
35 import org.onosproject.store.cluster.messaging.Endpoint; 36 import org.onosproject.store.cluster.messaging.Endpoint;
36 import org.onosproject.store.cluster.messaging.MessagingService; 37 import org.onosproject.store.cluster.messaging.MessagingService;
37 38
...@@ -77,7 +78,7 @@ public class CopycatTransportConnection implements Connection { ...@@ -77,7 +78,7 @@ public class CopycatTransportConnection implements Connection {
77 78
78 CopycatTransportConnection(long connectionId, 79 CopycatTransportConnection(long connectionId,
79 CopycatTransport.Mode mode, 80 CopycatTransport.Mode mode,
80 - String clusterName, 81 + PartitionId partitionId,
81 Address address, 82 Address address,
82 MessagingService messagingService, 83 MessagingService messagingService,
83 ThreadContext context) { 84 ThreadContext context) {
...@@ -86,11 +87,11 @@ public class CopycatTransportConnection implements Connection { ...@@ -86,11 +87,11 @@ public class CopycatTransportConnection implements Connection {
86 this.remoteAddress = checkNotNull(address); 87 this.remoteAddress = checkNotNull(address);
87 this.messagingService = checkNotNull(messagingService); 88 this.messagingService = checkNotNull(messagingService);
88 if (mode == CopycatTransport.Mode.CLIENT) { 89 if (mode == CopycatTransport.Mode.CLIENT) {
89 - this.outboundMessageSubject = String.format("onos-copycat-%s", clusterName); 90 + this.outboundMessageSubject = String.format("onos-copycat-%s", partitionId);
90 - this.inboundMessageSubject = String.format("onos-copycat-%s-%d", clusterName, connectionId); 91 + this.inboundMessageSubject = String.format("onos-copycat-%s-%d", partitionId, connectionId);
91 } else { 92 } else {
92 - this.outboundMessageSubject = String.format("onos-copycat-%s-%d", clusterName, connectionId); 93 + this.outboundMessageSubject = String.format("onos-copycat-%s-%d", partitionId, connectionId);
93 - this.inboundMessageSubject = String.format("onos-copycat-%s", clusterName); 94 + this.inboundMessageSubject = String.format("onos-copycat-%s", partitionId);
94 } 95 }
95 this.context = checkNotNull(context); 96 this.context = checkNotNull(context);
96 } 97 }
......
...@@ -29,6 +29,7 @@ import java.util.function.Consumer; ...@@ -29,6 +29,7 @@ import java.util.function.Consumer;
29 29
30 import org.apache.commons.io.IOUtils; 30 import org.apache.commons.io.IOUtils;
31 import org.onlab.util.Tools; 31 import org.onlab.util.Tools;
32 +import org.onosproject.cluster.PartitionId;
32 import org.onosproject.store.cluster.messaging.MessagingService; 33 import org.onosproject.store.cluster.messaging.MessagingService;
33 34
34 import com.google.common.collect.Maps; 35 import com.google.common.collect.Maps;
...@@ -46,15 +47,15 @@ public class CopycatTransportServer implements Server { ...@@ -46,15 +47,15 @@ public class CopycatTransportServer implements Server {
46 47
47 private final AtomicBoolean listening = new AtomicBoolean(false); 48 private final AtomicBoolean listening = new AtomicBoolean(false);
48 private CompletableFuture<Void> listenFuture = new CompletableFuture<>(); 49 private CompletableFuture<Void> listenFuture = new CompletableFuture<>();
49 - private final String clusterName; 50 + private final PartitionId partitionId;
50 private final MessagingService messagingService; 51 private final MessagingService messagingService;
51 private final String messageSubject; 52 private final String messageSubject;
52 private final Map<Long, CopycatTransportConnection> connections = Maps.newConcurrentMap(); 53 private final Map<Long, CopycatTransportConnection> connections = Maps.newConcurrentMap();
53 54
54 - CopycatTransportServer(String clusterName, MessagingService messagingService) { 55 + CopycatTransportServer(PartitionId partitionId, MessagingService messagingService) {
55 - this.clusterName = checkNotNull(clusterName); 56 + this.partitionId = checkNotNull(partitionId);
56 this.messagingService = checkNotNull(messagingService); 57 this.messagingService = checkNotNull(messagingService);
57 - this.messageSubject = String.format("onos-copycat-%s", clusterName); 58 + this.messageSubject = String.format("onos-copycat-%s", partitionId);
58 } 59 }
59 60
60 @Override 61 @Override
...@@ -78,7 +79,7 @@ public class CopycatTransportServer implements Server { ...@@ -78,7 +79,7 @@ public class CopycatTransportServer implements Server {
78 newConnection.set(true); 79 newConnection.set(true);
79 return new CopycatTransportConnection(connectionId, 80 return new CopycatTransportConnection(connectionId,
80 CopycatTransport.Mode.SERVER, 81 CopycatTransport.Mode.SERVER,
81 - clusterName, 82 + partitionId,
82 senderAddress, 83 senderAddress,
83 messagingService, 84 messagingService,
84 getOrCreateContext(context)); 85 getOrCreateContext(context));
...@@ -114,6 +115,6 @@ public class CopycatTransportServer implements Server { ...@@ -114,6 +115,6 @@ public class CopycatTransportServer implements Server {
114 if (context != null) { 115 if (context != null) {
115 return context; 116 return context;
116 } 117 }
117 - return new SingleThreadContext("copycat-transport-server-" + clusterName, parentContext.serializer().clone()); 118 + return new SingleThreadContext("copycat-transport-server-" + partitionId, parentContext.serializer().clone());
118 } 119 }
119 } 120 }
......