123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354 |
- "use strict";
- var checkCollectionName = require('./utils').checkCollectionName
- , ObjectID = require('mongodb-core').BSON.ObjectID
- , Long = require('mongodb-core').BSON.Long
- , Code = require('mongodb-core').BSON.Code
- , f = require('util').format
- , AggregationCursor = require('./aggregation_cursor')
- , MongoError = require('mongodb-core').MongoError
- , shallowClone = require('./utils').shallowClone
- , isObject = require('./utils').isObject
- , toError = require('./utils').toError
- , normalizeHintField = require('./utils').normalizeHintField
- , handleCallback = require('./utils').handleCallback
- , decorateCommand = require('./utils').decorateCommand
- , formattedOrderClause = require('./utils').formattedOrderClause
- , ReadPreference = require('./read_preference')
- , CoreReadPreference = require('mongodb-core').ReadPreference
- , CommandCursor = require('./command_cursor')
- , Define = require('./metadata')
- , Cursor = require('./cursor')
- , unordered = require('./bulk/unordered')
- , ordered = require('./bulk/ordered')
- , assign = require('./utils').assign;
- var Collection = function(db, topology, dbName, name, pkFactory, options) {
- checkCollectionName(name);
-
- var internalHint = null;
- var slaveOk = options == null || options.slaveOk == null ? db.slaveOk : options.slaveOk;
- var serializeFunctions = options == null || options.serializeFunctions == null ? db.s.options.serializeFunctions : options.serializeFunctions;
- var raw = options == null || options.raw == null ? db.s.options.raw : options.raw;
- var promoteLongs = options == null || options.promoteLongs == null ? db.s.options.promoteLongs : options.promoteLongs;
- var promoteValues = options == null || options.promoteValues == null ? db.s.options.promoteValues : options.promoteValues;
- var promoteBuffers = options == null || options.promoteBuffers == null ? db.s.options.promoteBuffers : options.promoteBuffers;
- var readPreference = null;
- var collectionHint = null;
- var namespace = f("%s.%s", dbName, name);
-
- var promiseLibrary = options.promiseLibrary;
-
- if(!promiseLibrary) {
- promiseLibrary = typeof global.Promise == 'function' ?
- global.Promise : require('es6-promise').Promise;
- }
-
- if(options && options.readPreference) {
- readPreference = options.readPreference;
- } else if(db.options.readPreference) {
- readPreference = db.options.readPreference;
- }
-
- pkFactory = pkFactory == null
- ? ObjectID
- : pkFactory;
-
- this.s = {
-
- pkFactory: pkFactory
-
- , db: db
-
- , topology: topology
-
- , dbName: dbName
-
- , options: options
-
- , namespace: namespace
-
- , readPreference: readPreference
-
- , slaveOk: slaveOk
-
- , serializeFunctions: serializeFunctions
-
- , raw: raw
-
- , promoteLongs: promoteLongs
-
- , promoteValues: promoteValues
-
- , promoteBuffers: promoteBuffers
-
- , internalHint: internalHint
-
- , collectionHint: collectionHint
-
- , name: name
-
- , promiseLibrary: promiseLibrary
-
- , readConcern: options.readConcern
- }
- }
- var define = Collection.define = new Define('Collection', Collection, false);
- Object.defineProperty(Collection.prototype, 'collectionName', {
- enumerable: true, get: function() { return this.s.name; }
- });
- Object.defineProperty(Collection.prototype, 'namespace', {
- enumerable: true, get: function() { return this.s.namespace; }
- });
- Object.defineProperty(Collection.prototype, 'readConcern', {
- enumerable: true, get: function() { return this.s.readConcern || {level: 'local'}; }
- });
- Object.defineProperty(Collection.prototype, 'writeConcern', {
- enumerable:true,
- get: function() {
- var ops = {};
- if(this.s.options.w != null) ops.w = this.s.options.w;
- if(this.s.options.j != null) ops.j = this.s.options.j;
- if(this.s.options.fsync != null) ops.fsync = this.s.options.fsync;
- if(this.s.options.wtimeout != null) ops.wtimeout = this.s.options.wtimeout;
- return ops;
- }
- });
- Object.defineProperty(Collection.prototype, "hint", {
- enumerable: true
- , get: function () { return this.s.collectionHint; }
- , set: function (v) { this.s.collectionHint = normalizeHintField(v); }
- });
- Collection.prototype.find = function() {
- var options
- , args = Array.prototype.slice.call(arguments, 0)
- , has_callback = typeof args[args.length - 1] === 'function'
- , has_weird_callback = typeof args[0] === 'function'
- , callback = has_callback ? args.pop() : (has_weird_callback ? args.shift() : null)
- , len = args.length
- , selector = len >= 1 ? args[0] : {}
- , fields = len >= 2 ? args[1] : undefined;
- if(len === 1 && has_weird_callback) {
-
- selector = {};
- options = args[0];
- }
- if(len === 2 && fields !== undefined && !Array.isArray(fields)) {
- var fieldKeys = Object.keys(fields);
- var is_option = false;
- for(var i = 0; i < fieldKeys.length; i++) {
- if(testForFields[fieldKeys[i]] != null) {
- is_option = true;
- break;
- }
- }
- if(is_option) {
- options = fields;
- fields = undefined;
- } else {
- options = {};
- }
- } else if(len === 2 && Array.isArray(fields) && !Array.isArray(fields[0])) {
- var newFields = {};
-
- for(i = 0; i < fields.length; i++) {
- newFields[fields[i]] = 1;
- }
-
- fields = newFields;
- }
- if(3 === len) {
- options = args[2];
- }
-
- selector = selector == null ? {} : selector;
-
- var object = selector;
- if(Buffer.isBuffer(object)) {
- var object_size = object[0] | object[1] << 8 | object[2] << 16 | object[3] << 24;
- if(object_size != object.length) {
- var error = new Error("query selector raw message size does not match message header size [" + object.length + "] != [" + object_size + "]");
- error.name = 'MongoError';
- throw error;
- }
- }
-
- object = fields;
- if(Buffer.isBuffer(object)) {
- object_size = object[0] | object[1] << 8 | object[2] << 16 | object[3] << 24;
- if(object_size != object.length) {
- error = new Error("query fields raw message size does not match message header size [" + object.length + "] != [" + object_size + "]");
- error.name = 'MongoError';
- throw error;
- }
- }
-
- if(selector != null && selector._bsontype == 'ObjectID') {
- selector = {_id:selector};
- }
-
-
- if(options && options.fields && !(Buffer.isBuffer(options.fields))) {
- fields = {};
- if(Array.isArray(options.fields)) {
- if(!options.fields.length) {
- fields['_id'] = 1;
- } else {
- var l = options.fields.length;
- for (i = 0; i < l; i++) {
- fields[options.fields[i]] = 1;
- }
- }
- } else {
- fields = options.fields;
- }
- }
- if (!options) options = {};
- var newOptions = {};
-
- for (var key in options) {
- newOptions[key] = options[key];
- }
-
- newOptions.skip = len > 3 ? args[2] : options.skip ? options.skip : 0;
- newOptions.limit = len > 3 ? args[3] : options.limit ? options.limit : 0;
- newOptions.raw = options.raw != null && typeof options.raw === 'boolean' ? options.raw : this.s.raw;
- newOptions.hint = options.hint != null ? normalizeHintField(options.hint) : this.s.collectionHint;
- newOptions.timeout = len == 5 ? args[4] : typeof options.timeout === 'undefined' ? undefined : options.timeout;
-
- newOptions.slaveOk = options.slaveOk != null ? options.slaveOk : this.s.db.slaveOk;
-
- newOptions = getReadPreference(this, newOptions, this.s.db, this);
-
- if(newOptions.readPreference != null
- && (newOptions.readPreference != 'primary' || newOptions.readPreference.mode != 'primary')) {
- newOptions.slaveOk = true;
- }
-
- if(selector != null && typeof selector != 'object') {
- throw MongoError.create({message: "query selector must be an object", driver:true });
- }
-
- var findCommand = {
- find: this.s.namespace
- , limit: newOptions.limit
- , skip: newOptions.skip
- , query: selector
- }
-
- if(typeof newOptions.awaitdata == 'boolean') {
- newOptions.awaitData = newOptions.awaitdata
- }
-
- if(typeof newOptions.timeout == 'boolean') newOptions.noCursorTimeout = newOptions.timeout;
-
- for(var name in newOptions) {
- if(newOptions[name] != null) findCommand[name] = newOptions[name];
- }
-
- var formatFields = function(fields) {
- var object = {};
- if(Array.isArray(fields)) {
- for(var i = 0; i < fields.length; i++) {
- if(Array.isArray(fields[i])) {
- object[fields[i][0]] = fields[i][1];
- } else {
- object[fields[i][0]] = 1;
- }
- }
- } else {
- object = fields;
- }
- return object;
- }
-
- if(fields) findCommand.fields = formatFields(fields);
-
- newOptions.db = this.s.db;
-
- newOptions.promiseLibrary = this.s.promiseLibrary;
-
- if(newOptions.raw == null && typeof this.s.raw == 'boolean') newOptions.raw = this.s.raw;
-
- if(newOptions.promoteLongs == null && typeof this.s.promoteLongs == 'boolean') newOptions.promoteLongs = this.s.promoteLongs;
- if(newOptions.promoteValues == null && typeof this.s.promoteValues == 'boolean') newOptions.promoteValues = this.s.promoteValues;
- if(newOptions.promoteBuffers == null && typeof this.s.promoteBuffers == 'boolean') newOptions.promoteBuffers = this.s.promoteBuffers;
-
- if(findCommand.sort) {
- findCommand.sort = formattedOrderClause(findCommand.sort);
- }
-
- if(this.s.readConcern) {
- findCommand.readConcern = this.s.readConcern;
- }
-
- decorateWithCollation(findCommand, this, options);
-
- if(typeof callback == 'function') return handleCallback(callback, null, this.s.topology.cursor(this.s.namespace, findCommand, newOptions));
- return this.s.topology.cursor(this.s.namespace, findCommand, newOptions);
- }
- define.classMethod('find', {callback: false, promise:false, returns: [Cursor]});
- Collection.prototype.insertOne = function(doc, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
- if(Array.isArray(doc) && typeof callback == 'function') {
- return callback(MongoError.create({message: 'doc parameter must be an object', driver:true }));
- } else if(Array.isArray(doc)) {
- return new this.s.promiseLibrary(function(resolve, reject) {
- reject(MongoError.create({message: 'doc parameter must be an object', driver:true }));
- });
- }
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return insertOne(self, doc, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- insertOne(self, doc, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var insertOne = function(self, doc, options, callback) {
- insertDocuments(self, [doc], options, function(err, r) {
- if(callback == null) return;
- if(err && callback) return callback(err);
-
- if(r == null) return callback(null, {result: {ok:1}});
-
- r.insertedCount = r.result.n;
- r.insertedId = doc._id;
- if(callback) callback(null, r);
- });
- }
- var mapInserManyResults = function(docs, r) {
- var ids = r.getInsertedIds();
- var keys = Object.keys(ids);
- var finalIds = new Array(keys.length);
- for(var i = 0; i < keys.length; i++) {
- if(ids[keys[i]]._id) {
- finalIds[ids[keys[i]].index] = ids[keys[i]]._id;
- }
- }
- var finalResult = {
- result: {ok: 1, n: r.insertedCount},
- ops: docs,
- insertedCount: r.insertedCount,
- insertedIds: finalIds
- };
- if(r.getLastOp()) {
- finalResult.result.opTime = r.getLastOp();
- }
- return finalResult;
- }
- define.classMethod('insertOne', {callback: true, promise:true});
- Collection.prototype.insertMany = function(docs, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {ordered:true};
- if(!Array.isArray(docs) && typeof callback == 'function') {
- return callback(MongoError.create({message: 'docs parameter must be an array of documents', driver:true }));
- } else if(!Array.isArray(docs)) {
- return new this.s.promiseLibrary(function(resolve, reject) {
- reject(MongoError.create({message: 'docs parameter must be an array of documents', driver:true }));
- });
- }
-
- if(typeof options.checkKeys != 'boolean') {
- options.checkKeys = true;
- }
-
- options['serializeFunctions'] = options['serializeFunctions'] || self.s.serializeFunctions;
-
- var forceServerObjectId = typeof options.forceServerObjectId == 'boolean'
- ? options.forceServerObjectId : self.s.db.options.forceServerObjectId;
-
- if(forceServerObjectId !== true) {
-
- for(var i = 0; i < docs.length; i++) {
- if(docs[i]._id == null) docs[i]._id = self.s.pkFactory.createPk();
- }
- }
-
- var operations = [{
- insertMany: docs
- }];
-
- if(typeof callback == 'function') return bulkWrite(self, operations, options, function(err, r) {
- if(err) return callback(err, r);
- callback(null, mapInserManyResults(docs, r));
- });
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- bulkWrite(self, operations, options, function(err, r) {
- if(err) return reject(err);
- resolve(mapInserManyResults(docs, r));
- });
- });
- }
- define.classMethod('insertMany', {callback: true, promise:true});
- Collection.prototype.bulkWrite = function(operations, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {ordered:true};
- if(!Array.isArray(operations)) {
- throw MongoError.create({message: "operations must be an array of documents", driver:true });
- }
-
- if(typeof callback == 'function') return bulkWrite(self, operations, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- bulkWrite(self, operations, options, function(err, r) {
- if(err && r == null) return reject(err);
- resolve(r);
- });
- });
- }
- var bulkWrite = function(self, operations, options, callback) {
-
- if(self.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = self.s.options.ignoreUndefined;
- }
-
- var bulk = options.ordered == true || options.ordered == null ? self.initializeOrderedBulkOp(options) : self.initializeUnorderedBulkOp(options);
-
- var collation = false;
-
- try {
- for(var i = 0; i < operations.length; i++) {
-
- var key = Object.keys(operations[i])[0];
-
- if(operations[i][key].collation) {
- collation = true;
- }
-
- bulk.raw(operations[i]);
- }
- } catch(err) {
- return callback(err, null);
- }
-
- var finalOptions = writeConcern(shallowClone(options), self.s.db, self, options);
- var writeCon = finalOptions.writeConcern ? finalOptions.writeConcern : {};
- var capabilities = self.s.topology.capabilities();
-
- if(collation && capabilities && !capabilities.commandsTakeCollation) {
- return callback(new MongoError(f('server/primary/mongos does not support collation')));
- }
-
- bulk.execute(writeCon, function(err, r) {
-
- if(!r && err) return callback(err, null);
-
- if(r && r.hasWriteErrors() && r.getWriteErrorCount() == 1) {
- return callback(toError(r.getWriteErrorAt(0)), r);
- }
- r.insertedCount = r.nInserted;
- r.matchedCount = r.nMatched;
- r.modifiedCount = r.nModified || 0;
- r.deletedCount = r.nRemoved;
- r.upsertedCount = r.getUpsertedIds().length;
- r.upsertedIds = {};
- r.insertedIds = {};
-
- r.n = r.insertedCount;
-
- var inserted = r.getInsertedIds();
-
- for(var i = 0; i < inserted.length; i++) {
- r.insertedIds[inserted[i].index] = inserted[i]._id;
- }
-
- var upserted = r.getUpsertedIds();
-
- for(i = 0; i < upserted.length; i++) {
- r.upsertedIds[upserted[i].index] = upserted[i]._id;
- }
-
- if(r.hasWriteErrors()) {
-
- var errors = r.getWriteErrors();
-
- return callback(toError({
- message: 'write operation failed', code: errors[0].code, writeErrors: errors
- }), r);
- }
-
- if(r.getWriteConcernError()) {
-
- return callback(toError(r.getWriteConcernError()), r);
- }
-
- callback(null, r);
- });
- }
- var insertDocuments = function(self, docs, options, callback) {
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- docs = Array.isArray(docs) ? docs : [docs];
-
- var finalOptions = writeConcern(shallowClone(options), self.s.db, self, options);
- if(typeof finalOptions.checkKeys != 'boolean') finalOptions.checkKeys = true;
-
- if(finalOptions.keepGoing == true) finalOptions.ordered = false;
- finalOptions['serializeFunctions'] = options['serializeFunctions'] || self.s.serializeFunctions;
-
- var forceServerObjectId = typeof options.forceServerObjectId == 'boolean'
- ? options.forceServerObjectId : self.s.db.options.forceServerObjectId;
-
- if(forceServerObjectId !== true){
- for(var i = 0; i < docs.length; i++) {
- if(docs[i]._id == null) docs[i]._id = self.s.pkFactory.createPk();
- }
- }
-
- self.s.topology.insert(self.s.namespace, docs, finalOptions, function(err, result) {
- if(callback == null) return;
- if(err) return handleCallback(callback, err);
- if(result == null) return handleCallback(callback, null, null);
- if(result.result.code) return handleCallback(callback, toError(result.result));
- if(result.result.writeErrors) return handleCallback(callback, toError(result.result.writeErrors[0]));
-
- result.ops = docs;
-
- handleCallback(callback, null, result);
- });
- }
- define.classMethod('bulkWrite', {callback: true, promise:true});
- Collection.prototype.insert = function(docs, options, callback) {
- if(typeof options == 'function') callback = options, options = {};
- options = options || {ordered:false};
- docs = !Array.isArray(docs) ? [docs] : docs;
- if(options.keepGoing == true) {
- options.ordered = false;
- }
- return this.insertMany(docs, options, callback);
- }
- define.classMethod('insert', {callback: true, promise:true});
- Collection.prototype.updateOne = function(filter, update, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = shallowClone(options)
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return updateOne(self, filter, update, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- updateOne(self, filter, update, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var updateOne = function(self, filter, update, options, callback) {
-
- options.multi = false;
-
- updateDocuments(self, filter, update, options, function(err, r) {
- if(callback == null) return;
- if(err && callback) return callback(err);
- if(r == null) return callback(null, {result: {ok:1}});
- r.modifiedCount = r.result.nModified != null ? r.result.nModified : r.result.n;
- r.upsertedId = Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? r.result.upserted[0] : null;
- r.upsertedCount = Array.isArray(r.result.upserted) && r.result.upserted.length ? r.result.upserted.length : 0;
- r.matchedCount = Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? 0 : r.result.n;
- if(callback) callback(null, r);
- });
- }
- define.classMethod('updateOne', {callback: true, promise:true});
- Collection.prototype.replaceOne = function(filter, doc, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = shallowClone(options)
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return replaceOne(self, filter, doc, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- replaceOne(self, filter, doc, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var replaceOne = function(self, filter, doc, options, callback) {
-
- options.multi = false;
-
- updateDocuments(self, filter, doc, options, function(err, r) {
- if(callback == null) return;
- if(err && callback) return callback(err);
- if(r == null) return callback(null, {result: {ok:1}});
- r.modifiedCount = r.result.nModified != null ? r.result.nModified : r.result.n;
- r.upsertedId = Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? r.result.upserted[0] : null;
- r.upsertedCount = Array.isArray(r.result.upserted) && r.result.upserted.length ? r.result.upserted.length : 0;
- r.matchedCount = Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? 0 : r.result.n;
- r.ops = [doc];
- if(callback) callback(null, r);
- });
- }
- define.classMethod('replaceOne', {callback: true, promise:true});
- Collection.prototype.updateMany = function(filter, update, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = shallowClone(options)
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return updateMany(self, filter, update, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- updateMany(self, filter, update, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var updateMany = function(self, filter, update, options, callback) {
-
- options.multi = true;
-
- updateDocuments(self, filter, update, options, function(err, r) {
- if(callback == null) return;
- if(err && callback) return callback(err);
- if(r == null) return callback(null, {result: {ok:1}});
- r.modifiedCount = r.result.nModified != null ? r.result.nModified : r.result.n;
- r.upsertedId = Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? r.result.upserted[0] : null;
- r.upsertedCount = Array.isArray(r.result.upserted) && r.result.upserted.length ? r.result.upserted.length : 0;
- r.matchedCount = Array.isArray(r.result.upserted) && r.result.upserted.length > 0 ? 0 : r.result.n;
- if(callback) callback(null, r);
- });
- }
- define.classMethod('updateMany', {callback: true, promise:true});
- var updateDocuments = function(self, selector, document, options, callback) {
- if('function' === typeof options) callback = options, options = null;
- if(options == null) options = {};
- if(!('function' === typeof callback)) callback = null;
-
- if(selector == null || typeof selector != 'object') return callback(toError("selector must be a valid JavaScript object"));
- if(document == null || typeof document != 'object') return callback(toError("document must be a valid JavaScript object"));
-
- var finalOptions = writeConcern(shallowClone(options), self.s.db, self, options);
-
-
-
- finalOptions['serializeFunctions'] = options['serializeFunctions'] || self.s.serializeFunctions;
-
- var op = {q: selector, u: document};
- op.upsert = typeof options.upsert == 'boolean' ? options.upsert : false;
- op.multi = typeof options.multi == 'boolean' ? options.multi : false;
-
- decorateWithCollation(finalOptions, self, options);
-
- self.s.topology.update(self.s.namespace, [op], finalOptions, function(err, result) {
- if(callback == null) return;
- if(err) return handleCallback(callback, err, null);
- if(result == null) return handleCallback(callback, null, null);
- if(result.result.code) return handleCallback(callback, toError(result.result));
- if(result.result.writeErrors) return handleCallback(callback, toError(result.result.writeErrors[0]));
-
- handleCallback(callback, null, result);
- });
- }
- Collection.prototype.update = function(selector, document, options, callback) {
- var self = this;
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return updateDocuments(self, selector, document, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- updateDocuments(self, selector, document, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- define.classMethod('update', {callback: true, promise:true});
- Collection.prototype.deleteOne = function(filter, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = shallowClone(options);
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return deleteOne(self, filter, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- deleteOne(self, filter, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var deleteOne = function(self, filter, options, callback) {
- options.single = true;
- removeDocuments(self, filter, options, function(err, r) {
- if(callback == null) return;
- if(err && callback) return callback(err);
- if(r == null) return callback(null, {result: {ok:1}});
- r.deletedCount = r.result.n;
- if(callback) callback(null, r);
- });
- }
- define.classMethod('deleteOne', {callback: true, promise:true});
- Collection.prototype.removeOne = Collection.prototype.deleteOne;
- define.classMethod('removeOne', {callback: true, promise:true});
- Collection.prototype.deleteMany = function(filter, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = shallowClone(options);
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return deleteMany(self, filter, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- deleteMany(self, filter, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var deleteMany = function(self, filter, options, callback) {
- options.single = false;
- removeDocuments(self, filter, options, function(err, r) {
- if(callback == null) return;
- if(err && callback) return callback(err);
- if(r == null) return callback(null, {result: {ok:1}});
- r.deletedCount = r.result.n;
- if(callback) callback(null, r);
- });
- }
- var removeDocuments = function(self, selector, options, callback) {
- if(typeof options == 'function') {
- callback = options, options = {};
- } else if (typeof selector === 'function') {
- callback = selector;
- options = {};
- selector = {};
- }
-
- options = options || {};
-
- var finalOptions = writeConcern(shallowClone(options), self.s.db, self, options);
-
- if(selector == null) selector = {};
-
- var op = {q: selector, limit: 0};
- if(options.single) op.limit = 1;
-
- decorateWithCollation(finalOptions, self, options);
-
- self.s.topology.remove(self.s.namespace, [op], finalOptions, function(err, result) {
- if(callback == null) return;
- if(err) return handleCallback(callback, err, null);
- if(result == null) return handleCallback(callback, null, null);
- if(result.result.code) return handleCallback(callback, toError(result.result));
- if(result.result.writeErrors) return handleCallback(callback, toError(result.result.writeErrors[0]));
-
- handleCallback(callback, null, result);
- });
- }
- define.classMethod('deleteMany', {callback: true, promise:true});
- Collection.prototype.removeMany = Collection.prototype.deleteMany;
- define.classMethod('removeMany', {callback: true, promise:true});
- Collection.prototype.remove = function(selector, options, callback) {
- var self = this;
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return removeDocuments(self, selector, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- removeDocuments(self, selector, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- define.classMethod('remove', {callback: true, promise:true});
- Collection.prototype.save = function(doc, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(this.s.options.ignoreUndefined) {
- options = shallowClone(options);
- options.ignoreUndefined = this.s.options.ignoreUndefined;
- }
-
- if(typeof callback == 'function') return save(self, doc, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- save(self, doc, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var save = function(self, doc, options, callback) {
-
- var finalOptions = writeConcern(shallowClone(options), self.s.db, self, options);
-
- if(doc._id != null) {
- finalOptions.upsert = true;
- return updateDocuments(self, {_id: doc._id}, doc, finalOptions, callback);
- }
-
- insertDocuments(self, [doc], options, function(err, r) {
- if(callback == null) return;
- if(doc == null) return handleCallback(callback, null, null);
- if(err) return handleCallback(callback, err, null);
- handleCallback(callback, null, r);
- });
- }
- define.classMethod('save', {callback: true, promise:true});
- Collection.prototype.findOne = function() {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 0);
- var callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
-
- if(typeof callback == 'function') return findOne(self, args, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- findOne(self, args, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var findOne = function(self, args, callback) {
- var cursor = self.find.apply(self, args).limit(-1).batchSize(1);
-
- cursor.next(function(err, item) {
- if(err != null) return handleCallback(callback, toError(err), null);
- handleCallback(callback, null, item);
- });
- }
- define.classMethod('findOne', {callback: true, promise:true});
- Collection.prototype.rename = function(newName, opt, callback) {
- var self = this;
- if(typeof opt == 'function') callback = opt, opt = {};
- opt = assign({}, opt, {readPreference: ReadPreference.PRIMARY});
-
- if(typeof callback == 'function') return rename(self, newName, opt, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- rename(self, newName, opt, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var rename = function(self, newName, opt, callback) {
-
- checkCollectionName(newName);
-
- var renameCollection = f("%s.%s", self.s.dbName, self.s.name);
- var toCollection = f("%s.%s", self.s.dbName, newName);
- var dropTarget = typeof opt.dropTarget == 'boolean' ? opt.dropTarget : false;
- var cmd = {'renameCollection':renameCollection, 'to':toCollection, 'dropTarget':dropTarget};
-
- decorateWithWriteConcern(cmd, self, opt);
-
- self.s.db.admin().command(cmd, opt, function(err, doc) {
- if(err) return handleCallback(callback, err, null);
-
- if(doc.errmsg) return handleCallback(callback, toError(doc), null);
- try {
- return handleCallback(callback, null, new Collection(self.s.db, self.s.topology, self.s.dbName, newName, self.s.pkFactory, self.s.options));
- } catch(err) {
- return handleCallback(callback, toError(err), null);
- }
- });
- }
- define.classMethod('rename', {callback: true, promise:true});
- Collection.prototype.drop = function(options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(typeof callback == 'function') return self.s.db.dropCollection(self.s.name, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- self.s.db.dropCollection(self.s.name, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- define.classMethod('drop', {callback: true, promise:true});
- Collection.prototype.options = function(callback) {
- var self = this;
-
- if(typeof callback == 'function') return options(self, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- options(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var options = function(self, callback) {
- self.s.db.listCollections({name: self.s.name}).toArray(function(err, collections) {
- if(err) return handleCallback(callback, err);
- if(collections.length == 0) {
- return handleCallback(callback, MongoError.create({message: f("collection %s not found", self.s.namespace), driver:true }));
- }
- handleCallback(callback, err, collections[0].options || null);
- });
- }
- define.classMethod('options', {callback: true, promise:true});
- Collection.prototype.isCapped = function(callback) {
- var self = this;
-
- if(typeof callback == 'function') return isCapped(self, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- isCapped(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var isCapped = function(self, callback) {
- self.options(function(err, document) {
- if(err) return handleCallback(callback, err);
- handleCallback(callback, null, document && document.capped);
- });
- }
- define.classMethod('isCapped', {callback: true, promise:true});
- Collection.prototype.createIndex = function(fieldOrSpec, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 1);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- options = args.length ? args.shift() || {} : {};
- options = typeof callback === 'function' ? options : callback;
- options = options == null ? {} : options;
-
- if(typeof callback == 'function') return createIndex(self, fieldOrSpec, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- createIndex(self, fieldOrSpec, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var createIndex = function(self, fieldOrSpec, options, callback) {
- self.s.db.createIndex(self.s.name, fieldOrSpec, options, callback);
- }
- define.classMethod('createIndex', {callback: true, promise:true});
- Collection.prototype.createIndexes = function(indexSpecs, callback) {
- var self = this;
-
- if(typeof callback == 'function') return createIndexes(self, indexSpecs, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- createIndexes(self, indexSpecs, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var createIndexes = function(self, indexSpecs, callback) {
- var capabilities = self.s.topology.capabilities();
-
- for(var i = 0; i < indexSpecs.length; i++) {
- if(indexSpecs[i].name == null) {
- var keys = [];
-
- if(indexSpecs[i].collation && capabilities && !capabilities.commandsTakeCollation) {
- return callback(new MongoError(f('server/primary/mongos does not support collation')));
- }
- for(var name in indexSpecs[i].key) {
- keys.push(f('%s_%s', name, indexSpecs[i].key[name]));
- }
-
- indexSpecs[i].name = keys.join('_');
- }
- }
-
- self.s.db.command({
- createIndexes: self.s.name, indexes: indexSpecs
- }, { readPreference: ReadPreference.PRIMARY }, callback);
- }
- define.classMethod('createIndexes', {callback: true, promise:true});
- Collection.prototype.dropIndex = function(indexName, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 1);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- options = args.length ? args.shift() || {} : {};
-
- options.readPreference = ReadPreference.PRIMARY;
-
- if(typeof callback == 'function') return dropIndex(self, indexName, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- dropIndex(self, indexName, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var dropIndex = function(self, indexName, options, callback) {
-
- var cmd = {'dropIndexes':self.s.name, 'index':indexName};
-
- decorateWithWriteConcern(cmd, self, options);
-
- self.s.db.command(cmd, options, function(err, result) {
- if(typeof callback != 'function') return;
- if(err) return handleCallback(callback, err, null);
- handleCallback(callback, null, result);
- });
- }
- define.classMethod('dropIndex', {callback: true, promise:true});
- Collection.prototype.dropIndexes = function(options, callback) {
- var self = this;
-
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(typeof callback == 'function') return dropIndexes(self, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- dropIndexes(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var dropIndexes = function(self, options, callback) {
- self.dropIndex('*', options, function(err) {
- if(err) return handleCallback(callback, err, false);
- handleCallback(callback, null, true);
- });
- }
- define.classMethod('dropIndexes', {callback: true, promise:true});
- Collection.prototype.dropAllIndexes = Collection.prototype.dropIndexes;
- define.classMethod('dropAllIndexes', {callback: true, promise:true});
- Collection.prototype.reIndex = function(options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(typeof callback == 'function') return reIndex(self, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- reIndex(self, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var reIndex = function(self, options, callback) {
-
- var cmd = {'reIndex':self.s.name};
-
- decorateWithWriteConcern(cmd, self, options);
-
- self.s.db.command(cmd, options, function(err, result) {
- if(callback == null) return;
- if(err) return handleCallback(callback, err, null);
- handleCallback(callback, null, result.ok ? true : false);
- });
- }
- define.classMethod('reIndex', {callback: true, promise:true});
- Collection.prototype.listIndexes = function(options) {
- options = options || {};
-
- options = shallowClone(options);
-
- options = getReadPreference(this, options, this.s.db, this);
-
- options.cursorFactory = CommandCursor;
-
- options.promiseLibrary = this.s.promiseLibrary;
- if(!this.s.topology.capabilities()) {
- throw new MongoError('cannot connect to server');
- }
-
- if(this.s.topology.capabilities().hasListIndexesCommand) {
-
- var cursor = options.batchSize ? {batchSize: options.batchSize} : {}
-
- var command = { listIndexes: this.s.name, cursor: cursor };
-
- cursor = this.s.topology.cursor(f('%s.$cmd', this.s.dbName), command, options);
-
- if(options.readPreference) cursor.setReadPreference(options.readPreference);
-
- return cursor;
- }
-
- var ns = f('%s.system.indexes', this.s.dbName);
-
- cursor = this.s.topology.cursor(ns, {find: ns, query: {ns: this.s.namespace}}, options);
-
- if(options.readPreference) cursor.setReadPreference(options.readPreference);
-
- if(options.batchSize) cursor = cursor.batchSize(options.batchSize);
-
- return cursor;
- };
- define.classMethod('listIndexes', {callback: false, promise:false, returns: [CommandCursor]});
- Collection.prototype.ensureIndex = function(fieldOrSpec, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(typeof callback == 'function') return ensureIndex(self, fieldOrSpec, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- ensureIndex(self, fieldOrSpec, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var ensureIndex = function(self, fieldOrSpec, options, callback) {
- self.s.db.ensureIndex(self.s.name, fieldOrSpec, options, callback);
- }
- define.classMethod('ensureIndex', {callback: true, promise:true});
- Collection.prototype.indexExists = function(indexes, callback) {
- var self = this;
-
- if(typeof callback == 'function') return indexExists(self, indexes, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- indexExists(self, indexes, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var indexExists = function(self, indexes, callback) {
- self.indexInformation(function(err, indexInformation) {
-
- if(err != null) return handleCallback(callback, err, null);
-
- if(!Array.isArray(indexes)) return handleCallback(callback, null, indexInformation[indexes] != null);
-
- for(var i = 0; i < indexes.length; i++) {
- if(indexInformation[indexes[i]] == null) {
- return handleCallback(callback, null, false);
- }
- }
-
- return handleCallback(callback, null, true);
- });
- }
- define.classMethod('indexExists', {callback: true, promise:true});
- Collection.prototype.indexInformation = function(options, callback) {
- var self = this;
-
- var args = Array.prototype.slice.call(arguments, 0);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- options = args.length ? args.shift() || {} : {};
-
- if(typeof callback == 'function') return indexInformation(self, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- indexInformation(self, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var indexInformation = function(self, options, callback) {
- self.s.db.indexInformation(self.s.name, options, callback);
- }
- define.classMethod('indexInformation', {callback: true, promise:true});
- Collection.prototype.count = function(query, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 0);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- var queryOption = args.length ? args.shift() || {} : {};
- var optionsOption = args.length ? args.shift() || {} : {};
-
- if(typeof callback == 'function') return count(self, queryOption, optionsOption, callback);
-
- query = query || {};
- options = options || {};
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- count(self, query, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- };
- var count = function(self, query, options, callback) {
- var skip = options.skip;
- var limit = options.limit;
- var hint = options.hint;
- var maxTimeMS = options.maxTimeMS;
-
- var cmd = {
- 'count': self.s.name, 'query': query
- };
-
- if(typeof skip == 'number') cmd.skip = skip;
- if(typeof limit == 'number') cmd.limit = limit;
- if(typeof maxTimeMS == 'number') cmd.maxTimeMS = maxTimeMS;
- if(hint) options.hint = hint;
- options = shallowClone(options);
-
- options = getReadPreference(self, options, self.s.db, self);
-
- if(self.s.readConcern) {
- cmd.readConcern = self.s.readConcern;
- }
-
- decorateWithCollation(cmd, self, options);
-
- self.s.db.command(cmd, options, function(err, result) {
- if(err) return handleCallback(callback, err);
- handleCallback(callback, null, result.n);
- });
- }
- define.classMethod('count', {callback: true, promise:true});
- Collection.prototype.distinct = function(key, query, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 1);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- var queryOption = args.length ? args.shift() || {} : {};
- var optionsOption = args.length ? args.shift() || {} : {};
-
- if(typeof callback == 'function') return distinct(self, key, queryOption, optionsOption, callback);
-
- query = query || {};
- options = options || {};
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- distinct(self, key, query, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- };
- var distinct = function(self, key, query, options, callback) {
-
- var maxTimeMS = options.maxTimeMS;
-
- var cmd = {
- 'distinct': self.s.name, 'key': key, 'query': query
- };
- options = shallowClone(options);
-
- options = getReadPreference(self, options, self.s.db, self);
-
- if(typeof maxTimeMS == 'number')
- cmd.maxTimeMS = maxTimeMS;
-
- if(self.s.readConcern) {
- cmd.readConcern = self.s.readConcern;
- }
-
- decorateWithCollation(cmd, self, options);
-
- self.s.db.command(cmd, options, function(err, result) {
- if(err) return handleCallback(callback, err);
- handleCallback(callback, null, result.values);
- });
- }
- define.classMethod('distinct', {callback: true, promise:true});
- Collection.prototype.indexes = function(callback) {
- var self = this;
-
- if(typeof callback == 'function') return indexes(self, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- indexes(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var indexes = function(self, callback) {
- self.s.db.indexInformation(self.s.name, {full:true}, callback);
- }
- define.classMethod('indexes', {callback: true, promise:true});
- Collection.prototype.stats = function(options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 0);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
-
- options = args.length ? args.shift() || {} : {};
-
- if(typeof callback == 'function') return stats(self, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- stats(self, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var stats = function(self, options, callback) {
-
- var commandObject = {
- collStats:self.s.name
- }
-
- if(options['scale'] != null) commandObject['scale'] = options['scale'];
- options = shallowClone(options);
-
- options = getReadPreference(self, options, self.s.db, self);
-
- self.s.db.command(commandObject, options, callback);
- }
- define.classMethod('stats', {callback: true, promise:true});
- Collection.prototype.findOneAndDelete = function(filter, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(filter == null || typeof filter != 'object') throw toError('filter parameter must be an object');
-
- if(typeof callback == 'function') return findOneAndDelete(self, filter, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- options = options || {};
- findOneAndDelete(self, filter, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var findOneAndDelete = function(self, filter, options, callback) {
-
- var finalOptions = shallowClone(options);
- finalOptions['fields'] = options.projection;
- finalOptions['remove'] = true;
-
- self.findAndModify(
- filter
- , options.sort
- , null
- , finalOptions
- , callback
- );
- }
- define.classMethod('findOneAndDelete', {callback: true, promise:true});
- Collection.prototype.findOneAndReplace = function(filter, replacement, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(filter == null || typeof filter != 'object') throw toError('filter parameter must be an object');
- if(replacement == null || typeof replacement != 'object') throw toError('replacement parameter must be an object');
-
- if(typeof callback == 'function') return findOneAndReplace(self, filter, replacement, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- options = options || {};
- findOneAndReplace(self, filter, replacement, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var findOneAndReplace = function(self, filter, replacement, options, callback) {
-
- var finalOptions = shallowClone(options);
- finalOptions['fields'] = options.projection;
- finalOptions['update'] = true;
- finalOptions['new'] = typeof options.returnOriginal == 'boolean' ? !options.returnOriginal : false;
- finalOptions['upsert'] = typeof options.upsert == 'boolean' ? options.upsert : false;
-
- self.findAndModify(
- filter
- , options.sort
- , replacement
- , finalOptions
- , callback
- );
- }
- define.classMethod('findOneAndReplace', {callback: true, promise:true});
- Collection.prototype.findOneAndUpdate = function(filter, update, options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {};
- options = options || {};
-
- if(filter == null || typeof filter != 'object') throw toError('filter parameter must be an object');
- if(update == null || typeof update != 'object') throw toError('update parameter must be an object');
-
- if(typeof callback == 'function') return findOneAndUpdate(self, filter, update, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- options = options || {};
- findOneAndUpdate(self, filter, update, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var findOneAndUpdate = function(self, filter, update, options, callback) {
-
- var finalOptions = shallowClone(options);
- finalOptions['fields'] = options.projection;
- finalOptions['update'] = true;
- finalOptions['new'] = typeof options.returnOriginal == 'boolean' ? !options.returnOriginal : false;
- finalOptions['upsert'] = typeof options.upsert == 'boolean' ? options.upsert : false;
-
- self.findAndModify(
- filter
- , options.sort
- , update
- , finalOptions
- , callback
- );
- }
- define.classMethod('findOneAndUpdate', {callback: true, promise:true});
- Collection.prototype.findAndModify = function(query, sort, doc, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 1);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- sort = args.length ? args.shift() || [] : [];
- doc = args.length ? args.shift() : null;
- options = args.length ? args.shift() || {} : {};
-
- options = shallowClone(options);
-
- options.readPreference = ReadPreference.PRIMARY;
-
- if(typeof callback == 'function') return findAndModify(self, query, sort, doc, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- options = options || {};
- findAndModify(self, query, sort, doc, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var findAndModify = function(self, query, sort, doc, options, callback) {
-
- var queryObject = {
- 'findandmodify': self.s.name
- , 'query': query
- };
- sort = formattedOrderClause(sort);
- if(sort) {
- queryObject.sort = sort;
- }
- queryObject.new = options.new ? true : false;
- queryObject.remove = options.remove ? true : false;
- queryObject.upsert = options.upsert ? true : false;
- if(options.fields) {
- queryObject.fields = options.fields;
- }
- if(doc && !options.remove) {
- queryObject.update = doc;
- }
- if(options.maxTimeMS)
- queryObject.maxTimeMS = options.maxTimeMS;
-
-
- if(options['serializeFunctions'] != null) {
- options['serializeFunctions'] = options['serializeFunctions'];
- } else {
- options['serializeFunctions'] = self.s.serializeFunctions;
- }
-
- options.checkKeys = false;
-
- var finalOptions = writeConcern(options, self.s.db, self, options);
-
- if(finalOptions.writeConcern) {
- queryObject.writeConcern = finalOptions.writeConcern;
- }
-
- if(typeof finalOptions.bypassDocumentValidation == 'boolean') {
- queryObject.bypassDocumentValidation = finalOptions.bypassDocumentValidation;
- }
-
- decorateWithCollation(queryObject, self, options);
-
- self.s.db.command(queryObject
- , options, function(err, result) {
- if(err) return handleCallback(callback, err, null);
- return handleCallback(callback, null, result);
- });
- }
- define.classMethod('findAndModify', {callback: true, promise:true});
- Collection.prototype.findAndRemove = function(query, sort, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 1);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- sort = args.length ? args.shift() || [] : [];
- options = args.length ? args.shift() || {} : {};
-
- if(typeof callback == 'function') return findAndRemove(self, query, sort, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- findAndRemove(self, query, sort, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var findAndRemove = function(self, query, sort, options, callback) {
-
- options['remove'] = true;
-
- self.findAndModify(query, sort, null, options, callback);
- }
- define.classMethod('findAndRemove', {callback: true, promise:true});
- function decorateWithWriteConcern(command, self, options) {
-
- var capabilities = self.s.topology.capabilities();
-
- if(capabilities && capabilities.commandsTakeWriteConcern) {
-
- var finalOptions = writeConcern(shallowClone(options), self.s.db, self, options);
-
- if(finalOptions.writeConcern) {
- command.writeConcern = finalOptions.writeConcern;
- }
- }
- }
- function decorateWithCollation(command, self, options) {
-
- var capabilities = self.s.topology.capabilities();
-
- if(capabilities && capabilities.commandsTakeCollation) {
- if(options.collation && typeof options.collation == 'object') {
- command.collation = options.collation;
- }
- }
- }
- Collection.prototype.aggregate = function(pipeline, options, callback) {
- var self = this;
- if(Array.isArray(pipeline)) {
-
- if(typeof options == 'function') {
- callback = options;
- options = {};
- }
-
-
- if(options == null && callback == null) {
- options = {};
- }
- } else {
-
- var args = Array.prototype.slice.call(arguments, 0);
-
- callback = args.pop();
-
- var opts = args[args.length - 1];
-
- options = opts && (opts.readPreference
- || opts.explain || opts.cursor || opts.out
- || opts.maxTimeMS || opts.allowDiskUse) ? args.pop() : {};
-
- pipeline = args;
- }
-
- var ignoreReadConcern = false;
-
- var command = { aggregate : this.s.name, pipeline : pipeline};
-
- if(typeof options.out == 'string') {
- pipeline.push({$out: options.out});
-
- ignoreReadConcern = true;
- } else if(pipeline.length > 0 && pipeline[pipeline.length - 1]['$out']) {
- ignoreReadConcern = true;
- }
-
- if(pipeline.length > 0 && pipeline[pipeline.length - 1]['$out']) {
- decorateWithWriteConcern(command, self, options);
- }
-
- decorateWithCollation(command, self, options);
-
- if(typeof options.bypassDocumentValidation == 'boolean') {
- command.bypassDocumentValidation = options.bypassDocumentValidation;
- }
-
- if(!ignoreReadConcern && this.s.readConcern) {
- command.readConcern = this.s.readConcern;
- }
-
- if(options.allowDiskUse) command.allowDiskUse = options.allowDiskUse;
- if(typeof options.maxTimeMS == 'number') command.maxTimeMS = options.maxTimeMS;
- options = shallowClone(options);
-
- options = getReadPreference(this, options, this.s.db, this);
-
- if(options.explain) command.explain = options.explain;
-
- if(options.cursor != null && typeof options.cursor != 'object') {
- throw toError('cursor options must be an object');
- }
-
- options.promiseLibrary = this.s.promiseLibrary;
-
- options.cursorFactory = AggregationCursor;
- if(typeof callback != 'function') {
- if(!this.s.topology.capabilities()) {
- throw new MongoError('cannot connect to server');
- }
- if(this.s.topology.capabilities().hasAggregationCursor) {
- options.cursor = options.cursor || { batchSize : 1000 };
- command.cursor = options.cursor;
- }
-
- if(typeof options.allowDiskUse == 'boolean') command.allowDiskUse = options.allowDiskUse;
- if(typeof options.maxTimeMS == 'number') command.maxTimeMS = options.maxTimeMS;
-
- return this.s.topology.cursor(this.s.namespace, command, options);
- }
-
- if(options.cursor) {
- return this.s.topology.cursor(this.s.namespace, command, options);
- }
-
- this.s.db.command(command, options, function(err, result) {
- if(err) {
- handleCallback(callback, err);
- } else if(result['err'] || result['errmsg']) {
- handleCallback(callback, toError(result));
- } else if(typeof result == 'object' && result['serverPipeline']) {
- handleCallback(callback, null, result['serverPipeline']);
- } else if(typeof result == 'object' && result['stages']) {
- handleCallback(callback, null, result['stages']);
- } else {
- handleCallback(callback, null, result.result);
- }
- });
- }
- define.classMethod('aggregate', {callback: true, promise:false});
- Collection.prototype.parallelCollectionScan = function(options, callback) {
- var self = this;
- if(typeof options == 'function') callback = options, options = {numCursors: 1};
-
- options.numCursors = options.numCursors || 1;
- options.batchSize = options.batchSize || 1000;
- options = shallowClone(options);
-
- options = getReadPreference(this, options, this.s.db, this);
-
- options.promiseLibrary = this.s.promiseLibrary;
-
- if(typeof callback == 'function') return parallelCollectionScan(self, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- parallelCollectionScan(self, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var parallelCollectionScan = function(self, options, callback) {
-
- var commandObject = {
- parallelCollectionScan: self.s.name
- , numCursors: options.numCursors
- }
-
- if(self.s.readConcern) {
- commandObject.readConcern = self.s.readConcern;
- }
-
- var raw = options.raw;
- delete options['raw'];
-
- self.s.db.command(commandObject, options, function(err, result) {
- if(err) return handleCallback(callback, err, null);
- if(result == null) return handleCallback(callback, new Error("no result returned for parallelCollectionScan"), null);
- var cursors = [];
-
- if(raw) options.raw = raw;
-
- for(var i = 0; i < result.cursors.length; i++) {
- var rawId = result.cursors[i].cursor.id
-
- var cursorId = typeof rawId == 'number' ? Long.fromNumber(rawId) : rawId;
-
- cursors.push(self.s.topology.cursor(self.s.namespace, cursorId, options));
- }
- handleCallback(callback, null, cursors);
- });
- }
- define.classMethod('parallelCollectionScan', {callback: true, promise:true});
- Collection.prototype.geoNear = function(x, y, options, callback) {
- var self = this;
- var point = typeof(x) == 'object' && x
- , args = Array.prototype.slice.call(arguments, point?1:2);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
-
- options = args.length ? args.shift() || {} : {};
-
- if(typeof callback == 'function') return geoNear(self, x, y, point, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- geoNear(self, x, y, point, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var geoNear = function(self, x, y, point, options, callback) {
-
- var commandObject = {
- geoNear:self.s.name,
- near: point || [x, y]
- }
- options = shallowClone(options);
-
- options = getReadPreference(self, options, self.s.db, self);
-
-
- var exclude = {
- readPreference: true,
- geoNear: true,
- near: true
- };
-
- commandObject = decorateCommand(commandObject, options, exclude);
-
- if(self.s.readConcern) {
- commandObject.readConcern = self.s.readConcern;
- }
-
- decorateWithCollation(commandObject, self, options);
-
- self.s.db.command(commandObject, options, function (err, res) {
- if(err) return handleCallback(callback, err);
- if(res.err || res.errmsg) return handleCallback(callback, toError(res));
-
-
- handleCallback(callback, null, res);
- });
- }
- define.classMethod('geoNear', {callback: true, promise:true});
- Collection.prototype.geoHaystackSearch = function(x, y, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 2);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
-
- options = args.length ? args.shift() || {} : {};
-
- if(typeof callback == 'function') return geoHaystackSearch(self, x, y, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- geoHaystackSearch(self, x, y, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var geoHaystackSearch = function(self, x, y, options, callback) {
-
- var commandObject = {
- geoSearch: self.s.name,
- near: [x, y]
- }
-
- commandObject = decorateCommand(commandObject, options, {readPreference: true});
- options = shallowClone(options);
-
- options = getReadPreference(self, options, self.s.db, self);
-
- if(self.s.readConcern) {
- commandObject.readConcern = self.s.readConcern;
- }
-
- self.s.db.command(commandObject, options, function (err, res) {
- if(err) return handleCallback(callback, err);
- if(res.err || res.errmsg) handleCallback(callback, toError(res));
-
-
- handleCallback(callback, null, res);
- });
- }
- define.classMethod('geoHaystackSearch', {callback: true, promise:true});
- var groupFunction = 'function () {\nvar c = db[ns].find(condition);\nvar map = new Map();\nvar reduce_function = reduce;\n\nwhile (c.hasNext()) {\nvar obj = c.next();\nvar key = {};\n\nfor (var i = 0, len = keys.length; i < len; ++i) {\nvar k = keys[i];\nkey[k] = obj[k];\n}\n\nvar aggObj = map.get(key);\n\nif (aggObj == null) {\nvar newObj = Object.extend({}, key);\naggObj = Object.extend(newObj, initial);\nmap.put(key, aggObj);\n}\n\nreduce_function(obj, aggObj);\n}\n\nreturn { "result": map.values() };\n}';
- Collection.prototype.group = function(keys, condition, initial, reduce, finalize, command, options, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 3);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
-
- reduce = args.length ? args.shift() : null;
- finalize = args.length ? args.shift() : null;
- command = args.length ? args.shift() : null;
- options = args.length ? args.shift() || {} : {};
-
- if(!(typeof finalize == 'function')) {
- command = finalize;
- finalize = null;
- }
- if (!Array.isArray(keys) && keys instanceof Object && typeof(keys) !== 'function' && !(keys._bsontype == 'Code')) {
- keys = Object.keys(keys);
- }
- if(typeof reduce === 'function') {
- reduce = reduce.toString();
- }
- if(typeof finalize === 'function') {
- finalize = finalize.toString();
- }
-
- command = command == null ? true : command;
-
- if(typeof callback == 'function') return group(self, keys, condition, initial, reduce, finalize, command, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- group(self, keys, condition, initial, reduce, finalize, command, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- });
- });
- }
- var group = function(self, keys, condition, initial, reduce, finalize, command, options, callback) {
-
- if(command) {
- var reduceFunction = reduce && reduce._bsontype == 'Code'
- ? reduce
- : new Code(reduce);
- var selector = {
- group: {
- 'ns': self.s.name
- , '$reduce': reduceFunction
- , 'cond': condition
- , 'initial': initial
- , 'out': "inline"
- }
- };
-
- if(finalize != null) selector.group['finalize'] = finalize;
-
- if ('function' === typeof keys || (keys && keys._bsontype == 'Code')) {
- selector.group.$keyf = keys && keys._bsontype == 'Code'
- ? keys
- : new Code(keys);
- } else {
- var hash = {};
- keys.forEach(function (key) {
- hash[key] = 1;
- });
- selector.group.key = hash;
- }
- options = shallowClone(options);
-
- options = getReadPreference(self, options, self.s.db, self);
-
- if(self.s.readConcern) {
- selector.readConcern = self.s.readConcern;
- }
-
- decorateWithCollation(selector, self, options);
-
- self.s.db.command(selector, options, function(err, result) {
- if(err) return handleCallback(callback, err, null);
- handleCallback(callback, null, result.retval);
- });
- } else {
-
- var scope = reduce != null && reduce._bsontype == 'Code'
- ? reduce.scope
- : {};
- scope.ns = self.s.name;
- scope.keys = keys;
- scope.condition = condition;
- scope.initial = initial;
-
- var groupfn = groupFunction.replace(/ reduce;/, reduce.toString() + ';');
- self.s.db.eval(new Code(groupfn, scope), function (err, results) {
- if (err) return handleCallback(callback, err, null);
- handleCallback(callback, null, results.result || results);
- });
- }
- }
- define.classMethod('group', {callback: true, promise:true});
- function processScope (scope) {
- if(!isObject(scope) || scope._bsontype == 'ObjectID') {
- return scope;
- }
- var keys = Object.keys(scope);
- var i = keys.length;
- var key;
- var new_scope = {};
- while (i--) {
- key = keys[i];
- if ('function' == typeof scope[key]) {
- new_scope[key] = new Code(String(scope[key]));
- } else {
- new_scope[key] = processScope(scope[key]);
- }
- }
- return new_scope;
- }
- Collection.prototype.mapReduce = function(map, reduce, options, callback) {
- var self = this;
- if('function' === typeof options) callback = options, options = {};
-
- if(null == options.out) {
- throw new Error("the out option parameter must be defined, see mongodb docs for possible values");
- }
- if('function' === typeof map) {
- map = map.toString();
- }
- if('function' === typeof reduce) {
- reduce = reduce.toString();
- }
- if('function' === typeof options.finalize) {
- options.finalize = options.finalize.toString();
- }
-
- if(typeof callback == 'function') return mapReduce(self, map, reduce, options, callback);
-
- return new this.s.promiseLibrary(function(resolve, reject) {
- mapReduce(self, map, reduce, options, function(err, r, r1) {
- if(err) return reject(err);
- if(!r1) return resolve(r);
- resolve({results: r, stats: r1});
- });
- });
- }
- var mapReduce = function(self, map, reduce, options, callback) {
- var mapCommandHash = {
- mapreduce: self.s.name
- , map: map
- , reduce: reduce
- };
-
- for(var n in options) {
- if('scope' == n) {
- mapCommandHash[n] = processScope(options[n]);
- } else {
- mapCommandHash[n] = options[n];
- }
- }
- options = shallowClone(options);
-
- options = getReadPreference(self, options, self.s.db, self);
-
- if((options.readPreference != false && options.readPreference != 'primary')
- && options['out'] && (options['out'].inline != 1 && options['out'] != 'inline')) {
-
- options.readPreference = 'primary';
-
- decorateWithWriteConcern(mapCommandHash, self, options);
- } else if(self.s.readConcern) {
- mapCommandHash.readConcern = self.s.readConcern;
- }
-
- if(typeof options.bypassDocumentValidation == 'boolean') {
- mapCommandHash.bypassDocumentValidation = options.bypassDocumentValidation;
- }
-
- decorateWithCollation(mapCommandHash, self, options);
-
- self.s.db.command(mapCommandHash, {readPreference:options.readPreference}, function (err, result) {
- if(err) return handleCallback(callback, err);
-
- if(1 != result.ok || result.err || result.errmsg) {
- return handleCallback(callback, toError(result));
- }
-
- var stats = {};
- if(result.timeMillis) stats['processtime'] = result.timeMillis;
- if(result.counts) stats['counts'] = result.counts;
- if(result.timing) stats['timing'] = result.timing;
-
- if(result.results) {
-
- if(options['verbose'] == null || !options['verbose']) {
- return handleCallback(callback, null, result.results);
- }
- return handleCallback(callback, null, result.results, stats);
- }
-
- var collection = null;
-
- if(result.result != null && typeof result.result == 'object') {
- var doc = result.result;
- collection = self.s.db.db(doc.db).collection(doc.collection);
- } else {
-
- collection = self.s.db.collection(result.result)
- }
-
- if(options['verbose'] == null || !options['verbose']) {
- return handleCallback(callback, err, collection);
- }
-
- handleCallback(callback, err, collection, stats);
- });
- }
- define.classMethod('mapReduce', {callback: true, promise:true});
- Collection.prototype.initializeUnorderedBulkOp = function(options) {
- options = options || {};
- options.promiseLibrary = this.s.promiseLibrary;
- return unordered(this.s.topology, this, options);
- }
- define.classMethod('initializeUnorderedBulkOp', {callback: false, promise:false, returns: [ordered.UnorderedBulkOperation]});
- Collection.prototype.initializeOrderedBulkOp = function(options) {
- options = options || {};
- options.promiseLibrary = this.s.promiseLibrary;
- return ordered(this.s.topology, this, options);
- }
- define.classMethod('initializeOrderedBulkOp', {callback: false, promise:false, returns: [ordered.OrderedBulkOperation]});
- var writeConcern = function(target, db, col, options) {
- if(options.w != null || options.j != null || options.fsync != null) {
- var opts = {};
- if(options.w != null) opts.w = options.w;
- if(options.wtimeout != null) opts.wtimeout = options.wtimeout;
- if(options.j != null) opts.j = options.j;
- if(options.fsync != null) opts.fsync = options.fsync;
- target.writeConcern = opts;
- } else if(col.writeConcern.w != null || col.writeConcern.j != null || col.writeConcern.fsync != null) {
- target.writeConcern = col.writeConcern;
- } else if(db.writeConcern.w != null || db.writeConcern.j != null || db.writeConcern.fsync != null) {
- target.writeConcern = db.writeConcern;
- }
- return target
- }
- var getReadPreference = function(self, options, db) {
- var r = null
- if(options.readPreference) {
- r = options.readPreference
- } else if(self.s.readPreference) {
- r = self.s.readPreference
- } else if(db.s.readPreference) {
- r = db.s.readPreference;
- }
- if(r instanceof ReadPreference) {
- options.readPreference = new CoreReadPreference(r.mode, r.tags, {maxStalenessSeconds: r.maxStalenessSeconds});
- } else if(typeof r == 'string') {
- options.readPreference = new CoreReadPreference(r);
- } else if(r && !(r instanceof ReadPreference) && typeof r == 'object') {
- var mode = r.mode || r.preference;
- if (mode && typeof mode == 'string') {
- options.readPreference = new CoreReadPreference(mode, r.tags, {maxStalenessSeconds: r.maxStalenessSeconds});
- }
- }
- return options;
- }
- var testForFields = {
- limit: 1, sort: 1, fields:1, skip: 1, hint: 1, explain: 1, snapshot: 1, timeout: 1, tailable: 1, tailableRetryInterval: 1
- , numberOfRetries: 1, awaitdata: 1, awaitData: 1, exhaust: 1, batchSize: 1, returnKey: 1, maxScan: 1, min: 1, max: 1, showDiskLoc: 1
- , comment: 1, raw: 1, readPreference: 1, partial: 1, read: 1, dbName: 1, oplogReplay: 1, connection: 1, maxTimeMS: 1, transforms: 1
- , collation: 1
- }
- module.exports = Collection;
|