Bug#1854 - Exit command in console causing OOM.
[controller.git] / opendaylight / md-sal / sal-dom-broker / src / main / java / org / opendaylight / controller / md / sal / dom / broker / impl / DOMDataBrokerImpl.java
1 /*
2  * Copyright (c) 2014 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.controller.md.sal.dom.broker.impl;
9
10 import static com.google.common.base.Preconditions.checkState;
11
12 import java.util.Map.Entry;
13 import java.util.concurrent.atomic.AtomicLong;
14
15 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
16 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
17 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
18 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
19 import org.opendaylight.controller.md.sal.dom.api.DOMDataChangeListener;
20 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
21 import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
22 import org.opendaylight.controller.sal.core.spi.data.DOMStore;
23 import org.opendaylight.controller.sal.core.spi.data.DOMStoreThreePhaseCommitCohort;
24 import org.opendaylight.controller.sal.core.spi.data.DOMStoreTransactionChain;
25 import org.opendaylight.yangtools.concepts.ListenerRegistration;
26 import org.opendaylight.yangtools.util.DurationStatsTracker;
27 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30
31 import com.google.common.base.Optional;
32 import com.google.common.collect.ImmutableMap;
33 import com.google.common.util.concurrent.CheckedFuture;
34 import com.google.common.util.concurrent.ListeningExecutorService;
35
36 public class DOMDataBrokerImpl extends AbstractDOMForwardedTransactionFactory<DOMStore> implements DOMDataBroker,
37         AutoCloseable {
38
39     private static final Logger LOG = LoggerFactory.getLogger(DOMDataBrokerImpl.class);
40
41     private final DOMDataCommitCoordinatorImpl coordinator;
42     private final AtomicLong txNum = new AtomicLong();
43     private final AtomicLong chainNum = new AtomicLong();
44     private volatile AutoCloseable closeable;
45
46     public DOMDataBrokerImpl(final ImmutableMap<LogicalDatastoreType, DOMStore> datastores,
47             final ListeningExecutorService executor) {
48         super(datastores);
49         this.coordinator = new DOMDataCommitCoordinatorImpl(executor);
50     }
51
52     public void setCloseable(AutoCloseable closeable) {
53         this.closeable = closeable;
54     }
55
56     public DurationStatsTracker getCommitStatsTracker() {
57         return coordinator.getCommitStatsTracker();
58     }
59
60     @Override
61     public void close() {
62         super.close();
63
64         if(closeable != null) {
65             try {
66                 closeable.close();
67             } catch(Exception e) {
68                 LOG.debug("Error closing instance", e);
69             }
70         }
71     }
72
73     @Override
74     protected Object newTransactionIdentifier() {
75         return "DOM-" + txNum.getAndIncrement();
76     }
77
78     @Override
79     public ListenerRegistration<DOMDataChangeListener> registerDataChangeListener(final LogicalDatastoreType store,
80             final YangInstanceIdentifier path, final DOMDataChangeListener listener, final DataChangeScope triggeringScope) {
81
82         DOMStore potentialStore = getTxFactories().get(store);
83         checkState(potentialStore != null, "Requested logical data store is not available.");
84         return potentialStore.registerChangeListener(path, listener, triggeringScope);
85     }
86
87     @Override
88     public DOMTransactionChain createTransactionChain(final TransactionChainListener listener) {
89         ImmutableMap.Builder<LogicalDatastoreType, DOMStoreTransactionChain> backingChainsBuilder = ImmutableMap
90                 .builder();
91         for (Entry<LogicalDatastoreType, DOMStore> entry : getTxFactories().entrySet()) {
92             backingChainsBuilder.put(entry.getKey(), entry.getValue().createTransactionChain());
93         }
94         long chainId = chainNum.getAndIncrement();
95         ImmutableMap<LogicalDatastoreType, DOMStoreTransactionChain> backingChains = backingChainsBuilder.build();
96         LOG.debug("Transactoin chain {} created with listener {}, backing store chains {}", chainId, listener,
97                 backingChains);
98         return new DOMDataBrokerTransactionChainImpl(chainId, backingChains, coordinator, listener);
99
100     }
101
102     @Override
103     public CheckedFuture<Void,TransactionCommitFailedException> submit(final DOMDataWriteTransaction transaction,
104             final Iterable<DOMStoreThreePhaseCommitCohort> cohorts) {
105         LOG.debug("Transaction: {} submitted with cohorts {}.", transaction.getIdentifier(), cohorts);
106         return coordinator.submit(transaction, cohorts, Optional.<DOMDataCommitErrorListener> absent());
107     }
108 }