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

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