Bug 5510 - Modifying DataTreeChangeHandler
[groupbasedpolicy.git] / renderers / iovisor / src / main / java / org / opendaylight / groupbasedpolicy / renderer / iovisor / sf / ActionDefinitionListener.java
1 /*
2  * Copyright (c) 2015 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.iovisor.sf;
10
11 import static com.google.common.base.Preconditions.checkNotNull;
12
13 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
14 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
15 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
18 import org.opendaylight.groupbasedpolicy.renderer.iovisor.IovisorRenderer;
19 import org.opendaylight.groupbasedpolicy.util.DataTreeChangeHandler;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.SubjectFeatureDefinitions;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.policy.rev140421.subject.feature.definitions.ActionDefinition;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.Renderers;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.Renderer;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.RendererKey;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.Capabilities;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.capabilities.SupportedActionDefinition;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.capabilities.SupportedActionDefinitionBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.capabilities.SupportedActionDefinitionKey;
29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32
33 import com.google.common.util.concurrent.FutureCallback;
34 import com.google.common.util.concurrent.Futures;
35
36 public class ActionDefinitionListener extends DataTreeChangeHandler<ActionDefinition> {
37
38     private static final Logger LOG = LoggerFactory.getLogger(ActionDefinitionListener.class);
39     private static final InstanceIdentifier<Capabilities> CAPABILITIES_IID = InstanceIdentifier.builder(Renderers.class)
40         .child(Renderer.class, new RendererKey(IovisorRenderer.RENDERER_NAME))
41         .child(Capabilities.class)
42         .build();
43     private static String PUT = "stored";
44     private static String DELETED = "removed";
45
46     public ActionDefinitionListener(DataBroker dataBroker) {
47         super(dataBroker);
48         registerDataTreeChangeListener(new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION,
49                 InstanceIdentifier.builder(SubjectFeatureDefinitions.class).child(ActionDefinition.class).build()));
50     }
51
52     private SupportedActionDefinition createSupportedActionDefinition(Action action) {
53         return new SupportedActionDefinitionBuilder().setActionDefinitionId(action.getId())
54             .setSupportedParameterValues(action.getSupportedParameterValues())
55             .build();
56     }
57
58     private FutureCallback<Void> logDebugResult(final SupportedActionDefinitionKey supportedActionDefinitionKey,
59             final String putOrDeleted) {
60         return new FutureCallback<Void>() {
61
62             @Override
63             public void onSuccess(Void result) {
64                 LOG.debug("Capability of renerer {} was {}: {}", IovisorRenderer.RENDERER_NAME.getValue(), putOrDeleted,
65                         supportedActionDefinitionKey.getActionDefinitionId().getValue());
66             }
67
68             @Override
69             public void onFailure(Throwable t) {
70                 LOG.error("Capability of renderer {} was NOT {}: {}", IovisorRenderer.RENDERER_NAME.getValue(),
71                         putOrDeleted, supportedActionDefinitionKey.getActionDefinitionId().getValue(), t);
72             }
73         };
74     }
75
76     @Override
77     protected void onWrite(DataObjectModification<ActionDefinition> rootNode,
78             InstanceIdentifier<ActionDefinition> rootIdentifier) {
79         onSubtreeModified(rootNode, rootIdentifier);
80     }
81
82     @Override
83     protected void onDelete(DataObjectModification<ActionDefinition> rootNode,
84             InstanceIdentifier<ActionDefinition> rootIdentifier) {
85         ActionDefinition actionDefinitionBefore = checkNotNull(rootNode.getDataBefore());
86         Action ourAction = SubjectFeatures.getAction(actionDefinitionBefore.getId());
87         if (ourAction != null) {
88             SupportedActionDefinitionKey supportedActionDefinitionKey =
89                     new SupportedActionDefinitionKey(ourAction.getId());
90             WriteTransaction wTx = dataProvider.newWriteOnlyTransaction();
91             wTx.delete(LogicalDatastoreType.OPERATIONAL,
92                     CAPABILITIES_IID.child(SupportedActionDefinition.class, supportedActionDefinitionKey));
93             Futures.addCallback(wTx.submit(), logDebugResult(supportedActionDefinitionKey, DELETED));
94         }
95     }
96
97     @Override
98     protected void onSubtreeModified(DataObjectModification<ActionDefinition> rootNode,
99             InstanceIdentifier<ActionDefinition> rootIdentifier) {
100         ActionDefinition actionDefinitionAfter = checkNotNull(rootNode.getDataAfter());
101         Action ourAction = SubjectFeatures.getAction(actionDefinitionAfter.getId());
102         if (ourAction != null) {
103             SupportedActionDefinition supportedActionDefinition = createSupportedActionDefinition(ourAction);
104             WriteTransaction wTx = dataProvider.newWriteOnlyTransaction();
105             wTx.put(LogicalDatastoreType.OPERATIONAL,
106                     CAPABILITIES_IID.child(SupportedActionDefinition.class, supportedActionDefinition.getKey()),
107                     supportedActionDefinition, true);
108             Futures.addCallback(wTx.submit(), logDebugResult(supportedActionDefinition.getKey(), PUT));
109         }
110     }
111 }