Move merge preserved copp logic to dedicated function

sonic-net · Dec 1, 2022 · 8268df9 · 8268df9
1 parent dd42c77
commit 8268df9
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 16 deletions.
diff --git a/cfgmgr/coppmgr.cpp b/cfgmgr/coppmgr.cpp
@@ -195,7 +195,6 @@ bool CoppMgr::isTrapIdDisabled(string trap_id)
 
 void CoppMgr::mergeConfig(CoppCfg &init_cfg, CoppCfg &m_cfg, std::vector<std::string> &cfg_keys, Table &cfgTable)
 {
-    CoppCfg helper;
     /* Read the init configuration first. If the same key is present in
      * user configuration, override the init fields with user fields
      */
@@ -235,12 +234,12 @@ void CoppMgr::mergeConfig(CoppCfg &init_cfg, CoppCfg &m_cfg, std::vector<std::st
             }
             if (!null_cfg)
             {
-                helper[i.first] = merged_fvs;
+                m_cfg[i.first] = merged_fvs;
             }
         }
         else
         {
-            helper[i.first] = init_cfg[i.first];
+            m_cfg[i.first] = init_cfg[i.first];
         }
     }
 
@@ -253,27 +252,30 @@ void CoppMgr::mergeConfig(CoppCfg &init_cfg, CoppCfg &m_cfg, std::vector<std::st
         {
             std::vector<FieldValueTuple> cfg_fvs;
             cfgTable.get(i, cfg_fvs);
-            helper[i] = cfg_fvs;
+            m_cfg[i] = cfg_fvs;
         }
     }
+}
 
-    /* Compare with the existing content of copp tables, in case values merged in helper map
+void CoppMgr::mergePreservedConfig(CoppCfg &merged_cfg, CoppCfg &m_cfg)
+{
+    /* Compare with the existing content of copp tables, in case values merged in merged map
      * will be ignored as they are duplicates. In case the value in helper (merge of init and
      * user config) will overwrie the copp table entries, current entry will be deleted from copp
      * tables.
      */
-    for (auto i : helper)
+    for (auto i : merged_cfg)
     {
         string table_key = i.first;
-        std::vector<FieldValueTuple> helper_fvs = i.second;
+        std::vector<FieldValueTuple> merged_fvs = i.second;
         std::vector<FieldValueTuple> preserved_fvs;
         std::vector<FieldValueTuple> new_fvs;
 
         bool key_found = m_coppTable.get(table_key, preserved_fvs);
         if (!key_found)
         {
             // key is not found in preserved copp table -> add
-            m_cfg[table_key] = helper[table_key];
+            m_cfg[table_key] = merged_cfg[table_key];
         }
         else
         {
@@ -282,10 +284,10 @@ void CoppMgr::mergeConfig(CoppCfg &init_cfg, CoppCfg &m_cfg, std::vector<std::st
             {
                 preserved_copp_entry[fvField(prev_fv)] = fvValue(prev_fv);
             }
-            for (auto helper_fv: helper_fvs)
+            for (auto merged_fv: merged_fvs)
             {
-                string field = fvField(helper_fv);
-                string value = fvValue(helper_fv);
+                string field = fvField(merged_fv);
+                string value = fvValue(merged_fv);
                 unordered_map<string, string>::const_iterator preserved_copp_it = preserved_copp_entry.find(field);
                 bool field_found = (preserved_copp_it != preserved_copp_entry.end());
                 bool overwrite = false;
@@ -298,21 +300,21 @@ void CoppMgr::mergeConfig(CoppCfg &init_cfg, CoppCfg &m_cfg, std::vector<std::st
                     }
                     else
                     {
-                        // overwrite -> delete preserved FV tuple from copp table and add helper fvs to m_cfg
+                        // overwrite -> delete preserved FV tuple from copp table and add merged fvs to m_cfg
                         overwrite = true;
                         m_coppTable.hdel(table_key, field);
                     }
                 }
                 if (!field_found || overwrite)
                 {
-                    new_fvs.push_back(helper_fv);
+                    new_fvs.push_back(merged_fv);
                 }
             }
             if (!new_fvs.empty())
             {
                 m_cfg[table_key] = new_fvs;
             }
-        }
+        }   
     }
 }
 
@@ -336,6 +338,7 @@ CoppMgr::CoppMgr(DBConnector *cfgDb, DBConnector *appDb, DBConnector *stateDb, c
     std::vector<string> group_cfg_keys;
     std::vector<string> trap_cfg_keys;
 
+    CoppCfg merged_group_cfg;
     CoppCfg group_cfg;
     CoppCfg trap_cfg;
 
@@ -390,8 +393,9 @@ CoppMgr::CoppMgr(DBConnector *cfgDb, DBConnector *appDb, DBConnector *stateDb, c
         }
     }
 
-    mergeConfig(m_coppGroupInitCfg, group_cfg, group_cfg_keys, m_cfgCoppGroupTable);
-
+    mergeConfig(m_coppGroupInitCfg, merged_group_cfg, group_cfg_keys, m_cfgCoppGroupTable);
+    mergePreservedConfig(merged_group_cfg, group_cfg);
+
     for (auto i: group_cfg)
     {
         string trap_ids;

diff --git a/cfgmgr/coppmgr.h b/cfgmgr/coppmgr.h
@@ -100,6 +100,7 @@ class CoppMgr : public Orch
     bool isTrapGroupInstalled(std::string key);
     bool isFeatureEnabled(std::string feature);
     void mergeConfig(CoppCfg &init_cfg, CoppCfg &m_cfg, std::vector<std::string> &cfg_keys, Table &cfgTable);
+    void mergePreservedConfig(CoppCfg &merged_cfg, CoppCfg &m_cfg);
 
     void removeTrap(std::string key);
     void addTrap(std::string trap_ids, std::string trap_group);