2 * Copyright (c) 2015 Brocade Communications Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.cluster.datastore.shardmanager;
10 import akka.actor.Address;
11 import akka.actor.AddressFromURIString;
12 import com.google.common.base.Preconditions;
13 import java.util.ArrayList;
14 import java.util.Collection;
17 import java.util.concurrent.ConcurrentHashMap;
18 import java.util.concurrent.ConcurrentMap;
19 import org.opendaylight.controller.cluster.access.concepts.MemberName;
20 import org.opendaylight.controller.cluster.datastore.identifiers.ShardIdentifier;
21 import org.opendaylight.controller.cluster.datastore.identifiers.ShardManagerIdentifier;
22 import org.opendaylight.controller.cluster.raft.PeerAddressResolver;
25 * Implementation PeerAddressResolver that resolves address for shard peer ids. This class is owned by the
26 * ShardMaanager and passed to Shard actors via the ConfigParams.
28 * @author Thomas Pantelis
30 class ShardPeerAddressResolver implements PeerAddressResolver {
31 // Stores a mapping between a member name and the address of the member. The map is concurrent as it
32 // will be accessed by multiple threads via the public resolve method.
33 private final ConcurrentMap<MemberName, Address> memberNameToAddress = new ConcurrentHashMap<>();
34 private final String shardManagerIdentifier;
35 private final String shardManagerType;
36 private final MemberName localMemberName;
38 ShardPeerAddressResolver(String shardManagerType, MemberName localMemberName) {
39 this.shardManagerIdentifier = ShardManagerIdentifier.builder().type(shardManagerType).build().toString();
40 this.shardManagerType = shardManagerType;
41 this.localMemberName = Preconditions.checkNotNull(localMemberName);
44 void addPeerAddress(MemberName memberName, Address address) {
45 memberNameToAddress.put(memberName, address);
48 void removePeerAddress(MemberName memberName) {
49 memberNameToAddress.remove(memberName);
52 Set<MemberName> getPeerMembers() {
53 return this.memberNameToAddress.keySet();
56 Address getPeerAddress(MemberName memberName) {
57 return memberNameToAddress.get(memberName);
60 Collection<String> getShardManagerPeerActorAddresses() {
61 Collection<String> peerAddresses = new ArrayList<>();
62 for (Map.Entry<MemberName, Address> entry: memberNameToAddress.entrySet()) {
63 if (!localMemberName.equals(entry.getKey())) {
64 peerAddresses.add(getShardManagerActorPathBuilder(entry.getValue()).toString());
71 ShardIdentifier getShardIdentifier(MemberName memberName, String shardName) {
72 return ShardIdentifier.create(shardName, memberName, shardManagerType);
75 String getShardActorAddress(String shardName, MemberName memberName) {
76 Address memberAddress = memberNameToAddress.get(memberName);
77 if (memberAddress != null) {
78 return getShardManagerActorPathBuilder(memberAddress).append("/").append(
79 getShardIdentifier(memberName, shardName)).toString();
85 StringBuilder getShardManagerActorPathBuilder(Address address) {
86 return new StringBuilder().append(address.toString()).append("/user/").append(shardManagerIdentifier);
90 public String resolve(String peerId) {
95 ShardIdentifier shardId = ShardIdentifier.fromShardIdString(peerId);
96 return getShardActorAddress(shardId.getShardName(), shardId.getMemberName());
100 public void setResolved(String peerId, String address) {
101 memberNameToAddress.put(ShardIdentifier.fromShardIdString(peerId).getMemberName(),
102 AddressFromURIString.parse(address));