Ecapsulate saidump.cpp with class and add unittest codes.

.gitignore

@@ -118,6 +118,7 @@ unittest/vslib/tests
 unittest/vslib/testslibsaivs
 unittest/proxylib/tests
 unittest/proxylib/testslibsaiproxy
+unittest/saidump/tests
 vslib/tests
 stub/tests
 unittest/stub/stub/testslibsaistub

configure.ac

@@ -297,6 +297,7 @@ AC_OUTPUT(Makefile
           unittest/vslib/Makefile
           unittest/syncd/Makefile
           unittest/proxylib/Makefile
+          unittest/saidump/Makefile
           pyext/Makefile
           pyext/py2/Makefile
           pyext/py3/Makefile)

saidump/Makefile.am

@@ -2,8 +2,13 @@ AM_CXXFLAGS = $(SAIINC) -I$(top_srcdir)/lib
 
 bin_PROGRAMS = saidump
 
-saidump_SOURCES = saidump.cpp
+saidump_SOURCES = main.cpp saidump.cpp
 saidump_CPPFLAGS = $(CODE_COVERAGE_CPPFLAGS)
 saidump_CXXFLAGS = $(DBGFLAGS) $(AM_CXXFLAGS) $(CXXFLAGS_COMMON) $(CODE_COVERAGE_CXXFLAGS)
 saidump_LDADD = -lhiredis -lswsscommon -lpthread -L$(top_srcdir)/meta/.libs -lsaimetadata -lsaimeta \
 				-L$(top_srcdir)/lib/.libs -lsairedis -lzmq $(CODE_COVERAGE_LIBS)
+
+noinst_LIBRARIES = libsaidump.a
+libsaidump_a_SOURCES = saidump.cpp
+libsaidump_a_CPPFLAGS = $(CODE_COVERAGE_CPPFLAGS)
+libsaidump_a_CXXFLAGS = $(DBGFLAGS) $(AM_CXXFLAGS) $(CXXFLAGS_COMMON) $(CODE_COVERAGE_CXXFLAGS)

saidump/saidump.cpp → saidump/SaiDump.cpp

@@ -1,45 +1,27 @@
+#include "saidump.h"
+extern "C" {
+#include <sai.h>
+}
 #include <inttypes.h>
-#include <string>
 #include <set>
-#include <sstream>
-#include <iostream>
-#include <fstream>
 #include <regex>
 #include <climits>
-
-extern "C" {
-#include <sai.h>
-}
-
-#include "swss/table.h"
-#include "meta/sai_serialize.h"
-#include "sairediscommon.h"
-#include <nlohmann/json.hpp>
-
 #include <getopt.h>
-
-// TODO split to multiple cpp
+#include "sairediscommon.h"
 
 using namespace swss;
 using json = nlohmann::json;
+using namespace syncd;
 
-// Default value: 100 MB
-constexpr int64_t RDB_JSON_MAX_SIZE = 1024 * 1024 * 100;
-
-struct CmdOptions
+SaiDump::SaiDump()
 {
-    bool skipAttributes;
-    bool dumpTempView;
-    bool dumpGraph;
-    std::string rdbJsonFile;
-    uint64_t rdbJSonSizeLimit;
-};
-
+}
 
-static CmdOptions g_cmdOptions;
-static std::map<sai_object_id_t, const TableMap*> g_oid_map;
+SaiDump::~SaiDump()
+{
+}
 
-void printUsage()
+void SaiDump::printUsage()
 {
     SWSS_LOG_ENTER();
 
@@ -56,15 +38,16 @@ void printUsage()
     std::cout << "        Print out this message" << std::endl;
 }
 
