]>
Commit | Line | Data |
---|---|---|
7c673cae | 1 | // Copyright (c) 2011-present, Facebook, Inc. All rights reserved. |
11fdf7f2 TL |
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). | |
7c673cae FG |
5 | |
6 | #ifndef __STDC_FORMAT_MACROS | |
7 | #define __STDC_FORMAT_MACROS | |
8 | #endif | |
9 | ||
10 | #include "util/event_logger.h" | |
11 | ||
12 | #include <inttypes.h> | |
13 | #include <cassert> | |
14 | #include <sstream> | |
15 | #include <string> | |
16 | ||
17 | #include "util/logging.h" | |
18 | #include "util/string_util.h" | |
19 | ||
20 | namespace rocksdb { | |
21 | ||
22 | ||
23 | EventLoggerStream::EventLoggerStream(Logger* logger) | |
24 | : logger_(logger), log_buffer_(nullptr), json_writer_(nullptr) {} | |
25 | ||
26 | EventLoggerStream::EventLoggerStream(LogBuffer* log_buffer) | |
27 | : logger_(nullptr), log_buffer_(log_buffer), json_writer_(nullptr) {} | |
28 | ||
29 | EventLoggerStream::~EventLoggerStream() { | |
30 | if (json_writer_) { | |
31 | json_writer_->EndObject(); | |
32 | #ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT | |
33 | printf("%s\n", json_writer_->Get().c_str()); | |
34 | #else | |
35 | if (logger_) { | |
36 | EventLogger::Log(logger_, *json_writer_); | |
37 | } else if (log_buffer_) { | |
38 | EventLogger::LogToBuffer(log_buffer_, *json_writer_); | |
39 | } | |
40 | #endif | |
41 | delete json_writer_; | |
42 | } | |
43 | } | |
44 | ||
45 | void EventLogger::Log(const JSONWriter& jwriter) { | |
46 | Log(logger_, jwriter); | |
47 | } | |
48 | ||
49 | void EventLogger::Log(Logger* logger, const JSONWriter& jwriter) { | |
50 | #ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT | |
51 | printf("%s\n", jwriter.Get().c_str()); | |
52 | #else | |
53 | rocksdb::Log(logger, "%s %s", Prefix(), jwriter.Get().c_str()); | |
54 | #endif | |
55 | } | |
56 | ||
57 | void EventLogger::LogToBuffer( | |
58 | LogBuffer* log_buffer, const JSONWriter& jwriter) { | |
59 | #ifdef ROCKSDB_PRINT_EVENTS_TO_STDOUT | |
60 | printf("%s\n", jwriter.Get().c_str()); | |
61 | #else | |
62 | assert(log_buffer); | |
63 | rocksdb::LogToBuffer(log_buffer, "%s %s", Prefix(), jwriter.Get().c_str()); | |
64 | #endif | |
65 | } | |
66 | ||
67 | } // namespace rocksdb |