xuemingqiang/mongo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add waitMongoProgramOnPort to shell utils

Browse Source
This commit is contained in:
Aaron
2011-01-04 14:40:02 -08:00
parent 0ca0ce9b2a
commit 42ebc1c037
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
shell/shell_utils.cpp
View File

@@ -622,11 +622,22 @@ namespace mongo {
}
BSONObj WaitProgram( const BSONObj& a ) {
int pid = a.firstElement().numberInt();
int pid = oneArg( a ).numberInt();
BSONObj x = BSON( "" << wait_for_pid( pid ) );
shells.erase( pid );
return x;
}
BSONObj WaitMongoProgramOnPort( const BSONObj &a ) {
int port = oneArg( a ).numberInt();
uassert( 13621, "no known mongo program on port", dbs.count( port ) != 0 );
log() << "waiting port: " << port << ", pid: " << dbs[ port ].first << endl;
bool ret = wait_for_pid( dbs[ port ].first );
if ( ret ) {
dbs.erase( port );
}
return BSON( "" << ret );
}
BSONObj StartMongoProgram( const BSONObj &a ) {
_nokillop = true;
@@ -902,6 +913,7 @@ namespace mongo {
scope.injectNative( "rawMongoProgramOutput", RawMongoProgramOutput );
scope.injectNative( "clearRawMongoProgramOutput", ClearRawMongoProgramOutput );
scope.injectNative( "waitProgram" , WaitProgram );
scope.injectNative( "waitMongoProgramOnPort" , WaitMongoProgramOnPort );
scope.injectNative( "getHostName" , getHostName );
scope.injectNative( "removeFile" , removeFile );