Implement the gRPC APIs for the job stats

Add the job stats APIs in the rdc_api_service at the server side rdcd Add the job stats APIs for the RdcStandaloneHandler at the client side Make the load librdc.so and librdc_client.so thread safe. Impelement async update all fields in RdcEmbeddedHandler. Change-Id: I659d91efb32d1094d3b7f0f2cec39518cd7336ce [ROCm/rdc commit: fe3e75edfa]
2020-04-15 12:20:22 -04:00
parent 0813e7052f
commit dc48d8c977
13 changed files with 495 additions and 66 deletions
@@ -0,0 +1,86 @@
+/*
+Copyright (c) 2020 - present Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+#ifndef RDC_LIB_RDCLIBRARYLOADER_H_
+#define RDC_LIB_RDCLIBRARYLOADER_H_
+#include <dlfcn.h>
+#include <string.h>
+#include <map>
+#include <mutex>   //  NOLINT(build/c++11)
+#include "rdc/rdc.h"
+#include "rdc_lib/RdcLogger.h"
+#include "rdc_lib/rdc_common.h"
+
+
+namespace amd {
+namespace rdc {
+class RdcLibraryLoader {
+ public:
+     RdcLibraryLoader();
+
+     template<typename T> rdc_status_t load(const char* filename,
+                T* func_make_handler);
+
+     rdc_status_t unload();
+
+     ~RdcLibraryLoader();
+
+ private:
+     void* libHandler_;
+     std::mutex library_mutex_;
+};
+
+template<typename T> rdc_status_t RdcLibraryLoader::load(const char* filename,
+                T* func_make_handler) {
+    if (filename == nullptr || func_make_handler == nullptr) {
+            return RDC_ST_FAIL_LOAD_MODULE;
+    }
+
+    if (libHandler_) {
+        unload();
+    }
+
+    std::lock_guard<std::mutex> guard(library_mutex_);
+    libHandler_ = dlopen(filename, RTLD_LAZY);
+    if (!libHandler_) {
+        char* error = dlerror();
+        RDC_LOG(RDC_ERROR, "Fail to open " << filename <<": " << error);
+        return RDC_ST_FAIL_LOAD_MODULE;
+    }
+
+    *reinterpret_cast<void**>(func_make_handler) =
+            dlsym(libHandler_, "make_handler");
+    if (*func_make_handler == nullptr) {
+            char* error = dlerror();
+            RDC_LOG(RDC_ERROR,
+            "Fail to find function make_handler from file "
+                    << filename <<": "  << error);
+            return RDC_ST_FAIL_LOAD_MODULE;
+    }
+
+    return RDC_ST_OK;
+}
+
+}  // namespace rdc
+}  // namespace amd
+
+
+#endif  // RDC_LIB_RDCLIBRARYLOADER_H_