2 * Copyright 2004-present Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 #include <folly/experimental/logging/LogLevel.h>
21 #include <folly/Conv.h>
27 LogLevel stringToLogLevel(StringPiece name) {
29 lowerNameStr.reserve(name.size());
31 lowerNameStr.push_back(static_cast<char>(std::tolower(c)));
33 StringPiece lowerName{lowerNameStr};
35 // If the string is of the form "LogLevel::foo" or "LogLevel(foo)"
36 // strip it down just to "foo". This makes sure we can process both
37 // the "LogLevel::DEBUG" and "LogLevel(1234)" formats produced by
38 // logLevelToString().
39 constexpr StringPiece lowercasePrefix{"loglevel::"};
40 constexpr StringPiece wrapperPrefix{"loglevel("};
41 if (lowerName.startsWith(lowercasePrefix)) {
42 lowerName.advance(lowercasePrefix.size());
43 } else if (lowerName.startsWith(wrapperPrefix) && lowerName.endsWith(")")) {
44 lowerName.advance(wrapperPrefix.size());
45 lowerName.subtract(1);
48 if (lowerName == "uninitialized") {
49 return LogLevel::UNINITIALIZED;
50 } else if (lowerName == "none") {
51 return LogLevel::NONE;
52 } else if (lowerName == "debug") {
53 return LogLevel::DEBUG;
54 } else if (lowerName == "info") {
55 return LogLevel::INFO;
56 } else if (lowerName == "warn" || lowerName == "warning") {
57 return LogLevel::WARN;
58 } else if (lowerName == "error" || lowerName == "err") {
60 } else if (lowerName == "critical") {
61 return LogLevel::CRITICAL;
62 } else if (lowerName == "max" || lowerName == "max_level") {
63 return LogLevel::MAX_LEVEL;
66 if (lowerName.startsWith("dbg")) {
67 auto remainder = lowerName.subpiece(3);
68 auto level = folly::tryTo<int>(remainder).value_or(-1);
69 if (level < 0 || level > 100) {
70 throw std::range_error("invalid dbg logger level: " + name.str());
72 return LogLevel::DBG0 - level;
75 // Try as an plain integer if all else fails
77 auto level = folly::to<uint32_t>(lowerName);
78 return static_cast<LogLevel>(level);
79 } catch (const std::exception&) {
80 throw std::range_error("invalid logger name " + name.str());
84 string logLevelToString(LogLevel level) {
85 if (level == LogLevel::UNINITIALIZED) {
86 return "LogLevel::UNINITIALIZED";
87 } else if (level == LogLevel::NONE) {
88 return "LogLevel::NONE";
89 } else if (level == LogLevel::DEBUG) {
90 return "LogLevel::DEBUG";
91 } else if (level == LogLevel::INFO) {
92 return "LogLevel::INFO";
93 } else if (level == LogLevel::WARN) {
94 return "LogLevel::WARN";
95 } else if (level == LogLevel::ERR) {
96 return "LogLevel::ERR";
97 } else if (level == LogLevel::CRITICAL) {
98 return "LogLevel::CRITICAL";
99 } else if (level == LogLevel::MAX_LEVEL) {
100 return "LogLevel::MAX_LEVEL";
103 if (static_cast<uint32_t>(level) <= static_cast<uint32_t>(LogLevel::DBG0) &&
104 static_cast<uint32_t>(level) > static_cast<uint32_t>(LogLevel::DEBUG)) {
106 static_cast<uint32_t>(LogLevel::DBG0) - static_cast<uint32_t>(level);
107 return folly::to<string>("LogLevel::DBG", num);
109 return folly::to<string>("LogLevel(", static_cast<uint32_t>(level), ")");
112 std::ostream& operator<<(std::ostream& os, LogLevel level) {
113 os << logLevelToString(level);