-CmdOptions handleCmdLine(int argc, char **argv)
+
+void SaiDump::handleCmdLine(int argc, char **argv)
 {
+    // Default value: 100MB
+    static constexpr int64_t RDB_JSON_MAX_SIZE = 1024 * 1024 * 100;    
     SWSS_LOG_ENTER();
 
-    CmdOptions options;
-
-    options.dumpTempView = false;
-    options.dumpGraph = false;
-    options.rdbJSonSizeLimit = RDB_JSON_MAX_SIZE;
+    dumpTempView = false;
+    dumpGraph = false;
+    rdbJSonSizeLimit = RDB_JSON_MAX_SIZE;
 
     const char* const optstring = "gtr:m:h";
     uint64_t result = 0;
@@ -94,17 +77,17 @@ CmdOptions handleCmdLine(int argc, char **argv)
         {
             case 'g':
                 SWSS_LOG_NOTICE("Dumping graph");
-                options.dumpGraph = true;
+                dumpGraph = true;
                 break;
 
             case 't':
                 SWSS_LOG_NOTICE("Dumping temp view");
-                options.dumpTempView = true;
+                dumpTempView = true;
                 break;
 
             case 'r':
                 SWSS_LOG_NOTICE("Dumping from %s", optarg);
-                options.rdbJsonFile = std::string(optarg);
+                rdbJsonFile = std::string(optarg);
                 break;
 
             case 'm':
@@ -124,9 +107,8 @@ CmdOptions handleCmdLine(int argc, char **argv)
                     exit(EXIT_SUCCESS);
                 }
 
-                options.rdbJSonSizeLimit = result * 1024 * 1024;
-                SWSS_LOG_NOTICE("Configure the RDB JSON MAX size to %llu MB", options.rdbJSonSizeLimit / 1024 / 1024);
-
+                rdbJSonSizeLimit = result * 1024 * 1024;
+                SWSS_LOG_NOTICE("Configure the RDB JSON MAX size to %llu MB", rdbJSonSizeLimit / 1024 / 1024);
                 break;
 
             case 'h':
@@ -143,11 +125,9 @@ CmdOptions handleCmdLine(int argc, char **argv)
                 exit(EXIT_FAILURE);
         }
     }
-
-    return options;
 }
 
-size_t get_max_attr_len(const TableMap& map)
+size_t SaiDump::get_max_attr_len(const TableMap& map)
 {
     SWSS_LOG_ENTER();
 
@@ -161,7 +141,7 @@ size_t get_max_attr_len(const TableMap& map)
     return max;
 }
 
-std::string pad_string(std::string s, size_t pad)
+std::string SaiDump::pad_string(std::string s, size_t pad)
 {
     SWSS_LOG_ENTER();
 
@@ -175,22 +155,7 @@ std::string pad_string(std::string s, size_t pad)
     return s;
 }
 
-const TableMap* get_table_map(sai_object_id_t object_id)
-{
-    SWSS_LOG_ENTER();
-
-    auto it = g_oid_map.find(object_id);
-
-    if (it == g_oid_map.end())
-    {
-        SWSS_LOG_THROW("unable to find oid %s in oid map",
-                sai_serialize_object_id(object_id).c_str());
-    }
-
-    return it->second;
-}
-
-void print_attributes(size_t indent, const TableMap& map)
+void SaiDump::print_attributes(size_t indent, const TableMap& map)
 {
     SWSS_LOG_ENTER();
 
@@ -218,7 +183,7 @@ void print_attributes(size_t indent, const TableMap& map)
 #define GV_ROOT_COLOR   "0.650 0.200 1.000"
 #define GV_NODE_COLOR   "0.650 0.500 1.000"
 
-void dumpGraph(const TableDump& td)
+void SaiDump::dumpGraphFun(const TableDump& td)
 {
     SWSS_LOG_ENTER();
 
@@ -448,61 +413,7 @@ void dumpGraph(const TableDump& td)
 
 #define SWSS_LOG_ERROR_AND_STDERR(format, ...) { fprintf(stderr, format"\n", ##__VA_ARGS__); SWSS_LOG_ERROR(format, ##__VA_ARGS__); }
 
-/**
- * @brief Process the input JSON file to make sure it's a valid JSON file for the JSON library.
- */
-static sai_status_t preProcessFile(const std::string file_name)
-{
-    SWSS_LOG_ENTER();
-
-    std::ifstream input_file(file_name);
-
-    if (!input_file.is_open())
-    {
-        SWSS_LOG_ERROR_AND_STDERR("Failed to open the input file %s.", file_name.c_str());
-        return SAI_STATUS_FAILURE;
-    }
-
-    input_file.seekg(0, std::ios::end);     // Move to the end of the file
-    uint64_t file_size = input_file.tellg(); // Get the current position
-    SWSS_LOG_NOTICE("Get %s's size %" PRIu64 " Bytes, limit: %" PRIu64 " MB.", file_name.c_str(), file_size, g_cmdOptions.rdbJSonSizeLimit / 1024 / 1024);
-
-    if (file_size >= g_cmdOptions.rdbJSonSizeLimit)
-    {
-        SWSS_LOG_ERROR_AND_STDERR("Get %s's size failure or its size %" PRIu64 " >= %" PRIu64 " MB.", file_name.c_str(), file_size, g_cmdOptions.rdbJSonSizeLimit / 1024 / 1024);
-        return SAI_STATUS_FAILURE;
-    }
-
-    input_file.seekg(0);     // Move to the begin of the file
-
-    // Read the content of the input file into a string
-    std::string content((std::istreambuf_iterator<char>(input_file)),
-                   std::istreambuf_iterator<char>());
-
-    content = regex_replace(content, std::regex("\\},\\{\\r"), ",");
-
-    std::ofstream outputFile(file_name);
-
-    if (!outputFile.is_open())
-    {
-        SWSS_LOG_ERROR_AND_STDERR("Failed to open the output file %s.", file_name.c_str());
-        return SAI_STATUS_FAILURE;
-    }
-
-    //Remove the 1st and last char to make sure its format is same as previous output
-    if (content.size() >= 2 && content[0] == '[' && content[content.length()-1] == ']')
-    {
-        outputFile << content.substr(1, content.size()-2);
-    }
-    else
-    {
-        outputFile << content;
-    }
-
-    return SAI_STATUS_SUCCESS;
-}
-
-static void traverseJson(const json & jsn)
+void SaiDump::traverseJson(const json & jsn)
 {
     SWSS_LOG_ENTER();
     if (jsn.is_object())
@@ -574,15 +485,15 @@ static void traverseJson(const json & jsn)
     }
 }
 
-static sai_status_t dumpFromRedisRdbJson(const std::string file_name)
+sai_status_t SaiDump::dumpFromRedisRdbJson()
 {
     SWSS_LOG_ENTER();
 
-    std::ifstream input_file(file_name);
+    std::ifstream input_file(rdbJsonFile);
 
     if (!input_file.is_open())
     {
-        SWSS_LOG_ERROR_AND_STDERR("The file %s does not exist for dumping from Redis RDB JSON file.", file_name.c_str());
+        SWSS_LOG_ERROR_AND_STDERR("The file %s does not exist for dumping from Redis RDB JSON file.", rdbJsonFile.c_str());
         return SAI_STATUS_FAILURE;
     }
 
@@ -596,49 +507,33 @@ static sai_status_t dumpFromRedisRdbJson(const std::string file_name)
     }
     catch (std::exception &ex)
     {
-        SWSS_LOG_ERROR_AND_STDERR("JSON file %s is invalid.", file_name.c_str());
         SWSS_LOG_ERROR_AND_STDERR("JSON parsing error: %s.", ex.what());
     }
 
     return SAI_STATUS_FAILURE;
 }
 
