From 71c9e9dfa87bf2bba5650c97012de6b923f2b973 Mon Sep 17 00:00:00 2001
From: Vishal Thapar <vthapar@redhat.com>
Date: Mon, 28 May 2018 11:48:36 +0530
Subject: [PATCH] QoS: Add support for direction

* Add direction support for Bandwidth Limit rules

JIRA: NETVIRT-1242
Change-Id: I3a42344e909213ba6af10f5b7d7250ad7a8213b7
Signed-off-by: Vishal Thapar <vthapar@redhat.com>
---
 .../neutron/constants/rev150712/NeutronUtils.java |  2 ++
 model/src/main/yang/neutron-qos.yang              |  6 ++++++
 .../neutron/spi/NeutronQosBandwidthLimitRule.java | 15 +++++++++++++++
 .../transcriber/NeutronQosPolicyInterface.java    |  4 +++-
 4 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/model/src/main/java/org/opendaylight/yang/gen/v1/urn/opendaylight/neutron/constants/rev150712/NeutronUtils.java b/model/src/main/java/org/opendaylight/yang/gen/v1/urn/opendaylight/neutron/constants/rev150712/NeutronUtils.java
index e71d74b8f..37fd64676 100644
--- a/model/src/main/java/org/opendaylight/yang/gen/v1/urn/opendaylight/neutron/constants/rev150712/NeutronUtils.java
+++ b/model/src/main/java/org/opendaylight/yang/gen/v1/urn/opendaylight/neutron/constants/rev150712/NeutronUtils.java
@@ -81,8 +81,10 @@ public final class NeutronUtils {
         public static Class<? extends DirectionBase> get(String key) {
             return MAPPER.get(key);
         }
+
         public static String getDirectionString(Class<? extends DirectionBase> direction) {
             return MAPPER_INVERSE.get(direction);
         }
+
     }
 }
diff --git a/model/src/main/yang/neutron-qos.yang b/model/src/main/yang/neutron-qos.yang
index 29e522bd5..879e9cf52 100644
--- a/model/src/main/yang/neutron-qos.yang
+++ b/model/src/main/yang/neutron-qos.yang
@@ -52,6 +52,12 @@ module neutron-qos {
             type uint64;
             description "The burst over the maximum KBPS value";
         }
+        leaf direction {
+            type identityref {
+                base "constants:direction-base";
+            }
+            description "The traffic direction";
+        }
     }
 
     grouping dscpmarking-rule-attributes {
diff --git a/neutron-spi/src/main/java/org/opendaylight/neutron/spi/NeutronQosBandwidthLimitRule.java b/neutron-spi/src/main/java/org/opendaylight/neutron/spi/NeutronQosBandwidthLimitRule.java
index 52c4a1e9f..86ee5b6df 100644
--- a/neutron-spi/src/main/java/org/opendaylight/neutron/spi/NeutronQosBandwidthLimitRule.java
+++ b/neutron-spi/src/main/java/org/opendaylight/neutron/spi/NeutronQosBandwidthLimitRule.java
@@ -25,6 +25,9 @@ public final class NeutronQosBandwidthLimitRule extends NeutronObject<NeutronQos
     @XmlElement(name = "max_burst_kbps")
     BigInteger maxBurstKbps;
 
+    @XmlElement(defaultValue = "egress", name = "direction")
+    String direction;
+
     public BigInteger getMaxKbps() {
         return maxKbps;
     }
@@ -41,6 +44,14 @@ public final class NeutronQosBandwidthLimitRule extends NeutronObject<NeutronQos
         this.maxBurstKbps = maxBurstKbps;
     }
 
+    public String getDirection() {
+        return direction;
+    }
+
+    public void setDirection(String direction) {
+        this.direction = direction;
+    }
+
     @Override
     public boolean extractField(String field, NeutronQosBandwidthLimitRule ans) {
         switch (field) {
@@ -50,6 +61,9 @@ public final class NeutronQosBandwidthLimitRule extends NeutronObject<NeutronQos
             case "max_burst_kbps":
                 ans.setMaxBurstKbps(this.getMaxBurstKbps());
                 break;
+            case "direction":
+                ans.setDirection(this.getDirection());
+                break;
             default:
                 return super.extractField(field, ans);
         }
@@ -61,6 +75,7 @@ public final class NeutronQosBandwidthLimitRule extends NeutronObject<NeutronQos
         return "qosBandwidthLimitRules{" + "qosBandwidthLimitRuleUUID='" + uuid + '\''
             + ", qosBandwidthLimitRuleTenantID='" + tenantID + '\'' + ", qosBandwidthLimitRuleMaxValue='" + maxKbps
             + '\'' + ", qosBandwidthLimitRuleMaxBurst='" + maxBurstKbps
+            + '\'' + ", qosBandwidthLimitRuleDirection='" + direction
             + '\''
             + '}';
     }
diff --git a/transcriber/src/main/java/org/opendaylight/neutron/transcriber/NeutronQosPolicyInterface.java b/transcriber/src/main/java/org/opendaylight/neutron/transcriber/NeutronQosPolicyInterface.java
index 7b549989e..3cc30234f 100644
--- a/transcriber/src/main/java/org/opendaylight/neutron/transcriber/NeutronQosPolicyInterface.java
+++ b/transcriber/src/main/java/org/opendaylight/neutron/transcriber/NeutronQosPolicyInterface.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2016 Intel Corporation and others.  All rights reserved.
+ * Copyright (c) 2016, 2018 Intel Corporation and others.  All rights reserved.
  *
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
@@ -60,6 +60,7 @@ public final class NeutronQosPolicyInterface
                         toMdIds(bandwidthLimitRule, BandwidthLimitRulesBuilder.class);
                 bandwidthLimitRulesBuilder.setMaxKbps(bandwidthLimitRule.getMaxKbps());
                 bandwidthLimitRulesBuilder.setMaxBurstKbps(bandwidthLimitRule.getMaxBurstKbps());
+                bandwidthLimitRulesBuilder.setDirection(DirectionMapper.get(bandwidthLimitRule.getDirection()));
                 listBandwith.add(bandwidthLimitRulesBuilder.build());
             }
             qosPolicyBuilder.setBandwidthLimitRules(listBandwith);
@@ -104,6 +105,7 @@ public final class NeutronQosPolicyInterface
                 NeutronQosBandwidthLimitRule opt = new NeutronQosBandwidthLimitRule();
                 opt.setMaxKbps(rule.getMaxKbps());
                 opt.setMaxBurstKbps(rule.getMaxBurstKbps());
+                opt.setDirection(DirectionMapper.getDirectionString(rule.getDirection()));
                 bandwidthLimitRules.add(opt);
             }
             result.setQosBandwidthLimitRules(bandwidthLimitRules);
-- 
2.36.6