shared libs for env_server

Author: rozgo

.gitignore

@@ -2,4 +2,6 @@ build/
 .vscode/
 .vs/
 .pytest_cache/
-*.pyc
+*.pyc
+
+/scratch.sh

.gitmodules

@@ -1,9 +1,9 @@
-[submodule "example/lib/spdlog"]
-	path = example/lib/spdlog
+[submodule "deps/lib/spdlog"]
+	path = deps/lib/spdlog
 	url = [email protected]:gabime/spdlog.git
-[submodule "example/lib/msgpack-c"]
-	path = example/lib/msgpack-c
+[submodule "deps/lib/msgpack-c"]
+	path = deps/lib/msgpack-c
 	url = [email protected]:msgpack/msgpack-c.git
-[submodule "example/lib/libzmq"]
-	path = example/lib/libzmq
+[submodule "deps/lib/libzmq"]
+	path = deps/lib/libzmq
 	url = [email protected]:zeromq/libzmq.git

CMakeLists.txt

@@ -38,7 +38,15 @@ if (CMAKE_SOURCE_DIR STREQUAL CMAKE_CURRENT_SOURCE_DIR)
 endif(CMAKE_SOURCE_DIR STREQUAL CMAKE_CURRENT_SOURCE_DIR)
 
 # Dependencies
+
+# Spdlog
+add_subdirectory(deps/lib/spdlog)
+# ZMQ
+option(ZMQ_BUILD_TESTS "" OFF)
+add_subdirectory(deps/lib/libzmq)
+
 ## PyTorch
+set(Torch_DIR /home/rozgo/deeprl/libtorch/libtorch-cxx11-abi-shared-with-deps-1.3.1/libtorch/share/cmake/Torch)
 if (NOT TORCH_FOUND)
     find_package(Torch REQUIRED)
     if (TORCH_CXX_FLAGS)
@@ -85,5 +93,7 @@ if (CPPRL_BUILD_EXAMPLE)
     add_subdirectory(example)
 endif(CPPRL_BUILD_EXAMPLE)
 
+add_subdirectory(env_server)
+
 # Recurse into source tree
 add_subdirectory(src)

deps/lib/libzmq

@@ -0,0 +1,14 @@
+add_executable(env_server env_server.cpp communicator.cpp)
+
+target_include_directories(env_server 
+PRIVATE
+.
+../include
+../deps
+../deps/lib/spdlog/include
+../deps/lib/msgpack-c/include
+../deps/lib/spdlog/include
+../deps/lib/libzmq/include
+)
+
+target_link_libraries(env_server PRIVATE libzmq-static cpprl)

deps/lib/msgpack-c

@@ -0,0 +1,34 @@
+#include <memory>
+#include <string>
+
+#include <spdlog/spdlog.h>
+
+#include "communicator.h"
+#include "requests.h"
+#include "third_party/zmq.hpp"
+
+namespace gym_client
+{
+Communicator::Communicator(const std::string &url)
+{
+    context = std::make_unique<zmq::context_t>(1);
+    socket = std::make_unique<zmq::socket_t>(*context, ZMQ_PAIR);
+
+    socket->connect(url.c_str());
+    spdlog::info(get_raw_response());
+}
+
+Communicator::~Communicator() {}
+
+std::string Communicator::get_raw_response()
+{
+    // Receive message
+    zmq::message_t zmq_msg;
+    socket->recv(&zmq_msg);
+
+    // Cast message to string
+    std::string response = std::string(static_cast<char *>(zmq_msg.data()), zmq_msg.size());
+
+    return response;
+}
+}

deps/lib/spdlog

@@ -0,0 +1,65 @@
+#pragma once
+
+#include <iostream>
+#include <memory>
+#include <sstream>
+#include <string>
+
+#include <msgpack.hpp>
+#include <spdlog/spdlog.h>
+#include <spdlog/fmt/bundled/ostream.h>
+
+#include "requests.h"
+#include "third_party/zmq.hpp"
+
+namespace gym_client
+{
+class Communicator
+{
+  public:
+    Communicator(const std::string &url);
+    ~Communicator();
+
+    std::string get_raw_response();
+
+    template <class T>
+    std::unique_ptr<T> get_response()
+    {
+        // Receive message
+        zmq::message_t packed_msg;
+        socket->recv(&packed_msg);
+
+        // Desrialize message
+        msgpack::object_handle object_handle = msgpack::unpack(static_cast<char *>(packed_msg.data()), packed_msg.size());
+        msgpack::object object = object_handle.get();
+
+        // Fill out response object
+        std::unique_ptr<T> response = std::make_unique<T>();
+        try
+        {
+            object.convert(response);
+        }
+        catch (...)
+        {
+            spdlog::error("Communication error: {}", object);
+        }
+
+        return response;
+    }
+
+    template <class T>
+    void send_request(const Request<T> &request)
+    {
+        msgpack::sbuffer buffer;
+        msgpack::pack(buffer, request);
+
+        zmq::message_t message(buffer.size());
+        std::memcpy(message.data(), buffer.data(), buffer.size());
+        socket->send(message);
+    }
+
+  private:
+    std::unique_ptr<zmq::context_t> context;
+    std::unique_ptr<zmq::socket_t> socket;
+};
+}