logchecker.cc

Go to the documentation of this file.

#include "ion/base/logchecker.h"

#include "ion/base/stringutils.h"

namespace ion {
namespace base {

LogChecker::LogChecker()
    : previous_writer_(base::GetLogEntryWriter()) {
  SetLogEntryWriter(this);
}

LogChecker::~LogChecker() {
  SetLogEntryWriter(previous_writer_);

  if (HasAnyMessages()) {
    LOG(ERROR) << "LogChecker destroyed with messages: " << GetLogString();
  }
}

bool LogChecker::HasMessage(const std::string& severity_string,
                            const std::string& substring) {
#if ION_PRODUCTION
  return true;
#else
  const std::string log = GetLogString();
  const std::vector<std::string> messages = SplitString(log, "\n");
  const size_t count = messages.size();
  for (size_t i = 0; i < count; ++i) {
    const bool check_dcheck = severity_string == "DFATAL" &&
        messages[i].find("DCHECK failed") != std::string::npos &&
        i < count - 1U;
    if ((messages[i].find(severity_string) == 0 &&
        messages[i].find(substring) != std::string::npos) ||
        (check_dcheck &&
         messages[i + 1U].find(substring) != std::string::npos)) {
      ClearLog();
      return true;
    }
  }
  return false;
#endif
}

bool LogChecker::HasNoMessage(const std::string& severity_string,
                              const std::string& substring) {
#if ION_PRODUCTION
  return true;
#else
  const std::string log = GetLogString();
  const std::vector<std::string> messages = SplitString(log, "\n");
  const size_t count = messages.size();
  for (size_t i = 0; i < count; ++i) {
    if (messages[i].find(severity_string) == 0 &&
        messages[i].find(substring) != std::string::npos)
      return false;
  }
  return true;
#endif
}

const std::vector<std::string> LogChecker::GetAllMessages() {
  const std::string log = GetLogString();
  ClearLog();
  return SplitString(log, "\n");
}

void LogChecker::Write(port::LogSeverity severity, const std::string& message) {
  stream_ << GetSeverityName(severity) << " " << message << std::endl;
}

}  // namespace base
}  // namespace ion