#10 get rid of slow stringstream and add in place formating of string

pull/11/head
Geoffrey Merck 4 years ago
parent 9517169491
commit 0914d3145d

35
Log.h

@ -19,11 +19,11 @@
#pragma once #pragma once
#include <ctime>
#include <string> #include <string>
#include <vector> #include <vector>
#include <boost/algorithm/string.hpp> #include <boost/algorithm/string.hpp>
#include <mutex> #include <mutex>
#include <sstream>
#include <cassert> #include <cassert>
#include "StringUtils.h" #include "StringUtils.h"
@ -36,45 +36,46 @@ private:
static bool m_addedTargets; static bool m_addedTargets;
static std::recursive_mutex m_targetsMutex; static std::recursive_mutex m_targetsMutex;
static void getTimeStamp(std::string & s); static void getTimeStamp(std::string& s);
template<typename... Args> static void formatLogMessage(std::string& output, LOG_SEVERITY severity, const std::string & f, Args... args) template<typename... Args>
static void formatLogMessage(std::string& output, LOG_SEVERITY severity, const std::string & f, Args... args)
{ {
assert(severity != LOG_NONE); assert(severity != LOG_NONE);
std::string severityStr; std::string severityStr(" ");
switch (severity) switch (severity)
{ {
case LOG_DEBUG: case LOG_DEBUG:
severityStr = "DEBUG "; severityStr.assign("DEBUG ");
break; break;
case LOG_ERROR: case LOG_ERROR:
severityStr = "ERROR "; severityStr.assign("ERROR ");
break; break;
case LOG_FATAL: case LOG_FATAL:
severityStr = "FATAL "; severityStr.assign("FATAL ");
break; break;
case LOG_INFO : case LOG_INFO :
severityStr = "INFO "; severityStr.assign("INFO ");
break; break;
case LOG_WARNING: case LOG_WARNING:
severityStr = "WARNING"; severityStr.assign("WARNING");
break; break;
case LOG_TRACE: case LOG_TRACE:
severityStr = "TRACE "; severityStr.assign("TRACE ");
break; break;
default: default:
break; break;
} }
std::string message = CStringUtils::string_format(f, args...); std::string timestamp;
boost::trim(message); getTimeStamp(timestamp);
std::string timeUtc;
getTimeStamp(timeUtc);
std::stringstream s;
s << "[" << timeUtc << "] [" << severityStr << "] " << message << std::endl;
output = s.str(); std::string f2("[%s] [%s] ");
f2.append(f);
CStringUtils::string_format_in_place(output, f2, timestamp.c_str(), severityStr.c_str(), args...);
boost::trim_if(output, [](char c){ return c == '\n' || c == '\r' || c == ' ' || c == '\t'; });
output.push_back('\n');
} }
public: public:

@ -30,12 +30,23 @@ class CStringUtils {
public: public:
template<typename ... Args> template<typename ... Args>
static std::string string_format( const std::string& format, Args ... args ) static std::string string_format( const std::string& format, Args ... args )
{
std::string ret;
string_format_in_place(ret, format, args...);
return ret;
}
template<typename ... Args>
static void string_format_in_place(std::string& output, const std::string& format, Args ... args )
{ {
int size_s = std::snprintf( nullptr, 0, format.c_str(), args ... ) + 1; // Extra space for '\0' int size_s = std::snprintf( nullptr, 0, format.c_str(), args ... ) + 1; // Extra space for '\0'
if( size_s <= 0 ){ throw std::runtime_error( "Error during formatting." ); } if( size_s <= 0 ){ throw std::runtime_error( "Error during formatting." ); }
auto size = static_cast<size_t>( size_s ); auto size = static_cast<size_t>( size_s );
auto buf = std::make_unique<char[]>( size ); auto buf = std::make_unique<char[]>( size );
std::snprintf( buf.get(), size, format.c_str(), args ... ); std::snprintf( buf.get(), size, format.c_str(), args ... );
return std::string( buf.get(), buf.get() + size - 1 ); // We don't want the '\0' inside
output.reserve(size);
output.assign(buf.get(), size - 1); // -1 because we do not need trailing '\0'
} }
}; };

Loading…
Cancel
Save

Powered by TurnKey Linux.