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

   1 package org.opendaylight.bgpmanager;
   2
   3 import com.google.common.util.concurrent.Futures;
   4 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
   5 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
   6 import org.opendaylight.yangtools.yang.binding.DataObject;
   7 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
   8 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
   9 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  10 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  11 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  12 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.VrfEntries;
  13 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntry;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntryBuilder;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTables;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTablesBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTablesKey;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntryKey;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.FibEntries;
  20
  21 import org.slf4j.Logger;
  22 import org.slf4j.LoggerFactory;
  23
  24 import java.util.ArrayList;
  25 import java.util.List;
  26 import java.util.Iterator;
  27
  28 import com.google.common.util.concurrent.FutureCallback;
  29 import com.google.common.base.Optional;
  30
  31 /**
  32  * Created by emhamla on 4/14/2015.
  33  */
  34 public class FibDSWriter {
  35     private static final Logger logger = LoggerFactory.getLogger(FibDSWriter.class);
  36     private final DataBroker broker;
  37
  38     public FibDSWriter(final DataBroker db) {
  39         broker = db;
  40     }
  41
  42     public synchronized void addFibEntryToDS(String rd, String prefix,
  43                                        String nexthop, int label) {
  44
  45         VrfEntry vrfEntry = new VrfEntryBuilder().setDestPrefix(prefix).
  46             setNextHopAddress(nexthop).setLabel((long)label).build();
  47         logger.debug("Created vrfEntry for {} nexthop {} label {}", prefix, nexthop, label);
  48
  49         List<VrfEntry> vrfEntryList = new ArrayList<VrfEntry>();
  50         vrfEntryList.add(vrfEntry);
  51
  52         InstanceIdentifierBuilder<VrfTables> idBuilder =
  53             InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
  54         InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
  55
  56         VrfTables vrfTableNew = new VrfTablesBuilder().setRouteDistinguisher(rd).
  57             setVrfEntry(vrfEntryList).build();
  58
  59         write(LogicalDatastoreType.CONFIGURATION, vrfTableId, vrfTableNew);
  60
  61     }
  62
  63     public synchronized void removeFibEntryFromDS(String rd, String prefix) {
  64
  65         logger.debug("Removing fib entry with destination prefix {} from vrf table for rd {}", prefix, rd);
  66
  67         InstanceIdentifierBuilder<VrfEntry> idBuilder =
  68             InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd)).child(VrfEntry.class, new VrfEntryKey(prefix));
  69         InstanceIdentifier<VrfEntry> vrfEntryId = idBuilder.build();
  70         delete(LogicalDatastoreType.CONFIGURATION, vrfEntryId);
  71
  72     }
  73
  74     public synchronized void removeVrfFromDS(String rd) {
  75         logger.debug("Removing vrf table for rd {}", rd);
  76
  77         InstanceIdentifierBuilder<VrfTables> idBuilder =
  78                 InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
  79         InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
  80
  81         delete(LogicalDatastoreType.CONFIGURATION, vrfTableId);
  82
  83     }
  84
  85
  86     private <T extends DataObject> Optional<T> read(LogicalDatastoreType datastoreType,
  87                                                     InstanceIdentifier<T> path) {
  88
  89         ReadOnlyTransaction tx = broker.newReadOnlyTransaction();
  90
  91         Optional<T> result = Optional.absent();
  92         try {
  93             result = tx.read(datastoreType, path).get();
  94         } catch (Exception e) {
  95             throw new RuntimeException(e);
  96         }
  97
  98         return result;
  99     }
 100
 101     private <T extends DataObject> void write(LogicalDatastoreType datastoreType,
 102                                                    InstanceIdentifier<T> path, T data) {
 103         WriteTransaction tx = broker.newWriteOnlyTransaction();
 104         tx.merge(datastoreType, path, data, true);
 105         tx.submit();
 106     }
 107
 108     private <T extends DataObject> void delete(LogicalDatastoreType datastoreType, InstanceIdentifier<T> path) {
 109         WriteTransaction tx = broker.newWriteOnlyTransaction();
 110         tx.delete(datastoreType, path);
 111         tx.submit();
 112     }
 113 }