2 * Copyright (c) 2017 Lumina Networks, Inc. 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.openflowplugin.applications.lldpspeaker;
10 import com.google.common.base.Optional;
11 import java.util.ArrayList;
12 import java.util.List;
13 import java.util.concurrent.ConcurrentHashMap;
14 import java.util.concurrent.ConcurrentMap;
15 import java.util.regex.Pattern;
16 import org.opendaylight.mdsal.eos.binding.api.EntityOwnershipChange;
17 import org.opendaylight.mdsal.eos.binding.api.EntityOwnershipListener;
18 import org.opendaylight.mdsal.eos.binding.api.EntityOwnershipService;
19 import org.opendaylight.mdsal.eos.common.api.EntityOwnershipState;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.core.general.entity.rev150930.Entity;
21 import org.slf4j.Logger;
22 import org.slf4j.LoggerFactory;
24 public class DeviceOwnershipStatusService implements EntityOwnershipListener {
25 private static final Logger LOG = LoggerFactory.getLogger(DeviceOwnershipStatusService.class);
26 private static final String SERVICE_ENTITY_TYPE = "org.opendaylight.mdsal.ServiceEntityType";
27 private static final Pattern NODE_ID_PATTERN = Pattern.compile("^openflow:\\d+");
29 private final EntityOwnershipService eos;
30 private final ConcurrentMap<String, EntityOwnershipState> ownershipStateCache = new ConcurrentHashMap<>();
32 public DeviceOwnershipStatusService(final EntityOwnershipService entityOwnershipService) {
33 this.eos = entityOwnershipService;
34 registerEntityOwnershipListener();
37 public boolean isEntityOwned(final String nodeId) {
38 EntityOwnershipState state = ownershipStateCache.get(nodeId);
40 java.util.Optional<EntityOwnershipState> status = getCurrentOwnershipStatus(nodeId);
41 if (status.isPresent()) {
43 ownershipStateCache.put(nodeId, state);
45 LOG.warn("Fetching ownership status failed for node {}", nodeId);
48 return state != null && state.equals(EntityOwnershipState.IS_OWNER);
51 public List<String> getOwnedNodes() {
52 List<String> nodes = new ArrayList<>();
53 ownershipStateCache.forEach((node, change) -> {
54 if (isEntityOwned(node)) {
62 public void ownershipChanged(final EntityOwnershipChange ownershipChange) {
63 final String entityName = ownershipChange.getEntity().getIdentifier().firstKeyOf(Entity.class).getName();
64 if (entityName != null && isOpenFlowEntity(entityName)) {
65 LOG.info("Entity ownership change received for node : {} : {}", entityName, ownershipChange);
66 if (!ownershipChange.getState().isOwner() && !ownershipChange.getState().hasOwner()
67 && !ownershipChange.inJeopardy()) {
68 LOG.debug("Entity for node {} is unregistered.", entityName);
69 ownershipStateCache.remove(entityName);
70 } else if (!ownershipChange.getState().isOwner() && ownershipChange.getState().hasOwner()) {
71 ownershipStateCache.put(entityName, EntityOwnershipState.OWNED_BY_OTHER);
72 } else if (ownershipChange.getState().isOwner()) {
73 ownershipStateCache.put(entityName, EntityOwnershipState.IS_OWNER);
78 private java.util.Optional<EntityOwnershipState> getCurrentOwnershipStatus(final String nodeId) {
79 org.opendaylight.mdsal.eos.binding.api.Entity entity = createNodeEntity(nodeId);
80 Optional<EntityOwnershipState> ownershipStatus = eos.getOwnershipState(entity);
82 if (ownershipStatus.isPresent()) {
83 LOG.debug("Fetched ownership status for node {} is {}", nodeId, ownershipStatus.get());
84 return java.util.Optional.of(ownershipStatus.get());
86 return java.util.Optional.empty();
89 private org.opendaylight.mdsal.eos.binding.api.Entity createNodeEntity(final String nodeId) {
90 return new org.opendaylight.mdsal.eos.binding.api.Entity(SERVICE_ENTITY_TYPE, nodeId);
93 private void registerEntityOwnershipListener() {
94 this.eos.registerListener(SERVICE_ENTITY_TYPE, this);
97 private boolean isOpenFlowEntity(String entity) {
98 return NODE_ID_PATTERN.matcher(entity).matches();