mirror of
https://github.com/mongodb/mongo.git
synced 2024-12-01 01:21:03 +01:00
78 lines
2.3 KiB
C++
78 lines
2.3 KiB
C++
// utils.cpp
|
|
/*
|
|
* Copyright (C) 2010 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 "pch.h"
|
|
#include "engine.h"
|
|
#include "../util/md5.hpp"
|
|
#include "../util/version.h"
|
|
|
|
namespace mongo {
|
|
|
|
void installBenchmarkSystem( Scope& scope );
|
|
|
|
BSONObj jsmd5( const BSONObj &a, void* data ) {
|
|
uassert( 10261 , "js md5 needs a string" , a.firstElement().type() == String );
|
|
const char * s = a.firstElement().valuestrsafe();
|
|
|
|
md5digest d;
|
|
md5_state_t st;
|
|
md5_init(&st);
|
|
md5_append( &st , (const md5_byte_t*)s , strlen( s ) );
|
|
md5_finish(&st, d);
|
|
|
|
return BSON( "" << digestToString( d ) );
|
|
}
|
|
|
|
BSONObj JSVersion( const BSONObj& args, void* data ) {
|
|
cout << "version: " << versionString << endl;
|
|
if ( strstr( versionString , "+" ) )
|
|
printGitVersion();
|
|
return BSONObj();
|
|
}
|
|
|
|
|
|
BSONObj JSSleep(const mongo::BSONObj &args, void* data) {
|
|
assert( args.nFields() == 1 );
|
|
assert( args.firstElement().isNumber() );
|
|
int ms = int( args.firstElement().number() );
|
|
{
|
|
auto_ptr< ScriptEngine::Unlocker > u = globalScriptEngine->newThreadUnlocker();
|
|
sleepmillis( ms );
|
|
}
|
|
|
|
BSONObjBuilder b;
|
|
b.appendUndefined( "" );
|
|
return b.obj();
|
|
}
|
|
|
|
// ---------------------------------
|
|
// ---- installer --------
|
|
// ---------------------------------
|
|
|
|
void installGlobalUtils( Scope& scope ) {
|
|
scope.injectNative( "hex_md5" , jsmd5 );
|
|
scope.injectNative( "version" , JSVersion );
|
|
scope.injectNative( "sleep" , JSSleep );
|
|
|
|
installBenchmarkSystem( scope );
|
|
}
|
|
|
|
}
|
|
|
|
|