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