Bug 9060: TracingBroker with transaction-debug-context-enabled

[controller.git] / opendaylight / md-sal / mdsal-trace / dom-impl / src / main / java / org / opendaylight / controller / md / sal / trace / dom / impl / TracingWriteTransaction.java

diff --git a/opendaylight/md-sal/mdsal-trace/dom-impl/src/main/java/org/opendaylight/controller/md/sal/trace/dom/impl/TracingWriteTransaction.java b/opendaylight/md-sal/mdsal-trace/dom-impl/src/main/java/org/opendaylight/controller/md/sal/trace/dom/impl/TracingWriteTransaction.java
index a895e983c26693dbed48619adea0e33f15893131..123129ce156eeb08781987cf5e4a848f1d46cb52 100644 (file)
--- a/opendaylight/md-sal/mdsal-trace/dom-impl/src/main/java/org/opendaylight/controller/md/sal/trace/dom/impl/TracingWriteTransaction.java
+++ b/opendaylight/md-sal/mdsal-trace/dom-impl/src/main/java/org/opendaylight/controller/md/sal/trace/dom/impl/TracingWriteTransaction.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016 Red Hat, Inc. and others.  All rights reserved.
+ * Copyright (c) 2017 Red Hat, Inc. and others. All rights reserved.
  *
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
@@ -7,111 +7,54 @@
  */
 package org.opendaylight.controller.md.sal.trace.dom.impl;
 
-import com.google.common.collect.ImmutableSet;
 import com.google.common.util.concurrent.CheckedFuture;
 import com.google.common.util.concurrent.ListenableFuture;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Objects;
-import java.util.Set;
-
+import java.time.Instant;
 import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
-import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
+import org.opendaylight.controller.md.sal.trace.closetracker.impl.CloseTracked;
+import org.opendaylight.controller.md.sal.trace.closetracker.impl.CloseTrackedRegistry;
+import org.opendaylight.controller.md.sal.trace.closetracker.impl.CloseTrackedTrait;
 import org.opendaylight.yangtools.yang.common.RpcResult;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
-import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
-
-class TracingWriteTransaction implements DOMDataWriteTransaction {
-
-    private final DOMDataWriteTransaction delegate;
-    private final TracingBroker tracingBroker;
-    private final List<String> logs = new ArrayList<>();
-
-    TracingWriteTransaction(DOMDataWriteTransaction delegate, TracingBroker tracingBroker) {
-        this.delegate = Objects.requireNonNull(delegate);
-        this.tracingBroker = Objects.requireNonNull(tracingBroker);
-    }
-
-    private void recordOp(LogicalDatastoreType store, YangInstanceIdentifier yiid,
-                                                                            String method, NormalizedNode<?,?> node) {
-        if (!tracingBroker.isWriteWatched(yiid, store)) {
-            return;
-        }
 
-        Object value = null;
-        if (node != null) {
-            value = node.getValue();
-        }
+class TracingWriteTransaction extends AbstractTracingWriteTransaction
+        implements CloseTracked<TracingWriteTransaction> {
 
-        if (value != null && value instanceof ImmutableSet && ((Set)value).isEmpty()) {
-            if (tracingBroker.LOG.isDebugEnabled()) {
-                tracingBroker.LOG.debug("Empty data set write to {}", tracingBroker.toPathString(yiid));
-            }
-        } else {
-            StringBuilder sb = new StringBuilder();
-            sb.append("Method \"").append(method);
-            sb.append("\" to ").append(store);
-            sb.append(" at ").append(tracingBroker.toPathString(yiid)).append('.');
-            sb.append(" Data: ");
-            if (node != null) {
-                sb.append(node.getValue());
-            } else {
-                sb.append("null");
-            }
-            sb.append(" Stack:").append(tracingBroker.getStackSummary());
-            synchronized (this) {
-                logs.add(sb.toString());
-            }
-        }
-    }
+    private final CloseTrackedTrait<TracingWriteTransaction> closeTracker;
 
-    private synchronized void logOps() {
-        for (String log : logs) {
-            tracingBroker.LOG.warn(log);
-        }
-        logs.clear();
+    TracingWriteTransaction(DOMDataWriteTransaction delegate, TracingBroker tracingBroker,
+            CloseTrackedRegistry<TracingWriteTransaction> writeTransactionsRegistry) {
+        super(delegate, tracingBroker);
+        this.closeTracker = new CloseTrackedTrait<>(writeTransactionsRegistry);
     }
 
     @Override
-    public void put(LogicalDatastoreType store, YangInstanceIdentifier yiid, NormalizedNode<?, ?> node) {
-        recordOp(store, yiid, "put", node);
-        delegate.put(store, yiid, node);
+    public CheckedFuture<Void, TransactionCommitFailedException> submit() {
+        closeTracker.removeFromTrackedRegistry();
+        return super.submit();
     }
 
     @Override
-    public void merge(LogicalDatastoreType store, YangInstanceIdentifier yiid, NormalizedNode<?, ?> node) {
-        recordOp(store, yiid, "merge", node);
-        delegate.merge(store, yiid, node);
+    public ListenableFuture<RpcResult<TransactionStatus>> commit() {
+        closeTracker.removeFromTrackedRegistry();
+        return super.commit();
     }
 
     @Override
     public boolean cancel() {
-        logs.clear();
-        return delegate.cancel();
-    }
-
-    @Override
-    public void delete(LogicalDatastoreType store, YangInstanceIdentifier yiid) {
-        recordOp(store, yiid, "delete", null);
-        delegate.delete(store, yiid);
+        closeTracker.removeFromTrackedRegistry();
+        return super.cancel();
     }
 
     @Override
-    public CheckedFuture<Void, TransactionCommitFailedException> submit() {
-        logOps();
-        return delegate.submit();
+    public Instant getObjectCreated() {
+        return closeTracker.getObjectCreated();
     }
 
     @Override
-    public ListenableFuture<RpcResult<TransactionStatus>> commit() {
-        logOps();
-        return delegate.commit();
+    public Throwable getAllocationContext() {
+        return closeTracker.getAllocationContext();
     }
 
-    @Override
-    public Object getIdentifier() {
-        return delegate.getIdentifier();
-    }
 }