neutronvpn/impl/src/main/java/org/opendaylight/netvirt/neutronvpn/evpn/utils/NeutronEvpnUtils.java

   1 /*
   2  * Copyright © 2017, 2018 Ericsson India Global Services Pvt Ltd. 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.netvirt.neutronvpn.evpn.utils;
   9
  10 import static org.opendaylight.genius.infra.Datastore.CONFIGURATION;
  11
  12 import com.google.common.base.Optional;
  13 import java.util.Collections;
  14 import javax.annotation.Nullable;
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  18 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  19 import org.opendaylight.genius.datastoreutils.SingleTransactionDataBroker;
  20 import org.opendaylight.genius.infra.ManagedNewTransactionRunner;
  21 import org.opendaylight.genius.infra.ManagedNewTransactionRunnerImpl;
  22 import org.opendaylight.infrautils.jobcoordinator.JobCoordinator;
  23 import org.opendaylight.infrautils.utils.concurrent.ListenableFutures;
  24 import org.opendaylight.netvirt.elanmanager.api.ElanHelper;
  25 import org.opendaylight.netvirt.vpnmanager.api.IVpnManager;
  26 import org.opendaylight.netvirt.vpnmanager.api.VpnHelper;
  27 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.VpnInstance;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.EvpnAugmentation;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.EvpnAugmentationBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstance;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstanceBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.EvpnRdToNetworks;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetwork;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetworkBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetworkKey;
  37 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  38 import org.slf4j.Logger;
  39 import org.slf4j.LoggerFactory;
  40
  41 public class NeutronEvpnUtils {
  42
  43     private static final Logger LOG = LoggerFactory.getLogger(NeutronEvpnUtils.class);
  44
  45     public enum Operation {
  46         ADD,
  47         DELETE
  48     }
  49
  50     private final DataBroker dataBroker;
  51     private final ManagedNewTransactionRunner txRunner;
  52     private final IVpnManager vpnManager;
  53     private final JobCoordinator jobCoordinator;
  54
  55     public NeutronEvpnUtils(DataBroker broker, IVpnManager vpnManager, JobCoordinator jobCoordinator) {
  56         this.dataBroker = broker;
  57         this.txRunner = new ManagedNewTransactionRunnerImpl(dataBroker);
  58         this.vpnManager = vpnManager;
  59         this.jobCoordinator = jobCoordinator;
  60     }
  61
  62     @Nullable
  63     public VpnInstance getVpnInstance(Uuid vpnId) {
  64         return VpnHelper.getVpnInstance(dataBroker, vpnId.getValue());
  65     }
  66
  67     public boolean isVpnAssociatedWithNetwork(VpnInstance vpnInstance) throws ReadFailedException {
  68         String rd = vpnManager.getPrimaryRdFromVpnInstance(vpnInstance);
  69         InstanceIdentifier<EvpnRdToNetwork> id = InstanceIdentifier.builder(EvpnRdToNetworks.class)
  70                 .child(EvpnRdToNetwork.class, new EvpnRdToNetworkKey(rd)).build();
  71         Optional<EvpnRdToNetwork> optionalEvpnRdToNetwork =
  72                 SingleTransactionDataBroker.syncReadOptional(dataBroker, LogicalDatastoreType.CONFIGURATION, id);
  73         if (optionalEvpnRdToNetwork.isPresent()) {
  74             LOG.debug("vpn is associated with network {}", optionalEvpnRdToNetwork);
  75             return true;
  76         }
  77         return false;
  78     }
  79
  80     public InstanceIdentifier<EvpnRdToNetwork> getRdToNetworkIdentifier(String vrfId) {
  81         return InstanceIdentifier.builder(EvpnRdToNetworks.class)
  82                 .child(EvpnRdToNetwork.class, new EvpnRdToNetworkKey(vrfId)).build();
  83     }
  84
  85     public void updateElanWithVpnInfo(String elanInstanceName, VpnInstance vpnInstance, Operation operation) {
  86         String vpnName = vpnInstance.getVpnInstanceName();
  87         InstanceIdentifier<ElanInstance> elanIid = ElanHelper.getElanInstanceConfigurationDataPath(elanInstanceName);
  88         ListenableFutures.addErrorLogging(txRunner.callWithNewReadWriteTransactionAndSubmit(CONFIGURATION, tx -> {
  89             Optional<ElanInstance> elanInstanceOptional =
  90                     tx.read(elanIid).get();
  91             if (!elanInstanceOptional.isPresent()) {
  92                 return;
  93             }
  94
  95             EvpnAugmentationBuilder evpnAugmentationBuilder = new EvpnAugmentationBuilder();
  96             ElanInstanceBuilder elanInstanceBuilder = new ElanInstanceBuilder(elanInstanceOptional.get());
  97             if (elanInstanceBuilder.augmentation(EvpnAugmentation.class) != null) {
  98                 evpnAugmentationBuilder =
  99                         new EvpnAugmentationBuilder(elanInstanceBuilder.augmentation(EvpnAugmentation.class));
 100             }
 101             if (operation == Operation.ADD) {
 102                 evpnAugmentationBuilder.setEvpnName(vpnName);
 103                 LOG.debug("Writing Elan-EvpnAugmentation with key {}", elanInstanceName);
 104             } else {
 105                 evpnAugmentationBuilder.setEvpnName(null);
 106                 LOG.debug("Deleting Elan-EvpnAugmentation with key {}", elanInstanceName);
 107             }
 108
 109             elanInstanceBuilder.addAugmentation(EvpnAugmentation.class, evpnAugmentationBuilder.build());
 110             tx.put(elanIid, elanInstanceBuilder.build(), WriteTransaction.CREATE_MISSING_PARENTS);
 111         }), LOG, "Error updating ELAN with VPN info {}, {}, {}", elanInstanceName, vpnInstance, operation);
 112     }
 113
 114     public void updateVpnWithElanInfo(VpnInstance vpnInstance, String elanInstanceName, Operation operation) {
 115         String rd = vpnManager.getPrimaryRdFromVpnInstance(vpnInstance);
 116
 117         InstanceIdentifier<EvpnRdToNetwork> rdToNetworkIdentifier = getRdToNetworkIdentifier(rd);
 118
 119         jobCoordinator.enqueueJob("EVPN_ASSOCIATE-" + rd,
 120             () -> Collections.singletonList(txRunner.callWithNewReadWriteTransactionAndSubmit(CONFIGURATION, tx -> {
 121                 if (operation == Operation.DELETE) {
 122                     LOG.debug("Deleting Evpn-Network with key {}", rd);
 123                     tx.delete(rdToNetworkIdentifier);
 124                 } else {
 125                     EvpnRdToNetworkBuilder evpnRdToNetworkBuilder = new EvpnRdToNetworkBuilder().withKey(
 126                             new EvpnRdToNetworkKey(rd));
 127                     evpnRdToNetworkBuilder.setRd(rd);
 128                     evpnRdToNetworkBuilder.setNetworkId(elanInstanceName);
 129                     LOG.info("updating Evpn {} with elaninstance {} and rd {}",
 130                             vpnInstance.getVpnInstanceName(), elanInstanceName, rd);
 131                     tx.put(rdToNetworkIdentifier,
 132                             evpnRdToNetworkBuilder.build(), WriteTransaction.CREATE_MISSING_PARENTS);
 133                 }
 134             })));
 135     }
 136
 137     public void updateElanAndVpn(VpnInstance vpnInstance, String subnetVpn, Operation operation) {
 138         LOG.debug("updating elan {} in vpn {}, operation {} ", subnetVpn, vpnInstance.getVpnInstanceName(), operation);
 139         updateVpnWithElanInfo(vpnInstance, subnetVpn, operation);
 140
 141         LOG.debug("updating vpn {}, in elan {} operation {} ", subnetVpn, vpnInstance.getVpnInstanceName(), operation);
 142         // this data store update has to be done for l3vpn as well once routing use case for rt2 is supported.
 143         updateElanWithVpnInfo(subnetVpn, vpnInstance, operation);
 144         return;
 145     }
 146 }