Cleanup + logging env variable (#387)

* [CP] Update tests/common/serialization.hpp - remove duplication in rocprofiler_callback_tracing_code_object_load_data_t * [CP] Update lib/rocprofiler-sdk/tests - create common.hpp - update registration.cpp to use common.hpp * [CP] Add lib/common/logging.{hpp,cpp} - generic init_logging function * [CP] Update lib/rocprofiler-sdk/hsa/async_copy.cpp - remove excess logging * [CP] Update lib/rocprofiler-sdk/registration.cpp - use common::init_logging(...) - enforce ROCPROFILER_REGISTER_FORCE_LOAD in rocprofiler_force_configure - logging updates in rocprofiler_set_api_table * Update include/rocprofiler-sdk/buffer_tracing.h - rocprofiler_buffer_tracing_marker_record_t -> rocprofiler_buffer_tracing_marker_api_record_t * Update lib/common/utility.hpp - remove active_capacity_gate * Update lib/rocprofiler-sdk/tests/common.hpp - fix get_{callback,buffer}_tracing_names() * Update lib/rocprofiler-sdk/counters/xml/{basic,derived}_counters.xml - add entries for gfx1102 [ROCm/rocprofiler-sdk commit: dc8b8aa448]
2024-01-17 00:28:20 -06:00
parent 871048b7b9
commit 12bc6c93d7
12 changed files with 341 additions and 239 deletions
@@ -0,0 +1,86 @@
+// MIT License
+//
+// Copyright (c) 2023 Advanced Micro Devices, Inc.
+//
+// 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.
+
+#include "lib/common/logging.hpp"
+#include "lib/common/environment.hpp"
+
+#include <fmt/format.h>
+#include <glog/logging.h>
+
+#include <fstream>
+#include <mutex>
+#include <unordered_map>
+
+namespace rocprofiler
+{
+namespace common
+{
+void
+init_logging(std::string_view env_var)
+{
+    static auto _once = std::once_flag{};
+    std::call_once(_once, [env_var]() {
+        auto get_argv0 = []() {
+            auto ifs  = std::ifstream{"/proc/self/cmdline"};
+            auto sarg = std::string{};
+            while(ifs && !ifs.eof())
+            {
+                ifs >> sarg;
+                if(!sarg.empty()) break;
+            }
+            return sarg;
+        };
+
+        static auto argv0 = get_argv0();
+        google::InitGoogleLogging(argv0.c_str());
+        auto loglvl = common::get_env(env_var, "error");
+        for(auto& itr : loglvl)
+            itr = tolower(itr);
+        // default to warning
+        auto loglvl_v = google::WARNING;
+        if(loglvl.find_first_not_of("0123456789") == std::string::npos)
+        {
+            loglvl_v = std::stoul(loglvl);
+        }
+        else
+        {
+            const auto opts =
+                std::unordered_map<std::string_view, uint32_t>{{"info", google::INFO},
+                                                               {"warning", google::WARNING},
+                                                               {"error", google::ERROR},
+                                                               {"fatal", google::FATAL}};
+            if(opts.find(loglvl) == opts.end())
+                throw std::runtime_error{
+                    fmt::format("invalid specifier for ROCPROFILER_LOG_LEVEL: {}. Supported: info, "
+                                "warning, error, fatal",
+                                loglvl)};
+            else
+                loglvl_v = opts.at(loglvl);
+        }
+
+        FLAGS_minloglevel     = loglvl_v;
+        FLAGS_stderrthreshold = loglvl_v;
+        LOG(INFO) << "logging initialized";
+    });
+}
+}  // namespace common
+}  // namespace rocprofiler