Add set, get and listkeys -sdltool CLI commands

[ric-plt/sdl.git] / src / cli / setcommand.cpp

diff --git a/src/cli/setcommand.cpp b/src/cli/setcommand.cpp
new file mode 100644 (file)
index 0000000..425be4d
--- /dev/null
+++ b/src/cli/setcommand.cpp
@@ -0,0 +1,87 @@
+#include <ostream>
+#include <cstdlib>
+#include "private/cli/commandmap.hpp"
+#include "private/cli/common.hpp"
+#include <sdl/syncstorage.hpp>
+
+using namespace shareddatalayer;
+using namespace shareddatalayer::cli;
+
+namespace
+{
+    std::shared_ptr<shareddatalayer::SyncStorage> createSyncStorage(const SyncStorage::Namespace& ns,
+                                                                    std::ostream& out)
+    {
+        try
+        {
+            auto sdl(shareddatalayer::SyncStorage::create());
+            sdl->waitReady(ns, std::chrono::minutes(1));
+            return sdl;
+        }
+        catch (const shareddatalayer::Exception& error)
+        {
+            out << "SyncStorage create failed: " << error.what() << std::endl;
+        }
+
+        return nullptr;
+    }
+
+    void set(shareddatalayer::SyncStorage& sdl,
+             const SyncStorage::Namespace& ns,
+             const SyncStorage::DataMap& dataMap,
+             std::ostream& out)
+    {
+        try
+        {
+            sdl.set(ns, dataMap);
+            out << "set {" << ns << "}," << dataMap << std::endl;
+        }
+        catch (const shareddatalayer::Exception& error)
+        {
+            out << "Error in set(" << ns << ", " << dataMap << ") failed: "
+                << error.what() << std::endl;
+        }
+    }
+
+    int setCommand(std::ostream& out, const boost::program_options::variables_map& map)
+    {
+        auto ns(map["ns"].as<std::string>());
+        auto key(map["key"].as<std::string>());
+        auto data(map["data"].as<std::string>());
+        auto debug(map["debug"].as<bool>());
+        SyncStorage::Data dataVector(data.begin(), data.end());
+
+        auto sdl(createSyncStorage(ns, out));
+        if (nullptr == sdl)
+            return EXIT_FAILURE;
+        sdl->setOperationTimeout(std::chrono::seconds(5));
+
+        if (debug)
+        {
+            out << "DEBUG setCommand ns: " << ns
+                << " key: " << key << " data: " << data << std::endl;
+        }
+
+        SyncStorage::DataMap datamap({{key, dataVector}});
+        set(std::ref(*sdl), ns, datamap, out);
+
+        return EXIT_SUCCESS;
+    }
+}
+
+const char *longHelpSetCmd =
+    "Use set SDL API to write data to storage under the namespace.\n\n"
+    "Example: sdltool set --ns 'sdltool' --key 'key1' --data '1'";
+
+AUTO_REGISTER_COMMAND(std::bind(setCommand, std::placeholders::_1, std::placeholders::_3),
+                      "set",
+                      "set key value with SDL API",
+                      longHelpSetCmd,
+                      CommandMap::Category::UTIL,
+                      30040,
+                      ("ns,n", boost::program_options::value<std::string>()->default_value("sdltoolns"), "namespace to use")
+                      ("key,k", boost::program_options::value<std::string>()->default_value("key1"), "key value")
+                      ("data,d", boost::program_options::value<std::string>()->default_value("1"), "data")
+                      ("debug", boost::program_options::bool_switch()->default_value(false), "Enable debug logs")
+                     );
+