about summary refs log tree commit diff
path: root/third_party/nix/src/libutil/logging.cc
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/nix/src/libutil/logging.cc')
-rw-r--r--third_party/nix/src/libutil/logging.cc379
1 files changed, 182 insertions, 197 deletions
diff --git a/third_party/nix/src/libutil/logging.cc b/third_party/nix/src/libutil/logging.cc
index b379306f6ec0..4f88044757cd 100644
--- a/third_party/nix/src/libutil/logging.cc
+++ b/third_party/nix/src/libutil/logging.cc
@@ -1,242 +1,227 @@
 #include "logging.hh"
-#include "util.hh"
-
 #include <atomic>
 #include <nlohmann/json.hpp>
+#include "util.hh"
 
 namespace nix {
 
 static thread_local ActivityId curActivity = 0;
 
-ActivityId getCurActivity()
-{
-    return curActivity;
-}
-void setCurActivity(const ActivityId activityId)
-{
-    curActivity = activityId;
-}
+ActivityId getCurActivity() { return curActivity; }
+void setCurActivity(const ActivityId activityId) { curActivity = activityId; }
 
-Logger * logger = makeDefaultLogger();
+Logger* logger = makeDefaultLogger();
 
-void Logger::warn(const std::string & msg)
-{
-    log(lvlWarn, ANSI_RED "warning:" ANSI_NORMAL " " + msg);
+void Logger::warn(const std::string& msg) {
+  log(lvlWarn, ANSI_RED "warning:" ANSI_NORMAL " " + msg);
 }
 
-class SimpleLogger : public Logger
-{
-public:
-
-    bool systemd, tty;
-
-    SimpleLogger()
-    {
-        systemd = getEnv("IN_SYSTEMD") == "1";
-        tty = isatty(STDERR_FILENO);
+class SimpleLogger : public Logger {
+ public:
+  bool systemd, tty;
+
+  SimpleLogger() {
+    systemd = getEnv("IN_SYSTEMD") == "1";
+    tty = isatty(STDERR_FILENO);
+  }
+
+  void log(Verbosity lvl, const FormatOrString& fs) override {
+    if (lvl > verbosity) return;
+
+    std::string prefix;
+
+    if (systemd) {
+      char c;
+      switch (lvl) {
+        case lvlError:
+          c = '3';
+          break;
+        case lvlWarn:
+          c = '4';
+          break;
+        case lvlInfo:
+          c = '5';
+          break;
+        case lvlTalkative:
+        case lvlChatty:
+          c = '6';
+          break;
+        default:
+          c = '7';
+      }
+      prefix = std::string("<") + c + ">";
     }
 
-    void log(Verbosity lvl, const FormatOrString & fs) override
-    {
-        if (lvl > verbosity) return;
-
-        std::string prefix;
-
-        if (systemd) {
-            char c;
-            switch (lvl) {
-            case lvlError: c = '3'; break;
-            case lvlWarn: c = '4'; break;
-            case lvlInfo: c = '5'; break;
-            case lvlTalkative: case lvlChatty: c = '6'; break;
-            default: c = '7';
-            }
-            prefix = std::string("<") + c + ">";
-        }
-
-        writeToStderr(prefix + filterANSIEscapes(fs.s, !tty) + "\n");
-    }
+    writeToStderr(prefix + filterANSIEscapes(fs.s, !tty) + "\n");
+  }
 
-    void startActivity(ActivityId act, Verbosity lvl, ActivityType type,
-        const std::string & s, const Fields & fields, ActivityId parent)
-        override
-    {
-        if (lvl <= verbosity && !s.empty())
-            log(lvl, s + "...");
-    }
+  void startActivity(ActivityId act, Verbosity lvl, ActivityType type,
+                     const std::string& s, const Fields& fields,
+                     ActivityId parent) override {
+    if (lvl <= verbosity && !s.empty()) log(lvl, s + "...");
+  }
 };
 
 Verbosity verbosity = lvlInfo;
 
-void warnOnce(bool & haveWarned, const FormatOrString & fs)
-{
-    if (!haveWarned) {
-        warn(fs.s);
-        haveWarned = true;
-    }
+void warnOnce(bool& haveWarned, const FormatOrString& fs) {
+  if (!haveWarned) {
+    warn(fs.s);
+    haveWarned = true;
+  }
 }
 
-void writeToStderr(const string & s)
-{
-    try {
-        writeFull(STDERR_FILENO, s, false);
-    } catch (SysError & e) {
-        /* Ignore failing writes to stderr.  We need to ignore write
-           errors to ensure that cleanup code that logs to stderr runs
-           to completion if the other side of stderr has been closed
-           unexpectedly. */
-    }
+void writeToStderr(const string& s) {
+  try {
+    writeFull(STDERR_FILENO, s, false);
+  } catch (SysError& e) {
+    /* Ignore failing writes to stderr.  We need to ignore write
+       errors to ensure that cleanup code that logs to stderr runs
+       to completion if the other side of stderr has been closed
+       unexpectedly. */
+  }
 }
 
-Logger * makeDefaultLogger()
-{
-    return new SimpleLogger();
+Logger* makeDefaultLogger() { return new SimpleLogger(); }
+
+std::atomic<uint64_t> nextId{(uint64_t)getpid() << 32};
+
+Activity::Activity(Logger& logger, Verbosity lvl, ActivityType type,
+                   const std::string& s, const Logger::Fields& fields,
+                   ActivityId parent)
+    : logger(logger), id(nextId++) {
+  logger.startActivity(id, lvl, type, s, fields, parent);
 }
 
-std::atomic<uint64_t> nextId{(uint64_t) getpid() << 32};
+struct JSONLogger : Logger {
+  Logger& prevLogger;
+
+  JSONLogger(Logger& prevLogger) : prevLogger(prevLogger) {}
+
+  void addFields(nlohmann::json& json, const Fields& fields) {
+    if (fields.empty()) return;
+    auto& arr = json["fields"] = nlohmann::json::array();
+    for (auto& f : fields)
+      if (f.type == Logger::Field::tInt)
+        arr.push_back(f.i);
+      else if (f.type == Logger::Field::tString)
+        arr.push_back(f.s);
+      else
+        abort();
+  }
+
+  void write(const nlohmann::json& json) {
+    prevLogger.log(lvlError, "@nix " + json.dump());
+  }
+
+  void log(Verbosity lvl, const FormatOrString& fs) override {
+    nlohmann::json json;
+    json["action"] = "msg";
+    json["level"] = lvl;
+    json["msg"] = fs.s;
+    write(json);
+  }
+
+  void startActivity(ActivityId act, Verbosity lvl, ActivityType type,
+                     const std::string& s, const Fields& fields,
+                     ActivityId parent) override {
+    nlohmann::json json;
+    json["action"] = "start";
+    json["id"] = act;
+    json["level"] = lvl;
+    json["type"] = type;
+    json["text"] = s;
+    addFields(json, fields);
+    // FIXME: handle parent
+    write(json);
+  }
+
+  void stopActivity(ActivityId act) override {
+    nlohmann::json json;
+    json["action"] = "stop";
+    json["id"] = act;
+    write(json);
+  }
+
+  void result(ActivityId act, ResultType type, const Fields& fields) override {
+    nlohmann::json json;
+    json["action"] = "result";
+    json["id"] = act;
+    json["type"] = type;
+    addFields(json, fields);
+    write(json);
+  }
+};
 
-Activity::Activity(Logger & logger, Verbosity lvl, ActivityType type,
-    const std::string & s, const Logger::Fields & fields, ActivityId parent)
-    : logger(logger), id(nextId++)
-{
-    logger.startActivity(id, lvl, type, s, fields, parent);
+Logger* makeJSONLogger(Logger& prevLogger) {
+  return new JSONLogger(prevLogger);
 }
 
-struct JSONLogger : Logger
-{
-    Logger & prevLogger;
-
-    JSONLogger(Logger & prevLogger) : prevLogger(prevLogger) { }
-
-    void addFields(nlohmann::json & json, const Fields & fields)
-    {
-        if (fields.empty()) return;
-        auto & arr = json["fields"] = nlohmann::json::array();
-        for (auto & f : fields)
-            if (f.type == Logger::Field::tInt)
-                arr.push_back(f.i);
-            else if (f.type == Logger::Field::tString)
-                arr.push_back(f.s);
-            else
-                abort();
-    }
+static Logger::Fields getFields(nlohmann::json& json) {
+  Logger::Fields fields;
+  for (auto& f : json) {
+    if (f.type() == nlohmann::json::value_t::number_unsigned)
+      fields.emplace_back(Logger::Field(f.get<uint64_t>()));
+    else if (f.type() == nlohmann::json::value_t::string)
+      fields.emplace_back(Logger::Field(f.get<std::string>()));
+    else
+      throw Error("unsupported JSON type %d", (int)f.type());
+  }
+  return fields;
+}
 
-    void write(const nlohmann::json & json)
-    {
-        prevLogger.log(lvlError, "@nix " + json.dump());
-    }
+bool handleJSONLogMessage(const std::string& msg, const Activity& act,
+                          std::map<ActivityId, Activity>& activities,
+                          bool trusted) {
+  if (!hasPrefix(msg, "@nix ")) return false;
 
-    void log(Verbosity lvl, const FormatOrString & fs) override
-    {
-        nlohmann::json json;
-        json["action"] = "msg";
-        json["level"] = lvl;
-        json["msg"] = fs.s;
-        write(json);
-    }
+  try {
+    auto json = nlohmann::json::parse(std::string(msg, 5));
 
-    void startActivity(ActivityId act, Verbosity lvl, ActivityType type,
-        const std::string & s, const Fields & fields, ActivityId parent) override
-    {
-        nlohmann::json json;
-        json["action"] = "start";
-        json["id"] = act;
-        json["level"] = lvl;
-        json["type"] = type;
-        json["text"] = s;
-        addFields(json, fields);
-        // FIXME: handle parent
-        write(json);
-    }
+    std::string action = json["action"];
 
-    void stopActivity(ActivityId act) override
-    {
-        nlohmann::json json;
-        json["action"] = "stop";
-        json["id"] = act;
-        write(json);
+    if (action == "start") {
+      auto type = (ActivityType)json["type"];
+      if (trusted || type == actDownload)
+        activities.emplace(
+            std::piecewise_construct, std::forward_as_tuple(json["id"]),
+            std::forward_as_tuple(*logger, (Verbosity)json["level"], type,
+                                  json["text"], getFields(json["fields"]),
+                                  act.id));
     }
 
-    void result(ActivityId act, ResultType type, const Fields & fields) override
-    {
-        nlohmann::json json;
-        json["action"] = "result";
-        json["id"] = act;
-        json["type"] = type;
-        addFields(json, fields);
-        write(json);
-    }
-};
+    else if (action == "stop")
+      activities.erase((ActivityId)json["id"]);
 
-Logger * makeJSONLogger(Logger & prevLogger)
-{
-    return new JSONLogger(prevLogger);
-}
+    else if (action == "result") {
+      auto i = activities.find((ActivityId)json["id"]);
+      if (i != activities.end())
+        i->second.result((ResultType)json["type"], getFields(json["fields"]));
+    }
 
-static Logger::Fields getFields(nlohmann::json & json)
-{
-    Logger::Fields fields;
-    for (auto & f : json) {
-        if (f.type() == nlohmann::json::value_t::number_unsigned)
-            fields.emplace_back(Logger::Field(f.get<uint64_t>()));
-        else if (f.type() == nlohmann::json::value_t::string)
-            fields.emplace_back(Logger::Field(f.get<std::string>()));
-        else throw Error("unsupported JSON type %d", (int) f.type());
+    else if (action == "setPhase") {
+      std::string phase = json["phase"];
+      act.result(resSetPhase, phase);
     }
-    return fields;
-}
 
-bool handleJSONLogMessage(const std::string & msg,
-    const Activity & act, std::map<ActivityId, Activity> & activities, bool trusted)
-{
-    if (!hasPrefix(msg, "@nix ")) return false;
-
-    try {
-        auto json = nlohmann::json::parse(std::string(msg, 5));
-
-        std::string action = json["action"];
-
-        if (action == "start") {
-            auto type = (ActivityType) json["type"];
-            if (trusted || type == actDownload)
-                activities.emplace(std::piecewise_construct,
-                    std::forward_as_tuple(json["id"]),
-                    std::forward_as_tuple(*logger, (Verbosity) json["level"], type,
-                        json["text"], getFields(json["fields"]), act.id));
-        }
-
-        else if (action == "stop")
-            activities.erase((ActivityId) json["id"]);
-
-        else if (action == "result") {
-            auto i = activities.find((ActivityId) json["id"]);
-            if (i != activities.end())
-                i->second.result((ResultType) json["type"], getFields(json["fields"]));
-        }
-
-        else if (action == "setPhase") {
-            std::string phase = json["phase"];
-            act.result(resSetPhase, phase);
-        }
-
-        else if (action == "msg") {
-            std::string msg = json["msg"];
-            logger->log((Verbosity) json["level"], msg);
-        }
-
-    } catch (std::exception & e) {
-        printError("bad log message from builder: %s", e.what());
+    else if (action == "msg") {
+      std::string msg = json["msg"];
+      logger->log((Verbosity)json["level"], msg);
     }
 
-    return true;
+  } catch (std::exception& e) {
+    printError("bad log message from builder: %s", e.what());
+  }
+
+  return true;
 }
 
 Activity::~Activity() {
-    try {
-        logger.stopActivity(id);
-    } catch (...) {
-        ignoreException();
-    }
+  try {
+    logger.stopActivity(id);
+  } catch (...) {
+    ignoreException();
+  }
 }
 
-}
+}  // namespace nix