ceph/src/rocksdb/util/event_logger.cc

   1 //  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
   2 //  This source code is licensed under the BSD-style license found in the
   3 //  LICENSE file in the root directory of this source tree. An additional grant
   4 //  of patent rights can be found in the PATENTS file in the same directory.
   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