[pipelined]Fix NIL Delete in QOS Restart Recovery (magma#4186)

Signed-off-by: Nick Yurchenko <koolzz@fb.com>
pshelar · Jan 12, 2021 · b9eb60b · b9eb60b
1 parent e736aef
commit b9eb60b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lte/gateway/python/magma/pipelined/qos/common.py b/lte/gateway/python/magma/pipelined/qos/common.py
@@ -116,8 +116,9 @@ def remove_rule(self, rule_num: int) -> None:
         if session_with_rule:
             for (d, _) in self.rules[rule_num]:
                 k = get_subscriber_key(self.imsi, session_with_rule.ip_addr,
-                                        rule_num, d)
-                del self._qos_store[get_json(k)]
+                                       rule_num, d)
+                if get_json(k) in self._qos_store:
+                    del self._qos_store[get_json(k)]
 
         del self.rules[rule_num]
         session_with_rule.rules.remove(rule_num)