67 lines
1.8 KiB
C++
67 lines
1.8 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
#ifndef __STDC_FORMAT_MACROS
|
|
#define __STDC_FORMAT_MACROS
|
|
#endif
|
|
|
|
#include "util/event_logger.h"
|
|
|
|
#include <inttypes.h>
|
|
#include <cassert>
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
#include "util/string_util.h"
|
|
|
|
namespace rocksdb {
|
|
|
|
|
|
EventLoggerStream::EventLoggerStream(Logger* logger)
|
|
: logger_(logger), log_buffer_(nullptr), json_writer_(nullptr) {}
|
|
|
|
EventLoggerStream::EventLoggerStream(LogBuffer* log_buffer)
|
|
: logger_(nullptr), log_buffer_(log_buffer), json_writer_(nullptr) {}
|
|
|
|
EventLoggerStream::~EventLoggerStream() {
|
|
if (json_writer_) {
|
|
json_writer_->EndObject();
|
|
#ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT
|
|
printf("%s\n", json_writer_->Get().c_str());
|
|
#else
|
|
if (logger_) {
|
|
EventLogger::Log(logger_, *json_writer_);
|
|
} else if (log_buffer_) {
|
|
EventLogger::LogToBuffer(log_buffer_, *json_writer_);
|
|
}
|
|
#endif
|
|
delete json_writer_;
|
|
}
|
|
}
|
|
|
|
void EventLogger::Log(const JSONWriter& jwriter) {
|
|
Log(logger_, jwriter);
|
|
}
|
|
|
|
void EventLogger::Log(Logger* logger, const JSONWriter& jwriter) {
|
|
#ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT
|
|
printf("%s\n", jwriter.Get().c_str());
|
|
#else
|
|
rocksdb::Log(logger, "%s %s", Prefix(), jwriter.Get().c_str());
|
|
#endif
|
|
}
|
|
|
|
void EventLogger::LogToBuffer(
|
|
LogBuffer* log_buffer, const JSONWriter& jwriter) {
|
|
#ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT
|
|
printf("%s\n", jwriter.Get().c_str());
|
|
#else
|
|
assert(log_buffer);
|
|
rocksdb::LogToBuffer(log_buffer, "%s %s", Prefix(), jwriter.Get().c_str());
|
|
#endif
|
|
}
|
|
|
|
} // namespace rocksdb
|