2 * Copyright (c) 2016 Cisco 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.sharding;
10 import static com.google.common.base.Preconditions.checkNotNull;
11 import static java.util.Objects.requireNonNull;
13 import akka.actor.Address;
14 import java.util.Collection;
15 import java.util.concurrent.ConcurrentHashMap;
16 import java.util.concurrent.ConcurrentMap;
17 import java.util.stream.Collectors;
18 import org.opendaylight.controller.cluster.access.concepts.MemberName;
21 * Resolver for remote {@link ShardedDataTreeActor}'s.
23 public class ShardingServiceAddressResolver {
25 private final ConcurrentMap<MemberName, Address> memberNameToAddress = new ConcurrentHashMap<>();
26 private final String shardingServiceActorIdentifier;
27 private final MemberName localMemberName;
29 public ShardingServiceAddressResolver(final String shardingServiceActorIdentifier,
30 final MemberName localMemberName) {
31 this.shardingServiceActorIdentifier = shardingServiceActorIdentifier;
32 this.localMemberName = localMemberName;
35 void addPeerAddress(final MemberName memberName, final Address address) {
36 memberNameToAddress.put(memberName, address);
39 void removePeerAddress(final MemberName memberName) {
40 memberNameToAddress.remove(memberName);
43 Address getPeerAddress(final MemberName memberName) {
44 return memberNameToAddress.get(memberName);
47 StringBuilder getActorPathBuilder(final Address address) {
48 return new StringBuilder().append(address.toString()).append("/user/").append(shardingServiceActorIdentifier);
51 Collection<String> getShardingServicePeerActorAddresses() {
52 final Collection<String> peerAddresses =
56 .filter(entry -> !localMemberName.equals(entry.getKey()))
57 .map(entry -> getActorPathBuilder(entry.getValue()).toString())
58 .collect(Collectors.toList());
63 public String resolve(final MemberName memberName) {
64 final Address address = memberNameToAddress.get(requireNonNull(memberName));
65 checkNotNull(address, "Requested member[%s] is not present in the resolver", memberName);
66 return getActorPathBuilder(address).toString();