vpnservice/bgpmanager/bgpmanager-impl/src/main/java/org/opendaylight/netvirt/bgpmanager/FibDSWriter.java

   1 /*
   2  * Copyright (c) 2015 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
   9 package org.opendaylight.netvirt.bgpmanager;
  10
  11 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  12 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  13 import org.opendaylight.netvirt.fibmanager.api.RouteOrigin;
  14 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  15 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentries.VrfEntry;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentries.VrfEntryBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTables;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTablesKey;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentries.VrfEntryKey;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.FibEntries;
  22
  23 import org.slf4j.Logger;
  24 import org.slf4j.LoggerFactory;
  25
  26 import com.google.common.base.Optional;
  27 import com.google.common.base.Preconditions;
  28
  29 import java.util.*;
  30
  31
  32 public class FibDSWriter {
  33     private static final Logger logger = LoggerFactory.getLogger(FibDSWriter.class);
  34     private final DataBroker broker;
  35
  36     public FibDSWriter(final DataBroker db) {
  37         broker = db;
  38     }
  39
  40     public synchronized void addFibEntryToDS(String rd, String prefix, List<String> nextHopList,
  41                                              int label, RouteOrigin origin) {
  42         if (rd == null || rd.isEmpty() ) {
  43             logger.error("Prefix {} not associated with vpn", prefix);
  44             return;
  45         }
  46
  47         Preconditions.checkNotNull(nextHopList, "NextHopList can't be null");
  48
  49         for ( String nextHop: nextHopList){
  50             if (nextHop == null || nextHop.isEmpty()){
  51                 logger.error("nextHop list contains null element");
  52                 return;
  53             }
  54             if (logger.isDebugEnabled()) {
  55                 logger.debug("Created vrfEntry for {} nexthop {} label {}", prefix, nextHop, label);
  56             }
  57
  58         }
  59
  60
  61         // Looking for existing prefix in MDSAL database
  62         Optional<FibEntries> fibEntries = Optional.absent();
  63         try{
  64             InstanceIdentifier<FibEntries> idRead = InstanceIdentifier.create(FibEntries.class);
  65             fibEntries = BgpUtil.read(broker, LogicalDatastoreType.CONFIGURATION, idRead);
  66
  67             InstanceIdentifier<VrfEntry> vrfEntryId =
  68                     InstanceIdentifier.builder(FibEntries.class)
  69                             .child(VrfTables.class, new VrfTablesKey(rd))
  70                             .child(VrfEntry.class, new VrfEntryKey(prefix)).build();
  71             Optional<VrfEntry> entry = BgpUtil.read(broker, LogicalDatastoreType.CONFIGURATION, vrfEntryId);
  72
  73             if (! entry.isPresent()) {
  74                 VrfEntry vrfEntry = new VrfEntryBuilder().setDestPrefix(prefix).setNextHopAddressList(nextHopList)
  75                         .setLabel((long)label).setOrigin(origin.getValue()).build();
  76
  77                 BgpUtil.write(broker, LogicalDatastoreType.CONFIGURATION, vrfEntryId, vrfEntry);
  78
  79             } else { // Found in MDSAL database
  80                 List<String> nh = entry.get().getNextHopAddressList();
  81                 for (String nextHop : nextHopList) {
  82                     if (!nh.contains(nextHop))
  83                         nh.add(nextHop);
  84                 }
  85                 VrfEntry vrfEntry = new VrfEntryBuilder().setDestPrefix(prefix).setNextHopAddressList(nh)
  86                         .setLabel((long) label).setOrigin(origin.getValue()).build();
  87
  88                 BgpUtil.update(broker, LogicalDatastoreType.CONFIGURATION, vrfEntryId, vrfEntry);
  89             }
  90         } catch (Exception e) {
  91             logger.error("addFibEntryToDS: error ", e);
  92         }
  93
  94     }
  95
  96     public synchronized void removeFibEntryFromDS(String rd, String prefix) {
  97
  98         if (rd == null || rd.isEmpty()) {
  99             logger.error("Prefix {} not associated with vpn", prefix);
 100             return;
 101         }
 102         logger.debug("Removing fib entry with destination prefix {} from vrf table for rd {}", prefix, rd);
 103
 104         InstanceIdentifierBuilder<VrfEntry> idBuilder =
 105             InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd)).child(VrfEntry.class, new VrfEntryKey(prefix));
 106         InstanceIdentifier<VrfEntry> vrfEntryId = idBuilder.build();
 107         BgpUtil.delete(broker, LogicalDatastoreType.CONFIGURATION, vrfEntryId);
 108
 109     }
 110
 111     public synchronized void removeVrfFromDS(String rd) {
 112         logger.debug("Removing vrf table for  rd {}", rd);
 113
 114         InstanceIdentifierBuilder<VrfTables> idBuilder =
 115                 InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
 116         InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
 117
 118         BgpUtil.delete(broker, LogicalDatastoreType.CONFIGURATION, vrfTableId);
 119
 120     }
 121
 122
 123
 124 }