Merge "Fixed main issues of SwitchManager"
[groupbasedpolicy.git] / renderers / ofoverlay / src / main / java / org / opendaylight / groupbasedpolicy / renderer / ofoverlay / OFOverlayRenderer.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
9 package org.opendaylight.groupbasedpolicy.renderer.ofoverlay;
10
11 import java.util.concurrent.Executors;
12 import java.util.concurrent.ScheduledExecutorService;
13
14 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
15 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
16 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
17 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
19 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
20 import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.node.SwitchManager;
21 import org.opendaylight.groupbasedpolicy.resolver.PolicyResolver;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.OfOverlayConfig;
23 import org.opendaylight.yangtools.concepts.ListenerRegistration;
24 import org.opendaylight.yangtools.yang.binding.DataObject;
25 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
26 import org.slf4j.Logger;
27 import org.slf4j.LoggerFactory;
28
29 import com.google.common.base.Optional;
30 import com.google.common.util.concurrent.FutureCallback;
31 import com.google.common.util.concurrent.Futures;
32 import com.google.common.util.concurrent.ListenableFuture;
33
34 /**
35  * Renderer that uses OpenFlow and OVSDB to implement an overlay network
36  * using Open vSwitch.
37  * @author readams
38  */
39 public class OFOverlayRenderer implements AutoCloseable, DataChangeListener {
40     private static final Logger LOG =
41             LoggerFactory.getLogger(OFOverlayRenderer.class);
42
43     private final DataBroker dataBroker;
44     private final PolicyResolver policyResolver;
45     private final SwitchManager switchManager;
46     private final EndpointManager endpointManager;
47     private final PolicyManager policyManager;
48
49     private final ScheduledExecutorService executor;
50
51     private static final InstanceIdentifier<OfOverlayConfig> configIid =
52             InstanceIdentifier.builder(OfOverlayConfig.class).build();
53
54     private OfOverlayConfig config;
55     ListenerRegistration<DataChangeListener> configReg;
56
57     public OFOverlayRenderer(DataBroker dataProvider,
58                              RpcProviderRegistry rpcRegistry) {
59         super();
60         this.dataBroker = dataProvider;
61
62         int numCPU = Runtime.getRuntime().availableProcessors();
63         //TODO: Consider moving to groupbasedpolicy-ofoverlay-config so as to be user configurable in distribution.
64         executor = Executors.newScheduledThreadPool(numCPU * 2);
65
66         switchManager = new SwitchManager(dataProvider);
67         endpointManager = new EndpointManager(dataProvider, rpcRegistry,
68                                               executor, switchManager);
69         policyResolver = new PolicyResolver(dataProvider, executor);
70
71         policyManager = new PolicyManager(dataProvider,
72                                           policyResolver,
73                                           switchManager,
74                                           endpointManager,
75                                           rpcRegistry,
76                                           executor);
77
78         configReg =
79                 dataProvider.registerDataChangeListener(LogicalDatastoreType.CONFIGURATION,
80                                                         configIid,
81                                                         this,
82                                                         DataChangeScope.SUBTREE);
83         readConfig();
84         LOG.info("Initialized OFOverlay renderer");
85
86     }
87
88     // *************
89     // AutoCloseable
90     // *************
91
92     @Override
93     public void close() throws Exception {
94         executor.shutdownNow();
95         if (configReg != null) configReg.close();
96         if (policyResolver != null) policyResolver.close();
97         if (switchManager != null) switchManager.close();
98         if (endpointManager != null) endpointManager.close();
99     }
100
101     // ******************
102     // DataChangeListener
103     // ******************
104
105     @Override
106     public void onDataChanged(AsyncDataChangeEvent<InstanceIdentifier<?>,
107                                                    DataObject> change) {
108         readConfig();
109     }
110
111     // **************
112     // Implementation
113     // **************
114
115     private void readConfig() {
116         ListenableFuture<Optional<OfOverlayConfig>> dao =
117                 dataBroker.newReadOnlyTransaction()
118                     .read(LogicalDatastoreType.CONFIGURATION, configIid);
119         Futures.addCallback(dao, new FutureCallback<Optional<OfOverlayConfig>>() {
120             @Override
121             public void onSuccess(final Optional<OfOverlayConfig> result) {
122                 if (!result.isPresent()) return;
123                 if (result.get() instanceof OfOverlayConfig) {
124                     config = result.get();
125                     applyConfig();
126                 }
127             }
128
129             @Override
130             public void onFailure(Throwable t) {
131                 LOG.error("Failed to read configuration", t);
132             }
133         }, executor);
134     }
135
136     private void applyConfig() {
137         switchManager.setEncapsulationFormat(config.getEncapsulationFormat());
138         endpointManager.setLearningMode(config.getLearningMode());
139         policyManager.setLearningMode(config.getLearningMode());
140     }
141 }