Fix transaction manager closing.
[openflowplugin.git] / applications / topology-manager / src / main / java / org / opendaylight / openflowplugin / applications / topology / manager / FlowCapableTopologyProvider.java
1 /*
2  * Copyright (c) 2013 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.openflowplugin.applications.topology.manager;
9
10 import com.google.common.base.Optional;
11 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
12 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
13 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
14 import org.opendaylight.controller.sal.binding.api.NotificationProviderService;
15 import org.opendaylight.openflowplugin.common.txchain.TransactionChainManager;
16 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
17 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
18 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
19 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
20 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
21 import org.opendaylight.yangtools.concepts.ListenerRegistration;
22 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
23 import org.opendaylight.yangtools.yang.binding.NotificationListener;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
26
27 public class FlowCapableTopologyProvider implements AutoCloseable {
28     private static final Logger LOG = LoggerFactory.getLogger(FlowCapableTopologyProvider.class);
29     private static final String TOPOLOGY_PROVIDER = "topology-provider";
30     static final String TOPOLOGY_ID = "flow:1";
31
32
33     private final DataBroker dataBroker;
34     private final NotificationProviderService notificationService;
35     private final OperationProcessor processor;
36     private TransactionChainManager transactionChainManager;
37     private ListenerRegistration<NotificationListener> listenerRegistration;
38
39     public FlowCapableTopologyProvider(DataBroker dataBroker, NotificationProviderService notificationService,
40             OperationProcessor processor) {
41         this.dataBroker = dataBroker;
42         this.notificationService = notificationService;
43         this.processor = processor;
44     }
45
46     /**
47      * Gets called on start of a bundle.
48      */
49     public void start() {
50         final TopologyKey key = new TopologyKey(new TopologyId(TOPOLOGY_ID));
51         final InstanceIdentifier<Topology> path = InstanceIdentifier
52                 .create(NetworkTopology.class)
53                 .child(Topology.class, key);
54
55         final FlowCapableTopologyExporter listener = new FlowCapableTopologyExporter(processor, path);
56         this.listenerRegistration = notificationService.registerNotificationListener(listener);
57         this.transactionChainManager = new TransactionChainManager(dataBroker, TOPOLOGY_PROVIDER);
58         this.transactionChainManager.activateTransactionManager();
59         this.transactionChainManager.initialSubmitWriteTransaction();
60
61         if(!isFlowTopologyExist(path)){
62             transactionChainManager.writeToTransaction(
63                     LogicalDatastoreType.OPERATIONAL,
64                     path,
65                     new TopologyBuilder().setKey(key).build(),
66                     true);
67             transactionChainManager.submitTransaction();
68         }
69
70         LOG.info("FlowCapableTopologyProvider started");
71     }
72
73     @Override
74     public void close() {
75         LOG.info("FlowCapableTopologyProvider stopped.");
76         this.transactionChainManager.close();
77         if (this.listenerRegistration != null) {
78             try {
79                 this.listenerRegistration.close();
80             } catch (Exception e) {
81                 LOG.warn("Failed to close listener registration: {}", e.getMessage());
82                 LOG.debug("Failed to close listener registration.. ", e);
83             }
84             listenerRegistration = null;
85         }
86     }
87
88     private boolean isFlowTopologyExist(final InstanceIdentifier<Topology> path) {
89         try {
90             Optional<Topology> ofTopology = this.transactionChainManager
91                     .readFromTransaction(LogicalDatastoreType.OPERATIONAL, path)
92                     .checkedGet();
93             LOG.debug("OpenFlow topology exist in the operational data store at {}",path);
94             if(ofTopology.isPresent()){
95                 return true;
96             }
97         } catch (ReadFailedException e) {
98             LOG.warn("OpenFlow topology read operation failed!", e);
99         }
100         return false;
101     }
102 }