Reorganize transactionChainHandler usage.
[netconf.git] / restconf / restconf-nb-rfc8040 / src / main / java / org / opendaylight / restconf / nb / rfc8040 / handlers / TransactionChainHandler.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, 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.restconf.nb.rfc8040.handlers;
9
10 import java.util.Objects;
11 import java.util.Queue;
12 import java.util.concurrent.ConcurrentLinkedQueue;
13 import javax.annotation.PreDestroy;
14 import javax.inject.Inject;
15 import javax.inject.Singleton;
16 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
17 import org.opendaylight.mdsal.dom.api.DOMDataTreeTransaction;
18 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
19 import org.opendaylight.mdsal.dom.api.DOMTransactionChainListener;
20 import org.opendaylight.restconf.common.errors.RestconfDocumentedException;
21 import org.slf4j.Logger;
22 import org.slf4j.LoggerFactory;
23
24 /**
25  * Implementation of {@link TransactionChainHandler}.
26  */
27 @Singleton
28 public class TransactionChainHandler implements Handler<DOMTransactionChain>, AutoCloseable {
29     private static final Logger LOG = LoggerFactory.getLogger(TransactionChainHandler.class);
30
31     private final DOMTransactionChainListener transactionChainListener = new DOMTransactionChainListener() {
32         @Override
33         public void onTransactionChainFailed(final DOMTransactionChain chain, final DOMDataTreeTransaction transaction,
34                 final Throwable cause) {
35             LOG.warn("TransactionChain({}) {} FAILED!", chain, transaction.getIdentifier(), cause);
36             transactionChainList.remove(chain);
37             throw new RestconfDocumentedException("TransactionChain(" + chain + ") not committed correctly", cause);
38         }
39
40         @Override
41         public void onTransactionChainSuccessful(final DOMTransactionChain chain) {
42             LOG.trace("TransactionChain({}) SUCCESSFUL", chain);
43             transactionChainList.remove(chain);
44         }
45     };
46
47     private final DOMDataBroker dataBroker;
48     private final Queue<DOMTransactionChain> transactionChainList;
49
50     /**
51      * Prepare transaction chain service for Restconf services.
52      */
53     @Inject
54     public TransactionChainHandler(final DOMDataBroker dataBroker) {
55         this.dataBroker = Objects.requireNonNull(dataBroker);
56         this.transactionChainList = new ConcurrentLinkedQueue<>();
57     }
58
59     /**
60      * Create and return new instance of object {@link DOMTransactionChain}.
61      * After use, is important to close transactionChain by method {@link DOMTransactionChain#close()}.
62      * @return new instance of object {@link DOMTransactionChain}
63      */
64     @Override
65     public DOMTransactionChain get() {
66         final DOMTransactionChain transactionChain = dataBroker.createTransactionChain(transactionChainListener);
67         this.transactionChainList.add(transactionChain);
68         LOG.trace("Started TransactionChain({})", transactionChain);
69         return transactionChain;
70     }
71
72     @Override
73     @PreDestroy
74     public synchronized void close() {
75         for (DOMTransactionChain transactionChain: this.transactionChainList) {
76             transactionChain.close();
77             LOG.trace("Closed TransactionChain({})", transactionChain);
78         }
79     }
80
81     /**
82      * Verify if {@link DOMTransactionChain} exist in {@link TransactionChainHandler} queue.
83      * @param transactionChain object to check.
84      * @return true if object still exist in {@link TransactionChainHandler}.
85      */
86     boolean verifyIfExistTransactionChain(DOMTransactionChain transactionChain) {
87         return this.transactionChainList.contains(transactionChain);
88     }
89 }