Tests for Iovisor root package
[groupbasedpolicy.git] / renderers / iovisor / src / main / java / org / opendaylight / groupbasedpolicy / renderer / iovisor / IovisorRenderer.java
1 /*
2  * Copyright (c) 2015 Cisco Systems. 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.iovisor;
10
11 import java.util.Map.Entry;
12
13 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
14 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
15 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
16 import org.opendaylight.groupbasedpolicy.api.EpRendererAugmentationRegistry;
17 import org.opendaylight.groupbasedpolicy.api.PolicyValidatorRegistry;
18 import org.opendaylight.groupbasedpolicy.renderer.iovisor.endpoint.EndpointListener;
19 import org.opendaylight.groupbasedpolicy.renderer.iovisor.endpoint.EndpointManager;
20 import org.opendaylight.groupbasedpolicy.renderer.iovisor.sf.Action;
21 import org.opendaylight.groupbasedpolicy.renderer.iovisor.sf.ActionDefinitionListener;
22 import org.opendaylight.groupbasedpolicy.renderer.iovisor.sf.ClassifierDefinitionListener;
23 import org.opendaylight.groupbasedpolicy.renderer.iovisor.sf.SubjectFeatures;
24 import org.opendaylight.groupbasedpolicy.renderer.iovisor.utils.IovisorIidFactory;
25 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ActionDefinitionId;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.iovisor.rev151030.IovisorModuleInstances;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.RendererName;
29 import org.slf4j.Logger;
30 import org.slf4j.LoggerFactory;
31
32 import com.google.common.base.Optional;
33
34 /**
35  * Renderer that converts GBP services to IOVisor Agents
36  */
37 public class IovisorRenderer implements AutoCloseable {
38
39     private static final Logger LOG = LoggerFactory.getLogger(IovisorRenderer.class);
40
41     public static final RendererName RENDERER_NAME = new RendererName("Iovisor");
42     public static String IOVISOR_MODULE_LISTENER_BASE_URL = "/policies/";
43
44     private final ClassifierDefinitionListener classifierDefinitionListener;
45     private EndpointManager endpointManager;
46     private EndpointListener endpointListener;
47     private final IovisorResolvedEndpointListener resolvedEndpointListener;
48     private ActionDefinitionListener actionDefinitionListener;
49     private ResolvedPolicyListener resolvedPolicyListener;
50
51     private DataBroker dataBroker;
52
53     public IovisorRenderer(DataBroker passedDataBroker, EpRendererAugmentationRegistry epRendererAugmentationRegistry,
54             PolicyValidatorRegistry policyValidatorRegistry) {
55         if (passedDataBroker != null)
56             dataBroker = passedDataBroker;
57         String baseUrl = getIovisorModuleBaseUrl();
58         if (baseUrl != null) {
59             // TODO FIXME Need to set listener on BaseUrl
60             IOVISOR_MODULE_LISTENER_BASE_URL = baseUrl;
61         }
62         endpointManager = new EndpointManager(dataBroker, epRendererAugmentationRegistry);
63         endpointListener = new EndpointListener(dataBroker, endpointManager);
64         resolvedPolicyListener = new ResolvedPolicyListener(dataBroker, endpointManager.getIovisorModuleManager());
65         resolvedEndpointListener = new IovisorResolvedEndpointListener(dataBroker);
66         classifierDefinitionListener = new ClassifierDefinitionListener(dataBroker);
67         actionDefinitionListener = new ActionDefinitionListener(dataBroker);
68
69         for (Entry<ActionDefinitionId, Action> entry : SubjectFeatures.getActions().entrySet()) {
70             policyValidatorRegistry.register(entry.getKey(), entry.getValue());
71         }
72         LOG.info("Iovisor Renderer has Started");
73
74     }
75
76     public IovisorResolvedEndpointListener getResolvedEndpointListener() {
77         return resolvedEndpointListener;
78     }
79
80     public EndpointManager getEndPointManager() {
81         return endpointManager;
82     }
83
84     private String getIovisorModuleBaseUrl() {
85         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
86         Optional<IovisorModuleInstances> readFromDs = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
87                 IovisorIidFactory.iovisorModuleInstanceWildCardIid(), rTx);
88         if (readFromDs.isPresent()) {
89             return readFromDs.get().getBaseUrl();
90         }
91         return null;
92     }
93
94     @Override
95     public void close() throws Exception {
96         if (endpointManager != null)
97             endpointManager.close();
98         if (endpointListener != null)
99             endpointListener.close();
100         if (classifierDefinitionListener != null)
101             classifierDefinitionListener.close();
102         if (actionDefinitionListener != null)
103             actionDefinitionListener.close();
104         if (resolvedPolicyListener != null)
105             resolvedPolicyListener.close();
106         if (resolvedEndpointListener != null)
107             resolvedEndpointListener.close();
108
109     }
110 }