2 * Copyright (c) 2015 Cisco Systems, Inc. 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
9 package org.opendaylight.ovsdb.southbound.transactions.md;
11 import com.google.common.util.concurrent.FutureCallback;
12 import com.google.common.util.concurrent.Futures;
13 import com.google.common.util.concurrent.ThreadFactoryBuilder;
14 import java.util.ArrayList;
15 import java.util.HashMap;
16 import java.util.List;
18 import java.util.concurrent.BlockingQueue;
19 import java.util.concurrent.ExecutorService;
20 import java.util.concurrent.Executors;
21 import java.util.concurrent.LinkedBlockingQueue;
22 import java.util.concurrent.ThreadFactory;
23 import java.util.concurrent.TimeUnit;
24 import java.util.concurrent.atomic.AtomicBoolean;
25 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
26 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
27 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
28 import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
29 import org.opendaylight.controller.md.sal.common.api.data.TransactionChain;
30 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
35 public class TransactionInvokerImpl implements TransactionInvoker,TransactionChainListener, Runnable, AutoCloseable {
36 private static final Logger LOG = LoggerFactory.getLogger(TransactionInvokerImpl.class);
37 private static final int QUEUE_SIZE = 10000;
38 private BindingTransactionChain chain;
39 private DataBroker db;
40 private BlockingQueue<TransactionCommand> inputQueue = new LinkedBlockingQueue<>(QUEUE_SIZE);
41 private BlockingQueue<ReadWriteTransaction> successfulTransactionQueue
42 = new LinkedBlockingQueue<>(QUEUE_SIZE);
43 private BlockingQueue<AsyncTransaction<?, ?>> failedTransactionQueue
44 = new LinkedBlockingQueue<>(QUEUE_SIZE);
45 private ExecutorService executor;
46 private Map<ReadWriteTransaction,TransactionCommand> transactionToCommand
48 private List<ReadWriteTransaction> pendingTransactions = new ArrayList<>();
49 private final AtomicBoolean runTask = new AtomicBoolean( true );
51 public TransactionInvokerImpl(DataBroker db) {
53 this.chain = db.createTransactionChain(this);
54 ThreadFactory threadFact = new ThreadFactoryBuilder().setNameFormat("transaction-invoker-impl-%d").build();
55 executor = Executors.newSingleThreadExecutor(threadFact);
56 executor.submit(this);
60 public void invoke(final TransactionCommand command) {
61 // TODO what do we do if queue is full?
62 inputQueue.offer(command);
66 public void onTransactionChainFailed(TransactionChain<?, ?> chain,
67 AsyncTransaction<?, ?> transaction, Throwable cause) {
68 failedTransactionQueue.offer(transaction);
72 public void onTransactionChainSuccessful(TransactionChain<?, ?> chain) {
79 while (runTask.get()) {
80 forgetSuccessfulTransactions();
82 List<TransactionCommand> commands = extractCommands();
83 for (TransactionCommand command: commands) {
84 final ReadWriteTransaction transaction = chain.newReadWriteTransaction();
85 recordPendingTransaction(command, transaction);
86 command.execute(transaction);
87 Futures.addCallback(transaction.submit(), new FutureCallback<Void>() {
89 public void onSuccess(final Void result) {
90 successfulTransactionQueue.offer(transaction);
94 public void onFailure(final Throwable throwable) {
95 // NOOP - handled by failure of transaction chain
99 } catch (Exception e) {
100 LOG.warn("Exception invoking Transaction: ", e);
105 private List<TransactionCommand> extractResubmitCommands() {
106 AsyncTransaction<?, ?> transaction = failedTransactionQueue.poll();
107 List<TransactionCommand> commands = new ArrayList<>();
108 if (transaction != null) {
109 int index = pendingTransactions.lastIndexOf(transaction);
110 List<ReadWriteTransaction> transactions =
111 pendingTransactions.subList(index, pendingTransactions.size() - 1);
112 for (ReadWriteTransaction tx: transactions) {
113 commands.add(transactionToCommand.get(tx));
115 resetTransactionQueue();
120 private void resetTransactionQueue() {
122 chain = db.createTransactionChain(this);
123 pendingTransactions = new ArrayList<>();
124 transactionToCommand = new HashMap<>();
125 failedTransactionQueue.clear();
126 successfulTransactionQueue.clear();
129 private void recordPendingTransaction(TransactionCommand command,
130 final ReadWriteTransaction transaction) {
131 transactionToCommand.put(transaction, command);
132 pendingTransactions.add(transaction);
135 private List<TransactionCommand> extractCommands() throws InterruptedException {
136 List<TransactionCommand> commands = extractResubmitCommands();
137 commands.addAll(extractCommandsFromQueue());
141 private List<TransactionCommand> extractCommandsFromQueue() throws InterruptedException {
142 List<TransactionCommand> result = new ArrayList<>();
143 TransactionCommand command = inputQueue.take();
144 while (command != null) {
146 command = inputQueue.poll();
151 private void forgetSuccessfulTransactions() {
152 ReadWriteTransaction transaction = successfulTransactionQueue.poll();
153 while (transaction != null) {
154 pendingTransactions.remove(transaction);
155 transactionToCommand.remove(transaction);
156 transaction = successfulTransactionQueue.poll();
161 public void close() throws Exception {
162 this.executor.shutdown();
163 if (!this.executor.awaitTermination(1, TimeUnit.SECONDS)) {
165 this.executor.shutdownNow();