Clear counter when dhcp6relay init

sonic-net · Dec 10, 2024 · 1bc4da2 · 1bc4da2
1 parent 84e4419
commit 1bc4da2
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 0 deletions.
diff --git a/src/relay.cpp b/src/relay.cpp
@@ -271,6 +271,7 @@ bool DHCPv6Msg::UnmarshalBinary(const uint8_t *packet, uint16_t len) {
  * @return              none
  */
 void initialize_counter(std::shared_ptr<swss::DBConnector> state_db, std::string &ifname) {
+    clear_counter(state_db);
     std::string table_name = counter_table + ifname;
     for (auto &intr : counterMap) {
         state_db->hset(table_name, intr.second, toString(0)); 
@@ -1334,3 +1335,19 @@ void shutdown_relay() {
     event_base_free(base);
     deinitialize_swss();
 }
+
+/**
+ * @code clear_counter(std::shared_ptr<swss::DBConnector> state_db);
+ * 
+ * @brief Clear all counter
+ * 
+ * @param state_db      state_db connector pointer
+ * 
+ */
+void clear_counter(std::shared_ptr<swss::DBConnector> state_db) {
+    std::string match_pattern = counter_table + std::string("*");
+    auto keys = state_db->keys(match_pattern);
+    for (auto &itr : keys) {
+        state_db->del(itr);
+    }
+}
diff --git a/src/relay.h b/src/relay.h
@@ -474,3 +474,12 @@ void client_packet_handler(uint8_t *buffer, ssize_t length, struct relay_config
  */
 void server_callback(evutil_socket_t fd, short event, void *arg);
 
+/**
+ * @code clear_counter(std::shared_ptr<swss::DBConnector> state_db);
+ * 
+ * @brief Clear all counter
+ * 
+ * @param state_db      state_db connector pointer
+ * 
+ */
+void clear_counter(std::shared_ptr<swss::DBConnector> state_db);
diff --git a/test/mock_relay.cpp b/test/mock_relay.cpp
@@ -345,6 +345,16 @@ TEST(counter, increase_counter)
   EXPECT_EQ(*ptr, "1");
 }
 
+TEST(counter, clear_counter)
+{
+  std::shared_ptr<swss::DBConnector> state_db = std::make_shared<swss::DBConnector> ("STATE_DB", 0);
+  std::string ifname = "Vlan1000";
+  initialize_counter(state_db, ifname);
+  EXPECT_FALSE(state_db->exists("DHCPv6_COUNTER_TABLE|Vlan1000"));
+  clear_counter(state_db);
+  EXPECT_FALSE(state_db->exists("DHCPv6_COUNTER_TABLE|Vlan1000"));
+}
+
 TEST(relay, relay_client) 
 {
   uint8_t msg[] = {