1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951 |
- "use strict";
- /**
- * @fileOverview GridFS is a tool for MongoDB to store files to the database.
- * Because of the restrictions of the object size the database can hold, a
- * facility to split a file into several chunks is needed. The {@link GridStore}
- * class offers a simplified api to interact with files while managing the
- * chunks of split files behind the scenes. More information about GridFS can be
- * found <a href="http://www.mongodb.org/display/DOCS/GridFS">here</a>.
- *
- * @example
- * var MongoClient = require('mongodb').MongoClient,
- * GridStore = require('mongodb').GridStore,
- * ObjectID = require('mongodb').ObjectID,
- * test = require('assert');
- *
- * // Connection url
- * var url = 'mongodb://localhost:27017/test';
- * // Connect using MongoClient
- * MongoClient.connect(url, function(err, db) {
- * var gridStore = new GridStore(db, null, "w");
- * gridStore.open(function(err, gridStore) {
- * gridStore.write("hello world!", function(err, gridStore) {
- * gridStore.close(function(err, result) {
- *
- * // Let's read the file using object Id
- * GridStore.read(db, result._id, function(err, data) {
- * test.equal('hello world!', data);
- * db.close();
- * test.done();
- * });
- * });
- * });
- * });
- * });
- */
- var Chunk = require('./chunk'),
- ObjectID = require('mongodb-core').BSON.ObjectID,
- ReadPreference = require('../read_preference'),
- Buffer = require('buffer').Buffer,
- Collection = require('../collection'),
- fs = require('fs'),
- f = require('util').format,
- util = require('util'),
- Define = require('../metadata'),
- MongoError = require('mongodb-core').MongoError,
- inherits = util.inherits,
- Duplex = require('stream').Duplex || require('readable-stream').Duplex,
- shallowClone = require('../utils').shallowClone;
- var REFERENCE_BY_FILENAME = 0,
- REFERENCE_BY_ID = 1;
- /**
- * Namespace provided by the mongodb-core and node.js
- * @external Duplex
- */
- /**
- * Create a new GridStore instance
- *
- * Modes
- * - **"r"** - read only. This is the default mode.
- * - **"w"** - write in truncate mode. Existing data will be overwriten.
- *
- * @class
- * @param {Db} db A database instance to interact with.
- * @param {object} [id] optional unique id for this file
- * @param {string} [filename] optional filename for this file, no unique constrain on the field
- * @param {string} mode set the mode for this file.
- * @param {object} [options=null] Optional settings.
- * @param {(number|string)} [options.w=null] The write concern.
- * @param {number} [options.wtimeout=null] The write concern timeout.
- * @param {boolean} [options.j=false] Specify a journal write concern.
- * @param {boolean} [options.fsync=false] Specify a file sync write concern.
- * @param {string} [options.root=null] Root collection to use. Defaults to **{GridStore.DEFAULT_ROOT_COLLECTION}**.
- * @param {string} [options.content_type=null] MIME type of the file. Defaults to **{GridStore.DEFAULT_CONTENT_TYPE}**.
- * @param {number} [options.chunk_size=261120] Size for the chunk. Defaults to **{Chunk.DEFAULT_CHUNK_SIZE}**.
- * @param {object} [options.metadata=null] Arbitrary data the user wants to store.
- * @param {object} [options.promiseLibrary=null] A Promise library class the application wishes to use such as Bluebird, must be ES6 compatible
- * @param {(ReadPreference|string)} [options.readPreference=null] The preferred read preference (ReadPreference.PRIMARY, ReadPreference.PRIMARY_PREFERRED, ReadPreference.SECONDARY, ReadPreference.SECONDARY_PREFERRED, ReadPreference.NEAREST).
- * @property {number} chunkSize Get the gridstore chunk size.
- * @property {number} md5 The md5 checksum for this file.
- * @property {number} chunkNumber The current chunk number the gridstore has materialized into memory
- * @return {GridStore} a GridStore instance.
- * @deprecated Use GridFSBucket API instead
- */
- var GridStore = function GridStore(db, id, filename, mode, options) {
- if(!(this instanceof GridStore)) return new GridStore(db, id, filename, mode, options);
- this.db = db;
- // Handle options
- if(typeof options === 'undefined') options = {};
- // Handle mode
- if(typeof mode === 'undefined') {
- mode = filename;
- filename = undefined;
- } else if(typeof mode == 'object') {
- options = mode;
- mode = filename;
- filename = undefined;
- }
- if(id && id._bsontype == 'ObjectID') {
- this.referenceBy = REFERENCE_BY_ID;
- this.fileId = id;
- this.filename = filename;
- } else if(typeof filename == 'undefined') {
- this.referenceBy = REFERENCE_BY_FILENAME;
- this.filename = id;
- if (mode.indexOf('w') != null) {
- this.fileId = new ObjectID();
- }
- } else {
- this.referenceBy = REFERENCE_BY_ID;
- this.fileId = id;
- this.filename = filename;
- }
- // Set up the rest
- this.mode = mode == null ? "r" : mode;
- this.options = options || {};
- // Opened
- this.isOpen = false;
- // Set the root if overridden
- this.root = this.options['root'] == null ? GridStore.DEFAULT_ROOT_COLLECTION : this.options['root'];
- this.position = 0;
- this.readPreference = this.options.readPreference || db.options.readPreference || ReadPreference.PRIMARY;
- this.writeConcern = _getWriteConcern(db, this.options);
- // Set default chunk size
- this.internalChunkSize = this.options['chunkSize'] == null ? Chunk.DEFAULT_CHUNK_SIZE : this.options['chunkSize'];
- // Get the promiseLibrary
- var promiseLibrary = this.options.promiseLibrary;
- // No promise library selected fall back
- if(!promiseLibrary) {
- promiseLibrary = typeof global.Promise == 'function' ?
- global.Promise : require('es6-promise').Promise;
- }
- // Set the promiseLibrary
- this.promiseLibrary = promiseLibrary;
- Object.defineProperty(this, "chunkSize", { enumerable: true
- , get: function () {
- return this.internalChunkSize;
- }
- , set: function(value) {
- if(!(this.mode[0] == "w" && this.position == 0 && this.uploadDate == null)) {
- this.internalChunkSize = this.internalChunkSize;
- } else {
- this.internalChunkSize = value;
- }
- }
- });
- Object.defineProperty(this, "md5", { enumerable: true
- , get: function () {
- return this.internalMd5;
- }
- });
- Object.defineProperty(this, "chunkNumber", { enumerable: true
- , get: function () {
- return this.currentChunk && this.currentChunk.chunkNumber ? this.currentChunk.chunkNumber : null;
- }
- });
- }
- var define = GridStore.define = new Define('Gridstore', GridStore, true);
- /**
- * The callback format for the Gridstore.open method
- * @callback GridStore~openCallback
- * @param {MongoError} error An error instance representing the error during the execution.
- * @param {GridStore} gridStore The GridStore instance if the open method was successful.
- */
- /**
- * Opens the file from the database and initialize this object. Also creates a
- * new one if file does not exist.
- *
- * @method
- * @param {GridStore~openCallback} [callback] this will be called after executing this method
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.open = function(callback) {
- var self = this;
- if( this.mode != "w" && this.mode != "w+" && this.mode != "r"){
- throw MongoError.create({message: "Illegal mode " + this.mode, driver:true});
- }
- // We provided a callback leg
- if(typeof callback == 'function') return open(self, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- open(self, function(err, store) {
- if(err) return reject(err);
- resolve(store);
- })
- });
- };
- var open = function(self, callback) {
- // Get the write concern
- var writeConcern = _getWriteConcern(self.db, self.options);
- // If we are writing we need to ensure we have the right indexes for md5's
- if((self.mode == "w" || self.mode == "w+")) {
- // Get files collection
- var collection = self.collection();
- // Put index on filename
- collection.ensureIndex([['filename', 1]], writeConcern, function() {
- // Get chunk collection
- var chunkCollection = self.chunkCollection();
- // Make an unique index for compatibility with mongo-cxx-driver:legacy
- var chunkIndexOptions = shallowClone(writeConcern);
- chunkIndexOptions.unique = true;
- // Ensure index on chunk collection
- chunkCollection.ensureIndex([['files_id', 1], ['n', 1]], chunkIndexOptions, function() {
- // Open the connection
- _open(self, writeConcern, function(err, r) {
- if(err) return callback(err);
- self.isOpen = true;
- callback(err, r);
- });
- });
- });
- } else {
- // Open the gridstore
- _open(self, writeConcern, function(err, r) {
- if(err) return callback(err);
- self.isOpen = true;
- callback(err, r);
- });
- }
- }
- // Push the definition for open
- define.classMethod('open', {callback: true, promise:true});
- /**
- * Verify if the file is at EOF.
- *
- * @method
- * @return {boolean} true if the read/write head is at the end of this file.
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.eof = function() {
- return this.position == this.length ? true : false;
- }
- define.classMethod('eof', {callback: false, promise:false, returns: [Boolean]});
- /**
- * The callback result format.
- * @callback GridStore~resultCallback
- * @param {MongoError} error An error instance representing the error during the execution.
- * @param {object} result The result from the callback.
- */
- /**
- * Retrieves a single character from this file.
- *
- * @method
- * @param {GridStore~resultCallback} [callback] this gets called after this method is executed. Passes null to the first parameter and the character read to the second or null to the second if the read/write head is at the end of the file.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.getc = function(callback) {
- var self = this;
- // We provided a callback leg
- if(typeof callback == 'function') return eof(self, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- eof(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- }
- var eof = function(self, callback) {
- if(self.eof()) {
- callback(null, null);
- } else if(self.currentChunk.eof()) {
- nthChunk(self, self.currentChunk.chunkNumber + 1, function(err, chunk) {
- self.currentChunk = chunk;
- self.position = self.position + 1;
- callback(err, self.currentChunk.getc());
- });
- } else {
- self.position = self.position + 1;
- callback(null, self.currentChunk.getc());
- }
- }
- define.classMethod('getc', {callback: true, promise:true});
- /**
- * Writes a string to the file with a newline character appended at the end if
- * the given string does not have one.
- *
- * @method
- * @param {string} string the string to write.
- * @param {GridStore~resultCallback} [callback] this will be called after executing this method. The first parameter will contain null and the second one will contain a reference to this object.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.puts = function(string, callback) {
- var self = this;
- var finalString = string.match(/\n$/) == null ? string + "\n" : string;
- // We provided a callback leg
- if(typeof callback == 'function') return this.write(finalString, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- self.write(finalString, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- }
- define.classMethod('puts', {callback: true, promise:true});
- /**
- * Return a modified Readable stream including a possible transform method.
- *
- * @method
- * @return {GridStoreStream}
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.stream = function() {
- return new GridStoreStream(this);
- }
- define.classMethod('stream', {callback: false, promise:false, returns: [GridStoreStream]});
- /**
- * Writes some data. This method will work properly only if initialized with mode "w" or "w+".
- *
- * @method
- * @param {(string|Buffer)} data the data to write.
- * @param {boolean} [close] closes this file after writing if set to true.
- * @param {GridStore~resultCallback} [callback] this will be called after executing this method. The first parameter will contain null and the second one will contain a reference to this object.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.write = function write(data, close, callback) {
- var self = this;
- // We provided a callback leg
- if(typeof callback == 'function') return _writeNormal(this, data, close, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- _writeNormal(self, data, close, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- }
- define.classMethod('write', {callback: true, promise:true});
- /**
- * Handles the destroy part of a stream
- *
- * @method
- * @result {null}
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.destroy = function destroy() {
- // close and do not emit any more events. queued data is not sent.
- if(!this.writable) return;
- this.readable = false;
- if(this.writable) {
- this.writable = false;
- this._q.length = 0;
- this.emit('close');
- }
- }
- define.classMethod('destroy', {callback: false, promise:false});
- /**
- * Stores a file from the file system to the GridFS database.
- *
- * @method
- * @param {(string|Buffer|FileHandle)} file the file to store.
- * @param {GridStore~resultCallback} [callback] this will be called after executing this method. The first parameter will contain null and the second one will contain a reference to this object.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.writeFile = function (file, callback) {
- var self = this;
- // We provided a callback leg
- if(typeof callback == 'function') return writeFile(self, file, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- writeFile(self, file, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var writeFile = function(self, file, callback) {
- if (typeof file === 'string') {
- fs.open(file, 'r', function (err, fd) {
- if(err) return callback(err);
- self.writeFile(fd, callback);
- });
- return;
- }
- self.open(function (err, self) {
- if(err) return callback(err, self);
- fs.fstat(file, function (err, stats) {
- if(err) return callback(err, self);
- var offset = 0;
- var index = 0;
- // Write a chunk
- var writeChunk = function() {
- fs.read(file, self.chunkSize, offset, 'binary', function(err, data, bytesRead) {
- if(err) return callback(err, self);
- offset = offset + bytesRead;
- // Create a new chunk for the data
- var chunk = new Chunk(self, {n:index++}, self.writeConcern);
- chunk.write(data, function(err, chunk) {
- if(err) return callback(err, self);
- chunk.save({}, function(err) {
- if(err) return callback(err, self);
- self.position = self.position + data.length;
- // Point to current chunk
- self.currentChunk = chunk;
- if(offset >= stats.size) {
- fs.close(file);
- self.close(function(err) {
- if(err) return callback(err, self);
- return callback(null, self);
- });
- } else {
- return process.nextTick(writeChunk);
- }
- });
- });
- });
- }
- // Process the first write
- process.nextTick(writeChunk);
- });
- });
- }
- define.classMethod('writeFile', {callback: true, promise:true});
- /**
- * Saves this file to the database. This will overwrite the old entry if it
- * already exists. This will work properly only if mode was initialized to
- * "w" or "w+".
- *
- * @method
- * @param {GridStore~resultCallback} [callback] this will be called after executing this method. The first parameter will contain null and the second one will contain a reference to this object.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.close = function(callback) {
- var self = this;
- // We provided a callback leg
- if(typeof callback == 'function') return close(self, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- close(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var close = function(self, callback) {
- if(self.mode[0] == "w") {
- // Set up options
- var options = self.writeConcern;
- if(self.currentChunk != null && self.currentChunk.position > 0) {
- self.currentChunk.save({}, function(err) {
- if(err && typeof callback == 'function') return callback(err);
- self.collection(function(err, files) {
- if(err && typeof callback == 'function') return callback(err);
- // Build the mongo object
- if(self.uploadDate != null) {
- buildMongoObject(self, function(err, mongoObject) {
- if(err) {
- if(typeof callback == 'function') return callback(err); else throw err;
- }
- files.save(mongoObject, options, function(err) {
- if(typeof callback == 'function')
- callback(err, mongoObject);
- });
- });
- } else {
- self.uploadDate = new Date();
- buildMongoObject(self, function(err, mongoObject) {
- if(err) {
- if(typeof callback == 'function') return callback(err); else throw err;
- }
- files.save(mongoObject, options, function(err) {
- if(typeof callback == 'function')
- callback(err, mongoObject);
- });
- });
- }
- });
- });
- } else {
- self.collection(function(err, files) {
- if(err && typeof callback == 'function') return callback(err);
- self.uploadDate = new Date();
- buildMongoObject(self, function(err, mongoObject) {
- if(err) {
- if(typeof callback == 'function') return callback(err); else throw err;
- }
- files.save(mongoObject, options, function(err) {
- if(typeof callback == 'function')
- callback(err, mongoObject);
- });
- });
- });
- }
- } else if(self.mode[0] == "r") {
- if(typeof callback == 'function')
- callback(null, null);
- } else {
- if(typeof callback == 'function')
- callback(MongoError.create({message: f("Illegal mode %s", self.mode), driver:true}));
- }
- }
- define.classMethod('close', {callback: true, promise:true});
- /**
- * The collection callback format.
- * @callback GridStore~collectionCallback
- * @param {MongoError} error An error instance representing the error during the execution.
- * @param {Collection} collection The collection from the command execution.
- */
- /**
- * Retrieve this file's chunks collection.
- *
- * @method
- * @param {GridStore~collectionCallback} callback the command callback.
- * @return {Collection}
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.chunkCollection = function(callback) {
- if(typeof callback == 'function')
- return this.db.collection((this.root + ".chunks"), callback);
- return this.db.collection((this.root + ".chunks"));
- };
- define.classMethod('chunkCollection', {callback: true, promise:false, returns: [Collection]});
- /**
- * Deletes all the chunks of this file in the database.
- *
- * @method
- * @param {GridStore~resultCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.unlink = function(callback) {
- var self = this;
- // We provided a callback leg
- if(typeof callback == 'function') return unlink(self, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- unlink(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var unlink = function(self, callback) {
- deleteChunks(self, function(err) {
- if(err!==null) {
- err.message = "at deleteChunks: " + err.message;
- return callback(err);
- }
- self.collection(function(err, collection) {
- if(err!==null) {
- err.message = "at collection: " + err.message;
- return callback(err);
- }
- collection.remove({'_id':self.fileId}, self.writeConcern, function(err) {
- callback(err, self);
- });
- });
- });
- }
- define.classMethod('unlink', {callback: true, promise:true});
- /**
- * Retrieves the file collection associated with this object.
- *
- * @method
- * @param {GridStore~collectionCallback} callback the command callback.
- * @return {Collection}
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.collection = function(callback) {
- if(typeof callback == 'function')
- this.db.collection(this.root + ".files", callback);
- return this.db.collection(this.root + ".files");
- };
- define.classMethod('collection', {callback: true, promise:false, returns: [Collection]});
- /**
- * The readlines callback format.
- * @callback GridStore~readlinesCallback
- * @param {MongoError} error An error instance representing the error during the execution.
- * @param {string[]} strings The array of strings returned.
- */
- /**
- * Read the entire file as a list of strings splitting by the provided separator.
- *
- * @method
- * @param {string} [separator] The character to be recognized as the newline separator.
- * @param {GridStore~readlinesCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.readlines = function(separator, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 0);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- separator = args.length ? args.shift() : "\n";
- separator = separator || "\n";
- // We provided a callback leg
- if(typeof callback == 'function') return readlines(self, separator, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- readlines(self, separator, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var readlines = function(self, separator, callback) {
- self.read(function(err, data) {
- if(err) return callback(err);
- var items = data.toString().split(separator);
- items = items.length > 0 ? items.splice(0, items.length - 1) : [];
- for(var i = 0; i < items.length; i++) {
- items[i] = items[i] + separator;
- }
- callback(null, items);
- });
- }
- define.classMethod('readlines', {callback: true, promise:true});
- /**
- * Deletes all the chunks of this file in the database if mode was set to "w" or
- * "w+" and resets the read/write head to the initial position.
- *
- * @method
- * @param {GridStore~resultCallback} [callback] this will be called after executing this method. The first parameter will contain null and the second one will contain a reference to this object.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.rewind = function(callback) {
- var self = this;
- // We provided a callback leg
- if(typeof callback == 'function') return rewind(self, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- rewind(self, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var rewind = function(self, callback) {
- if(self.currentChunk.chunkNumber != 0) {
- if(self.mode[0] == "w") {
- deleteChunks(self, function(err) {
- if(err) return callback(err);
- self.currentChunk = new Chunk(self, {'n': 0}, self.writeConcern);
- self.position = 0;
- callback(null, self);
- });
- } else {
- self.currentChunk(0, function(err, chunk) {
- if(err) return callback(err);
- self.currentChunk = chunk;
- self.currentChunk.rewind();
- self.position = 0;
- callback(null, self);
- });
- }
- } else {
- self.currentChunk.rewind();
- self.position = 0;
- callback(null, self);
- }
- }
- define.classMethod('rewind', {callback: true, promise:true});
- /**
- * The read callback format.
- * @callback GridStore~readCallback
- * @param {MongoError} error An error instance representing the error during the execution.
- * @param {Buffer} data The data read from the GridStore object
- */
- /**
- * Retrieves the contents of this file and advances the read/write head. Works with Buffers only.
- *
- * There are 3 signatures for this method:
- *
- * (callback)
- * (length, callback)
- * (length, buffer, callback)
- *
- * @method
- * @param {number} [length] the number of characters to read. Reads all the characters from the read/write head to the EOF if not specified.
- * @param {(string|Buffer)} [buffer] a string to hold temporary data. This is used for storing the string data read so far when recursively calling this method.
- * @param {GridStore~readCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.read = function(length, buffer, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 0);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- length = args.length ? args.shift() : null;
- buffer = args.length ? args.shift() : null;
- // We provided a callback leg
- if(typeof callback == 'function') return read(self, length, buffer, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- read(self, length, buffer, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- }
- var read = function(self, length, buffer, callback) {
- // The data is a c-terminated string and thus the length - 1
- var finalLength = length == null ? self.length - self.position : length;
- var finalBuffer = buffer == null ? new Buffer(finalLength) : buffer;
- // Add a index to buffer to keep track of writing position or apply current index
- finalBuffer._index = buffer != null && buffer._index != null ? buffer._index : 0;
- if((self.currentChunk.length() - self.currentChunk.position + finalBuffer._index) >= finalLength) {
- var slice = self.currentChunk.readSlice(finalLength - finalBuffer._index);
- // Copy content to final buffer
- slice.copy(finalBuffer, finalBuffer._index);
- // Update internal position
- self.position = self.position + finalBuffer.length;
- // Check if we don't have a file at all
- if(finalLength == 0 && finalBuffer.length == 0) return callback(MongoError.create({message: "File does not exist", driver:true}), null);
- // Else return data
- return callback(null, finalBuffer);
- }
- // Read the next chunk
- slice = self.currentChunk.readSlice(self.currentChunk.length() - self.currentChunk.position);
- // Copy content to final buffer
- slice.copy(finalBuffer, finalBuffer._index);
- // Update index position
- finalBuffer._index += slice.length;
- // Load next chunk and read more
- nthChunk(self, self.currentChunk.chunkNumber + 1, function(err, chunk) {
- if(err) return callback(err);
- if(chunk.length() > 0) {
- self.currentChunk = chunk;
- self.read(length, finalBuffer, callback);
- } else {
- if(finalBuffer._index > 0) {
- callback(null, finalBuffer)
- } else {
- callback(MongoError.create({message: "no chunks found for file, possibly corrupt", driver:true}), null);
- }
- }
- });
- }
- define.classMethod('read', {callback: true, promise:true});
- /**
- * The tell callback format.
- * @callback GridStore~tellCallback
- * @param {MongoError} error An error instance representing the error during the execution.
- * @param {number} position The current read position in the GridStore.
- */
- /**
- * Retrieves the position of the read/write head of this file.
- *
- * @method
- * @param {number} [length] the number of characters to read. Reads all the characters from the read/write head to the EOF if not specified.
- * @param {(string|Buffer)} [buffer] a string to hold temporary data. This is used for storing the string data read so far when recursively calling this method.
- * @param {GridStore~tellCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.tell = function(callback) {
- var self = this;
- // We provided a callback leg
- if(typeof callback == 'function') return callback(null, this.position);
- // Return promise
- return new self.promiseLibrary(function(resolve) {
- resolve(self.position);
- });
- };
- define.classMethod('tell', {callback: true, promise:true});
- /**
- * The tell callback format.
- * @callback GridStore~gridStoreCallback
- * @param {MongoError} error An error instance representing the error during the execution.
- * @param {GridStore} gridStore The gridStore.
- */
- /**
- * Moves the read/write head to a new location.
- *
- * There are 3 signatures for this method
- *
- * Seek Location Modes
- * - **GridStore.IO_SEEK_SET**, **(default)** set the position from the start of the file.
- * - **GridStore.IO_SEEK_CUR**, set the position from the current position in the file.
- * - **GridStore.IO_SEEK_END**, set the position from the end of the file.
- *
- * @method
- * @param {number} [position] the position to seek to
- * @param {number} [seekLocation] seek mode. Use one of the Seek Location modes.
- * @param {GridStore~gridStoreCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.prototype.seek = function(position, seekLocation, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 1);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- seekLocation = args.length ? args.shift() : null;
- // We provided a callback leg
- if(typeof callback == 'function') return seek(self, position, seekLocation, callback);
- // Return promise
- return new self.promiseLibrary(function(resolve, reject) {
- seek(self, position, seekLocation, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- }
- var seek = function(self, position, seekLocation, callback) {
- // Seek only supports read mode
- if(self.mode != 'r') {
- return callback(MongoError.create({message: "seek is only supported for mode r", driver:true}))
- }
- var seekLocationFinal = seekLocation == null ? GridStore.IO_SEEK_SET : seekLocation;
- var finalPosition = position;
- var targetPosition = 0;
- // Calculate the position
- if(seekLocationFinal == GridStore.IO_SEEK_CUR) {
- targetPosition = self.position + finalPosition;
- } else if(seekLocationFinal == GridStore.IO_SEEK_END) {
- targetPosition = self.length + finalPosition;
- } else {
- targetPosition = finalPosition;
- }
- // Get the chunk
- var newChunkNumber = Math.floor(targetPosition/self.chunkSize);
- var seekChunk = function() {
- nthChunk(self, newChunkNumber, function(err, chunk) {
- if(err) return callback(err, null);
- if(chunk == null) return callback(new Error('no chunk found'));
- // Set the current chunk
- self.currentChunk = chunk;
- self.position = targetPosition;
- self.currentChunk.position = (self.position % self.chunkSize);
- callback(err, self);
- });
- };
- seekChunk();
- }
- define.classMethod('seek', {callback: true, promise:true});
- /**
- * @ignore
- */
- var _open = function(self, options, callback) {
- var collection = self.collection();
- // Create the query
- var query = self.referenceBy == REFERENCE_BY_ID ? {_id:self.fileId} : {filename:self.filename};
- query = null == self.fileId && self.filename == null ? null : query;
- options.readPreference = self.readPreference;
- // Fetch the chunks
- if(query != null) {
- collection.findOne(query, options, function(err, doc) {
- if(err) return error(err);
- // Check if the collection for the files exists otherwise prepare the new one
- if(doc != null) {
- self.fileId = doc._id;
- // Prefer a new filename over the existing one if this is a write
- self.filename = ((self.mode == 'r') || (self.filename == undefined)) ? doc.filename : self.filename;
- self.contentType = doc.contentType;
- self.internalChunkSize = doc.chunkSize;
- self.uploadDate = doc.uploadDate;
- self.aliases = doc.aliases;
- self.length = doc.length;
- self.metadata = doc.metadata;
- self.internalMd5 = doc.md5;
- } else if (self.mode != 'r') {
- self.fileId = self.fileId == null ? new ObjectID() : self.fileId;
- self.contentType = GridStore.DEFAULT_CONTENT_TYPE;
- self.internalChunkSize = self.internalChunkSize == null ? Chunk.DEFAULT_CHUNK_SIZE : self.internalChunkSize;
- self.length = 0;
- } else {
- self.length = 0;
- var txtId = self.fileId._bsontype == "ObjectID" ? self.fileId.toHexString() : self.fileId;
- return error(MongoError.create({message: f("file with id %s not opened for writing", (self.referenceBy == REFERENCE_BY_ID ? txtId : self.filename)), driver:true}), self);
- }
- // Process the mode of the object
- if(self.mode == "r") {
- nthChunk(self, 0, options, function(err, chunk) {
- if(err) return error(err);
- self.currentChunk = chunk;
- self.position = 0;
- callback(null, self);
- });
- } else if(self.mode == "w" && doc) {
- // Delete any existing chunks
- deleteChunks(self, options, function(err) {
- if(err) return error(err);
- self.currentChunk = new Chunk(self, {'n':0}, self.writeConcern);
- self.contentType = self.options['content_type'] == null ? self.contentType : self.options['content_type'];
- self.internalChunkSize = self.options['chunk_size'] == null ? self.internalChunkSize : self.options['chunk_size'];
- self.metadata = self.options['metadata'] == null ? self.metadata : self.options['metadata'];
- self.aliases = self.options['aliases'] == null ? self.aliases : self.options['aliases'];
- self.position = 0;
- callback(null, self);
- });
- } else if(self.mode == "w") {
- self.currentChunk = new Chunk(self, {'n':0}, self.writeConcern);
- self.contentType = self.options['content_type'] == null ? self.contentType : self.options['content_type'];
- self.internalChunkSize = self.options['chunk_size'] == null ? self.internalChunkSize : self.options['chunk_size'];
- self.metadata = self.options['metadata'] == null ? self.metadata : self.options['metadata'];
- self.aliases = self.options['aliases'] == null ? self.aliases : self.options['aliases'];
- self.position = 0;
- callback(null, self);
- } else if(self.mode == "w+") {
- nthChunk(self, lastChunkNumber(self), options, function(err, chunk) {
- if(err) return error(err);
- // Set the current chunk
- self.currentChunk = chunk == null ? new Chunk(self, {'n':0}, self.writeConcern) : chunk;
- self.currentChunk.position = self.currentChunk.data.length();
- self.metadata = self.options['metadata'] == null ? self.metadata : self.options['metadata'];
- self.aliases = self.options['aliases'] == null ? self.aliases : self.options['aliases'];
- self.position = self.length;
- callback(null, self);
- });
- }
- });
- } else {
- // Write only mode
- self.fileId = null == self.fileId ? new ObjectID() : self.fileId;
- self.contentType = GridStore.DEFAULT_CONTENT_TYPE;
- self.internalChunkSize = self.internalChunkSize == null ? Chunk.DEFAULT_CHUNK_SIZE : self.internalChunkSize;
- self.length = 0;
- // No file exists set up write mode
- if(self.mode == "w") {
- // Delete any existing chunks
- deleteChunks(self, options, function(err) {
- if(err) return error(err);
- self.currentChunk = new Chunk(self, {'n':0}, self.writeConcern);
- self.contentType = self.options['content_type'] == null ? self.contentType : self.options['content_type'];
- self.internalChunkSize = self.options['chunk_size'] == null ? self.internalChunkSize : self.options['chunk_size'];
- self.metadata = self.options['metadata'] == null ? self.metadata : self.options['metadata'];
- self.aliases = self.options['aliases'] == null ? self.aliases : self.options['aliases'];
- self.position = 0;
- callback(null, self);
- });
- } else if(self.mode == "w+") {
- nthChunk(self, lastChunkNumber(self), options, function(err, chunk) {
- if(err) return error(err);
- // Set the current chunk
- self.currentChunk = chunk == null ? new Chunk(self, {'n':0}, self.writeConcern) : chunk;
- self.currentChunk.position = self.currentChunk.data.length();
- self.metadata = self.options['metadata'] == null ? self.metadata : self.options['metadata'];
- self.aliases = self.options['aliases'] == null ? self.aliases : self.options['aliases'];
- self.position = self.length;
- callback(null, self);
- });
- }
- }
- // only pass error to callback once
- function error (err) {
- if(error.err) return;
- callback(error.err = err);
- }
- };
- /**
- * @ignore
- */
- var writeBuffer = function(self, buffer, close, callback) {
- if(typeof close === "function") { callback = close; close = null; }
- var finalClose = typeof close == 'boolean' ? close : false;
- if(self.mode != "w") {
- callback(MongoError.create({message: f("file with id %s not opened for writing", (self.referenceBy == REFERENCE_BY_ID ? self.referenceBy : self.filename)), driver:true}), null);
- } else {
- if(self.currentChunk.position + buffer.length >= self.chunkSize) {
- // Write out the current Chunk and then keep writing until we have less data left than a chunkSize left
- // to a new chunk (recursively)
- var previousChunkNumber = self.currentChunk.chunkNumber;
- var leftOverDataSize = self.chunkSize - self.currentChunk.position;
- var firstChunkData = buffer.slice(0, leftOverDataSize);
- var leftOverData = buffer.slice(leftOverDataSize);
- // A list of chunks to write out
- var chunksToWrite = [self.currentChunk.write(firstChunkData)];
- // If we have more data left than the chunk size let's keep writing new chunks
- while(leftOverData.length >= self.chunkSize) {
- // Create a new chunk and write to it
- var newChunk = new Chunk(self, {'n': (previousChunkNumber + 1)}, self.writeConcern);
- firstChunkData = leftOverData.slice(0, self.chunkSize);
- leftOverData = leftOverData.slice(self.chunkSize);
- // Update chunk number
- previousChunkNumber = previousChunkNumber + 1;
- // Write data
- newChunk.write(firstChunkData);
- // Push chunk to save list
- chunksToWrite.push(newChunk);
- }
- // Set current chunk with remaining data
- self.currentChunk = new Chunk(self, {'n': (previousChunkNumber + 1)}, self.writeConcern);
- // If we have left over data write it
- if(leftOverData.length > 0) self.currentChunk.write(leftOverData);
- // Update the position for the gridstore
- self.position = self.position + buffer.length;
- // Total number of chunks to write
- var numberOfChunksToWrite = chunksToWrite.length;
- for(var i = 0; i < chunksToWrite.length; i++) {
- chunksToWrite[i].save({}, function(err) {
- if(err) return callback(err);
- numberOfChunksToWrite = numberOfChunksToWrite - 1;
- if(numberOfChunksToWrite <= 0) {
- // We care closing the file before returning
- if(finalClose) {
- return self.close(function(err) {
- callback(err, self);
- });
- }
- // Return normally
- return callback(null, self);
- }
- });
- }
- } else {
- // Update the position for the gridstore
- self.position = self.position + buffer.length;
- // We have less data than the chunk size just write it and callback
- self.currentChunk.write(buffer);
- // We care closing the file before returning
- if(finalClose) {
- return self.close(function(err) {
- callback(err, self);
- });
- }
- // Return normally
- return callback(null, self);
- }
- }
- };
- /**
- * Creates a mongoDB object representation of this object.
- *
- * <pre><code>
- * {
- * '_id' : , // {number} id for this file
- * 'filename' : , // {string} name for this file
- * 'contentType' : , // {string} mime type for this file
- * 'length' : , // {number} size of this file?
- * 'chunksize' : , // {number} chunk size used by this file
- * 'uploadDate' : , // {Date}
- * 'aliases' : , // {array of string}
- * 'metadata' : , // {string}
- * }
- * </code></pre>
- *
- * @ignore
- */
- var buildMongoObject = function(self, callback) {
- // Calcuate the length
- var mongoObject = {
- '_id': self.fileId,
- 'filename': self.filename,
- 'contentType': self.contentType,
- 'length': self.position ? self.position : 0,
- 'chunkSize': self.chunkSize,
- 'uploadDate': self.uploadDate,
- 'aliases': self.aliases,
- 'metadata': self.metadata
- };
- var md5Command = {filemd5:self.fileId, root:self.root};
- self.db.command(md5Command, function(err, results) {
- if(err) return callback(err);
- mongoObject.md5 = results.md5;
- callback(null, mongoObject);
- });
- };
- /**
- * Gets the nth chunk of this file.
- * @ignore
- */
- var nthChunk = function(self, chunkNumber, options, callback) {
- if(typeof options == 'function') {
- callback = options;
- options = {};
- }
- options = options || self.writeConcern;
- options.readPreference = self.readPreference;
- // Get the nth chunk
- self.chunkCollection().findOne({'files_id':self.fileId, 'n':chunkNumber}, options, function(err, chunk) {
- if(err) return callback(err);
- var finalChunk = chunk == null ? {} : chunk;
- callback(null, new Chunk(self, finalChunk, self.writeConcern));
- });
- };
- /**
- * @ignore
- */
- var lastChunkNumber = function(self) {
- return Math.floor((self.length ? self.length - 1 : 0)/self.chunkSize);
- };
- /**
- * Deletes all the chunks of this file in the database.
- *
- * @ignore
- */
- var deleteChunks = function(self, options, callback) {
- if(typeof options == 'function') {
- callback = options;
- options = {};
- }
- options = options || self.writeConcern;
- if(self.fileId != null) {
- self.chunkCollection().remove({'files_id':self.fileId}, options, function(err) {
- if(err) return callback(err, false);
- callback(null, true);
- });
- } else {
- callback(null, true);
- }
- };
- /**
- * The collection to be used for holding the files and chunks collection.
- *
- * @classconstant DEFAULT_ROOT_COLLECTION
- **/
- GridStore.DEFAULT_ROOT_COLLECTION = 'fs';
- /**
- * Default file mime type
- *
- * @classconstant DEFAULT_CONTENT_TYPE
- **/
- GridStore.DEFAULT_CONTENT_TYPE = 'binary/octet-stream';
- /**
- * Seek mode where the given length is absolute.
- *
- * @classconstant IO_SEEK_SET
- **/
- GridStore.IO_SEEK_SET = 0;
- /**
- * Seek mode where the given length is an offset to the current read/write head.
- *
- * @classconstant IO_SEEK_CUR
- **/
- GridStore.IO_SEEK_CUR = 1;
- /**
- * Seek mode where the given length is an offset to the end of the file.
- *
- * @classconstant IO_SEEK_END
- **/
- GridStore.IO_SEEK_END = 2;
- /**
- * Checks if a file exists in the database.
- *
- * @method
- * @static
- * @param {Db} db the database to query.
- * @param {string} name The name of the file to look for.
- * @param {string} [rootCollection] The root collection that holds the files and chunks collection. Defaults to **{GridStore.DEFAULT_ROOT_COLLECTION}**.
- * @param {object} [options=null] Optional settings.
- * @param {(ReadPreference|string)} [options.readPreference=null] The preferred read preference (ReadPreference.PRIMARY, ReadPreference.PRIMARY_PREFERRED, ReadPreference.SECONDARY, ReadPreference.SECONDARY_PREFERRED, ReadPreference.NEAREST).
- * @param {object} [options.promiseLibrary=null] A Promise library class the application wishes to use such as Bluebird, must be ES6 compatible
- * @param {GridStore~resultCallback} [callback] result from exists.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.exist = function(db, fileIdObject, rootCollection, options, callback) {
- var args = Array.prototype.slice.call(arguments, 2);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- rootCollection = args.length ? args.shift() : null;
- options = args.length ? args.shift() : {};
- options = options || {};
- // Get the promiseLibrary
- var promiseLibrary = options.promiseLibrary;
- // No promise library selected fall back
- if(!promiseLibrary) {
- promiseLibrary = typeof global.Promise == 'function' ?
- global.Promise : require('es6-promise').Promise;
- }
- // We provided a callback leg
- if(typeof callback == 'function') return exists(db, fileIdObject, rootCollection, options, callback);
- // Return promise
- return new promiseLibrary(function(resolve, reject) {
- exists(db, fileIdObject, rootCollection, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var exists = function(db, fileIdObject, rootCollection, options, callback) {
- // Establish read preference
- var readPreference = options.readPreference || ReadPreference.PRIMARY;
- // Fetch collection
- var rootCollectionFinal = rootCollection != null ? rootCollection : GridStore.DEFAULT_ROOT_COLLECTION;
- db.collection(rootCollectionFinal + ".files", function(err, collection) {
- if(err) return callback(err);
- // Build query
- var query = (typeof fileIdObject == 'string' || Object.prototype.toString.call(fileIdObject) == '[object RegExp]' )
- ? {'filename':fileIdObject}
- : {'_id':fileIdObject}; // Attempt to locate file
- // We have a specific query
- if(fileIdObject != null
- && typeof fileIdObject == 'object'
- && Object.prototype.toString.call(fileIdObject) != '[object RegExp]') {
- query = fileIdObject;
- }
- // Check if the entry exists
- collection.findOne(query, {readPreference:readPreference}, function(err, item) {
- if(err) return callback(err);
- callback(null, item == null ? false : true);
- });
- });
- }
- define.staticMethod('exist', {callback: true, promise:true});
- /**
- * Gets the list of files stored in the GridFS.
- *
- * @method
- * @static
- * @param {Db} db the database to query.
- * @param {string} [rootCollection] The root collection that holds the files and chunks collection. Defaults to **{GridStore.DEFAULT_ROOT_COLLECTION}**.
- * @param {object} [options=null] Optional settings.
- * @param {(ReadPreference|string)} [options.readPreference=null] The preferred read preference (ReadPreference.PRIMARY, ReadPreference.PRIMARY_PREFERRED, ReadPreference.SECONDARY, ReadPreference.SECONDARY_PREFERRED, ReadPreference.NEAREST).
- * @param {object} [options.promiseLibrary=null] A Promise library class the application wishes to use such as Bluebird, must be ES6 compatible
- * @param {GridStore~resultCallback} [callback] result from exists.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.list = function(db, rootCollection, options, callback) {
- var args = Array.prototype.slice.call(arguments, 1);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- rootCollection = args.length ? args.shift() : null;
- options = args.length ? args.shift() : {};
- options = options || {};
- // Get the promiseLibrary
- var promiseLibrary = options.promiseLibrary;
- // No promise library selected fall back
- if(!promiseLibrary) {
- promiseLibrary = typeof global.Promise == 'function' ?
- global.Promise : require('es6-promise').Promise;
- }
- // We provided a callback leg
- if(typeof callback == 'function') return list(db, rootCollection, options, callback);
- // Return promise
- return new promiseLibrary(function(resolve, reject) {
- list(db, rootCollection, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var list = function(db, rootCollection, options, callback) {
- // Ensure we have correct values
- if(rootCollection != null && typeof rootCollection == 'object') {
- options = rootCollection;
- rootCollection = null;
- }
- // Establish read preference
- var readPreference = options.readPreference || ReadPreference.PRIMARY;
- // Check if we are returning by id not filename
- var byId = options['id'] != null ? options['id'] : false;
- // Fetch item
- var rootCollectionFinal = rootCollection != null ? rootCollection : GridStore.DEFAULT_ROOT_COLLECTION;
- var items = [];
- db.collection((rootCollectionFinal + ".files"), function(err, collection) {
- if(err) return callback(err);
- collection.find({}, {readPreference:readPreference}, function(err, cursor) {
- if(err) return callback(err);
- cursor.each(function(err, item) {
- if(item != null) {
- items.push(byId ? item._id : item.filename);
- } else {
- callback(err, items);
- }
- });
- });
- });
- }
- define.staticMethod('list', {callback: true, promise:true});
- /**
- * Reads the contents of a file.
- *
- * This method has the following signatures
- *
- * (db, name, callback)
- * (db, name, length, callback)
- * (db, name, length, offset, callback)
- * (db, name, length, offset, options, callback)
- *
- * @method
- * @static
- * @param {Db} db the database to query.
- * @param {string} name The name of the file.
- * @param {number} [length] The size of data to read.
- * @param {number} [offset] The offset from the head of the file of which to start reading from.
- * @param {object} [options=null] Optional settings.
- * @param {(ReadPreference|string)} [options.readPreference=null] The preferred read preference (ReadPreference.PRIMARY, ReadPreference.PRIMARY_PREFERRED, ReadPreference.SECONDARY, ReadPreference.SECONDARY_PREFERRED, ReadPreference.NEAREST).
- * @param {object} [options.promiseLibrary=null] A Promise library class the application wishes to use such as Bluebird, must be ES6 compatible
- * @param {GridStore~readCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.read = function(db, name, length, offset, options, callback) {
- var args = Array.prototype.slice.call(arguments, 2);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- length = args.length ? args.shift() : null;
- offset = args.length ? args.shift() : null;
- options = args.length ? args.shift() : null;
- options = options || {};
- // Get the promiseLibrary
- var promiseLibrary = options ? options.promiseLibrary : null;
- // No promise library selected fall back
- if(!promiseLibrary) {
- promiseLibrary = typeof global.Promise == 'function' ?
- global.Promise : require('es6-promise').Promise;
- }
- // We provided a callback leg
- if(typeof callback == 'function') return readStatic(db, name, length, offset, options, callback);
- // Return promise
- return new promiseLibrary(function(resolve, reject) {
- readStatic(db, name, length, offset, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var readStatic = function(db, name, length, offset, options, callback) {
- new GridStore(db, name, "r", options).open(function(err, gridStore) {
- if(err) return callback(err);
- // Make sure we are not reading out of bounds
- if(offset && offset >= gridStore.length) return callback("offset larger than size of file", null);
- if(length && length > gridStore.length) return callback("length is larger than the size of the file", null);
- if(offset && length && (offset + length) > gridStore.length) return callback("offset and length is larger than the size of the file", null);
- if(offset != null) {
- gridStore.seek(offset, function(err, gridStore) {
- if(err) return callback(err);
- gridStore.read(length, callback);
- });
- } else {
- gridStore.read(length, callback);
- }
- });
- }
- define.staticMethod('read', {callback: true, promise:true});
- /**
- * Read the entire file as a list of strings splitting by the provided separator.
- *
- * @method
- * @static
- * @param {Db} db the database to query.
- * @param {(String|object)} name the name of the file.
- * @param {string} [separator] The character to be recognized as the newline separator.
- * @param {object} [options=null] Optional settings.
- * @param {(ReadPreference|string)} [options.readPreference=null] The preferred read preference (ReadPreference.PRIMARY, ReadPreference.PRIMARY_PREFERRED, ReadPreference.SECONDARY, ReadPreference.SECONDARY_PREFERRED, ReadPreference.NEAREST).
- * @param {object} [options.promiseLibrary=null] A Promise library class the application wishes to use such as Bluebird, must be ES6 compatible
- * @param {GridStore~readlinesCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.readlines = function(db, name, separator, options, callback) {
- var args = Array.prototype.slice.call(arguments, 2);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- separator = args.length ? args.shift() : null;
- options = args.length ? args.shift() : null;
- options = options || {};
- // Get the promiseLibrary
- var promiseLibrary = options ? options.promiseLibrary : null;
- // No promise library selected fall back
- if(!promiseLibrary) {
- promiseLibrary = typeof global.Promise == 'function' ?
- global.Promise : require('es6-promise').Promise;
- }
- // We provided a callback leg
- if(typeof callback == 'function') return readlinesStatic(db, name, separator, options, callback);
- // Return promise
- return new promiseLibrary(function(resolve, reject) {
- readlinesStatic(db, name, separator, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var readlinesStatic = function(db, name, separator, options, callback) {
- var finalSeperator = separator == null ? "\n" : separator;
- new GridStore(db, name, "r", options).open(function(err, gridStore) {
- if(err) return callback(err);
- gridStore.readlines(finalSeperator, callback);
- });
- }
- define.staticMethod('readlines', {callback: true, promise:true});
- /**
- * Deletes the chunks and metadata information of a file from GridFS.
- *
- * @method
- * @static
- * @param {Db} db The database to query.
- * @param {(string|array)} names The name/names of the files to delete.
- * @param {object} [options=null] Optional settings.
- * @param {object} [options.promiseLibrary=null] A Promise library class the application wishes to use such as Bluebird, must be ES6 compatible
- * @param {GridStore~resultCallback} [callback] the command callback.
- * @return {Promise} returns Promise if no callback passed
- * @deprecated Use GridFSBucket API instead
- */
- GridStore.unlink = function(db, names, 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() : {};
- options = options || {};
- // Get the promiseLibrary
- var promiseLibrary = options.promiseLibrary;
- // No promise library selected fall back
- if(!promiseLibrary) {
- promiseLibrary = typeof global.Promise == 'function' ?
- global.Promise : require('es6-promise').Promise;
- }
- // We provided a callback leg
- if(typeof callback == 'function') return unlinkStatic(self, db, names, options, callback);
- // Return promise
- return new promiseLibrary(function(resolve, reject) {
- unlinkStatic(self, db, names, options, function(err, r) {
- if(err) return reject(err);
- resolve(r);
- })
- });
- };
- var unlinkStatic = function(self, db, names, options, callback) {
- // Get the write concern
- var writeConcern = _getWriteConcern(db, options);
- // List of names
- if(names.constructor == Array) {
- var tc = 0;
- for(var i = 0; i < names.length; i++) {
- ++tc;
- GridStore.unlink(db, names[i], options, function() {
- if(--tc == 0) {
- callback(null, self);
- }
- });
- }
- } else {
- new GridStore(db, names, "w", options).open(function(err, gridStore) {
- if(err) return callback(err);
- deleteChunks(gridStore, function(err) {
- if(err) return callback(err);
- gridStore.collection(function(err, collection) {
- if(err) return callback(err);
- collection.remove({'_id':gridStore.fileId}, writeConcern, function(err) {
- callback(err, self);
- });
- });
- });
- });
- }
- }
- define.staticMethod('unlink', {callback: true, promise:true});
- /**
- * @ignore
- */
- var _writeNormal = function(self, data, close, callback) {
- // If we have a buffer write it using the writeBuffer method
- if(Buffer.isBuffer(data)) {
- return writeBuffer(self, data, close, callback);
- } else {
- return writeBuffer(self, new Buffer(data, 'binary'), close, callback);
- }
- }
- /**
- * @ignore
- */
- var _setWriteConcernHash = function(options) {
- var finalOptions = {};
- if(options.w != null) finalOptions.w = options.w;
- if(options.journal == true) finalOptions.j = options.journal;
- if(options.j == true) finalOptions.j = options.j;
- if(options.fsync == true) finalOptions.fsync = options.fsync;
- if(options.wtimeout != null) finalOptions.wtimeout = options.wtimeout;
- return finalOptions;
- }
- /**
- * @ignore
- */
- var _getWriteConcern = function(self, options) {
- // Final options
- var finalOptions = {w:1};
- options = options || {};
- // Local options verification
- if(options.w != null || typeof options.j == 'boolean' || typeof options.journal == 'boolean' || typeof options.fsync == 'boolean') {
- finalOptions = _setWriteConcernHash(options);
- } else if(options.safe != null && typeof options.safe == 'object') {
- finalOptions = _setWriteConcernHash(options.safe);
- } else if(typeof options.safe == "boolean") {
- finalOptions = {w: (options.safe ? 1 : 0)};
- } else if(self.options.w != null || typeof self.options.j == 'boolean' || typeof self.options.journal == 'boolean' || typeof self.options.fsync == 'boolean') {
- finalOptions = _setWriteConcernHash(self.options);
- } else if(self.safe && (self.safe.w != null || typeof self.safe.j == 'boolean' || typeof self.safe.journal == 'boolean' || typeof self.safe.fsync == 'boolean')) {
- finalOptions = _setWriteConcernHash(self.safe);
- } else if(typeof self.safe == "boolean") {
- finalOptions = {w: (self.safe ? 1 : 0)};
- }
- // Ensure we don't have an invalid combination of write concerns
- if(finalOptions.w < 1
- && (finalOptions.journal == true || finalOptions.j == true || finalOptions.fsync == true)) throw MongoError.create({message: "No acknowledgement using w < 1 cannot be combined with journal:true or fsync:true", driver:true});
- // Return the options
- return finalOptions;
- }
- /**
- * Create a new GridStoreStream instance (INTERNAL TYPE, do not instantiate directly)
- *
- * @class
- * @extends external:Duplex
- * @return {GridStoreStream} a GridStoreStream instance.
- * @deprecated Use GridFSBucket API instead
- */
- var GridStoreStream = function(gs) {
- // Initialize the duplex stream
- Duplex.call(this);
- // Get the gridstore
- this.gs = gs;
- // End called
- this.endCalled = false;
- // If we have a seek
- this.totalBytesToRead = this.gs.length - this.gs.position;
- this.seekPosition = this.gs.position;
- }
- //
- // Inherit duplex
- inherits(GridStoreStream, Duplex);
- GridStoreStream.prototype._pipe = GridStoreStream.prototype.pipe;
- // Set up override
- GridStoreStream.prototype.pipe = function(destination) {
- var self = this;
- // Only open gridstore if not already open
- if(!self.gs.isOpen) {
- self.gs.open(function(err) {
- if(err) return self.emit('error', err);
- self.totalBytesToRead = self.gs.length - self.gs.position;
- self._pipe.apply(self, [destination]);
- });
- } else {
- self.totalBytesToRead = self.gs.length - self.gs.position;
- self._pipe.apply(self, [destination]);
- }
- return destination;
- }
- // Called by stream
- GridStoreStream.prototype._read = function() {
- var self = this;
- var read = function() {
- // Read data
- self.gs.read(length, function(err, buffer) {
- if(err && !self.endCalled) return self.emit('error', err);
- // Stream is closed
- if(self.endCalled || buffer == null) return self.push(null);
- // Remove bytes read
- if(buffer.length <= self.totalBytesToRead) {
- self.totalBytesToRead = self.totalBytesToRead - buffer.length;
- self.push(buffer);
- } else if(buffer.length > self.totalBytesToRead) {
- self.totalBytesToRead = self.totalBytesToRead - buffer._index;
- self.push(buffer.slice(0, buffer._index));
- }
- // Finished reading
- if(self.totalBytesToRead <= 0) {
- self.endCalled = true;
- }
- });
- }
- // Set read length
- var length = self.gs.length < self.gs.chunkSize ? self.gs.length - self.seekPosition : self.gs.chunkSize;
- if(!self.gs.isOpen) {
- self.gs.open(function(err) {
- self.totalBytesToRead = self.gs.length - self.gs.position;
- if(err) return self.emit('error', err);
- read();
- });
- } else {
- read();
- }
- }
- GridStoreStream.prototype.destroy = function() {
- this.pause();
- this.endCalled = true;
- this.gs.close();
- this.emit('end');
- }
- GridStoreStream.prototype.write = function(chunk) {
- var self = this;
- if(self.endCalled) return self.emit('error', MongoError.create({message: 'attempting to write to stream after end called', driver:true}))
- // Do we have to open the gridstore
- if(!self.gs.isOpen) {
- self.gs.open(function() {
- self.gs.isOpen = true;
- self.gs.write(chunk, function() {
- process.nextTick(function() {
- self.emit('drain');
- });
- });
- });
- return false;
- } else {
- self.gs.write(chunk, function() {
- self.emit('drain');
- });
- return true;
- }
- }
- GridStoreStream.prototype.end = function(chunk, encoding, callback) {
- var self = this;
- var args = Array.prototype.slice.call(arguments, 0);
- callback = args.pop();
- if(typeof callback != 'function') args.push(callback);
- chunk = args.length ? args.shift() : null;
- encoding = args.length ? args.shift() : null;
- self.endCalled = true;
- if(chunk) {
- self.gs.write(chunk, function() {
- self.gs.close(function() {
- if(typeof callback == 'function') callback();
- self.emit('end')
- });
- });
- }
- self.gs.close(function() {
- if(typeof callback == 'function') callback();
- self.emit('end')
- });
- }
- /**
- * The read() method pulls some data out of the internal buffer and returns it. If there is no data available, then it will return null.
- * @function external:Duplex#read
- * @param {number} size Optional argument to specify how much data to read.
- * @return {(String | Buffer | null)}
- */
- /**
- * Call this function to cause the stream to return strings of the specified encoding instead of Buffer objects.
- * @function external:Duplex#setEncoding
- * @param {string} encoding The encoding to use.
- * @return {null}
- */
- /**
- * This method will cause the readable stream to resume emitting data events.
- * @function external:Duplex#resume
- * @return {null}
- */
- /**
- * This method will cause a stream in flowing-mode to stop emitting data events. Any data that becomes available will remain in the internal buffer.
- * @function external:Duplex#pause
- * @return {null}
- */
- /**
- * This method pulls all the data out of a readable stream, and writes it to the supplied destination, automatically managing the flow so that the destination is not overwhelmed by a fast readable stream.
- * @function external:Duplex#pipe
- * @param {Writable} destination The destination for writing data
- * @param {object} [options] Pipe options
- * @return {null}
- */
- /**
- * This method will remove the hooks set up for a previous pipe() call.
- * @function external:Duplex#unpipe
- * @param {Writable} [destination] The destination for writing data
- * @return {null}
- */
- /**
- * This is useful in certain cases where a stream is being consumed by a parser, which needs to "un-consume" some data that it has optimistically pulled out of the source, so that the stream can be passed on to some other party.
- * @function external:Duplex#unshift
- * @param {(Buffer|string)} chunk Chunk of data to unshift onto the read queue.
- * @return {null}
- */
- /**
- * Versions of Node prior to v0.10 had streams that did not implement the entire Streams API as it is today. (See "Compatibility" below for more information.)
- * @function external:Duplex#wrap
- * @param {Stream} stream An "old style" readable stream.
- * @return {null}
- */
- /**
- * This method writes some data to the underlying system, and calls the supplied callback once the data has been fully handled.
- * @function external:Duplex#write
- * @param {(string|Buffer)} chunk The data to write
- * @param {string} encoding The encoding, if chunk is a String
- * @param {function} callback Callback for when this chunk of data is flushed
- * @return {boolean}
- */
- /**
- * Call this method when no more data will be written to the stream. If supplied, the callback is attached as a listener on the finish event.
- * @function external:Duplex#end
- * @param {(string|Buffer)} chunk The data to write
- * @param {string} encoding The encoding, if chunk is a String
- * @param {function} callback Callback for when this chunk of data is flushed
- * @return {null}
- */
- /**
- * GridStoreStream stream data event, fired for each document in the cursor.
- *
- * @event GridStoreStream#data
- * @type {object}
- */
- /**
- * GridStoreStream stream end event
- *
- * @event GridStoreStream#end
- * @type {null}
- */
- /**
- * GridStoreStream stream close event
- *
- * @event GridStoreStream#close
- * @type {null}
- */
- /**
- * GridStoreStream stream readable event
- *
- * @event GridStoreStream#readable
- * @type {null}
- */
- /**
- * GridStoreStream stream drain event
- *
- * @event GridStoreStream#drain
- * @type {null}
- */
- /**
- * GridStoreStream stream finish event
- *
- * @event GridStoreStream#finish
- * @type {null}
- */
- /**
- * GridStoreStream stream pipe event
- *
- * @event GridStoreStream#pipe
- * @type {null}
- */
- /**
- * GridStoreStream stream unpipe event
- *
- * @event GridStoreStream#unpipe
- * @type {null}
- */
- /**
- * GridStoreStream stream error event
- *
- * @event GridStoreStream#error
- * @type {null}
- */
- /**
- * @ignore
- */
- module.exports = GridStore;
|