-int main(int argc, char **argv)
+void SaiDump::dumpFromRedisDb()
 {
-    swss::Logger::getInstance().setMinPrio(swss::Logger::SWSS_DEBUG);
-
     SWSS_LOG_ENTER();
-
-    swss::Logger::getInstance().setMinPrio(swss::Logger::SWSS_NOTICE);
-
-    swss::Logger::getInstance().setMinPrio(swss::Logger::SWSS_INFO);
-
-    g_cmdOptions = handleCmdLine(argc, argv);
-
-
-    if (g_cmdOptions.rdbJsonFile.size() > 0)
+    if (rdbJsonFile.size() > 0)
     {
-        if (SAI_STATUS_FAILURE == preProcessFile(g_cmdOptions.rdbJsonFile))
-        {
-            return EXIT_FAILURE;
-        }
-
-        return dumpFromRedisRdbJson(g_cmdOptions.rdbJsonFile);
+         dumpFromRedisRdbJson();
+         return;
     }
 
     swss::DBConnector db("ASIC_DB", 0);
 
     std::string table = ASIC_STATE_TABLE;
 
-    if (g_cmdOptions.dumpTempView)
+    if (dumpTempView)
     {
         table = TEMP_PREFIX + table;
     }
 
     swss::Table t(&db, table);
 
     TableDump dump;
-
     t.dump(dump);
 
     for (const auto&key: dump)
@@ -656,32 +551,25 @@ int main(int argc, char **argv)
         {
             sai_object_id_t object_id;
             sai_deserialize_object_id(str_object_id, object_id);
-
-            g_oid_map[object_id] = &key.second;
+            m_oid_map[object_id] = &key.second;
         }
     }
 
-    if (g_cmdOptions.dumpGraph)
+    if (dumpGraph)
     {
-        dumpGraph(dump);
-
-        return EXIT_SUCCESS;
+        dumpGraphFun(dump);
+        return;
     }
 
     for (const auto&key: dump)
     {
         auto start = key.first.find_first_of(":");
         auto str_object_type = key.first.substr(0, start);
         auto str_object_id  = key.first.substr(start + 1);
-
         std::cout << str_object_type << " " << str_object_id << " " << std::endl;
 
         size_t indent = 4;
-
         print_attributes(indent, key.second);
-
         std::cout << std::endl;
     }
-
-    return EXIT_SUCCESS;
 }