/Users/deen/code/yugabyte-db/src/yb/rocksutil/yb_rocksdb_logger.cc
Line | Count | Source (jump to first uncovered line) |
1 | | // Copyright (c) YugaByte, Inc. |
2 | | // |
3 | | // Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except |
4 | | // in compliance with the License. You may obtain a copy of the License at |
5 | | // |
6 | | // http://www.apache.org/licenses/LICENSE-2.0 |
7 | | // |
8 | | // Unless required by applicable law or agreed to in writing, software distributed under the License |
9 | | // is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express |
10 | | // or implied. See the License for the specific language governing permissions and limitations |
11 | | // under the License. |
12 | | // |
13 | | |
14 | | #include "yb/rocksutil/yb_rocksdb_logger.h" |
15 | | |
16 | | #include <stdarg.h> |
17 | | |
18 | | #include <glog/logging.h> |
19 | | |
20 | | using rocksdb::InfoLogLevel; |
21 | | |
22 | | namespace yb { |
23 | | |
24 | | void YBRocksDBLogger::Logv(const rocksdb::InfoLogLevel log_level, |
25 | | const char *format, |
26 | 0 | va_list ap) { |
27 | 0 | LogvWithContext(__FILE__, __LINE__, log_level, format, ap); |
28 | 0 | } |
29 | | |
30 | | void YBRocksDBLogger::LogvWithContext(const char* file, |
31 | | const int line, |
32 | | const rocksdb::InfoLogLevel log_level, |
33 | | const char* format, |
34 | 4.51M | va_list ap) { |
35 | | // We try first to use 1024-bytes buffer on stack and then dynamically allocated buffer of larger |
36 | | // size. |
37 | 4.51M | static constexpr size_t kInitialBufferSize = 1024; |
38 | 4.51M | static const size_t kMaxBufferSize = google::LogMessage::kMaxLogMessageLen; |
39 | | |
40 | 4.51M | char initial_buffer[kInitialBufferSize]; |
41 | 4.51M | std::unique_ptr<char[]> dynamic_buffer; |
42 | 4.51M | char* buffer = initial_buffer; |
43 | | |
44 | 4.52M | auto print = [&buffer, &ap, &format](size_t buffer_size) -> size_t { |
45 | 4.52M | va_list backup_ap; |
46 | 4.52M | va_copy(backup_ap, ap); |
47 | 4.52M | const size_t size = vsnprintf(buffer, buffer_size, format, backup_ap); |
48 | 4.52M | va_end(backup_ap); |
49 | 4.52M | return size; |
50 | 4.52M | }; |
51 | | |
52 | 4.51M | const size_t required_buffer_size = print(kInitialBufferSize) + 1; |
53 | 4.51M | if (required_buffer_size > kInitialBufferSize) { |
54 | 4.05k | const size_t buffer_size = std::min(required_buffer_size, kMaxBufferSize); |
55 | 4.05k | dynamic_buffer.reset(buffer = new char[buffer_size]); |
56 | 4.05k | print(buffer_size); |
57 | 4.05k | } |
58 | | |
59 | 4.51M | google::LogMessage(file, line, YBRocksDBLogger::ConvertToGLogLevel(log_level)).stream() |
60 | 4.51M | << prefix_ << buffer; |
61 | 4.51M | } |
62 | | |
63 | 4.51M | int YBRocksDBLogger::ConvertToGLogLevel(const rocksdb::InfoLogLevel rocksdb_log_level) { |
64 | 4.51M | switch (rocksdb_log_level) { |
65 | 271k | case InfoLogLevel::DEBUG_LEVEL: |
66 | 4.51M | case InfoLogLevel::INFO_LEVEL: |
67 | 4.51M | case InfoLogLevel::HEADER_LEVEL: |
68 | | // GLOG doesn't have separate levels for DEBUG or HEADER. Default those to INFO also. |
69 | 4.51M | return google::GLOG_INFO; |
70 | 2.67k | case InfoLogLevel::WARN_LEVEL: |
71 | 2.67k | return google::GLOG_WARNING; |
72 | 3 | case InfoLogLevel::ERROR_LEVEL: |
73 | 3 | return google::GLOG_ERROR; |
74 | 0 | case InfoLogLevel::FATAL_LEVEL: |
75 | 0 | return google::GLOG_FATAL; |
76 | 0 | default: |
77 | 0 | LOG(FATAL) << "Unknown rocksdb log level: " << rocksdb_log_level; |
78 | 4.51M | } |
79 | 4.51M | } |
80 | | |
81 | 527k | rocksdb::InfoLogLevel YBRocksDBLogger::ConvertToRocksDBLogLevel(const int glog_level) { |
82 | 527k | switch (glog_level) { |
83 | 527k | case google::GLOG_INFO: |
84 | 527k | return InfoLogLevel::INFO_LEVEL; |
85 | 0 | case google::GLOG_WARNING: |
86 | 0 | return InfoLogLevel::WARN_LEVEL; |
87 | 0 | case google::GLOG_ERROR: |
88 | 0 | return InfoLogLevel::ERROR_LEVEL; |
89 | 0 | case google::GLOG_FATAL: |
90 | 0 | return InfoLogLevel::FATAL_LEVEL; |
91 | 0 | default: |
92 | 0 | LOG(FATAL) << "Unknown glog level: " << glog_level; |
93 | 527k | } |
94 | 527k | } |
95 | | |
96 | | } // namespace yb |