1 // Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
2 // This source code is licensed under both the GPLv2 (found in the
3 // COPYING file in the root directory) and Apache 2.0 License
4 // (found in the LICENSE.Apache file in the root directory).
5
6 #include "logging/event_logger.h"
7
8 #include <cassert>
9 #include <cinttypes>
10 #include <sstream>
11 #include <string>
12
13 #include "logging/logging.h"
14 #include "util/string_util.h"
15
16 namespace ROCKSDB_NAMESPACE {
17
EventLoggerStream(Logger * logger)18 EventLoggerStream::EventLoggerStream(Logger* logger)
19 : logger_(logger),
20 log_buffer_(nullptr),
21 max_log_size_(0),
22 json_writer_(nullptr) {}
23
EventLoggerStream(LogBuffer * log_buffer,const size_t max_log_size)24 EventLoggerStream::EventLoggerStream(LogBuffer* log_buffer,
25 const size_t max_log_size)
26 : logger_(nullptr),
27 log_buffer_(log_buffer),
28 max_log_size_(max_log_size),
29 json_writer_(nullptr) {}
30
~EventLoggerStream()31 EventLoggerStream::~EventLoggerStream() {
32 if (json_writer_) {
33 json_writer_->EndObject();
34 #ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT
35 printf("%s\n", json_writer_->Get().c_str());
36 #else
37 if (logger_) {
38 EventLogger::Log(logger_, *json_writer_);
39 } else if (log_buffer_) {
40 assert(max_log_size_);
41 EventLogger::LogToBuffer(log_buffer_, *json_writer_, max_log_size_);
42 }
43 #endif
44 delete json_writer_;
45 }
46 }
47
Log(const JSONWriter & jwriter)48 void EventLogger::Log(const JSONWriter& jwriter) {
49 Log(logger_, jwriter);
50 }
51
Log(Logger * logger,const JSONWriter & jwriter)52 void EventLogger::Log(Logger* logger, const JSONWriter& jwriter) {
53 #ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT
54 printf("%s\n", jwriter.Get().c_str());
55 #else
56 ROCKSDB_NAMESPACE::Log(logger, "%s %s", Prefix(), jwriter.Get().c_str());
57 #endif
58 }
59
LogToBuffer(LogBuffer * log_buffer,const JSONWriter & jwriter,const size_t max_log_size)60 void EventLogger::LogToBuffer(LogBuffer* log_buffer, const JSONWriter& jwriter,
61 const size_t max_log_size) {
62 #ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT
63 printf("%s\n", jwriter.Get().c_str());
64 #else
65 assert(log_buffer);
66 ROCKSDB_NAMESPACE::LogToBuffer(log_buffer, max_log_size, "%s %s", Prefix(),
67 jwriter.Get().c_str());
68 #endif
69 }
70
71 } // namespace ROCKSDB_NAMESPACE
72