opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/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;
   9
  10 import akka.actor.Address;
  11 import java.util.ArrayList;
  12 import java.util.Collection;
  13 import java.util.Map;
  14 import java.util.concurrent.ConcurrentHashMap;
  15 import java.util.concurrent.ConcurrentMap;
  16 import org.opendaylight.controller.cluster.datastore.identifiers.ShardIdentifier;
  17 import org.opendaylight.controller.cluster.datastore.identifiers.ShardManagerIdentifier;
  18 import org.opendaylight.controller.cluster.raft.PeerAddressResolver;
  19
  20 /**
  21  * Implementation PeerAddressResolver that resolves address for shard peer ids. This class is owned by the
  22  * ShardMaanager and passed to Shard actors via the ConfigParams.
  23  *
  24  * @author Thomas Pantelis
  25  */
  26 class ShardPeerAddressResolver implements PeerAddressResolver {
  27     // Stores a mapping between a member name and the address of the member. The map is concurrent as it
  28     // will be accessed by multiple threads via the public resolve method.
  29     private final ConcurrentMap<String, Address> memberNameToAddress = new ConcurrentHashMap<>();
  30     private final String shardManagerIdentifier;
  31     private final String shardManagerType;
  32     private final String localMemberName;
  33
  34     public ShardPeerAddressResolver(String shardManagerType, String localMemberName) {
  35         this.shardManagerIdentifier = ShardManagerIdentifier.builder().type(shardManagerType).build().toString();
  36         this.shardManagerType = shardManagerType;
  37         this.localMemberName = localMemberName;
  38     }
  39
  40     void addPeerAddress(String memberName, Address address) {
  41         memberNameToAddress.put(memberName, address);
  42     }
  43
  44     void removePeerAddress(String memberName) {
  45         memberNameToAddress.remove(memberName);
  46     }
  47
  48     Address getPeerAddress(String memberName) {
  49         return memberNameToAddress.get(memberName);
  50     }
  51
  52     Collection<String> getShardManagerPeerActorAddresses() {
  53         Collection<String> peerAddresses = new ArrayList<>();
  54         for(Map.Entry<String, Address> entry: memberNameToAddress.entrySet()) {
  55             if(!localMemberName.equals(entry.getKey())) {
  56                 peerAddresses.add(getShardManagerActorPathBuilder(entry.getValue()).toString());
  57             }
  58         }
  59
  60         return peerAddresses;
  61     }
  62
  63     ShardIdentifier getShardIdentifier(String memberName, String shardName){
  64         return ShardIdentifier.builder().memberName(memberName).shardName(shardName).type(shardManagerType).build();
  65     }
  66
  67     String getShardActorAddress(String shardName, String memberName) {
  68         Address memberAddress = memberNameToAddress.get(memberName);
  69         if(memberAddress != null) {
  70             return getShardManagerActorPathBuilder(memberAddress).append("/").append(
  71                     getShardIdentifier(memberName, shardName)).toString();
  72         }
  73
  74         return null;
  75     }
  76
  77     private StringBuilder getShardManagerActorPathBuilder(Address address) {
  78         return new StringBuilder().append(address.toString()).append("/user/").append(shardManagerIdentifier);
  79     }
  80
  81     @Override
  82     public String resolve(String peerId) {
  83         ShardIdentifier shardId = ShardIdentifier.builder().fromShardIdString(peerId).build();
  84         return getShardActorAddress(shardId.getShardName(), shardId.getMemberName());
  85     }
  86 }