Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

436 lines (356 sloc) 13.109 kB
// engine.cpp
/* Copyright 2009 10gen Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "pch.h"
#include "engine.h"
#include "../util/file.h"
#include "../client/dbclient.h"
namespace mongo {
long long Scope::_lastVersion = 1;
int Scope::_numScopes = 0;
Scope::Scope() : _localDBName("") , _loadedVersion(0){
_numScopes++;
}
Scope::~Scope(){
_numScopes--;
}
ScriptEngine::ScriptEngine() : _scopeInitCallback() {
}
ScriptEngine::~ScriptEngine(){
}
void Scope::append( BSONObjBuilder & builder , const char * fieldName , const char * scopeName ){
int t = type( scopeName );
switch ( t ){
case Object:
builder.append( fieldName , getObject( scopeName ) );
break;
case Array:
builder.appendArray( fieldName , getObject( scopeName ) );
break;
case NumberDouble:
builder.append( fieldName , getNumber( scopeName ) );
break;
case NumberInt:
builder.append( fieldName , getNumberInt( scopeName ) );
break;
case NumberLong:
builder.append( fieldName , getNumberLongLong( scopeName ) );
break;
case String:
builder.append( fieldName , getString( scopeName ).c_str() );
break;
case Bool:
builder.appendBool( fieldName , getBoolean( scopeName ) );
break;
case jstNULL:
case Undefined:
builder.appendNull( fieldName );
break;
case Date:
// TODO: make signed
builder.appendDate( fieldName , Date_t((unsigned long long)getNumber( scopeName )) );
break;
default:
stringstream temp;
temp << "can't append type from:";
temp << t;
uassert( 10206 , temp.str() , 0 );
}
}
int Scope::invoke( const char* code , const BSONObj& args, int timeoutMs ){
ScriptingFunction func = createFunction( code );
uassert( 10207 , "compile failed" , func );
return invoke( func , args, timeoutMs );
}
bool Scope::execFile( const string& filename , bool printResult , bool reportError , bool assertOnError, int timeoutMs ){
path p( filename );
if ( ! exists( p ) ){
cout << "file [" << filename << "] doesn't exist" << endl;
if ( assertOnError )
assert( 0 );
return false;
}
// iterate directories and recurse using all *.js files in the directory
if ( is_directory( p ) ){
directory_iterator end;
bool empty = true;
for (directory_iterator it (p); it != end; it++){
empty = false;
path sub (*it);
if (!endsWith(sub.string().c_str(), ".js"))
continue;
if (!execFile(sub.string().c_str(), printResult, reportError, assertOnError, timeoutMs))
return false;
}
if (empty){
cout << "directory [" << filename << "] doesn't have any *.js files" << endl;
if ( assertOnError )
assert( 0 );
return false;
}
return true;
}
File f;
f.open( filename.c_str() , true );
fileofs L = f.len();
assert( L <= 0x7ffffffe );
char * data = (char*)malloc( (size_t) L+1 );
data[L] = 0;
f.read( 0 , data , (size_t) L );
return exec( data , filename , printResult , reportError , assertOnError, timeoutMs );
}
void Scope::storedFuncMod(){
_lastVersion++;
}
void Scope::validateObjectIdString( const string &str ) {
massert( 10448 , "invalid object id: length", str.size() == 24 );
for ( string::size_type i=0; i<str.size(); i++ ){
char c = str[i];
if ( ( c >= '0' && c <= '9' ) ||
( c >= 'a' && c <= 'f' ) ||
( c >= 'A' && c <= 'F' ) ){
continue;
}
massert( 10430 , "invalid object id: not hex", false );
}
}
void Scope::loadStored( bool ignoreNotConnected ){
if ( _localDBName.size() == 0 ){
if ( ignoreNotConnected )
return;
uassert( 10208 , "need to have locallyConnected already" , _localDBName.size() );
}
if ( _loadedVersion == _lastVersion )
return;
_loadedVersion = _lastVersion;
string coll = _localDBName + ".system.js";
static DBClientBase * db = createDirectClient();
auto_ptr<DBClientCursor> c = db->query( coll , Query() );
set<string> thisTime;
while ( c->more() ){
BSONObj o = c->next();
BSONElement n = o["_id"];
BSONElement v = o["value"];
uassert( 10209 , "name has to be a string" , n.type() == String );
uassert( 10210 , "value has to be set" , v.type() != EOO );
setElement( n.valuestr() , v );
thisTime.insert( n.valuestr() );
_storedNames.insert( n.valuestr() );
}
// --- remove things from scope that were removed
list<string> toremove;
for ( set<string>::iterator i=_storedNames.begin(); i!=_storedNames.end(); i++ ){
string n = *i;
if ( thisTime.count( n ) == 0 )
toremove.push_back( n );
}
for ( list<string>::iterator i=toremove.begin(); i!=toremove.end(); i++ ){
string n = *i;
_storedNames.erase( n );
execSetup( (string)"delete " + n , "clean up scope" );
}
}
ScriptingFunction Scope::createFunction( const char * code ){
if ( code[0] == '/' && code [1] == '*' ){
code += 2;
while ( code[0] && code[1] ){
if ( code[0] == '*' && code[1] == '/' ){
code += 2;
break;
}
code++;
}
}
map<string,ScriptingFunction>::iterator i = _cachedFunctions.find( code );
if ( i != _cachedFunctions.end() )
return i->second;
ScriptingFunction f = _createFunction( code );
_cachedFunctions[code] = f;
return f;
}
typedef map< string , list<Scope*> > PoolToScopes;
class ScopeCache {
public:
ScopeCache(){
_magic = 17;
}
~ScopeCache(){
assert( _magic == 17 );
_magic = 1;
if ( inShutdown() )
return;
clear();
}
void done( const string& pool , Scope * s ){
scoped_lock lk( _mutex );
list<Scope*> & l = _pools[pool];
if ( l.size() > 10 ){
delete s;
}
else {
l.push_back( s );
s->reset();
}
}
Scope * get( const string& pool ){
scoped_lock lk( _mutex );
list<Scope*> & l = _pools[pool];
if ( l.size() == 0 )
return 0;
Scope * s = l.back();
l.pop_back();
s->reset();
return s;
}
void clear(){
set<Scope*> seen;
for ( PoolToScopes::iterator i=_pools.begin() ; i != _pools.end(); i++ ){
for ( list<Scope*>::iterator j=i->second.begin(); j != i->second.end(); j++ ){
Scope * s = *j;
assert( ! seen.count( s ) );
delete s;
seen.insert( s );
}
}
_pools.clear();
}
private:
PoolToScopes _pools;
mongo::mutex _mutex;
int _magic;
};
thread_specific_ptr<ScopeCache> scopeCache;
class PooledScope : public Scope {
public:
PooledScope( const string pool , Scope * real ) : _pool( pool ) , _real( real ){
_real->loadStored( true );
};
virtual ~PooledScope(){
ScopeCache * sc = scopeCache.get();
if ( sc ){
sc->done( _pool , _real );
_real = 0;
}
else {
log() << "warning: scopeCache is empty!" << endl;
delete _real;
_real = 0;
}
}
void reset(){
_real->reset();
}
void init( BSONObj * data ){
_real->init( data );
}
void localConnect( const char * dbName ){
_real->localConnect( dbName );
}
void externalSetup(){
_real->externalSetup();
}
double getNumber( const char *field ){
return _real->getNumber( field );
}
string getString( const char *field ){
return _real->getString( field );
}
bool getBoolean( const char *field ){
return _real->getBoolean( field );
}
BSONObj getObject( const char *field ){
return _real->getObject( field );
}
int type( const char *field ){
return _real->type( field );
}
void setElement( const char *field , const BSONElement& val ){
_real->setElement( field , val );
}
void setNumber( const char *field , double val ){
_real->setNumber( field , val );
}
void setString( const char *field , const char * val ){
_real->setString( field , val );
}
void setObject( const char *field , const BSONObj& obj , bool readOnly=true ){
_real->setObject( field , obj , readOnly );
}
void setBoolean( const char *field , bool val ){
_real->setBoolean( field , val );
}
void setThis( const BSONObj * obj ){
_real->setThis( obj );
}
ScriptingFunction createFunction( const char * code ){
return _real->createFunction( code );
}
ScriptingFunction _createFunction( const char * code ){
return _real->createFunction( code );
}
/**
* @return 0 on success
*/
int invoke( ScriptingFunction func , const BSONObj& args, int timeoutMs , bool ignoreReturn ){
return _real->invoke( func , args , timeoutMs , ignoreReturn );
}
string getError(){
return _real->getError();
}
bool exec( const string& code , const string& name , bool printResult , bool reportError , bool assertOnError, int timeoutMs = 0 ){
return _real->exec( code , name , printResult , reportError , assertOnError , timeoutMs );
}
bool execFile( const string& filename , bool printResult , bool reportError , bool assertOnError, int timeoutMs = 0 ){
return _real->execFile( filename , printResult , reportError , assertOnError , timeoutMs );
}
void injectNative( const char *field, NativeFunction func ){
_real->injectNative( field , func );
}
void gc(){
_real->gc();
}
private:
string _pool;
Scope * _real;
};
auto_ptr<Scope> ScriptEngine::getPooledScope( const string& pool ){
if ( ! scopeCache.get() ){
scopeCache.reset( new ScopeCache() );
}
Scope * s = scopeCache->get( pool );
if ( ! s ){
s = newScope();
}
auto_ptr<Scope> p;
p.reset( new PooledScope( pool , s ) );
return p;
}
void ScriptEngine::threadDone(){
ScopeCache * sc = scopeCache.get();
if ( sc ){
sc->clear();
}
}
void ( *ScriptEngine::_connectCallback )( DBClientWithCommands & ) = 0;
ScriptEngine * globalScriptEngine;
bool hasJSReturn( const string& code ){
size_t x = code.find( "return" );
if ( x == string::npos )
return false;
return
( x == 0 || ! isalpha( code[x-1] ) ) &&
! isalpha( code[x+6] );
}
}
Jump to Line
Something went wrong with that request. Please try again.