renderers/ofoverlay/src/main/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/statistics/ResolvedPolicyClassifierListener.java

   1 /*
   2  * Copyright (c) 2016 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.groupbasedpolicy.renderer.ofoverlay.statistics;
   9
  10 import static com.google.common.base.Preconditions.checkNotNull;
  11
  12 import java.util.Collections;
  13 import java.util.HashMap;
  14 import java.util.List;
  15 import java.util.Map;
  16 import java.util.Map.Entry;
  17
  18 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  19 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
  20 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
  21 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  22 import org.opendaylight.groupbasedpolicy.util.DataTreeChangeHandler;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.ResolvedPolicies;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.has.classifiers.Classifier;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.has.resolved.rules.ResolvedRule;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.resolved.policies.ResolvedPolicy;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.resolved.policies.resolved.policy.PolicyRuleGroupWithEndpointConstraints;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.resolved.policy.rev150828.resolved.policies.resolved.policy.policy.rule.group.with.endpoint.constraints.PolicyRuleGroup;
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  30 import org.slf4j.Logger;
  31 import org.slf4j.LoggerFactory;
  32
  33 import com.google.common.collect.MapDifference;
  34 import com.google.common.collect.Maps;
  35
  36 public class ResolvedPolicyClassifierListener extends DataTreeChangeHandler<ResolvedPolicy> {
  37
  38     private static final Logger LOG = LoggerFactory.getLogger(ResolvedPolicyClassifierListener.class);
  39     private final OFStatisticsManager ofStatsManager;
  40
  41     public ResolvedPolicyClassifierListener(DataBroker dataProvider, OFStatisticsManager ofStatsManager) {
  42         super(dataProvider);
  43         this.ofStatsManager = checkNotNull(ofStatsManager);
  44         registerDataTreeChangeListener(new DataTreeIdentifier<>(LogicalDatastoreType.OPERATIONAL,
  45                 InstanceIdentifier.builder(ResolvedPolicies.class).child(ResolvedPolicy.class).build()));
  46     }
  47
  48     @Override
  49     protected void onWrite(DataObjectModification<ResolvedPolicy> rootNode,
  50             InstanceIdentifier<ResolvedPolicy> rootIdentifier) {
  51         ResolvedPolicy resolvedPolicy = rootNode.getDataAfter();
  52         Map<InstanceIdentifier<Classifier>, Classifier> classifierByIid =
  53                 resolveClassifiers(resolvedPolicy, rootIdentifier);
  54         for (Entry<InstanceIdentifier<Classifier>, Classifier> classifierEntry : classifierByIid.entrySet()) {
  55             LOG.trace("New classifier created: {}\n{}", classifierEntry.getKey(),
  56                     classifierEntry.getValue());
  57             ofStatsManager.pullStatsForClassifier(classifierEntry.getKey(),
  58                     classifierEntry.getValue());
  59         }
  60     }
  61
  62     @Override
  63     protected void onDelete(DataObjectModification<ResolvedPolicy> rootNode,
  64             InstanceIdentifier<ResolvedPolicy> rootIdentifier) {
  65         LOG.debug("Delete is not supported yet.");
  66     }
  67
  68     @Override
  69     protected void onSubtreeModified(DataObjectModification<ResolvedPolicy> rootNode,
  70             InstanceIdentifier<ResolvedPolicy> rootIdentifier) {
  71         ResolvedPolicy resolvedPolicyAfter = rootNode.getDataAfter();
  72         ResolvedPolicy resolvedPolicyBefore = rootNode.getDataBefore();
  73         Map<InstanceIdentifier<Classifier>, Classifier> classifierByIidAfter =
  74                 resolveClassifiers(resolvedPolicyAfter, rootIdentifier);
  75         Map<InstanceIdentifier<Classifier>, Classifier> classifierByIidBefore =
  76                 resolveClassifiers(resolvedPolicyBefore, rootIdentifier);
  77         MapDifference<InstanceIdentifier<Classifier>, Classifier> difference =
  78                 Maps.difference(classifierByIidBefore, classifierByIidAfter);
  79         Map<InstanceIdentifier<Classifier>, Classifier> createdClassifierByIid = difference.entriesOnlyOnRight();
  80         for (Entry<InstanceIdentifier<Classifier>, Classifier> createdClassfierEntry : createdClassifierByIid
  81             .entrySet()) {
  82             LOG.trace("New classifier created: {}\n{}", createdClassfierEntry.getKey(),
  83                     createdClassfierEntry.getValue());
  84             ofStatsManager.pullStatsForClassifier(createdClassfierEntry.getKey(), createdClassfierEntry.getValue());
  85         }
  86         // TODO missing impl for case when classifier is changed or removed
  87     }
  88
  89     // TODO move to some util class
  90     public static Map<InstanceIdentifier<Classifier>, Classifier> resolveClassifiers(ResolvedPolicy resolvedPolicy,
  91             InstanceIdentifier<ResolvedPolicy> resolvedPolicyIid) {
  92         List<PolicyRuleGroupWithEndpointConstraints> policyRgWithEcs =
  93                 resolvedPolicy.getPolicyRuleGroupWithEndpointConstraints();
  94         if (policyRgWithEcs == null) {
  95             return Collections.emptyMap();
  96         }
  97         Map<InstanceIdentifier<Classifier>, Classifier> result = new HashMap<>();
  98         for (PolicyRuleGroupWithEndpointConstraints policyRgWithEc : policyRgWithEcs) {
  99             List<PolicyRuleGroup> policyRuleGroups = policyRgWithEc.getPolicyRuleGroup();
 100             if (policyRuleGroups == null) {
 101                 continue;
 102             }
 103             for (PolicyRuleGroup policyRuleGroup : policyRuleGroups) {
 104                 List<ResolvedRule> resolvedRules = policyRuleGroup.getResolvedRule();
 105                 if (resolvedRules == null) {
 106                     continue;
 107                 }
 108                 for (ResolvedRule resolvedRule : resolvedRules) {
 109                     List<Classifier> classifiers = resolvedRule.getClassifier();
 110                     if (classifiers == null) {
 111                         continue;
 112                     }
 113                     for (Classifier classifier : classifiers) {
 114                         InstanceIdentifier<Classifier> classifierIid = resolvedPolicyIid.builder()
 115                             .child(PolicyRuleGroupWithEndpointConstraints.class)
 116                             .child(PolicyRuleGroup.class, policyRuleGroup.getKey())
 117                             .child(ResolvedRule.class, resolvedRule.getKey())
 118                             .child(Classifier.class, classifier.getKey())
 119                             .build();
 120                         result.put(classifierIid, classifier);
 121                     }
 122                 }
 123             }
 124         }
 125         return result;
 126     }
 127
 128 }