140 lines
4.5 KiB
C++
140 lines
4.5 KiB
C++
#include <fc/log/console_appender.hpp>
|
|
#include <fc/log/log_message.hpp>
|
|
#include <fc/thread/unique_lock.hpp>
|
|
#include <fc/string.hpp>
|
|
#include <fc/variant.hpp>
|
|
#include <fc/reflect/variant.hpp>
|
|
#ifndef WIN32
|
|
#include <unistd.h>
|
|
#endif
|
|
#include <boost/thread/mutex.hpp>
|
|
#define COLOR_CONSOLE 1
|
|
#include "console_defines.h"
|
|
#include <fc/io/stdio.hpp>
|
|
#include <fc/exception/exception.hpp>
|
|
#include <iomanip>
|
|
#include <sstream>
|
|
|
|
|
|
namespace fc {
|
|
|
|
class console_appender::impl {
|
|
public:
|
|
config cfg;
|
|
color::type lc[log_level::off+1];
|
|
#ifdef WIN32
|
|
HANDLE console_handle;
|
|
#endif
|
|
};
|
|
|
|
console_appender::console_appender( const variant& args ) :
|
|
my(new impl)
|
|
{
|
|
#ifdef WIN32
|
|
my->console_handle = INVALID_HANDLE_VALUE;
|
|
#endif
|
|
try
|
|
{
|
|
my->cfg = args.as<config>();//fc::variant_cast<config>(args);
|
|
|
|
#ifdef WIN32
|
|
if (my->cfg.stream = stream::std_error)
|
|
my->console_handle = GetStdHandle(STD_ERROR_HANDLE);
|
|
else if (my->cfg.stream = stream::std_out)
|
|
my->console_handle = GetStdHandle(STD_OUTPUT_HANDLE);
|
|
#endif
|
|
|
|
for( int i = 0; i < log_level::off+1; ++i )
|
|
my->lc[i] = color::console_default;
|
|
for( auto itr = my->cfg.level_colors.begin(); itr != my->cfg.level_colors.end(); ++itr )
|
|
my->lc[itr->level] = itr->color;
|
|
}
|
|
catch ( exception& e )
|
|
{
|
|
fc::cerr<<e.to_detail_string()<<"\n";
|
|
throw;
|
|
}
|
|
}
|
|
|
|
console_appender::~console_appender() {}
|
|
|
|
#ifdef WIN32
|
|
static WORD
|
|
#else
|
|
static const char*
|
|
#endif
|
|
get_console_color(console_appender::color::type t ) {
|
|
switch( t ) {
|
|
case console_appender::color::red: return CONSOLE_RED;
|
|
case console_appender::color::green: return CONSOLE_GREEN;
|
|
case console_appender::color::brown: return CONSOLE_BROWN;
|
|
case console_appender::color::blue: return CONSOLE_BLUE;
|
|
case console_appender::color::magenta: return CONSOLE_MAGENTA;
|
|
case console_appender::color::cyan: return CONSOLE_CYAN;
|
|
case console_appender::color::white: return CONSOLE_WHITE;
|
|
case console_appender::color::console_default:
|
|
default:
|
|
return CONSOLE_DEFAULT;
|
|
}
|
|
}
|
|
|
|
boost::mutex& log_mutex() {
|
|
static boost::mutex m; return m;
|
|
}
|
|
|
|
void console_appender::log( const log_message& m ) {
|
|
//fc::string message = fc::format_string( m.get_format(), m.get_data() );
|
|
//fc::variant lmsg(m);
|
|
|
|
FILE* out = stream::std_error ? stderr : stdout;
|
|
|
|
//fc::string fmt_str = fc::format_string( cfg.format, mutable_variant_object(m.get_context())( "message", message) );
|
|
std::stringstream file_line;
|
|
file_line << m.get_context().get_file() <<":"<<m.get_context().get_line_number() <<" ";
|
|
|
|
///////////////
|
|
std::stringstream line;
|
|
line << (m.get_context().get_timestamp().time_since_epoch().count() % (1000ll*1000ll*60ll*60))/1000 <<"ms ";
|
|
line << std::setw( 10 ) << std::left << m.get_context().get_thread_name().substr(0,9).c_str() <<" "<<std::setw(30)<< std::left <<file_line.str();
|
|
|
|
auto me = m.get_context().get_method();
|
|
// strip all leading scopes...
|
|
if( me.size() )
|
|
{
|
|
uint32_t p = 0;
|
|
for( uint32_t i = 0;i < me.size(); ++i )
|
|
{
|
|
if( me[i] == ':' ) p = i;
|
|
}
|
|
|
|
if( me[p] == ':' ) ++p;
|
|
line << std::setw( 20 ) << std::left << m.get_context().get_method().substr(p,20).c_str() <<" ";
|
|
}
|
|
line << "] ";
|
|
fc::string message = fc::format_string( m.get_format(), m.get_data() );
|
|
line << message;//.c_str();
|
|
//
|
|
//
|
|
|
|
|
|
fc::unique_lock<boost::mutex> lock(log_mutex());
|
|
#ifdef WIN32
|
|
if (my->console_handle != INVALID_HANDLE_VALUE)
|
|
SetConsoleTextAttribute(my->console_handle, get_console_color( my->lc[m.get_context().get_log_level()] ));
|
|
#else
|
|
if(isatty(fileno(out))) fprintf( out, "\r%s", get_console_color( my->lc[m.get_context().get_log_level()] ) );
|
|
#endif
|
|
|
|
fprintf( out, "%s", line.str().c_str()); //fmt_str.c_str() );
|
|
|
|
#ifdef WIN32
|
|
if (my->console_handle != INVALID_HANDLE_VALUE)
|
|
SetConsoleTextAttribute(my->console_handle, CONSOLE_DEFAULT);
|
|
#else
|
|
if(isatty(fileno(out))) fprintf( out, "\r%s", CONSOLE_DEFAULT );
|
|
#endif
|
|
fprintf( out, "\n" );
|
|
if( my->cfg.flush ) fflush( out );
|
|
}
|
|
|
|
}
|