123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514 |
- "use strict";
- var EventEmitter = require('events').EventEmitter
- , inherits = require('util').inherits
- , f = require('util').format
- , ServerCapabilities = require('./topology_base').ServerCapabilities
- , MongoError = require('mongodb-core').MongoError
- , CMongos = require('mongodb-core').Mongos
- , Cursor = require('./cursor')
- , AggregationCursor = require('./aggregation_cursor')
- , CommandCursor = require('./command_cursor')
- , Define = require('./metadata')
- , Server = require('./server')
- , Store = require('./topology_base').Store
- , MAX_JS_INT = require('./utils').MAX_JS_INT
- , translateOptions = require('./utils').translateOptions
- , filterOptions = require('./utils').filterOptions
- , mergeOptions = require('./utils').mergeOptions
- , os = require('os');
- var driverVersion = require(__dirname + '/../package.json').version;
- var nodejsversion = f('Node.js %s, %s', process.version, os.endianness());
- var type = os.type();
- var name = process.platform;
- var architecture = process.arch;
- var release = os.release();
-
- var legalOptionNames = ['ha', 'haInterval', 'acceptableLatencyMS'
- , 'poolSize', 'ssl', 'checkServerIdentity', 'sslValidate'
- , 'sslCA', 'sslCert', 'sslKey', 'sslPass', 'socketOptions', 'bufferMaxEntries'
- , 'store', 'auto_reconnect', 'autoReconnect', 'emitError'
- , 'keepAlive', 'noDelay', 'connectTimeoutMS', 'socketTimeoutMS'
- , 'loggerLevel', 'logger', 'reconnectTries', 'appname', 'domainsEnabled'
- , 'servername', 'promoteLongs', 'promoteValues', 'promoteBuffers'];
- var Mongos = function(servers, options) {
- if(!(this instanceof Mongos)) return new Mongos(servers, options);
- options = options || {};
- var self = this;
-
- options = filterOptions(options, legalOptionNames);
-
- for(var i = 0; i < servers.length; i++) {
- if(!(servers[i] instanceof Server)) {
- throw MongoError.create({message: "all seed list instances must be of the Server type", driver:true});
- }
- }
-
- var storeOptions = {
- force: false
- , bufferMaxEntries: typeof options.bufferMaxEntries == 'number' ? options.bufferMaxEntries : MAX_JS_INT
- }
-
- var store = options.store || new Store(self, storeOptions);
-
- EventEmitter.call(this);
-
- var seedlist = servers.map(function(x) {
- return {host: x.host, port: x.port}
- });
-
- var reconnect = typeof options.auto_reconnect == 'boolean' ? options.auto_reconnect : true;
- reconnect = typeof options.autoReconnect == 'boolean' ? options.autoReconnect : reconnect;
-
- var clonedOptions = mergeOptions({}, {
- disconnectHandler: store,
- cursorFactory: Cursor,
- reconnect: reconnect,
- emitError: typeof options.emitError == 'boolean' ? options.emitError : true,
- size: typeof options.poolSize == 'number' ? options.poolSize : 5
- });
-
- clonedOptions = translateOptions(clonedOptions, options);
-
- var socketOptions = options.socketOptions && Object.keys(options.socketOptions).length > 0
- ? options.socketOptions : options;
-
- clonedOptions = translateOptions(clonedOptions, socketOptions);
- if(typeof clonedOptions.keepAlive == 'number') {
- clonedOptions.keepAliveInitialDelay = clonedOptions.keepAlive;
- clonedOptions.keepAlive = clonedOptions.keepAlive > 0;
- }
-
- this.clientInfo = {
- driver: {
- name: "nodejs",
- version: driverVersion
- },
- os: {
- type: type,
- name: name,
- architecture: architecture,
- version: release
- },
- platform: nodejsversion
- }
-
- clonedOptions.clientInfo = this.clientInfo;
-
- if(options.appname) {
- clonedOptions.clientInfo.application = { name: options.appname };
- }
-
- var mongos = new CMongos(seedlist, clonedOptions)
-
- var sCapabilities = null;
-
- this.s = {
-
- mongos: mongos
-
- , sCapabilities: sCapabilities
-
- , debug: clonedOptions.debug
-
- , storeOptions: storeOptions
-
- , clonedOptions: clonedOptions
-
- , store: store
-
- , options: options
- }
- }
- var define = Mongos.define = new Define('Mongos', Mongos, false);
- inherits(Mongos, EventEmitter);
- Object.defineProperty(Mongos.prototype, 'isMasterDoc', {
- enumerable:true, get: function() { return this.s.mongos.lastIsMaster(); }
- });
- Object.defineProperty(Mongos.prototype, 'parserType', {
- enumerable:true, get: function() {
- return this.s.mongos.parserType;
- }
- });
- Object.defineProperty(Mongos.prototype, 'bson', {
- enumerable: true, get: function() {
- return this.s.mongos.s.bson;
- }
- });
- Object.defineProperty(Mongos.prototype, 'haInterval', {
- enumerable:true, get: function() { return this.s.mongos.s.haInterval; }
- });
- Mongos.prototype.connect = function(db, _options, callback) {
- var self = this;
- if('function' === typeof _options) callback = _options, _options = {};
- if(_options == null) _options = {};
- if(!('function' === typeof callback)) callback = null;
- self.s.options = _options;
-
- self.s.storeOptions.bufferMaxEntries = db.bufferMaxEntries;
-
- var connectErrorHandler = function() {
- return function(err) {
-
- var events = ['timeout', 'error', 'close'];
- events.forEach(function(e) {
- self.removeListener(e, connectErrorHandler);
- });
- self.s.mongos.removeListener('connect', connectErrorHandler);
-
- try {
- callback(err);
- } catch(err) {
- process.nextTick(function() { throw err; })
- }
- }
- }
-
- var errorHandler = function(event) {
- return function(err) {
- if(event != 'error') {
- self.emit(event, err);
- }
- }
- }
-
- var reconnectHandler = function() {
- self.emit('reconnect');
- self.s.store.execute();
- }
-
- var relay = function(event) {
- return function(t, server) {
- self.emit(event, t, server);
- }
- }
-
- var connectHandler = function() {
-
- ["timeout", "error", "close", 'serverOpening', 'serverDescriptionChanged', 'serverHeartbeatStarted',
- 'serverHeartbeatSucceeded', 'serverHeartbeatFailed', 'serverClosed', 'topologyOpening',
- 'topologyClosed', 'topologyDescriptionChanged'].forEach(function(e) {
- self.s.mongos.removeAllListeners(e);
- });
-
- self.s.mongos.once('timeout', errorHandler('timeout'));
- self.s.mongos.once('error', errorHandler('error'));
- self.s.mongos.once('close', errorHandler('close'));
-
- self.s.mongos.on('serverDescriptionChanged', relay('serverDescriptionChanged'));
- self.s.mongos.on('serverHeartbeatStarted', relay('serverHeartbeatStarted'));
- self.s.mongos.on('serverHeartbeatSucceeded', relay('serverHeartbeatSucceeded'));
- self.s.mongos.on('serverHeartbeatFailed', relay('serverHeartbeatFailed'));
- self.s.mongos.on('serverOpening', relay('serverOpening'));
- self.s.mongos.on('serverClosed', relay('serverClosed'));
- self.s.mongos.on('topologyOpening', relay('topologyOpening'));
- self.s.mongos.on('topologyClosed', relay('topologyClosed'));
- self.s.mongos.on('topologyDescriptionChanged', relay('topologyDescriptionChanged'));
-
- self.s.mongos.on('fullsetup', relay('fullsetup'));
-
- self.emit('open', null, self);
-
- try {
- callback(null, self);
- } catch(err) {
- process.nextTick(function() { throw err; })
- }
- }
-
- self.s.mongos.once('timeout', connectErrorHandler('timeout'));
- self.s.mongos.once('error', connectErrorHandler('error'));
- self.s.mongos.once('close', connectErrorHandler('close'));
- self.s.mongos.once('connect', connectHandler);
-
- self.s.mongos.on('joined', relay('joined'));
- self.s.mongos.on('left', relay('left'));
-
- self.s.mongos.on('reconnect', reconnectHandler);
-
- self.s.mongos.connect(_options);
- }
- Mongos.prototype.capabilities = function() {
- if(this.s.sCapabilities) return this.s.sCapabilities;
- if(this.s.mongos.lastIsMaster() == null) return null;
- this.s.sCapabilities = new ServerCapabilities(this.s.mongos.lastIsMaster());
- return this.s.sCapabilities;
- }
- define.classMethod('capabilities', {callback: false, promise:false, returns: [ServerCapabilities]});
- Mongos.prototype.command = function(ns, cmd, options, callback) {
- this.s.mongos.command(ns, cmd, options, callback);
- }
- define.classMethod('command', {callback: true, promise:false});
- Mongos.prototype.insert = function(ns, ops, options, callback) {
- this.s.mongos.insert(ns, ops, options, function(e, m) {
- callback(e, m)
- });
- }
- define.classMethod('insert', {callback: true, promise:false});
- Mongos.prototype.update = function(ns, ops, options, callback) {
- this.s.mongos.update(ns, ops, options, callback);
- }
- define.classMethod('update', {callback: true, promise:false});
- Mongos.prototype.remove = function(ns, ops, options, callback) {
- this.s.mongos.remove(ns, ops, options, callback);
- }
- define.classMethod('remove', {callback: true, promise:false});
- Mongos.prototype.isDestroyed = function() {
- return this.s.mongos.isDestroyed();
- }
- Mongos.prototype.isConnected = function() {
- return this.s.mongos.isConnected();
- }
- define.classMethod('isConnected', {callback: false, promise:false, returns: [Boolean]});
- Mongos.prototype.cursor = function(ns, cmd, options) {
- options.disconnectHandler = this.s.store;
- return this.s.mongos.cursor(ns, cmd, options);
- }
- define.classMethod('cursor', {callback: false, promise:false, returns: [Cursor, AggregationCursor, CommandCursor]});
- Mongos.prototype.lastIsMaster = function() {
- return this.s.mongos.lastIsMaster();
- }
- Mongos.prototype.close = function(forceClosed) {
- this.s.mongos.destroy();
-
- if(forceClosed == true) {
- this.s.storeOptions.force = forceClosed;
- this.s.store.flush();
- }
- }
- define.classMethod('close', {callback: false, promise:false});
- Mongos.prototype.auth = function() {
- var args = Array.prototype.slice.call(arguments, 0);
- this.s.mongos.auth.apply(this.s.mongos, args);
- }
- define.classMethod('auth', {callback: true, promise:false});
- Mongos.prototype.logout = function() {
- var args = Array.prototype.slice.call(arguments, 0);
- this.s.mongos.logout.apply(this.s.mongos, args);
- }
- define.classMethod('logout', {callback: true, promise:false});
- Mongos.prototype.connections = function() {
- return this.s.mongos.connections();
- }
- define.classMethod('connections', {callback: false, promise:false, returns:[Array]});
- module.exports = Mongos;
|