1 package org.opendaylight.bgpmanager;
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;
21 import org.slf4j.Logger;
22 import org.slf4j.LoggerFactory;
24 import java.util.ArrayList;
25 import java.util.List;
26 import java.util.Iterator;
28 import com.google.common.util.concurrent.FutureCallback;
29 import com.google.common.base.Optional;
32 * Created by emhamla on 4/14/2015.
34 public class FibDSWriter {
35 private static final Logger logger = LoggerFactory.getLogger(FibDSWriter.class);
36 private final DataBroker broker;
38 public FibDSWriter(final DataBroker db) {
42 public synchronized void addFibEntryToDS(String rd, String prefix,
43 String nexthop, int label) {
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);
49 List<VrfEntry> vrfEntryList = new ArrayList<VrfEntry>();
50 vrfEntryList.add(vrfEntry);
52 InstanceIdentifierBuilder<VrfTables> idBuilder =
53 InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
54 InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
56 VrfTables vrfTableNew = new VrfTablesBuilder().setRouteDistinguisher(rd).
57 setVrfEntry(vrfEntryList).build();
59 write(LogicalDatastoreType.CONFIGURATION, vrfTableId, vrfTableNew);
63 public synchronized void removeFibEntryFromDS(String rd, String prefix) {
65 logger.debug("Removing fib entry with destination prefix {} from vrf table for rd {}", prefix, rd);
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);
74 public synchronized void removeVrfFromDS(String rd) {
75 logger.debug("Removing vrf table for rd {}", rd);
77 InstanceIdentifierBuilder<VrfTables> idBuilder =
78 InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
79 InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
81 delete(LogicalDatastoreType.CONFIGURATION, vrfTableId);
86 private <T extends DataObject> Optional<T> read(LogicalDatastoreType datastoreType,
87 InstanceIdentifier<T> path) {
89 ReadOnlyTransaction tx = broker.newReadOnlyTransaction();
91 Optional<T> result = Optional.absent();
93 result = tx.read(datastoreType, path).get();
94 } catch (Exception e) {
95 throw new RuntimeException(e);
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);
108 private <T extends DataObject> void delete(LogicalDatastoreType datastoreType, InstanceIdentifier<T> path) {
109 WriteTransaction tx = broker.newWriteOnlyTransaction();
110 tx.delete(datastoreType, path);