2 * Copyright (c) 2015 - 2016 Ericsson India Global Services Pvt Ltd. 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.netvirt.vpnmanager;
10 import static org.opendaylight.genius.infra.Datastore.OPERATIONAL;
12 import java.util.ArrayList;
13 import java.util.List;
14 import java.util.stream.Collectors;
15 import javax.annotation.PostConstruct;
16 import javax.inject.Inject;
17 import javax.inject.Singleton;
18 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
19 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
20 import org.opendaylight.genius.datastoreutils.AsyncDataTreeChangeListenerBase;
21 import org.opendaylight.genius.infra.ManagedNewTransactionRunner;
22 import org.opendaylight.genius.infra.ManagedNewTransactionRunnerImpl;
23 import org.opendaylight.infrautils.utils.concurrent.ListenableFutures;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.FibEntries;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTables;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTablesKey;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentries.VrfEntry;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentrybase.RoutePaths;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntry;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntryBuilder;
31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
32 import org.slf4j.Logger;
33 import org.slf4j.LoggerFactory;
36 public class FibEntriesListener extends AsyncDataTreeChangeListenerBase<VrfEntry, FibEntriesListener> {
37 private static final Logger LOG = LoggerFactory.getLogger(FibEntriesListener.class);
38 private final DataBroker dataBroker;
39 private final ManagedNewTransactionRunner txRunner;
40 private final VpnInstanceListener vpnInstanceListener;
43 public FibEntriesListener(final DataBroker dataBroker, final VpnInstanceListener vpnInstanceListener) {
44 super(VrfEntry.class, FibEntriesListener.class);
45 this.dataBroker = dataBroker;
46 this.txRunner = new ManagedNewTransactionRunnerImpl(dataBroker);
47 this.vpnInstanceListener = vpnInstanceListener;
52 LOG.info("{} start", getClass().getSimpleName());
53 registerListener(LogicalDatastoreType.OPERATIONAL, dataBroker);
57 protected InstanceIdentifier<VrfEntry> getWildCardPath() {
58 return InstanceIdentifier.create(FibEntries.class).child(VrfTables.class).child(VrfEntry.class);
62 protected FibEntriesListener getDataTreeChangeListener() {
63 return FibEntriesListener.this;
68 protected void remove(InstanceIdentifier<VrfEntry> identifier,
70 LOG.trace("Remove Fib event - Key : {}, value : {} ", identifier, del);
71 final VrfTablesKey key = identifier.firstKeyOf(VrfTables.class);
72 String rd = key.getRouteDistinguisher();
73 List<RoutePaths> routePaths = del.getRoutePaths();
74 removeLabelFromVpnInstance(rd, routePaths);
78 protected void update(InstanceIdentifier<VrfEntry> identifier,
79 VrfEntry original, VrfEntry update) {
80 final VrfTablesKey key = identifier.firstKeyOf(VrfTables.class);
81 String rd = key.getRouteDistinguisher();
82 List<RoutePaths> originalRoutePaths = new ArrayList<>(original.getRoutePaths());
83 List<RoutePaths> updateRoutePaths = new ArrayList<>(update.getRoutePaths());
84 if (originalRoutePaths.size() < updateRoutePaths.size()) {
85 updateRoutePaths.removeAll(originalRoutePaths);
86 addLabelToVpnInstance(rd, updateRoutePaths);
87 } else if (originalRoutePaths.size() > updateRoutePaths.size()) {
88 originalRoutePaths.removeAll(updateRoutePaths);
89 removeLabelFromVpnInstance(rd, originalRoutePaths);
94 protected void add(InstanceIdentifier<VrfEntry> identifier,
96 LOG.trace("Add Vrf Entry event - Key : {}, value : {}", identifier, add);
97 final VrfTablesKey key = identifier.firstKeyOf(VrfTables.class);
98 String rd = key.getRouteDistinguisher();
99 addLabelToVpnInstance(rd, add.getRoutePaths());
102 private void addLabelToVpnInstance(String rd, List<RoutePaths> routePaths) {
103 List<Long> labels = routePaths.stream().map(RoutePaths::getLabel).distinct()
104 .collect(Collectors.toList());
105 VpnInstanceOpDataEntry vpnInstanceOpData = vpnInstanceListener.getVpnInstanceOpData(rd);
106 if (vpnInstanceOpData != null) {
107 List<Long> routeIds = vpnInstanceOpData.getRouteEntryId() == null ? new ArrayList<>()
108 : vpnInstanceOpData.getRouteEntryId();
109 labels.forEach(label -> {
110 LOG.debug("Adding label to vpn info - {}", label);
111 if (!routeIds.contains(label)) {
115 ListenableFutures.addErrorLogging(txRunner.callWithNewWriteOnlyTransactionAndSubmit(OPERATIONAL, tx ->
116 tx.put(VpnUtil.getVpnInstanceOpDataIdentifier(rd),
117 new VpnInstanceOpDataEntryBuilder(vpnInstanceOpData).setRouteEntryId(routeIds)
119 LOG, "Error adding label to VPN instance");
121 LOG.warn("No VPN Instance found for RD: {}", rd);
125 private void removeLabelFromVpnInstance(String rd, List<RoutePaths> routePaths) {
126 List<Long> labels = routePaths.stream().map(RoutePaths::getLabel).distinct()
127 .collect(Collectors.toList());
128 VpnInstanceOpDataEntry vpnInstanceOpData = vpnInstanceListener.getVpnInstanceOpData(rd);
129 if (vpnInstanceOpData != null) {
130 List<Long> routeIds = vpnInstanceOpData.getRouteEntryId();
131 if (routeIds == null) {
132 LOG.debug("Fib Route entry is empty.");
134 LOG.debug("Removing label from vpn info - {}", labels);
135 routeIds.removeAll(labels);
136 ListenableFutures.addErrorLogging(txRunner.callWithNewWriteOnlyTransactionAndSubmit(OPERATIONAL, tx ->
137 tx.put(VpnUtil.getVpnInstanceOpDataIdentifier(rd),
138 new VpnInstanceOpDataEntryBuilder(vpnInstanceOpData).setRouteEntryId(routeIds)
140 LOG, "Error removing label from VPN instance");
143 LOG.warn("No VPN Instance found for RD: {}", rd);