[[email protected] tmp]# cat logger.h #ifndef LOGGER_UTIL_INCLUDED #define LOGGER_UTIL_INCLUDED /* Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; version 2 of the License. This program 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 for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */ /** A trivial placeholder for inserting Time signatures into streams. @example cout << Datetime() << "[Info] Today was a sunny day" << endl; */ #include <ostream> #include <string> #include <sstream> using namespace std; struct Datetime {}; ostream &operator<<(ostream &os, const Datetime &dt); class Gen_spaces { public: Gen_spaces(int s) { m_spaces.assign(s,‘ ‘); } ostream &operator<<(ostream &os) { return os; } friend ostream &operator<<(ostream &os, const Gen_spaces &gen); private: string m_spaces; }; ostream &operator<<(ostream &os, const Gen_spaces &gen); class Log : public ostream { public: Log(ostream &str, string logclass) : ostream(NULL), m_buffer(str, logclass) { this->init(&m_buffer); } void enabled(bool s) { m_buffer.enabled(s); } private: class Log_buff : public stringbuf { public: Log_buff(ostream &str, string &logc) :m_os(str),m_logc(logc), m_enabled(true) {} void set_log_class(string &s) { m_logc= s; } void enabled(bool s) { m_enabled= s; } virtual int sync(); private: ostream &m_os; string m_logc; bool m_enabled; }; Log_buff m_buffer; }; #endif /* LOGGER_UTIL_INCLUDED */ [[email protected] tmp]# cat logger.cc /* Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; version 2 of the License. This program 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 for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */ #include "logger.h" #include <iostream> #include <locale> ostream &operator<<(ostream &os, const Datetime &dt) { const char format[]= "%Y-%m-%d %X"; time_t t(time(NULL)); tm tm(*localtime(&t)); std::locale loc(cout.getloc()); ostringstream sout; const std::time_put<char> &tput = std::use_facet<std::time_put<char> >(loc); tput.put(sout.rdbuf(), sout, ‘\0‘, &tm, &format[0], &format[11]); os << sout.str() << " "; return os; } ostream &operator<<(ostream &os, const Gen_spaces &gen) { return os << gen.m_spaces; } int Log::Log_buff::sync() { string sout(str()); if (m_enabled && sout.length() > 0) { m_os << Datetime() << "[" << m_logc << "]"<< Gen_spaces(8-m_logc.length()) << sout; } str(""); m_os.flush(); return 0; } [[email protected] tmp]# cat main.cpp #include <iostream> #include "logger.h" using namespace std; int main(void) { cout << Datetime() << "[测试] 今天的天气不错!" << endl; return 0; } [[email protected] tmp]# g++ main.cpp logger.cc && ./a.out 2016-11-03 17:44:09 [测试] 今天的天气不错! [[email protected] tmp]#
时间: 2024-10-06 02:18:00