opendaylight/md-sal/mdsal-trace/dom-impl/src/main/java/org/opendaylight/controller/md/sal/trace/dom/impl/TracingTransactionChain.java

   1 /*
   2  * Copyright (c) 2017 Red Hat, Inc. 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 package org.opendaylight.controller.md.sal.trace.dom.impl;
   9
  10 import java.util.Objects;
  11 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction;
  12 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
  13 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
  14 import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
  15 import org.opendaylight.controller.md.sal.trace.closetracker.impl.AbstractCloseTracked;
  16 import org.opendaylight.controller.md.sal.trace.closetracker.impl.CloseTrackedRegistry;
  17
  18 class TracingTransactionChain extends AbstractCloseTracked<TracingTransactionChain> implements DOMTransactionChain {
  19
  20     private final DOMTransactionChain delegate;
  21     private final TracingBroker tracingBroker;
  22     private final CloseTrackedRegistry<TracingReadOnlyTransaction> readOnlyTransactionsRegistry;
  23     private final CloseTrackedRegistry<TracingWriteTransaction> writeTransactionsRegistry;
  24     private final CloseTrackedRegistry<TracingReadWriteTransaction> readWriteTransactionsRegistry;
  25
  26     TracingTransactionChain(DOMTransactionChain delegate, TracingBroker tracingBroker,
  27             CloseTrackedRegistry<TracingTransactionChain> transactionChainsRegistry) {
  28         super(transactionChainsRegistry);
  29         this.delegate = Objects.requireNonNull(delegate);
  30         this.tracingBroker = Objects.requireNonNull(tracingBroker);
  31
  32         final boolean isDebugging = transactionChainsRegistry.isDebugContextEnabled();
  33         this.readOnlyTransactionsRegistry  = new CloseTrackedRegistry<>(this, "newReadOnlyTransaction", isDebugging);
  34         this.writeTransactionsRegistry     = new CloseTrackedRegistry<>(this, "newWriteOnlyTransaction", isDebugging);
  35         this.readWriteTransactionsRegistry = new CloseTrackedRegistry<>(this, "newReadWriteTransaction", isDebugging);
  36     }
  37
  38     @Override
  39     @SuppressWarnings("resource")
  40     public DOMDataReadOnlyTransaction newReadOnlyTransaction() {
  41         final DOMDataReadOnlyTransaction tx = delegate.newReadOnlyTransaction();
  42         return new TracingReadOnlyTransaction(tx, tracingBroker, readOnlyTransactionsRegistry);
  43     }
  44
  45     @Override
  46     public DOMDataReadWriteTransaction newReadWriteTransaction() {
  47         return new TracingReadWriteTransaction(delegate.newReadWriteTransaction(), tracingBroker,
  48                 readWriteTransactionsRegistry);
  49     }
  50
  51     @Override
  52     public DOMDataWriteTransaction newWriteOnlyTransaction() {
  53         final DOMDataWriteTransaction tx = delegate.newWriteOnlyTransaction();
  54         return new TracingWriteTransaction(tx, tracingBroker, writeTransactionsRegistry);
  55     }
  56
  57     @Override
  58     public void close() {
  59         delegate.close();
  60         super.removeFromTrackedRegistry();
  61     }
  62
  63 }