opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/shardmanager/ShardPeerAddressResolver.java

   1 /*
   2  * Copyright (c) 2015 Brocade Communications Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.cluster.datastore.shardmanager;
   9
  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;
  15 import java.util.Map;
  16 import java.util.Set;
  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;
  23
  24 /**
  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.
  27  *
  28  * @author Thomas Pantelis
  29  */
  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;
  37
  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);
  42     }
  43
  44     void addPeerAddress(MemberName memberName, Address address) {
  45         memberNameToAddress.put(memberName, address);
  46     }
  47
  48     void removePeerAddress(MemberName memberName) {
  49         memberNameToAddress.remove(memberName);
  50     }
  51
  52     Set<MemberName> getPeerMembers() {
  53         return this.memberNameToAddress.keySet();
  54     }
  55
  56     Address getPeerAddress(MemberName memberName) {
  57         return memberNameToAddress.get(memberName);
  58     }
  59
  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());
  65             }
  66         }
  67
  68         return peerAddresses;
  69     }
  70
  71     ShardIdentifier getShardIdentifier(MemberName memberName, String shardName) {
  72         return ShardIdentifier.create(shardName, memberName, shardManagerType);
  73     }
  74
  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();
  80         }
  81
  82         return null;
  83     }
  84
  85     StringBuilder getShardManagerActorPathBuilder(Address address) {
  86         return new StringBuilder().append(address.toString()).append("/user/").append(shardManagerIdentifier);
  87     }
  88
  89     @Override
  90     public String resolve(String peerId) {
  91         if (peerId == null) {
  92             return null;
  93         }
  94
  95         ShardIdentifier shardId = ShardIdentifier.fromShardIdString(peerId);
  96         return getShardActorAddress(shardId.getShardName(), shardId.getMemberName());
  97     }
  98
  99     @Override
 100     public void setResolved(String peerId, String address) {
 101         memberNameToAddress.put(ShardIdentifier.fromShardIdString(peerId).getMemberName(),
 102                 AddressFromURIString.parse(address));
 103     }
 104 }