1+ // Licensed to the Apache Software Foundation (ASF) under one
2+ // or more contributor license agreements. See the NOTICE file
3+ // distributed with this work for additional information
4+ // regarding copyright ownership. The ASF licenses this file
5+ // to you under the Apache License, Version 2.0 (the
6+ // "License"); you may not use this file except in compliance
7+ // with the License. You may obtain a copy of the License at
8+ //
9+ // http://www.apache.org/licenses/LICENSE-2.0
10+ //
11+ // Unless required by applicable law or agreed to in writing,
12+ // software distributed under the License is distributed on an
13+ // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14+ // KIND, either express or implied. See the License for the
15+ // specific language governing permissions and limitations
16+ // under the License.
17+
18+ #include " runtime/be_proc_monitor.h"
19+
20+ #include < fmt/format.h>
21+ #include < glog/logging.h>
22+ #include < sys/stat.h>
23+ #include < unistd.h>
24+
25+ #include < deque>
26+ #include < filesystem>
27+ #include < fstream>
28+ #include < map>
29+ #include < nlohmann/json.hpp>
30+
31+ std::string BeProcMonitor::get_be_thread_info () {
32+ int32_t pid = getpid ();
33+ std::string proc_path = fmt::format (" /proc/{}/task"
34+ if (access (proc_path.c_str (), F_OK) != 0 ) {
35+ LOG (WARNING) << " be proc path " " not exists."
36+ return " "
37+ }
38+
39+ std::map<std::string, int > thread_num_map;
40+
41+ int total_thread_count = 0 ;
42+ int distinct_thread_name_count = 0 ;
43+ for (const auto & entry : std::filesystem::directory_iterator (proc_path)) {
44+ const std::string tid_path = entry.path ().string ();
45+ std::string thread_name_path = tid_path + " /comm"
46+ struct stat st;
47+ // == 0 means exists
48+ if (stat (thread_name_path.c_str (), &st) == 0 ) {
49+ // NOTE: there is no need to close std::ifstream, it's called during deconstruction.
50+ // refer:https://stackoverflow.com/questions/748014/do-i-need-to-manually-close-an-ifstream
51+ std::ifstream file (thread_name_path.c_str ());
52+ if (!file.is_open ()) {
53+ continue ;
54+ }
55+ std::stringstream str_buf;
56+ str_buf << file.rdbuf ();
57+ std::string thread_name = str_buf.str ();
58+ thread_name.erase (std::remove (thread_name.begin (), thread_name.end (), ' \n '
59+ thread_name.end ());
60+
61+ if (thread_num_map.find (thread_name) != thread_num_map.end ()) {
62+ thread_num_map[thread_name]++;
63+ } else {
64+ distinct_thread_name_count++;
65+ thread_num_map.emplace (thread_name, 1 );
66+ }
67+ total_thread_count++;
68+ }
69+ }
70+
71+ std::deque<std::pair<std::string, int >> ordered_list (thread_num_map.begin (),
72+ thread_num_map.end ());
73+ std::sort (ordered_list.begin (), ordered_list.end (),
74+ [](const auto & lhs, const auto & rhs) { return lhs.second > rhs.second ; });
75+
76+ ordered_list.push_front (
77+ std::make_pair (" distinct_thread_name_count"
78+ ordered_list.push_front (std::make_pair (" total_thread_count"
79+ ordered_list.push_front (std::make_pair (" be_process_id"
80+
81+ nlohmann::json js = nlohmann::json::array ();
82+ for (const auto & p : ordered_list) {
83+ js.push_back ({p.first , p.second });
84+ }
85+
86+ std::string output_json_str = js.dump ();
87+ return output_json_str;
88+ }
0 commit comments