148 lines
4.3 KiB
C++
148 lines
4.3 KiB
C++
// assert_util.cpp
|
|
|
|
/**
|
|
* Copyright (C) 2008 10gen Inc.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3,
|
|
* as published by the Free Software Foundation.
|
|
*
|
|
* 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 Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "assert_util.h"
|
|
#include "assert.h"
|
|
#include "file.h"
|
|
|
|
namespace mongo {
|
|
|
|
string getDbContext();
|
|
|
|
Assertion lastAssert[4];
|
|
|
|
/* "warning" assert -- safe to continue, so we don't throw exception. */
|
|
void wasserted(const char *msg, const char *file, unsigned line) {
|
|
problem() << "Assertion failure " << msg << ' ' << file << ' ' << dec << line << endl;
|
|
sayDbContext();
|
|
raiseError(msg && *msg ? msg : "wassertion failure");
|
|
lastAssert[1].set(msg, getDbContext().c_str(), file, line);
|
|
}
|
|
|
|
void asserted(const char *msg, const char *file, unsigned line) {
|
|
problem() << "Assertion failure " << msg << ' ' << file << ' ' << dec << line << endl;
|
|
sayDbContext();
|
|
raiseError(msg && *msg ? msg : "assertion failure");
|
|
lastAssert[0].set(msg, getDbContext().c_str(), file, line);
|
|
stringstream temp;
|
|
temp << "assertion " << file << ":" << line;
|
|
AssertionException e;
|
|
e.msg = temp.str();
|
|
throw e;
|
|
}
|
|
|
|
void uassert_nothrow(const char *msg) {
|
|
lastAssert[3].set(msg, getDbContext().c_str(), "", 0);
|
|
raiseError(msg);
|
|
}
|
|
|
|
int uacount = 0;
|
|
void uasserted(const char *msg) {
|
|
if ( ++uacount < 100 )
|
|
log() << "User Exception " << msg << endl;
|
|
else
|
|
RARELY log() << "User Exception " << msg << endl;
|
|
lastAssert[3].set(msg, getDbContext().c_str(), "", 0);
|
|
raiseError(msg);
|
|
throw UserException(msg);
|
|
}
|
|
|
|
void msgasserted(const char *msg) {
|
|
log() << "Assertion: " << msg << '\n';
|
|
lastAssert[2].set(msg, getDbContext().c_str(), "", 0);
|
|
raiseError(msg && *msg ? msg : "massert failure");
|
|
throw MsgAssertionException(msg);
|
|
}
|
|
|
|
string Assertion::toString() {
|
|
if ( !isSet() )
|
|
return "";
|
|
|
|
stringstream ss;
|
|
ss << msg << '\n';
|
|
if ( *context )
|
|
ss << context << '\n';
|
|
if ( *file )
|
|
ss << file << ' ' << line << '\n';
|
|
return ss.str();
|
|
}
|
|
|
|
|
|
class LoggingManager {
|
|
public:
|
|
LoggingManager()
|
|
: _enabled(0) , _file(0) {
|
|
}
|
|
|
|
void start( const string& lp , bool append ){
|
|
uassert( "LoggingManager already started" , ! _enabled );
|
|
_path = lp;
|
|
_append = append;
|
|
_enabled = 1;
|
|
rotate();
|
|
}
|
|
|
|
void rotate(){
|
|
if ( ! _enabled ){
|
|
cout << "LoggingManager not enabled" << endl;
|
|
return;
|
|
}
|
|
|
|
if ( _file ){
|
|
#ifdef _WIN32
|
|
cout << "log rotationd doesn't work on windows" << endl;
|
|
return;
|
|
#else
|
|
struct tm t;
|
|
localtime_r( &_opened , &t );
|
|
|
|
stringstream ss;
|
|
ss << _path << "." << ( 1900 + t.tm_year ) << "-" << t.tm_mon << "-" << t.tm_mday
|
|
<< "_" << t.tm_hour << "-" << t.tm_min << "-" << t.tm_sec;
|
|
string s = ss.str();
|
|
rename( _path.c_str() , s.c_str() );
|
|
#endif
|
|
}
|
|
|
|
_file = freopen( _path.c_str() , _append ? "a" : "w" , stdout );
|
|
_opened = time(0);
|
|
}
|
|
|
|
private:
|
|
|
|
bool _enabled;
|
|
string _path;
|
|
bool _append;
|
|
|
|
FILE * _file;
|
|
time_t _opened;
|
|
|
|
} loggingManager;
|
|
|
|
void initLogging( const string& lp , bool append ){
|
|
cout << "all output going to: " << lp << endl;
|
|
loggingManager.start( lp , append );
|
|
}
|
|
|
|
void rotateLogs( int signal ){
|
|
loggingManager.rotate();
|
|
}
|
|
}
|
|
|