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