Bug 5510 - Modifying DataTreeChangeHandler
[groupbasedpolicy.git] / renderers / iovisor / src / main / java / org / opendaylight / groupbasedpolicy / renderer / iovisor / sf / ClassifierDefinitionListener.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.ClassifierDefinition;
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.SupportedClassifierDefinition;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.capabilities.SupportedClassifierDefinitionBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.capabilities.SupportedClassifierDefinitionKey;
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 ClassifierDefinitionListener extends DataTreeChangeHandler<ClassifierDefinition> {
37
38     private static final Logger LOG = LoggerFactory.getLogger(ClassifierDefinitionListener.class);
39
40     private static final String PUT = "stored";
41     private static final String DELETED = "removed";
42
43     private static final InstanceIdentifier<Capabilities> CAPABILITIES_IID = InstanceIdentifier.builder(Renderers.class)
44         .child(Renderer.class, new RendererKey(IovisorRenderer.RENDERER_NAME))
45         .child(Capabilities.class)
46         .build();
47
48     public ClassifierDefinitionListener(DataBroker dataBroker) {
49         super(dataBroker);
50         registerDataTreeChangeListener(new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION,
51                 InstanceIdentifier.builder(SubjectFeatureDefinitions.class).child(ClassifierDefinition.class).build()));
52     }
53
54     private SupportedClassifierDefinition createSupportedClassifierDefinition(Classifier classifier) {
55         SupportedClassifierDefinitionBuilder scdBuilder =
56                 new SupportedClassifierDefinitionBuilder().setClassifierDefinitionId(classifier.getId())
57                     .setSupportedParameterValues(classifier.getSupportedParameterValues());
58         if (classifier.getParent() != null) {
59             scdBuilder.setParentClassifierDefinitionId(classifier.getParent().getId());
60         }
61         return scdBuilder.build();
62     }
63
64     private FutureCallback<Void> logDebugResult(final SupportedClassifierDefinitionKey supportedClassifierDefinitionKey,
65             final String putOrDeleted) {
66         return new FutureCallback<Void>() {
67
68             @Override
69             public void onSuccess(Void result) {
70                 LOG.debug("Capability of renerer {} was {}: {}", IovisorRenderer.RENDERER_NAME.getValue(), putOrDeleted,
71                         supportedClassifierDefinitionKey.getClassifierDefinitionId().getValue());
72             }
73
74             @Override
75             public void onFailure(Throwable t) {
76                 LOG.error("Capability of renderer {} was NOT {}: {}", IovisorRenderer.RENDERER_NAME.getValue(),
77                         putOrDeleted, supportedClassifierDefinitionKey.getClassifierDefinitionId().getValue(), t);
78             }
79         };
80     }
81
82     @Override
83     protected void onWrite(DataObjectModification<ClassifierDefinition> rootNode,
84             InstanceIdentifier<ClassifierDefinition> rootIdentifier) {
85         onSubtreeModified(rootNode, rootIdentifier);
86     }
87
88     @Override
89     protected void onDelete(DataObjectModification<ClassifierDefinition> rootNode,
90             InstanceIdentifier<ClassifierDefinition> rootIdentifier) {
91         ClassifierDefinition classifierDefinitionBefore = checkNotNull(rootNode.getDataBefore());
92         Classifier ourClassifier = SubjectFeatures.getClassifier(classifierDefinitionBefore.getId());
93         if (ourClassifier != null) {
94             SupportedClassifierDefinitionKey supportedClassifierDefinitionKey =
95                     new SupportedClassifierDefinitionKey(ourClassifier.getId());
96             WriteTransaction wTx = dataProvider.newWriteOnlyTransaction();
97
98             wTx.delete(LogicalDatastoreType.OPERATIONAL,
99                     CAPABILITIES_IID.child(SupportedClassifierDefinition.class, supportedClassifierDefinitionKey));
100             Futures.addCallback(wTx.submit(), logDebugResult(supportedClassifierDefinitionKey, DELETED));
101         }
102     }
103
104     @Override
105     protected void onSubtreeModified(DataObjectModification<ClassifierDefinition> rootNode,
106             InstanceIdentifier<ClassifierDefinition> rootIdentifier) {
107         ClassifierDefinition classifierDefinitionAfter = checkNotNull(rootNode.getDataAfter());
108         Classifier ourClassifier = SubjectFeatures.getClassifier(classifierDefinitionAfter.getId());
109         if (ourClassifier != null) {
110             SupportedClassifierDefinition supportedClassifierDefinition =
111                     createSupportedClassifierDefinition(ourClassifier);
112             WriteTransaction wTx = dataProvider.newWriteOnlyTransaction();
113             wTx.put(LogicalDatastoreType.OPERATIONAL,
114                     CAPABILITIES_IID.child(SupportedClassifierDefinition.class, supportedClassifierDefinition.getKey()),
115                     supportedClassifierDefinition, true);
116             Futures.addCallback(wTx.submit(), logDebugResult(supportedClassifierDefinition.getKey(), PUT));
117         }
118     }
119 }