2 * Copyright (c) 2015 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
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
8 package org.opendaylight.netvirt.bgpmanager;
10 import com.google.common.base.Optional;
11 import com.google.common.util.concurrent.ThreadFactoryBuilder;
13 import java.util.concurrent.BlockingQueue;
14 import java.util.concurrent.ExecutionException;
15 import java.util.concurrent.ExecutorService;
16 import java.util.concurrent.Executors;
17 import java.util.concurrent.LinkedBlockingQueue;
18 import java.util.concurrent.ThreadFactory;
19 import java.util.concurrent.TimeoutException;
20 import java.util.concurrent.atomic.AtomicInteger;
22 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
23 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
24 import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
26 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
27 import org.opendaylight.controller.md.sal.common.api.data.TransactionChain;
28 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
29 import org.opendaylight.genius.datastoreutils.SingleTransactionDataBroker;
30 import org.opendaylight.genius.mdsalutil.MDSALUtil;
31 import org.opendaylight.genius.utils.batching.ActionableResource;
32 import org.opendaylight.genius.utils.batching.ActionableResourceImpl;
33 import org.opendaylight.genius.utils.batching.ResourceBatchingManager;
34 import org.opendaylight.genius.utils.batching.ResourceHandler;
35 import org.opendaylight.netvirt.bgpmanager.thrift.gen.encap_type;
36 import org.opendaylight.netvirt.bgpmanager.thrift.gen.protocol_type;
37 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.BgpControlPlaneType;
38 import org.opendaylight.yang.gen.v1.urn.ericsson.params.xml.ns.yang.ebgp.rev150901.EncapType;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.VpnInstanceOpData;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntry;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntryKey;
42 import org.opendaylight.yangtools.yang.binding.DataObject;
43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
44 import org.slf4j.Logger;
45 import org.slf4j.LoggerFactory;
47 public class BgpUtil {
48 private static final Logger LOG = LoggerFactory.getLogger(BgpUtil.class);
49 private static DataBroker dataBroker;
50 private static BindingTransactionChain fibTransact;
51 public static final int PERIODICITY = 500;
52 private static AtomicInteger pendingWrTransaction = new AtomicInteger(0);
53 public static final int BATCH_SIZE = 1000;
54 public static Integer batchSize;
55 public static Integer batchInterval;
56 private static int txChainAttempts = 0;
58 private static BlockingQueue<ActionableResource> bgpResourcesBufferQ = new LinkedBlockingQueue<>();
60 // return number of pending Write Transactions with BGP-Util (no read)
61 public static int getGetPendingWrTransaction() {
62 return pendingWrTransaction.get();
65 static ThreadFactory namedThreadFactory = new ThreadFactoryBuilder()
66 .setNameFormat("bgp-util-mdsal-%d").build();
68 static ExecutorService threadPool = Executors.newFixedThreadPool(1, namedThreadFactory);
71 static synchronized BindingTransactionChain getTransactionChain() {
75 static void registerWithBatchManager(ResourceHandler resourceHandler) {
76 ResourceBatchingManager resBatchingManager = ResourceBatchingManager.getInstance();
77 resBatchingManager.registerBatchableResource("BGP-RESOURCES", bgpResourcesBufferQ, resourceHandler);
80 static <T extends DataObject> void update(DataBroker broker, final LogicalDatastoreType datastoreType,
81 final InstanceIdentifier<T> path, final T data) {
82 ActionableResource actResource = new ActionableResourceImpl(path.toString());
83 actResource.setAction(ActionableResource.UPDATE);
84 actResource.setInstanceIdentifier(path);
85 actResource.setInstance(data);
86 bgpResourcesBufferQ.add(actResource);
89 public static <T extends DataObject> void write(DataBroker broker, final LogicalDatastoreType datastoreType,
90 final InstanceIdentifier<T> path, final T data) {
91 ActionableResource actResource = new ActionableResourceImpl(path.toString());
92 actResource.setAction(ActionableResource.CREATE);
93 actResource.setInstanceIdentifier(path);
94 actResource.setInstance(data);
95 bgpResourcesBufferQ.add(actResource);
98 static <T extends DataObject> void delete(DataBroker broker, final LogicalDatastoreType datastoreType,
99 final InstanceIdentifier<T> path) {
100 ActionableResource actResource = new ActionableResourceImpl(path.toString());
101 actResource.setAction(ActionableResource.DELETE);
102 actResource.setInstanceIdentifier(path);
103 actResource.setInstance(null);
104 bgpResourcesBufferQ.add(actResource);
107 public static void setBroker(final DataBroker broker) {
108 BgpUtil.dataBroker = broker;
109 initTransactionChain();
112 static synchronized void initTransactionChain() {
113 if (fibTransact != null) {
115 LOG.error("*** TxChain Close, *** Attempts: {}", txChainAttempts);
118 BgpUtil.fibTransact = dataBroker.createTransactionChain(new BgpUtilTransactionChainListener());
122 static class BgpUtilTransactionChainListener implements TransactionChainListener {
124 public void onTransactionChainFailed(TransactionChain<?, ?> transactionChain,
125 AsyncTransaction<?, ?> asyncTransaction, Throwable throwable) {
126 LOG.error("*** TxChain Creation Failed *** Attempts: {}", txChainAttempts);
127 initTransactionChain();
131 public void onTransactionChainSuccessful(TransactionChain<?, ?> transactionChain) {
132 LOG.trace("TxChain Creation Success");
136 public static DataBroker getBroker() {
140 // Convert ProtocolType to thrift protocol_type
141 public static protocol_type convertToThriftProtocolType(BgpControlPlaneType protocolType) {
142 // TODO: add implementation
143 return protocol_type.PROTOCOL_ANY;
146 // Convert EncapType to thrift encap_type
147 public static encap_type convertToThriftEncapType(EncapType encapType) {
148 // TODO: add implementation
149 return encap_type.MPLS;
152 static VpnInstanceOpDataEntry getVpnInstanceOpData(DataBroker broker, String rd) throws InterruptedException,
153 ExecutionException, TimeoutException {
154 InstanceIdentifier<VpnInstanceOpDataEntry> id = getVpnInstanceOpDataIdentifier(rd);
155 Optional<VpnInstanceOpDataEntry> vpnInstanceOpData = MDSALUtil.read(broker,
156 LogicalDatastoreType.OPERATIONAL, id);
157 if (vpnInstanceOpData.isPresent()) {
158 return vpnInstanceOpData.get();
163 public static InstanceIdentifier<VpnInstanceOpDataEntry> getVpnInstanceOpDataIdentifier(String rd) {
164 return InstanceIdentifier.builder(VpnInstanceOpData.class)
165 .child(VpnInstanceOpDataEntry.class, new VpnInstanceOpDataEntryKey(rd)).build();
168 public static String getVpnNameFromRd(DataBroker dataBroker2, String rd) {
169 InstanceIdentifier<VpnInstanceOpDataEntry> id = InstanceIdentifier.create(VpnInstanceOpData.class)
170 .child(VpnInstanceOpDataEntry.class,
171 new VpnInstanceOpDataEntryKey(rd));
173 Optional<VpnInstanceOpDataEntry> vpnInstanceOpData =
174 SingleTransactionDataBroker.syncReadOptional(dataBroker2, LogicalDatastoreType.OPERATIONAL, id);
175 return vpnInstanceOpData.isPresent() ? vpnInstanceOpData.get().getVpnInstanceName() : null;
176 } catch (ReadFailedException e) {
177 LOG.warn("Exception while retrieving VpnInstance name for RD {}", rd, e);