Permalink
Fetching contributors…
Cannot retrieve contributors at this time
131 lines (112 sloc) 3.36 KB
// bridge.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 "../util/message.h"
#include "../client/dbclient.h"
using namespace mongo;
using namespace std;
int port = 0;
string destUri;
class Forwarder {
public:
Forwarder( MessagingPort &mp ) : mp_( mp ) {
}
void operator()() const {
DBClientConnection dest;
string errmsg;
while( !dest.connect( destUri, errmsg ) )
sleepmillis( 500 );
Message m;
while( 1 ) {
m.reset();
if ( !mp_.recv( m ) ) {
cout << "end connection " << mp_.farEnd.toString() << endl;
mp_.shutdown();
break;
}
int oldId = m.data->id;
if ( m.data->operation() == dbQuery || m.data->operation() == dbMsg || m.data->operation() == dbGetMore ) {
Message response;
dest.port().call( m, response );
mp_.reply( m, response, oldId );
} else {
dest.port().say( m, oldId );
}
}
}
private:
MessagingPort &mp_;
};
set<MessagingPort*> ports;
class MyListener : public Listener {
public:
MyListener( int port ) : Listener( "", port ) {}
virtual void accepted(MessagingPort *mp) {
ports.insert( mp );
Forwarder f( *mp );
boost::thread t( f );
}
};
auto_ptr< MyListener > listener;
#if !defined(_WIN32)
void cleanup( int sig ) {
close( listener->socket() );
for ( set<MessagingPort*>::iterator i = ports.begin(); i != ports.end(); i++ )
(*i)->shutdown();
::exit( 0 );
}
void setupSignals() {
signal( SIGINT , cleanup );
signal( SIGTERM , cleanup );
signal( SIGPIPE , cleanup );
signal( SIGABRT , cleanup );
signal( SIGSEGV , cleanup );
signal( SIGBUS , cleanup );
signal( SIGFPE , cleanup );
}
#else
inline void setupSignals() {}
#endif
void helpExit() {
cout << "usage mongobridge --port <port> --dest <destUri>" << endl;
cout << " port: port to listen for mongo messages" << endl;
cout << " destUri: uri of remote mongod instance" << endl;
::exit( -1 );
}
void check( bool b ) {
if ( !b )
helpExit();
}
int main( int argc, char **argv ) {
setupSignals();
check( argc == 5 );
for( int i = 1; i < 5; ++i ) {
check( i % 2 != 0 );
if ( strcmp( argv[ i ], "--port" ) == 0 ) {
port = strtol( argv[ ++i ], 0, 10 );
} else if ( strcmp( argv[ i ], "--dest" ) == 0 ) {
destUri = argv[ ++i ];
} else {
check( false );
}
}
check( port != 0 && !destUri.empty() );
listener.reset( new MyListener( port ) );
listener->init();
listener->listen();
return 0;
}