2 * Copyright (c) 2016 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
8 package org.opendaylight.controller.md.sal.dom.broker.impl.legacy.sharded.adapter;
10 import static java.util.Objects.requireNonNull;
12 import com.google.common.base.Preconditions;
13 import java.util.HashMap;
15 import java.util.concurrent.atomic.AtomicLong;
16 import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
17 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
18 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction;
19 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
20 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
21 import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
22 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction;
23 import org.opendaylight.mdsal.dom.api.DOMDataTreeTransaction;
24 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
25 import org.opendaylight.mdsal.dom.api.DOMTransactionChainListener;
26 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
29 class ShardedDOMDataBrokerDelegatingTransactionChain implements DOMTransactionChain, DOMTransactionChainListener {
30 private final Map<Object, AsyncTransaction<?, ?>> transactionMap = new HashMap<>();
31 private final org.opendaylight.mdsal.dom.api.DOMTransactionChain txChainDelegate;
32 private final SchemaContext schemaContext;
33 private final TransactionChainListener txChainListener;
34 private final Object txChainIdentifier;
35 private final AtomicLong txNum = new AtomicLong();
37 ShardedDOMDataBrokerDelegatingTransactionChain(final Object txChainIdentifier,
38 final SchemaContext schemaContext,
39 final org.opendaylight.mdsal.dom.api.DOMDataBroker
41 final TransactionChainListener txChainListener) {
42 requireNonNull(brokerDelegate);
43 this.schemaContext = requireNonNull(schemaContext);
44 this.txChainIdentifier = requireNonNull(txChainIdentifier);
45 this.txChainListener = requireNonNull(txChainListener);
46 this.txChainDelegate = brokerDelegate.createTransactionChain(this);
50 public DOMDataReadOnlyTransaction newReadOnlyTransaction() {
51 final DOMDataTreeReadTransaction readTxDelegate = txChainDelegate.newReadOnlyTransaction();
52 final DOMDataReadOnlyTransaction readTx = new ShardedDOMDataBrokerDelegatingReadTransaction(
53 newTransactionIdentifier(), readTxDelegate);
54 transactionMap.put(readTxDelegate.getIdentifier(), readTx);
60 public DOMDataReadWriteTransaction newReadWriteTransaction() {
61 final Object readWriteTxId = newTransactionIdentifier();
62 final DOMDataTreeReadTransaction readTxDelegate = txChainDelegate.newReadOnlyTransaction();
63 final DOMDataReadOnlyTransaction readTx = new ShardedDOMDataBrokerDelegatingReadTransaction(readWriteTxId,
66 final DOMDataTreeWriteTransaction writeTxDelegate = txChainDelegate.newWriteOnlyTransaction();
67 final DOMDataWriteTransaction writeTx = new ShardedDOMDataBrokerDelegatingWriteTransaction(readWriteTxId,
70 final DOMDataReadWriteTransaction readWriteTx = new ShardedDOMDataBrokerDelegatingReadWriteTransaction(
71 readWriteTxId, schemaContext, readTx, writeTx);
72 transactionMap.put(readTxDelegate.getIdentifier(), readWriteTx);
73 transactionMap.put(writeTxDelegate.getIdentifier(), readWriteTx);
79 public DOMDataWriteTransaction newWriteOnlyTransaction() {
80 final DOMDataTreeWriteTransaction writeTxDelegate = txChainDelegate.newWriteOnlyTransaction();
81 final DOMDataWriteTransaction writeTx = new ShardedDOMDataBrokerDelegatingWriteTransaction(
82 newTransactionIdentifier(), writeTxDelegate);
83 transactionMap.put(writeTxDelegate.getIdentifier(), writeTx);
90 txChainDelegate.close();
94 public void onTransactionChainFailed(final org.opendaylight.mdsal.dom.api.DOMTransactionChain chain,
95 final DOMDataTreeTransaction transaction, final Throwable cause) {
96 txChainListener.onTransactionChainFailed(this, transactionFromDelegate(transaction.getIdentifier()), cause);
100 public void onTransactionChainSuccessful(final org.opendaylight.mdsal.dom.api.DOMTransactionChain chain) {
101 txChainListener.onTransactionChainSuccessful(this);
104 private AsyncTransaction<?, ?> transactionFromDelegate(final Object delegateId) {
105 Preconditions.checkState(transactionMap.containsKey(delegateId),
106 "Delegate transaction {} is not present in transaction chain history", delegateId);
107 return transactionMap.get(delegateId);
110 private Object newTransactionIdentifier() {
111 return "DOM-CHAIN-" + txChainIdentifier + "-" + txNum.getAndIncrement();