123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511 |
- "use strict";
- var EventEmitter = require('events').EventEmitter
- , inherits = require('util').inherits
- , CServer = require('mongodb-core').Server
- , Cursor = require('./cursor')
- , AggregationCursor = require('./aggregation_cursor')
- , CommandCursor = require('./command_cursor')
- , f = require('util').format
- , ServerCapabilities = require('./topology_base').ServerCapabilities
- , Store = require('./topology_base').Store
- , Define = require('./metadata')
- , MongoError = require('mongodb-core').MongoError
- , 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', 'reconnectInterval', 'monitoring'
- , 'appname', 'domainsEnabled'
- , 'servername', 'promoteLongs', 'promoteValues', 'promoteBuffers'];
- var Server = function(host, port, options) {
- options = options || {};
- if(!(this instanceof Server)) return new Server(host, port, options);
- EventEmitter.call(this);
- var self = this;
-
- options = filterOptions(options, legalOptionNames);
-
- var storeOptions = {
- force: false
- , bufferMaxEntries: typeof options.bufferMaxEntries == 'number' ? options.bufferMaxEntries : MAX_JS_INT
- }
-
- var store = options.store || new Store(self, storeOptions);
-
- if(host.indexOf('\/') != -1) {
- if(port != null && typeof port == 'object') {
- options = port;
- port = null;
- }
- } else if(port == null) {
- throw MongoError.create({message: 'port must be specified', driver:true});
- }
-
- var reconnect = typeof options.auto_reconnect == 'boolean' ? options.auto_reconnect : true;
- reconnect = typeof options.autoReconnect == 'boolean' ? options.autoReconnect : reconnect;
-
- var clonedOptions = mergeOptions({}, {
- host: host, port: port, 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 server = new CServer(clonedOptions);
-
- this.s = {
-
- server: server
-
- , sCapabilities: null
-
- , clonedOptions: clonedOptions
-
- , reconnect: clonedOptions.reconnect
-
- , emitError: clonedOptions.emitError
-
- , poolSize: clonedOptions.size
-
- , storeOptions: storeOptions
-
- , store: store
-
- , host: host
-
- , port: port
-
- , options: options
- }
- }
- inherits(Server, EventEmitter);
- var define = Server.define = new Define('Server', Server, false);
- Object.defineProperty(Server.prototype, 'bson', {
- enumerable: true, get: function() {
- return this.s.server.s.bson;
- }
- });
- Object.defineProperty(Server.prototype, 'isMasterDoc', {
- enumerable:true, get: function() {
- return this.s.server.lastIsMaster();
- }
- });
- Object.defineProperty(Server.prototype, 'parserType', {
- enumerable:true, get: function() {
- return this.s.server.parserType;
- }
- });
- Object.defineProperty(Server.prototype, 'poolSize', {
- enumerable:true, get: function() { return this.s.server.connections().length; }
- });
- Object.defineProperty(Server.prototype, 'autoReconnect', {
- enumerable:true, get: function() { return this.s.reconnect; }
- });
- Object.defineProperty(Server.prototype, 'host', {
- enumerable:true, get: function() { return this.s.host; }
- });
- Object.defineProperty(Server.prototype, 'port', {
- enumerable:true, get: function() { return this.s.port; }
- });
- Server.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.s.server.removeListener(e, connectHandlers[e]);
- });
- self.s.server.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);
- self.s.store.execute();
- }
-
- var reconnectFailedHandler = function(err) {
- self.emit('reconnectFailed', err);
- self.s.store.flush(err);
- }
-
- var destroyHandler = function() {
- self.s.store.flush();
- }
-
- var connectHandler = function() {
-
- ["timeout", "error", "close", 'serverOpening', 'serverDescriptionChanged', 'serverHeartbeatStarted',
- 'serverHeartbeatSucceeded', 'serverHeartbeatFailed', 'serverClosed', 'topologyOpening',
- 'topologyClosed', 'topologyDescriptionChanged'].forEach(function(e) {
- self.s.server.removeAllListeners(e);
- });
-
- self.s.server.on('timeout', errorHandler('timeout'));
- self.s.server.once('error', errorHandler('error'));
- self.s.server.on('close', errorHandler('close'));
-
- self.s.server.on('destroy', destroyHandler);
-
- var relay = function(event) {
- return function(t, server) {
- self.emit(event, t, server);
- }
- }
-
- self.s.server.on('serverDescriptionChanged', relay('serverDescriptionChanged'));
- self.s.server.on('serverHeartbeatStarted', relay('serverHeartbeatStarted'));
- self.s.server.on('serverHeartbeatSucceeded', relay('serverHeartbeatSucceeded'));
- self.s.server.on('serverHeartbeatFailed', relay('serverHeartbeatFailed'));
- self.s.server.on('serverOpening', relay('serverOpening'));
- self.s.server.on('serverClosed', relay('serverClosed'));
- self.s.server.on('topologyOpening', relay('topologyOpening'));
- self.s.server.on('topologyClosed', relay('topologyClosed'));
- self.s.server.on('topologyDescriptionChanged', relay('topologyDescriptionChanged'));
- self.s.server.on('attemptReconnect', relay('attemptReconnect'));
- self.s.server.on('monitoring', relay('monitoring'));
-
- self.emit('open', null, self);
-
- try {
- callback(null, self);
- } catch(err) {
- console.log(err.stack)
- process.nextTick(function() { throw err; })
- }
- }
-
- var connectHandlers = {
- timeout: connectErrorHandler('timeout'),
- error: connectErrorHandler('error'),
- close: connectErrorHandler('close')
- };
-
- self.s.server.once('timeout', connectHandlers.timeout);
- self.s.server.once('error', connectHandlers.error);
- self.s.server.once('close', connectHandlers.close);
- self.s.server.once('connect', connectHandler);
-
- self.s.server.on('reconnect', reconnectHandler);
- self.s.server.on('reconnectFailed', reconnectFailedHandler);
-
- self.s.server.connect(_options);
- }
- Server.prototype.capabilities = function() {
- if(this.s.sCapabilities) return this.s.sCapabilities;
- if(this.s.server.lastIsMaster() == null) return null;
- this.s.sCapabilities = new ServerCapabilities(this.s.server.lastIsMaster());
- return this.s.sCapabilities;
- }
- define.classMethod('capabilities', {callback: false, promise:false, returns: [ServerCapabilities]});
- Server.prototype.command = function(ns, cmd, options, callback) {
- this.s.server.command(ns, cmd, options, callback);
- }
- define.classMethod('command', {callback: true, promise:false});
- Server.prototype.insert = function(ns, ops, options, callback) {
- this.s.server.insert(ns, ops, options, callback);
- }
- define.classMethod('insert', {callback: true, promise:false});
- Server.prototype.update = function(ns, ops, options, callback) {
- this.s.server.update(ns, ops, options, callback);
- }
- define.classMethod('update', {callback: true, promise:false});
- Server.prototype.remove = function(ns, ops, options, callback) {
- this.s.server.remove(ns, ops, options, callback);
- }
- define.classMethod('remove', {callback: true, promise:false});
- Server.prototype.isConnected = function() {
- return this.s.server.isConnected();
- }
- Server.prototype.isDestroyed = function() {
- return this.s.server.isDestroyed();
- }
- define.classMethod('isConnected', {callback: false, promise:false, returns: [Boolean]});
- Server.prototype.cursor = function(ns, cmd, options) {
- options.disconnectHandler = this.s.store;
- return this.s.server.cursor(ns, cmd, options);
- }
- define.classMethod('cursor', {callback: false, promise:false, returns: [Cursor, AggregationCursor, CommandCursor]});
- Server.prototype.lastIsMaster = function() {
- return this.s.server.lastIsMaster();
- }
- Server.prototype.unref = function() {
- this.s.server.unref();
- }
- Server.prototype.close = function(forceClosed) {
- this.s.server.destroy();
-
- if(forceClosed == true) {
- this.s.storeOptions.force = forceClosed;
- this.s.store.flush();
- }
- }
- define.classMethod('close', {callback: false, promise:false});
- Server.prototype.auth = function() {
- var args = Array.prototype.slice.call(arguments, 0);
- this.s.server.auth.apply(this.s.server, args);
- }
- define.classMethod('auth', {callback: true, promise:false});
- Server.prototype.logout = function() {
- var args = Array.prototype.slice.call(arguments, 0);
- this.s.server.logout.apply(this.s.server, args);
- }
- define.classMethod('logout', {callback: true, promise:false});
- Server.prototype.connections = function() {
- return this.s.server.connections();
- }
- define.classMethod('connections', {callback: false, promise:false, returns:[Array]});
- module.exports = Server;
|