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 org.opendaylight.controller.md.sal.binding.api.DataBroker;
  15 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  16 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  17 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  18 import org.opendaylight.genius.datastoreutils.SingleTransactionDataBroker;
  19 import org.opendaylight.genius.infra.ManagedNewTransactionRunner;
  20 import org.opendaylight.genius.infra.ManagedNewTransactionRunnerImpl;
  21 import org.opendaylight.infrautils.jobcoordinator.JobCoordinator;
  22 import org.opendaylight.infrautils.utils.concurrent.ListenableFutures;
  23 import org.opendaylight.netvirt.elanmanager.api.ElanHelper;
  24 import org.opendaylight.netvirt.vpnmanager.api.IVpnManager;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.EvpnAugmentation;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.EvpnAugmentationBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstance;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstanceBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.EvpnRdToNetworks;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetwork;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetworkBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.evpn.rd.to.networks.EvpnRdToNetworkKey;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.VpnInstance;
  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 boolean isVpnAssociatedWithNetwork(VpnInstance vpnInstance) throws ReadFailedException {
  60         String rd = vpnManager.getPrimaryRdFromVpnInstance(vpnInstance);
  61         InstanceIdentifier<EvpnRdToNetwork> id = InstanceIdentifier.builder(EvpnRdToNetworks.class)
  62                 .child(EvpnRdToNetwork.class, new EvpnRdToNetworkKey(rd)).build();
  63         Optional<EvpnRdToNetwork> optionalEvpnRdToNetwork =
  64                 SingleTransactionDataBroker.syncReadOptional(dataBroker, LogicalDatastoreType.CONFIGURATION, id);
  65         if (optionalEvpnRdToNetwork.isPresent()) {
  66             LOG.debug("vpn is associated with network {}", optionalEvpnRdToNetwork);
  67             return true;
  68         }
  69         return false;
  70     }
  71
  72     public InstanceIdentifier<EvpnRdToNetwork> getRdToNetworkIdentifier(String vrfId) {
  73         return InstanceIdentifier.builder(EvpnRdToNetworks.class)
  74                 .child(EvpnRdToNetwork.class, new EvpnRdToNetworkKey(vrfId)).build();
  75     }
  76
  77     public void updateElanWithVpnInfo(String elanInstanceName, VpnInstance vpnInstance, Operation operation) {
  78         String vpnName = vpnInstance.getVpnInstanceName();
  79         InstanceIdentifier<ElanInstance> elanIid = ElanHelper.getElanInstanceConfigurationDataPath(elanInstanceName);
  80         ListenableFutures.addErrorLogging(txRunner.callWithNewReadWriteTransactionAndSubmit(CONFIGURATION, tx -> {
  81             Optional<ElanInstance> elanInstanceOptional =
  82                     tx.read(elanIid).get();
  83             if (!elanInstanceOptional.isPresent()) {
  84                 return;
  85             }
  86
  87             EvpnAugmentationBuilder evpnAugmentationBuilder = new EvpnAugmentationBuilder();
  88             ElanInstanceBuilder elanInstanceBuilder = new ElanInstanceBuilder(elanInstanceOptional.get());
  89             if (elanInstanceBuilder.augmentation(EvpnAugmentation.class) != null) {
  90                 evpnAugmentationBuilder =
  91                         new EvpnAugmentationBuilder(elanInstanceBuilder.augmentation(EvpnAugmentation.class));
  92             }
  93             if (operation == Operation.ADD) {
  94                 evpnAugmentationBuilder.setEvpnName(vpnName);
  95                 LOG.debug("Writing Elan-EvpnAugmentation with key {}", elanInstanceName);
  96             } else {
  97                 evpnAugmentationBuilder.setEvpnName(null);
  98                 LOG.debug("Deleting Elan-EvpnAugmentation with key {}", elanInstanceName);
  99             }
 100
 101             elanInstanceBuilder.addAugmentation(EvpnAugmentation.class, evpnAugmentationBuilder.build());
 102             tx.put(elanIid, elanInstanceBuilder.build(), WriteTransaction.CREATE_MISSING_PARENTS);
 103         }), LOG, "Error updating ELAN with VPN info {}, {}, {}", elanInstanceName, vpnInstance, operation);
 104     }
 105
 106     public void updateVpnWithElanInfo(VpnInstance vpnInstance, String elanInstanceName, Operation operation) {
 107         String rd = vpnManager.getPrimaryRdFromVpnInstance(vpnInstance);
 108
 109         InstanceIdentifier<EvpnRdToNetwork> rdToNetworkIdentifier = getRdToNetworkIdentifier(rd);
 110
 111         jobCoordinator.enqueueJob("EVPN_ASSOCIATE-" + rd,
 112             () -> Collections.singletonList(txRunner.callWithNewReadWriteTransactionAndSubmit(CONFIGURATION, tx -> {
 113                 if (operation == Operation.DELETE) {
 114                     LOG.debug("Deleting Evpn-Network with key {}", rd);
 115                     tx.delete(rdToNetworkIdentifier);
 116                 } else {
 117                     EvpnRdToNetworkBuilder evpnRdToNetworkBuilder = new EvpnRdToNetworkBuilder().withKey(
 118                             new EvpnRdToNetworkKey(rd));
 119                     evpnRdToNetworkBuilder.setRd(rd);
 120                     evpnRdToNetworkBuilder.setNetworkId(elanInstanceName);
 121                     LOG.info("updating Evpn {} with elaninstance {} and rd {}",
 122                             vpnInstance.getVpnInstanceName(), elanInstanceName, rd);
 123                     tx.put(rdToNetworkIdentifier,
 124                             evpnRdToNetworkBuilder.build(), WriteTransaction.CREATE_MISSING_PARENTS);
 125                 }
 126             })));
 127     }
 128
 129     public void updateElanAndVpn(VpnInstance vpnInstance, String subnetVpn, Operation operation) {
 130         LOG.debug("updating elan {} in vpn {}, operation {} ", subnetVpn, vpnInstance.getVpnInstanceName(), operation);
 131         updateVpnWithElanInfo(vpnInstance, subnetVpn, operation);
 132
 133         LOG.debug("updating vpn {}, in elan {} operation {} ", subnetVpn, vpnInstance.getVpnInstanceName(), operation);
 134         // this data store update has to be done for l3vpn as well once routing use case for rt2 is supported.
 135         updateElanWithVpnInfo(subnetVpn, vpnInstance, operation);
 136         return;
 137     }
 138 }