Add support for caching warmboot state in WarmStart function.

Author: mint570

common/warm_restart.cpp

@@ -91,6 +91,7 @@ void WarmStart::initialize(const std::string &app_name,
             std::unique_ptr<Table>(new Table(warmStart.m_cfgDb.get(), CFG_WARM_RESTART_TABLE_NAME));
 
     warmStart.m_initialized = true;
+    warmStart.m_warmbootState = WSUNKNOWN;
 }
 
 /*
@@ -294,6 +295,13 @@ void WarmStart::getWarmStartState(const std::string &app_name, WarmStartState &s
         return;
     }
 
+    if (app_name == warmStart.m_appName &&
+        warmStart.m_warmbootState != WSUNKNOWN) {
+        /* Cache is up-to-date. Read state from cache. */
+        state = warmStart.m_warmbootState;
+        return;
+    }
+
     warmStart.m_stateWarmRestartTable->hget(app_name, "state", statestr);
 
     /* If warm-start is enabled, state cannot be assumed as Reconciled
@@ -309,7 +317,12 @@ void WarmStart::getWarmStartState(const std::string &app_name, WarmStartState &s
             break;
         }
     }
-        
+    if (app_name == warmStart.m_appName)
+    {
+        /* Update cache. */
+        warmStart.m_warmbootState = state;
+    }
+
     SWSS_LOG_INFO("%s warm start state get %s(%d)",
                     app_name.c_str(), statestr.c_str(), state);
 
@@ -325,6 +338,12 @@ void WarmStart::setWarmStartState(const std::string &app_name, WarmStartState st
                                             "state",
                                             warmStartStateNameMap.at(state).c_str());
 
+    if (app_name == warmStart.m_appName)
+    {
+        /* Update cache. */
+        warmStart.m_warmbootState = state;
+    }
+
     SWSS_LOG_NOTICE("%s warm start state changed to %s",
                     app_name.c_str(),
                     warmStartStateNameMap.at(state).c_str());

common/warm_restart.h

@@ -105,6 +105,7 @@ class WarmStart
     bool                                 m_systemWarmRebootEnabled;
     std::string                          m_appName;
     std::string                          m_dockerName;
+    WarmStartState                       m_warmbootState;
 };
 
 }