Add set, get and listkeys -sdltool CLI commands

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

diff --git a/src/cli/listkeyscommand.cpp b/src/cli/listkeyscommand.cpp
new file mode 100644 (file)
index 0000000..04f39c7
--- /dev/null
+++ b/src/cli/listkeyscommand.cpp
@@ -0,0 +1,77 @@
+#include <ostream>
+#include <cstdlib>
+#include "private/cli/commandmap.hpp"
+#include "private/cli/common.hpp"
+#include <sdl/syncstorage.hpp>
+#include <sdl/asyncstorage.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 listkeys(shareddatalayer::SyncStorage& sdl,
+                  const SyncStorage::Namespace& ns,
+                  const std::string& pattern,
+                  std::ostream& out)
+    {
+        try
+        {
+            auto keys(sdl.listKeys(ns, pattern));
+            for (auto key: keys)
+                out << key << std::endl;
+        }
+        catch (const shareddatalayer::Exception& error)
+        {
+            out << "listKeys(" << ns << ", " << pattern << ") failed: "
+                << error.what() << std::endl;
+        }
+    }
+
+    int listKeysCommand(std::ostream& out, const boost::program_options::variables_map& map)
+    {
+        auto ns(map["ns"].as<std::string>());
+        auto pattern(map["pattern"].as<std::string>());
+
+        auto sdl(createSyncStorage(ns, out));
+        if (nullptr == sdl)
+            return EXIT_FAILURE;
+        sdl->setOperationTimeout(std::chrono::seconds(5));
+
+        listkeys(std::ref(*sdl), ns, pattern, out);
+
+        return EXIT_SUCCESS;
+    }
+}
+
+const char *longHelpListkeysCmd =
+    "Use listKeys SDL API to list keys matching search pattern under the namespace.\n\n"
+    "Example: sdltool listkeys --ns 'sdltool' --pattern 'foo*'";
+
+AUTO_REGISTER_COMMAND(std::bind(listKeysCommand, std::placeholders::_1, std::placeholders::_3),
+                      "listkeys",
+                      "listKeys SDL API",
+                      longHelpListkeysCmd,
+                      CommandMap::Category::UTIL,
+                      30030,
+                      ("ns,n", boost::program_options::value<std::string>()->default_value("sdltoolns"), "namespace to use")
+                      ("pattern,p", boost::program_options::value<std::string>()->default_value("*"), "key search pattern")
+                     );
+