147 lines
4.8 KiB
C++
147 lines
4.8 KiB
C++
/*
|
|
* Copyright (c) 2016, 2018, Oracle and/or its affiliates. All rights reserved.
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
*
|
|
* This code is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 only, as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
|
* or visit www.oracle.com if you need additional information or have any
|
|
* questions.
|
|
*
|
|
*/
|
|
#include "precompiled.hpp"
|
|
#include "logging/logMessageBuffer.hpp"
|
|
#include "memory/allocation.inline.hpp"
|
|
#include "runtime/thread.inline.hpp"
|
|
|
|
template <typename T>
|
|
static void grow(T*& buffer, size_t& capacity, size_t minimum_length = 0) {
|
|
size_t new_size = capacity * 2;
|
|
if (new_size < minimum_length) {
|
|
new_size = minimum_length;
|
|
}
|
|
buffer = REALLOC_C_HEAP_ARRAY(T, buffer, new_size, mtLogging);
|
|
capacity = new_size;
|
|
}
|
|
|
|
LogMessageBuffer::LogMessageBuffer() : _message_buffer_size(0),
|
|
_message_buffer_capacity(0),
|
|
_message_buffer(NULL),
|
|
_line_count(0),
|
|
_line_capacity(0),
|
|
_lines(NULL),
|
|
_allocated(false),
|
|
_least_detailed_level(LogLevel::Off),
|
|
_prefix_fn(NULL) {
|
|
}
|
|
|
|
LogMessageBuffer::~LogMessageBuffer() {
|
|
if (_allocated) {
|
|
FREE_C_HEAP_ARRAY(char, _message_buffer);
|
|
FREE_C_HEAP_ARRAY(LogLine, _lines);
|
|
}
|
|
}
|
|
|
|
void LogMessageBuffer::reset() {
|
|
_message_buffer_size = 0;
|
|
_line_count = 0;
|
|
}
|
|
|
|
void LogMessageBuffer::initialize_buffers() {
|
|
assert(!_allocated, "buffer already initialized/allocated");
|
|
_allocated = true;
|
|
_message_buffer = NEW_C_HEAP_ARRAY(char, InitialMessageBufferCapacity, mtLogging);
|
|
_lines = NEW_C_HEAP_ARRAY(LogLine, InitialLineCapacity, mtLogging);
|
|
_message_buffer_capacity = InitialMessageBufferCapacity;
|
|
_line_capacity = InitialLineCapacity;
|
|
}
|
|
|
|
void LogMessageBuffer::Iterator::skip_messages_with_finer_level() {
|
|
for (; _current_line_index < _message._line_count; _current_line_index++) {
|
|
if (_message._lines[_current_line_index].level >= _level) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void LogMessageBuffer::write(LogLevelType level, const char* fmt, ...) {
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
vwrite(level, fmt, args);
|
|
va_end(args);
|
|
};
|
|
|
|
void LogMessageBuffer::vwrite(LogLevelType level, const char* fmt, va_list args) {
|
|
if (!_allocated) {
|
|
initialize_buffers();
|
|
}
|
|
|
|
if (level > _least_detailed_level) {
|
|
_least_detailed_level = level;
|
|
}
|
|
|
|
size_t written;
|
|
for (int attempts = 0; attempts < 2; attempts++) {
|
|
written = 0;
|
|
size_t remaining_buffer_length = _message_buffer_capacity - _message_buffer_size;
|
|
char* current_buffer_position = _message_buffer + _message_buffer_size;
|
|
|
|
if (_prefix_fn != NULL) {
|
|
written += _prefix_fn(current_buffer_position, remaining_buffer_length);
|
|
current_buffer_position += written;
|
|
if (remaining_buffer_length < written) {
|
|
remaining_buffer_length = 0;
|
|
} else {
|
|
remaining_buffer_length -= written;
|
|
}
|
|
}
|
|
|
|
va_list copy;
|
|
va_copy(copy, args);
|
|
written += (size_t)os::vsnprintf(current_buffer_position, remaining_buffer_length, fmt, copy) + 1;
|
|
va_end(copy);
|
|
if (written > _message_buffer_capacity - _message_buffer_size) {
|
|
assert(attempts == 0, "Second attempt should always have a sufficiently large buffer (resized to fit).");
|
|
grow(_message_buffer, _message_buffer_capacity, _message_buffer_size + written);
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (_line_count == _line_capacity) {
|
|
grow(_lines, _line_capacity);
|
|
}
|
|
|
|
_lines[_line_count].level = level;
|
|
_lines[_line_count].message_offset = _message_buffer_size;
|
|
_message_buffer_size += written;
|
|
_line_count++;
|
|
}
|
|
|
|
#define LOG_LEVEL(level, name) \
|
|
LogMessageBuffer& LogMessageBuffer::v##name(const char* fmt, va_list args) { \
|
|
vwrite(LogLevel::level, fmt, args); \
|
|
return *this; \
|
|
} \
|
|
LogMessageBuffer& LogMessageBuffer::name(const char* fmt, ...) { \
|
|
va_list args; \
|
|
va_start(args, fmt); \
|
|
vwrite(LogLevel::level, fmt, args); \
|
|
va_end(args); \
|
|
return *this; \
|
|
}
|
|
LOG_LEVEL_LIST
|
|
#undef LOG_LEVEL
|