// Copyright IBM Corp. 2013,2016. All Rights Reserved. // Node module: loopback-datasource-juggler // This file is licensed under the MIT License. // License text available at https://opensource.org/licenses/MIT // Turning on strict for this file breaks lots of test cases; // disabling strict for this file /* eslint-disable strict */ /*! * Module dependencies */ var ModelBuilder = require('./model-builder.js').ModelBuilder; var ModelDefinition = require('./model-definition.js'); var RelationDefinition = require('./relation-definition.js'); var OberserverMixin = require('./observer'); var jutil = require('./jutil'); var utils = require('./utils'); var ModelBaseClass = require('./model.js'); var DataAccessObject = require('./dao.js'); var defineScope = require('./scope.js').defineScope; var EventEmitter = require('events').EventEmitter; var util = require('util'); var assert = require('assert'); var async = require('async'); var traverse = require('traverse'); var g = require('strong-globalize')(); var juggler = require('..'); if (process.env.DEBUG === 'loopback') { // For back-compatibility process.env.DEBUG = 'loopback:*'; } var debug = require('debug')('loopback:datasource'); /*! * Export public API */ exports.DataSource = DataSource; /*! * Helpers */ var slice = Array.prototype.slice; /** * LoopBack models can manipulate data via the DataSource object. * Attaching a `DataSource` to a `Model` adds instance methods and static methods to the `Model`. * * Define a data source to persist model data. * To create a DataSource programmatically, call `createDataSource()` on the LoopBack object; for example: * ```js * var oracle = loopback.createDataSource({ * connector: 'oracle', * host: '111.22.333.44', * database: 'MYDB', * username: 'username', * password: 'password' * }); * ``` * * All classes in single dataSource share same the connector type and * one database connection. * * For example, the following creates a DataSource, and waits for a connection callback. * * ``` * var dataSource = new DataSource('mysql', { database: 'myapp_test' }); * dataSource.define(...); * dataSource.on('connected', function () { * // work with database * }); * ``` * @class DataSource * @param {String} [name] Optional name for datasource. * @options {Object} settings Database-specific settings to establish connection (settings depend on specific connector). * The table below lists a typical set for a relational database. * @property {String} connector Database connector to use. For any supported connector, can be any of: * * - The connector module from `require(connectorName)`. * - The full name of the connector module, such as 'loopback-connector-oracle'. * - The short name of the connector module, such as 'oracle'. * - A local module under `./connectors/` folder. * @property {String} host Database server host name. * @property {String} port Database server port number. * @property {String} username Database user name. * @property {String} password Database password. * @property {String} database Name of the database to use. * @property {Boolean} debug Display debugging information. Default is false. */ function DataSource(name, settings, modelBuilder) { if (!(this instanceof DataSource)) { return new DataSource(name, settings); } // Check if the settings object is passed as the first argument if (typeof name === 'object' && settings === undefined) { settings = name; name = undefined; } // Check if the first argument is a URL if (typeof name === 'string' && name.indexOf('://') !== -1) { name = utils.parseSettings(name); } // Check if the settings is in the form of URL string if (typeof settings === 'string' && settings.indexOf('://') !== -1) { settings = utils.parseSettings(settings); } this.modelBuilder = modelBuilder || new ModelBuilder(); this.models = this.modelBuilder.models; this.definitions = this.modelBuilder.definitions; this.juggler = juggler; // operation metadata // Initialize it before calling setup as the connector might register operations this._operations = {}; this.setup(name, settings); this._setupConnector(); // connector var connector = this.connector; // DataAccessObject - connector defined or supply the default var dao = (connector && connector.DataAccessObject) || this.constructor.DataAccessObject; this.DataAccessObject = function() { }; // define DataAccessObject methods Object.keys(dao).forEach(function(name) { var fn = dao[name]; this.DataAccessObject[name] = fn; if (typeof fn === 'function') { this.defineOperation(name, { accepts: fn.accepts, 'returns': fn.returns, http: fn.http, remoteEnabled: fn.shared ? true : false, scope: this.DataAccessObject, fnName: name, }); } }.bind(this)); // define DataAccessObject.prototype methods Object.keys(dao.prototype).forEach(function(name) { var fn = dao.prototype[name]; this.DataAccessObject.prototype[name] = fn; if (typeof fn === 'function') { this.defineOperation(name, { prototype: true, accepts: fn.accepts, 'returns': fn.returns, http: fn.http, remoteEnabled: fn.shared ? true : false, scope: this.DataAccessObject.prototype, fnName: name, }); } }.bind(this)); } util.inherits(DataSource, EventEmitter); // allow child classes to supply a data access object DataSource.DataAccessObject = DataAccessObject; /** * Set up the connector instance for backward compatibility with JugglingDB schema/adapter * @private */ DataSource.prototype._setupConnector = function() { this.connector = this.connector || this.adapter; // The legacy JugglingDB adapter will set up `adapter` property this.adapter = this.connector; // Keep the adapter as an alias to connector if (this.connector) { if (!this.connector.dataSource) { // Set up the dataSource if the connector doesn't do so this.connector.dataSource = this; } var dataSource = this; this.connector.log = function(query, start) { dataSource.log(query, start); }; this.connector.logger = function(query) { var t1 = Date.now(); var log = this.log; return function(q) { log(q || query, t1); }; }; // Configure the connector instance to mix in observer functions jutil.mixin(this.connector, OberserverMixin); } }; // List possible connector module names function connectorModuleNames(name) { var names = []; // Check the name as is if (!name.match(/^\//)) { names.push('./connectors/' + name); // Check built-in connectors if (name.indexOf('loopback-connector-') !== 0) { names.push('loopback-connector-' + name); // Try loopback-connector- } } // Only try the short name if the connector is not from StrongLoop if (['mongodb', 'oracle', 'mysql', 'postgresql', 'mssql', 'rest', 'soap', 'db2', 'cloudant'] .indexOf(name) === -1) { names.push(name); } return names; } // testable with DI function tryModules(names, loader) { var mod; loader = loader || require; for (var m = 0; m < names.length; m++) { try { mod = loader(names[m]); } catch (e) { var notFound = e.code === 'MODULE_NOT_FOUND' && e.message && e.message.indexOf(names[m]) > 0; if (notFound) { debug('Module %s not found, will try another candidate.', names[m]); continue; } debug('Cannot load connector %s: %s', names[m], e.stack || e); throw e; } if (mod) { break; } } return mod; } /*! * Resolve a connector by name * @param name The connector name * @returns {*} * @private */ DataSource._resolveConnector = function(name, loader) { var names = connectorModuleNames(name); var connector = tryModules(names, loader); var error = null; if (!connector) { error = g.f('\nWARNING: {{LoopBack}} connector "%s" is not installed ' + 'as any of the following modules:\n\n %s\n\nTo fix, run:\n\n {{npm install %s --save}}\n', name, names.join('\n'), names[names.length - 1]); } return { connector: connector, error: error, }; }; /** * Connect to the data source. * If no callback is provided, it will return a Promise. * Emits the 'connect' event. * @param callback * @returns {Promise} * @emits connected */ DataSource.prototype.connect = function(callback) { callback = callback || utils.createPromiseCallback(); var self = this; if (this.connected) { // The data source is already connected, return immediately process.nextTick(callback); return callback.promise; } if (typeof this.connector.connect !== 'function') { // Connector doesn't have the connect function // Assume no connect is needed self.connected = true; self.connecting = false; process.nextTick(function() { self.emit('connected'); callback(); }); return callback.promise; } // Queue the callback this.pendingConnectCallbacks = this.pendingConnectCallbacks || []; this.pendingConnectCallbacks.push(callback); // The connect is already in progress if (this.connecting) return callback.promise; // Set connecting flag to be true this.connecting = true; this.connector.connect(function(err, result) { self.connecting = false; if (!err) self.connected = true; var cbs = self.pendingConnectCallbacks; self.pendingConnectCallbacks = []; if (!err) { self.emit('connected'); } else { self.emit('error', err); } // Invoke all pending callbacks async.each(cbs, function(cb, done) { try { cb(err); } catch (e) { // Ignore error to make sure all callbacks are invoked debug('Uncaught error raised by connect callback function: ', e); } finally { done(); } }, function(err) { if (err) throw err; // It should not happen }); }); return callback.promise; }; /** * Set up the data source * @param {String} name The name * @param {Object} settings The settings * @returns {*} * @private */ DataSource.prototype.setup = function(name, settings) { var dataSource = this; var connector; // support single settings object if (name && typeof name === 'object' && !settings) { settings = name; name = undefined; } if (typeof settings === 'object') { if (settings.initialize) { connector = settings; } else if (settings.connector) { connector = settings.connector; } else if (settings.adapter) { connector = settings.adapter; } } // just save everything we get this.settings = settings || {}; this.settings.debug = this.settings.debug || debug.enabled; if (this.settings.debug) { debug('Settings: %j', this.settings); } // Disconnected by default this.connected = false; this.connecting = false; if (typeof connector === 'string') { name = connector; connector = undefined; } name = name || (connector && connector.name); this.name = name; if (name && !connector) { if (typeof name === 'object') { // The first argument might be the connector itself connector = name; this.name = connector.name; } else { // The connector has not been resolved var result = DataSource._resolveConnector(name); connector = result.connector; if (!connector) { console.error(result.error); this.emit('error', new Error(result.error)); return; } } } if (connector) { var postInit = function postInit(err, result) { this._setupConnector(); // we have an connector now? if (!this.connector) { throw new Error(g.f('Connector is not defined correctly: ' + 'it should create `{{connector}}` member of dataSource')); } this.connected = !err; // Connected now if (this.connected) { this.emit('connected'); } else { // The connection fails, let's report it and hope it will be recovered in the next call g.error('Connection fails: %s\nIt will be retried for the next request.', err); this.emit('error', err); this.connecting = false; } }.bind(this); try { if ('function' === typeof connector.initialize) { // Call the async initialize method connector.initialize(this, postInit); } else if ('function' === typeof connector) { // Use the connector constructor directly this.connector = new connector(this.settings); postInit(); } } catch (err) { if (err.message) { err.message = 'Cannot initialize connector ' + JSON.stringify(connector.name || name) + ': ' + err.message; } throw err; } } }; function isModelClass(cls) { if (!cls) { return false; } return cls.prototype instanceof ModelBaseClass; } DataSource.relationTypes = Object.keys(RelationDefinition.RelationTypes); function isModelDataSourceAttached(model) { return model && (!model.settings.unresolved) && (model.dataSource instanceof DataSource); } /*! * Define scopes for the model class from the scopes object. See * [scopes](./Model-definition-JSON-file.html#scopes) for more information on * scopes and valid options objects. * @param {Object} modelClass - The model class that corresponds to the model * definition that will be enhanced by the provided scopes. * @param {Object} scopes A key-value collection of names and their object * definitions * @property options The options defined on the scope object. */ DataSource.prototype.defineScopes = function(modelClass, scopes) { if (scopes) { for (var s in scopes) { defineScope(modelClass, modelClass, s, scopes[s], {}, scopes[s].options); } } }; /*! * Define relations for the model class from the relations object. See * [relations](./Model-definition-JSON-file.html#relations) for more information. * @param {Object} modelClass - The model class that corresponds to the model * definition that will be enhanced by the provided relations. * @param {Object} relations A key-value collection of relation names and their * object definitions. */ DataSource.prototype.defineRelations = function(modelClass, relations) { var self = this; // Create a function for the closure in the loop var createListener = function(name, relation, targetModel, throughModel) { if (!isModelDataSourceAttached(targetModel)) { targetModel.once('dataAccessConfigured', function(model) { // Check if the through model doesn't exist or resolved if (!throughModel || isModelDataSourceAttached(throughModel)) { // The target model is resolved var params = traverse(relation).clone(); params.as = name; params.model = model; if (throughModel) { params.through = throughModel; } modelClass[relation.type].call(modelClass, name, params); } }); } if (throughModel && !isModelDataSourceAttached(throughModel)) { // Set up a listener to the through model throughModel.once('dataAccessConfigured', function(model) { if (isModelDataSourceAttached(targetModel)) { // The target model is resolved var params = traverse(relation).clone(); params.as = name; params.model = targetModel; params.through = model; modelClass[relation.type].call(modelClass, name, params); } }); } }; // Set up the relations if (relations) { Object.keys(relations).forEach(function(rn) { var r = relations[rn]; assert(DataSource.relationTypes.indexOf(r.type) !== -1, 'Invalid relation type: ' + r.type); assert(isValidRelationName(rn), 'Invalid relation name: ' + rn); var targetModel, polymorphicName; if (r.polymorphic && r.type !== 'belongsTo' && !r.model) { throw new Error(g.f('No model specified for {{polymorphic}} %s: %s', r.type, rn)); } if (r.polymorphic) { polymorphicName = typeof r.model === 'string' ? r.model : rn; if (typeof r.polymorphic === 'string') { polymorphicName = r.polymorphic; } else if (typeof r.polymorphic === 'object' && typeof r.polymorphic.as === 'string') { polymorphicName = r.polymorphic.as; } } if (r.model) { targetModel = isModelClass(r.model) ? r.model : self.getModel(r.model, true); } var throughModel = null; if (r.through) { throughModel = isModelClass(r.through) ? r.through : self.getModel(r.through, true); } if ((targetModel && !isModelDataSourceAttached(targetModel)) || (throughModel && !isModelDataSourceAttached(throughModel))) { // Create a listener to defer the relation set up createListener(rn, r, targetModel, throughModel); } else { // The target model is resolved var params = traverse(r).clone(); params.as = rn; params.model = polymorphicName || targetModel; if (throughModel) { params.through = throughModel; } modelClass[r.type].call(modelClass, rn, params); } }); } }; function isValidRelationName(relationName) { var invalidRelationNames = ['trigger']; return invalidRelationNames.indexOf(relationName) === -1; } /*! * Set up the data access functions from the data source. Each data source will * expose a data access object (DAO), which will be mixed into the modelClass. * @param {Model} modelClass The model class that will receive DAO mixins. * @param {Object} settings The settings object; typically allows any settings * that would be valid for a typical Model object. */ DataSource.prototype.setupDataAccess = function(modelClass, settings) { if (this.connector) { // Check if the id property should be generated var idName = modelClass.definition.idName(); var idProp = modelClass.definition.rawProperties[idName]; if (idProp && idProp.generated && this.connector.getDefaultIdType) { // Set the default id type from connector's ability var idType = this.connector.getDefaultIdType() || String; idProp.type = idType; modelClass.definition.rawProperties[idName].type = idType; modelClass.definition.properties[idName].type = idType; var forceId = settings.forceId; if (idProp.generated && forceId !== false) { forceId = true; } if (forceId) { modelClass.validatesAbsenceOf(idName, {if: 'isNewRecord'}); } } if (this.connector.define) { // pass control to connector this.connector.define({ model: modelClass, properties: modelClass.definition.properties, settings: settings, }); } } // add data access objects this.mixin(modelClass); // define relations from LDL (options.relations) var relations = settings.relationships || settings.relations; this.defineRelations(modelClass, relations); // Emit the dataAccessConfigured event to indicate all the methods for data // access have been mixed into the model class modelClass.emit('dataAccessConfigured', modelClass); // define scopes from LDL (options.relations) var scopes = settings.scopes || {}; this.defineScopes(modelClass, scopes); }; /** * Define a model class. Returns newly created model object. * The first (String) argument specifying the model name is required. * You can provide one or two JSON object arguments, to provide configuration options. * See [Model definition reference](http://docs.strongloop.com/display/DOC/Model+definition+reference) for details. * * Simple example: * ``` * var User = dataSource.createModel('User', { * email: String, * password: String, * birthDate: Date, * activated: Boolean * }); * ``` * More advanced example * ``` * var User = dataSource.createModel('User', { * email: { type: String, limit: 150, index: true }, * password: { type: String, limit: 50 }, * birthDate: Date, * registrationDate: {type: Date, default: function () { return new Date }}, * activated: { type: Boolean, default: false } * }); * ``` * You can also define an ACL when you create a new data source with the `DataSource.create()` method. For example: * * ```js * var Customer = ds.createModel('Customer', { * name: { * type: String, * acls: [ * {principalType: ACL.USER, principalId: 'u001', accessType: ACL.WRITE, permission: ACL.DENY}, * {principalType: ACL.USER, principalId: 'u001', accessType: ACL.ALL, permission: ACL.ALLOW} * ] * } * }, { * acls: [ * {principalType: ACL.USER, principalId: 'u001', accessType: ACL.ALL, permission: ACL.ALLOW} * ] * }); * ``` * * @param {String} className Name of the model to create. * @param {Object} properties Hash of model properties in format `{property: Type, property2: Type2, ...}` or `{property: {type: Type}, property2: {type: Type2}, ...}` * @options {Object} properties Other configuration options. This corresponds to the options key in the config object. * @param {Object} settings A settings object that would typically be used for Model objects. */ DataSource.prototype.createModel = DataSource.prototype.define = function defineClass(className, properties, settings) { var args = slice.call(arguments); if (!className) { throw new Error(g.f('Class name required')); } if (args.length === 1) { properties = {}; args.push(properties); } if (args.length === 2) { settings = {}; args.push(settings); } properties = properties || {}; settings = settings || {}; if (this.isRelational()) { // Set the strict mode to be true for relational DBs by default if (settings.strict === undefined || settings.strict === null) { settings.strict = true; } if (settings.strict === false) { settings.strict = 'throw'; } } var modelClass = this.modelBuilder.define(className, properties, settings); modelClass.dataSource = this; if (settings.unresolved) { return modelClass; } this.setupDataAccess(modelClass, settings); modelClass.emit('dataSourceAttached', modelClass); return modelClass; }; /** * Mixin DataAccessObject methods. * * @param {Function} ModelCtor The model constructor * @private */ DataSource.prototype.mixin = function(ModelCtor) { var ops = this.operations(); var DAO = this.DataAccessObject; // mixin DAO jutil.mixin(ModelCtor, DAO, {proxyFunctions: true, override: true}); // decorate operations as alias functions Object.keys(ops).forEach(function(name) { var op = ops[name]; var scope; if (op.enabled) { scope = op.prototype ? ModelCtor.prototype : ModelCtor; // var sfn = scope[name] = function () { // op.scope[op.fnName].apply(self, arguments); // } Object.keys(op) .filter(function(key) { // filter out the following keys return ~[ 'scope', 'fnName', 'prototype', ].indexOf(key); }) .forEach(function(key) { if (typeof op[key] !== 'undefined') { op.scope[op.fnName][key] = op[key]; } }); } }); }; /*! Method will be deprecated in LoopBack.next */ /** * See [ModelBuilder.getModel](http://apidocs.strongloop.com/loopback-datasource-juggler/#modelbuilder-prototype-getmodel) * for details. */ DataSource.prototype.getModel = function(name, forceCreate) { return this.modelBuilder.getModel(name, forceCreate); }; /*! Method will be deprecated in LoopBack.next */ /** * See ModelBuilder.getModelDefinition * See [ModelBuilder.getModelDefinition](http://apidocs.strongloop.com/loopback-datasource-juggler/#modelbuilder-prototype-getmodeldefinition) * for details. */ DataSource.prototype.getModelDefinition = function(name) { return this.modelBuilder.getModelDefinition(name); }; /*! Method will be deprecated in LoopBack.next */ /** * Get the data source types collection. * @returns {String[]} The data source type array. * For example, ['db', 'nosql', 'mongodb'] would be represent a datasource of * type 'db', with a subtype of 'nosql', and would use the 'mongodb' connector. * * Alternatively, ['rest'] would be a different type altogether, and would have * no subtype. */ DataSource.prototype.getTypes = function() { var getTypes = this.connector && this.connector.getTypes; var types = getTypes && getTypes() || []; if (typeof types === 'string') { types = types.split(/[\s,\/]+/); } return types; }; /** * Check the data source supports the specified types. * @param {String|String[]} types Type name or an array of type names. * @returns {Boolean} true if all types are supported by the data source */ DataSource.prototype.supportTypes = function(types) { var supportedTypes = this.getTypes(); if (Array.isArray(types)) { // Check each of the types for (var i = 0; i < types.length; i++) { if (supportedTypes.indexOf(types[i]) === -1) { // Not supported return false; } } return true; } else { // The types is a string return supportedTypes.indexOf(types) !== -1; } }; /*! In future versions, this will not maintain a strict 1:1 relationship between datasources and model classes * Moving forward, we will allow a model to be attached to a datasource. The model itself becomes a template. */ /** * Attach an existing model to a data source. * This will mixin all of the data access object functions (DAO) into your * modelClass definition. * @param {Function} modelClass The model constructor that will be enhanced by * DAO mixins. */ DataSource.prototype.attach = function(modelClass) { if (modelClass.dataSource === this) { // Already attached to the data source return modelClass; } if (modelClass.modelBuilder !== this.modelBuilder) { this.modelBuilder.definitions[modelClass.modelName] = modelClass.definition; this.modelBuilder.models[modelClass.modelName] = modelClass; // reset the modelBuilder modelClass.modelBuilder = this.modelBuilder; } // redefine the dataSource modelClass.dataSource = this; this.setupDataAccess(modelClass, modelClass.settings); modelClass.emit('dataSourceAttached', modelClass); return modelClass; }; /*! Method will be deprecated in LoopBack.next */ /** * Define a property with name `prop` on a target `model`. See * [Properties](./Model-definition-JSON-file.html#properties) for more information * regarding valid options for `params`. * @param {String} model Name of model * @param {String} prop Name of property * @param {Property} params Property settings */ DataSource.prototype.defineProperty = function(model, prop, params) { this.modelBuilder.defineProperty(model, prop, params); var resolvedProp = this.getModelDefinition(model).properties[prop]; if (this.connector && this.connector.defineProperty) { this.connector.defineProperty(model, prop, resolvedProp); } }; /** * Drop schema objects such as tables, indexes, views, triggers, etc that correspond * to model definitions attached to this DataSource instance, specified by the `models` parameter. * * **WARNING**: In many situations, this will destroy data! `autoupdate()` will attempt to preserve * data while updating the schema on your target DataSource, but this is not guaranteed to be safe. * * Please check the documentation for your specific connector(s) for a detailed breakdown of * behaviors for automigrate! * * @param {String|String[]} [models] Model(s) to migrate. If not present, apply to all models. * @param {Function} [callback] Callback function. Optional. * */ DataSource.prototype.automigrate = function(models, cb) { this.freeze(); if ((!cb) && ('function' === typeof models)) { cb = models; models = undefined; } cb = cb || utils.createPromiseCallback(); if (!this.connector.automigrate) { // NOOP process.nextTick(cb); return cb.promise; } // First argument is a model name if ('string' === typeof models) { models = [models]; } var attachedModels = this.connector._models; if (attachedModels && typeof attachedModels === 'object') { models = models || Object.keys(attachedModels); if (models.length === 0) { process.nextTick(cb); return cb.promise; } var invalidModels = models.filter(function(m) { return !(m in attachedModels); }); if (invalidModels.length) { process.nextTick(function() { cb(new Error(g.f('Cannot migrate models not attached to this datasource: %s', invalidModels.join(' ')))); }); return cb.promise; } } this.connector.automigrate(models, cb); return cb.promise; }; /** * Update existing database tables. * This method applies only to database connectors. * * **WARNING**: `autoupdate()` will attempt to preserve data while updating the * schema on your target DataSource, but this is not guaranteed to be safe. * * Please check the documentation for your specific connector(s) for a detailed breakdown of * behaviors for automigrate!* * * @param {String|String[]} [models] Model(s) to migrate. If not present, apply to all models. * @param {Function} [cb] The callback function */ DataSource.prototype.autoupdate = function(models, cb) { this.freeze(); if ((!cb) && ('function' === typeof models)) { cb = models; models = undefined; } cb = cb || utils.createPromiseCallback(); if (!this.connector.autoupdate) { // NOOP process.nextTick(cb); return cb.promise; } // First argument is a model name if ('string' === typeof models) { models = [models]; } var attachedModels = this.connector._models; if (attachedModels && typeof attachedModels === 'object') { models = models || Object.keys(attachedModels); if (models.length === 0) { process.nextTick(cb); return cb.promise; } var invalidModels = models.filter(function(m) { return !(m in attachedModels); }); if (invalidModels.length) { process.nextTick(function() { cb(new Error(g.f('Cannot migrate models not attached to this datasource: %s', invalidModels.join(' ')))); }); return cb.promise; } } this.connector.autoupdate(models, cb); return cb.promise; }; /** * Discover existing database tables. * This method returns an array of model objects, including {type, name, onwer} * * @param {Object} options The options * @param {Function} Callback function. Optional. * @options {Object} options Discovery options. See below. * @property {String} owner/schema The owner or schema to discover from. * @property {Boolean} all If true, discover all models; if false, discover only models owned by the current user. * @property {Boolean} views If true, include views; if false, only tables. * @property {Number} limit Page size * @property {Number} offset Starting index * @returns {ModelDefinition[]} */ DataSource.prototype.discoverModelDefinitions = function(options, cb) { this.freeze(); if (cb === undefined && typeof options === 'function') { cb = options; options = {}; } options = options || {}; cb = cb || utils.createPromiseCallback(); if (this.connector.discoverModelDefinitions) { this.connector.discoverModelDefinitions(options, cb); } else if (cb) { process.nextTick(cb); } return cb.promise; }; /*! Method will be completely removed in LoopBack.next */ /** * The synchronous version of discoverModelDefinitions. * @options {Object} options The options * @property {Boolean} all If true, discover all models; if false, discover only models owned by the current user. * @property {Boolean} views If true, nclude views; if false, only tables. * @property {Number} limit Page size * @property {Number} offset Starting index * @returns {ModelDefinition[]} */ DataSource.prototype.discoverModelDefinitionsSync = function(options) { this.freeze(); if (this.connector.discoverModelDefinitionsSync) { return this.connector.discoverModelDefinitionsSync(options); } return null; }; /** * Discover properties for a given model. * * Callback function return value is an object that can have the following properties: * *| Key | Type | Description | *|-----|------|-------------| *|owner | String | Database owner or schema| *|tableName | String | Table/view name| *|columnName | String | Column name| *|dataType | String | Data type| *|dataLength | Number | Data length| *|dataPrecision | Number | Numeric data precision| *|dataScale |Number | Numeric data scale| *|nullable |Boolean | If true, then the data can be null| * See [Properties](./Model-definition-JSON-file.html#properties) for more * details on the Property return type. * @param {String} modelName The table/view name * @options {Object} options The options * @property {String} owner|schema The database owner or schema * @param {Function} cb Callback function. Optional * @callback cb * @returns {Promise} A promise that returns an array of Properties (Property[]) * */ DataSource.prototype.discoverModelProperties = function(modelName, options, cb) { this.freeze(); if (cb === undefined && typeof options === 'function') { cb = options; options = {}; } options = options || {}; cb = cb || utils.createPromiseCallback(); if (this.connector.discoverModelProperties) { this.connector.discoverModelProperties(modelName, options, cb); } else if (cb) { process.nextTick(cb); } return cb.promise; }; /*! Method will be completely removed in LoopBack.next */ /** * The synchronous version of discoverModelProperties * @param {String} modelName The table/view name * @param {Object} options The options * @returns {*} */ DataSource.prototype.discoverModelPropertiesSync = function(modelName, options) { this.freeze(); if (this.connector.discoverModelPropertiesSync) { return this.connector.discoverModelPropertiesSync(modelName, options); } return null; }; /** * Discover primary keys for a given owner/modelName. * Callback function return value is an object that can have the following properties: * *| Key | Type | Description | *|-----|------|-------------| *| owner |String | Table schema or owner (may be null). Owner defaults to current user. *| tableName |String| Table name *| columnName |String| Column name *| keySeq |Number| Sequence number within primary key (1 indicates the first column in the primary key; 2 indicates the second column in the primary key). *| pkName |String| Primary key name (may be null) * See [ID Properties](./Model-definition-JSON-file.html#id-properties) for more * information. * @param {String} modelName The model name * @options {Object} options The options * @property {String} owner|schema The database owner or schema * @param {Function} [cb] The callback function * @returns {Promise} A promise with an array of Primary Keys (Property[]) */ DataSource.prototype.discoverPrimaryKeys = function(modelName, options, cb) { this.freeze(); if (cb === undefined && typeof options === 'function') { cb = options; options = {}; } options = options || {}; cb = cb || utils.createPromiseCallback(); if (this.connector.discoverPrimaryKeys) { this.connector.discoverPrimaryKeys(modelName, options, cb); } else if (cb) { process.nextTick(cb); } return cb.promise; }; /*! Method will be completely removed in LoopBack.next */ /** * The synchronous version of discoverPrimaryKeys * @param {String} modelName The model name * @options {Object} options The options * @property {String} owner|schema The database owner or schema * @returns {*} */ DataSource.prototype.discoverPrimaryKeysSync = function(modelName, options) { this.freeze(); if (this.connector.discoverPrimaryKeysSync) { return this.connector.discoverPrimaryKeysSync(modelName, options); } return null; }; /** * Discover foreign keys for a given owner/modelName * * Callback function return value is an object that can have the following properties: * *| Key | Type | Description | *|-----|------|-------------| *|fkOwner |String | Foreign key table schema (may be null) *|fkName |String | Foreign key name (may be null) *|fkTableName |String | Foreign key table name *|fkColumnName |String | Foreign key column name *|keySeq |Number | Sequence number within a foreign key( a value of 1 represents the first column of the foreign key, a value of 2 would represent the second column within the foreign key). *|pkOwner |String | Primary key table schema being imported (may be null) *|pkName |String | Primary key name (may be null) *|pkTableName |String | Primary key table name being imported *|pkColumnName |String | Primary key column name being imported * See [Relations](/Model-definition-JSON-file.html#relations) for more * information. * @param {String} modelName The model name * @options {Object} options The options * @property {String} owner|schema The database owner or schema * @param {Function} [cb] The callback function * @returns {Promise} A Promise with an array of foreign key relations. * */ DataSource.prototype.discoverForeignKeys = function(modelName, options, cb) { this.freeze(); if (cb === undefined && typeof options === 'function') { cb = options; options = {}; } options = options || {}; cb = cb || utils.createPromiseCallback(); if (this.connector.discoverForeignKeys) { this.connector.discoverForeignKeys(modelName, options, cb); } else if (cb) { process.nextTick(cb); } return cb.promise; }; /*! Method will be completely removed in LoopBack.next */ /** * The synchronous version of discoverForeignKeys * * @param {String} modelName The model name * @param {Object} options The options * @returns {*} */ DataSource.prototype.discoverForeignKeysSync = function(modelName, options) { this.freeze(); if (this.connector.discoverForeignKeysSync) { return this.connector.discoverForeignKeysSync(modelName, options); } return null; }; /** * Retrieves a description of the foreign key columns that reference the given table's primary key columns * (the foreign keys exported by a table), ordered by fkTableOwner, fkTableName, and keySeq. * * Callback function return value is an object that can have the following properties: * *| Key | Type | Description | *|-----|------|-------------| *|fkOwner |String | Foreign key table schema (may be null) *|fkName |String | Foreign key name (may be null) *|fkTableName |String | Foreign key table name *|fkColumnName |String | Foreign key column name *|keySeq |Number | Sequence number within a foreign key( a value of 1 represents the first column of the foreign key, a value of 2 would represent the second column within the foreign key). *|pkOwner |String | Primary key table schema being imported (may be null) *|pkName |String | Primary key name (may be null) *|pkTableName |String | Primary key table name being imported *|pkColumnName |String | Primary key column name being imported * See [Relations](/Model-definition-JSON-file.html#relations) for more * information. * * @param {String} modelName The model name * @options {Object} options The options * @property {String} owner|schema The database owner or schema * @param {Function} [cb] The callback function * @returns {Promise} A Promise with an array of exported foreign key relations. */ DataSource.prototype.discoverExportedForeignKeys = function(modelName, options, cb) { this.freeze(); if (cb === undefined && typeof options === 'function') { cb = options; options = {}; } options = options || {}; cb = cb || utils.createPromiseCallback(); if (this.connector.discoverExportedForeignKeys) { this.connector.discoverExportedForeignKeys(modelName, options, cb); } else if (cb) { process.nextTick(cb); } return cb.promise; }; /*! Method will be completely removed in LoopBack.next */ /** * The synchronous version of discoverExportedForeignKeys * @param {String} modelName The model name * @param {Object} options The options * @returns {*} */ DataSource.prototype.discoverExportedForeignKeysSync = function(modelName, options) { this.freeze(); if (this.connector.discoverExportedForeignKeysSync) { return this.connector.discoverExportedForeignKeysSync(modelName, options); } return null; }; function capitalize(str) { if (!str) { return str; } return str.charAt(0).toUpperCase() + ((str.length > 1) ? str.slice(1).toLowerCase() : ''); } function fromDBName(dbName, camelCase) { if (!dbName) { return dbName; } var parts = dbName.split(/-|_/); parts[0] = camelCase ? parts[0].toLowerCase() : capitalize(parts[0]); for (var i = 1; i < parts.length; i++) { parts[i] = capitalize(parts[i]); } return parts.join(''); } /** * Discover one schema from the given model without following the relations. **Example schema from oracle connector:** * * ```js * { * "name": "Product", * "options": { * "idInjection": false, * "oracle": { * "schema": "BLACKPOOL", * "table": "PRODUCT" * } * }, * "properties": { * "id": { * "type": "String", * "required": true, * "length": 20, * "id": 1, * "oracle": { * "columnName": "ID", * "dataType": "VARCHAR2", * "dataLength": 20, * "nullable": "N" * } * }, * "name": { * "type": "String", * "required": false, * "length": 64, * "oracle": { * "columnName": "NAME", * "dataType": "VARCHAR2", * "dataLength": 64, * "nullable": "Y" * } * }, * ... * "fireModes": { * "type": "String", * "required": false, * "length": 64, * "oracle": { * "columnName": "FIRE_MODES", * "dataType": "VARCHAR2", * "dataLength": 64, * "nullable": "Y" * } * } * } * } * ``` * * @param {String} tableName The name of the table to discover. * @options {Object} [options] An options object typically used for Relations. * See [Relations](./Model-definition-JSON-file.html#relations) for more information. * @param {Function} [cb] The callback function * @returns {Promise} A promise object that resolves to a single schema. */ DataSource.prototype.discoverSchema = function(tableName, options, cb) { options = options || {}; if (!cb && 'function' === typeof options) { cb = options; options = {}; } options.visited = {}; options.relations = false; cb = cb || utils.createPromiseCallback(); this.discoverSchemas(tableName, options, function(err, schemas) { if (err || !schemas) { cb && cb(err, schemas); return; } for (var s in schemas) { cb && cb(null, schemas[s]); return; } }); return cb.promise; }; /** * Discover schema from a given tableName/viewName. * * @param {String} tableName The table name. * @options {Object} [options] Options; see below. * @property {String} owner|schema Database owner or schema name. * @property {Boolean} relations True if relations (primary key/foreign key) are navigated; false otherwise. * @property {Boolean} all True if all owners are included; false otherwise. * @property {Boolean} views True if views are included; false otherwise. * @param {Function} [cb] The callback function * @returns {Promise} A promise object that resolves to an array of schemas. */ DataSource.prototype.discoverSchemas = function(tableName, options, cb) { options = options || {}; if (!cb && 'function' === typeof options) { cb = options; options = {}; } cb = cb || utils.createPromiseCallback(); var self = this; var dbType = this.connector.name || this.name; var nameMapper; if (options.nameMapper === null) { // No mapping nameMapper = function(type, name) { return name; }; } else if (typeof options.nameMapper === 'function') { // Custom name mapper nameMapper = options.nameMapper; } else { // Default name mapper nameMapper = function mapName(type, name) { if (type === 'table' || type === 'model') { return fromDBName(name, false); } else if (type == 'fk') { return fromDBName(name + 'Rel', true); } else { return fromDBName(name, true); } }; } if (this.connector.discoverSchemas) { // Delegate to the connector implementation this.connector.discoverSchemas(tableName, options, cb); return cb.promise; } var tasks = [ this.discoverModelProperties.bind(this, tableName, options), this.discoverPrimaryKeys.bind(this, tableName, options)]; var followingRelations = options.associations || options.relations; if (followingRelations) { tasks.push(this.discoverForeignKeys.bind(this, tableName, options)); } async.parallel(tasks, function(err, results) { if (err) { cb(err); return cb.promise; } var columns = results[0]; if (!columns || columns.length === 0) { cb(new Error(g.f('Table \'%s\' does not exist.', tableName))); return cb.promise; } // Handle primary keys var primaryKeys = results[1] || []; var pks = {}; primaryKeys.forEach(function(pk) { pks[pk.columnName] = pk.keySeq; }); if (self.settings.debug) { debug('Primary keys: ', pks); } var schema = { name: nameMapper('table', tableName), options: { idInjection: false, // DO NOT add id property }, properties: {}, }; schema.options[dbType] = { schema: columns[0].owner, table: tableName, }; columns.forEach(function(item) { var propName = nameMapper('column', item.columnName); schema.properties[propName] = { type: item.type, required: (item.nullable === 'N' || item.nullable === 'NO' || item.nullable === 0 || item.nullable === false), length: item.dataLength, precision: item.dataPrecision, scale: item.dataScale, }; if (pks[item.columnName]) { schema.properties[propName].id = pks[item.columnName]; } var dbSpecific = schema.properties[propName][dbType] = { columnName: item.columnName, dataType: item.dataType, dataLength: item.dataLength, dataPrecision: item.dataPrecision, dataScale: item.dataScale, nullable: item.nullable, }; // merge connector-specific properties if (item[dbType]) { for (var k in item[dbType]) { dbSpecific[k] = item[dbType][k]; } } }); // Add current modelName to the visited tables options.visited = options.visited || {}; var schemaKey = columns[0].owner + '.' + tableName; if (!options.visited.hasOwnProperty(schemaKey)) { if (self.settings.debug) { debug('Adding schema for ' + schemaKey); } options.visited[schemaKey] = schema; } var otherTables = {}; if (followingRelations) { // Handle foreign keys var fks = {}; var foreignKeys = results[2] || []; foreignKeys.forEach(function(fk) { var fkInfo = { keySeq: fk.keySeq, owner: fk.pkOwner, tableName: fk.pkTableName, columnName: fk.pkColumnName, }; if (fks[fk.fkName]) { fks[fk.fkName].push(fkInfo); } else { fks[fk.fkName] = [fkInfo]; } }); if (self.settings.debug) { debug('Foreign keys: ', fks); } schema.options.relations = {}; foreignKeys.forEach(function(fk) { var propName = nameMapper('fk', (fk.fkName || fk.pkTableName)); schema.options.relations[propName] = { model: nameMapper('table', fk.pkTableName), type: 'belongsTo', foreignKey: nameMapper('column', fk.fkColumnName), }; var key = fk.pkOwner + '.' + fk.pkTableName; if (!options.visited.hasOwnProperty(key) && !otherTables.hasOwnProperty(key)) { otherTables[key] = {owner: fk.pkOwner, tableName: fk.pkTableName}; } }); } if (Object.keys(otherTables).length === 0) { cb(null, options.visited); } else { var moreTasks = []; for (var t in otherTables) { if (self.settings.debug) { debug('Discovering related schema for ' + schemaKey); } var newOptions = {}; for (var key in options) { newOptions[key] = options[key]; } newOptions.owner = otherTables[t].owner; moreTasks.push(DataSource.prototype.discoverSchemas.bind(self, otherTables[t].tableName, newOptions)); } async.parallel(moreTasks, function(err, results) { var result = results && results[0]; cb(err, result); }); } }); return cb.promise; }; /*! Method will be completely removed in LoopBack.next */ /** * Discover schema from a given table/view synchronously * @param {String} modelName The model name * @options {Object} [options] Options; see below. * @property {String} owner|schema Database owner or schema name. * @property {Boolean} relations True if relations (primary key/foreign key) are navigated; false otherwise. * @property {Boolean} all True if all owners are included; false otherwise. * @property {Boolean} views True if views are included; false otherwise. * @returns {Array} An array of schema definition objects. */ DataSource.prototype.discoverSchemasSync = function(modelName, options) { var self = this; var dbType = this.name || this.connector.name; var columns = this.discoverModelPropertiesSync(modelName, options); if (!columns || columns.length === 0) { return []; } var nameMapper = options.nameMapper || function mapName(type, name) { if (type === 'table' || type === 'model') { return fromDBName(name, false); } else { return fromDBName(name, true); } }; // Handle primary keys var primaryKeys = this.discoverPrimaryKeysSync(modelName, options); var pks = {}; primaryKeys.forEach(function(pk) { pks[pk.columnName] = pk.keySeq; }); if (self.settings.debug) { debug('Primary keys: ', pks); } var schema = { name: nameMapper('table', modelName), options: { idInjection: false, // DO NOT add id property }, properties: {}, }; schema.options[dbType] = { schema: columns.length > 0 && columns[0].owner, table: modelName, }; columns.forEach(function(item) { var i = item; var propName = nameMapper('column', item.columnName); schema.properties[propName] = { type: item.type, required: (item.nullable === 'N'), length: item.dataLength, precision: item.dataPrecision, scale: item.dataScale, }; if (pks[item.columnName]) { schema.properties[propName].id = pks[item.columnName]; } schema.properties[propName][dbType] = { columnName: i.columnName, dataType: i.dataType, dataLength: i.dataLength, dataPrecision: item.dataPrecision, dataScale: item.dataScale, nullable: i.nullable, }; }); // Add current modelName to the visited tables options.visited = options.visited || {}; var schemaKey = columns[0].owner + '.' + modelName; if (!options.visited.hasOwnProperty(schemaKey)) { if (self.settings.debug) { debug('Adding schema for ' + schemaKey); } options.visited[schemaKey] = schema; } var otherTables = {}; var followingRelations = options.associations || options.relations; if (followingRelations) { // Handle foreign keys var fks = {}; var foreignKeys = this.discoverForeignKeysSync(modelName, options); foreignKeys.forEach(function(fk) { var fkInfo = { keySeq: fk.keySeq, owner: fk.pkOwner, tableName: fk.pkTableName, columnName: fk.pkColumnName, }; if (fks[fk.fkName]) { fks[fk.fkName].push(fkInfo); } else { fks[fk.fkName] = [fkInfo]; } }); if (self.settings.debug) { debug('Foreign keys: ', fks); } schema.options.relations = {}; foreignKeys.forEach(function(fk) { var propName = nameMapper('column', fk.pkTableName); schema.options.relations[propName] = { model: nameMapper('table', fk.pkTableName), type: 'belongsTo', foreignKey: nameMapper('column', fk.fkColumnName), }; var key = fk.pkOwner + '.' + fk.pkTableName; if (!options.visited.hasOwnProperty(key) && !otherTables.hasOwnProperty(key)) { otherTables[key] = {owner: fk.pkOwner, tableName: fk.pkTableName}; } }); } if (Object.keys(otherTables).length === 0) { return options.visited; } else { var moreTasks = []; for (var t in otherTables) { if (self.settings.debug) { debug('Discovering related schema for ' + schemaKey); } var newOptions = {}; for (var key in options) { newOptions[key] = options[key]; } newOptions.owner = otherTables[t].owner; self.discoverSchemasSync(otherTables[t].tableName, newOptions); } return options.visited; } }; /** * Discover and build models from the specified owner/modelName. * * @param {String} modelName The model name. * @options {Object} [options] Options; see below. * @property {String} owner|schema Database owner or schema name. * @property {Boolean} relations True if relations (primary key/foreign key) are navigated; false otherwise. * @property {Boolean} all True if all owners are included; false otherwise. * @property {Boolean} views True if views are included; false otherwise. * @param {Function} [cb] The callback function * @returns {Promise} A Promise object that resolves with a map of model * constructors, keyed by model name */ DataSource.prototype.discoverAndBuildModels = function(modelName, options, cb) { var self = this; options = options || {}; this.discoverSchemas(modelName, options, function(err, schemas) { if (err) { cb && cb(err, schemas); return; } var schemaList = []; for (var s in schemas) { var schema = schemas[s]; if (options.base) { schema.options = schema.options || {}; schema.options.base = options.base; } schemaList.push(schema); } var models = self.modelBuilder.buildModels(schemaList, self.createModel.bind(self)); cb && cb(err, models); }); }; /*! Method will be completely removed in LoopBack.next */ /** * Discover and build models from the given owner/modelName synchronously. * @param {String} modelName The model name. * @options {Object} [options] Options; see below. * @property {String} owner|schema Database owner or schema name. * @property {Boolean} relations True if relations (primary key/foreign key) are navigated; false otherwise. * @property {Boolean} all True if all owners are included; false otherwise. * @property {Boolean} views True if views are included; false otherwise. * @param {String} modelName The model name * @param {Object} [options] The options * @returns {Object} A map of model constructors, keyed by model name */ DataSource.prototype.discoverAndBuildModelsSync = function(modelName, options) { options = options || {}; var schemas = this.discoverSchemasSync(modelName, options); var schemaList = []; for (var s in schemas) { var schema = schemas[s]; if (options.base) { schema.options = schema.options || {}; schema.options.base = options.base; } schemaList.push(schema); } var models = this.modelBuilder.buildModels(schemaList, this.createModel.bind(this)); return models; }; /** * Introspect a JSON object and build a model class * @param {String} name Name of the model * @param {Object} json The json object representing a model instance * @param {Object} options Options * @returns {Model} A Model class */ DataSource.prototype.buildModelFromInstance = function(name, json, options) { // Introspect the JSON document to generate a schema var schema = ModelBuilder.introspect(json); // Create a model for the generated schema return this.createModel(name, schema, options); }; /** * Check whether or not migrations are required for the database schema to match * the Model definitions attached to the DataSource. * Note: This method applies only to SQL connectors. * @param {String|String[]} [models] A model name or an array of model names. If not present, apply to all models. * @returns {Boolean} Whether or not migrations are required. */ DataSource.prototype.isActual = function(models, cb) { this.freeze(); if (this.connector.isActual) { this.connector.isActual(models, cb); } else { if ((!cb) && ('function' === typeof models)) { cb = models; models = undefined; } if (cb) { process.nextTick(function() { cb(null, true); }); } } }; /** * Log benchmarked message. Do not redefine this method, if you need to grab * schema logs, use `dataSource.on('log', ...)` emitter event * * @private used by connectors */ DataSource.prototype.log = function(sql, t) { debug(sql, t); this.emit('log', sql, t); }; /** * Freeze dataSource. Behavior depends on connector * To continuously add artifacts to datasource until it is frozen, but it is not really used in loopback. */ DataSource.prototype.freeze = function freeze() { if (!this.connector) { throw new Error(g.f('The connector has not been initialized.')); } if (this.connector.freezeDataSource) { this.connector.freezeDataSource(); } if (this.connector.freezeSchema) { this.connector.freezeSchema(); } }; /** * Return table name for specified `modelName`. * @param {String} modelName The model name. * @returns {String} The table name. */ DataSource.prototype.tableName = function(modelName) { return this.getModelDefinition(modelName).tableName(this.connector.name); }; /** * Return column name for specified modelName and propertyName * @param {String} modelName The model name * @param {String} propertyName The property name * @returns {String} columnName The column name. */ DataSource.prototype.columnName = function(modelName, propertyName) { return this.getModelDefinition(modelName).columnName(this.connector.name, propertyName); }; /** * Return column metadata for specified modelName and propertyName * @param {String} modelName The model name * @param {String} propertyName The property name * @returns {Object} column metadata */ DataSource.prototype.columnMetadata = function(modelName, propertyName) { return this.getModelDefinition(modelName).columnMetadata(this.connector.name, propertyName); }; /** * Return column names for specified modelName * @param {String} modelName The model name * @returns {String[]} column names */ DataSource.prototype.columnNames = function(modelName) { return this.getModelDefinition(modelName).columnNames(this.connector.name); }; /** * Find the ID column name * @param {String} modelName The model name * @returns {String} columnName for ID */ DataSource.prototype.idColumnName = function(modelName) { return this.getModelDefinition(modelName).idColumnName(this.connector.name); }; /** * Find the ID property name * @param {String} modelName The model name * @returns {String} property name for ID */ DataSource.prototype.idName = function(modelName) { if (!this.getModelDefinition(modelName).idName) { g.error('No {{id}} name %s', this.getModelDefinition(modelName)); } return this.getModelDefinition(modelName).idName(); }; /** * Find the ID property names sorted by the index * @param {String} modelName The model name * @returns {String[]} property names for IDs */ DataSource.prototype.idNames = function(modelName) { return this.getModelDefinition(modelName).idNames(); }; /** * Find the id property definition * @param {String} modelName The model name * @returns {Object} The id property definition */ DataSource.prototype.idProperty = function(modelName) { var def = this.getModelDefinition(modelName); var idProps = def && def.ids(); return idProps && idProps[0] && idProps[0].property; }; /** * Define foreign key to another model * @param {String} className The model name that owns the key * @param {String} key Name of key field * @param {String} foreignClassName The foreign model name * @param {String} pkName (optional) primary key used for foreignKey */ DataSource.prototype.defineForeignKey = function defineForeignKey(className, key, foreignClassName, pkName) { var pkType = null; var foreignModel = this.getModelDefinition(foreignClassName); pkName = pkName || foreignModel && foreignModel.idName(); if (pkName) { pkType = foreignModel.properties[pkName].type; } var model = this.getModelDefinition(className); if (model.properties[key]) { if (pkType) { // Reset the type of the foreign key model.rawProperties[key].type = model.properties[key].type = pkType; } return; } var fkDef = {type: pkType}; var foreignMeta = this.columnMetadata(foreignClassName, pkName); if (foreignMeta && (foreignMeta.dataType || foreignMeta.dataLength)) { fkDef[this.connector.name] = {}; if (foreignMeta.dataType) { fkDef[this.connector.name].dataType = foreignMeta.dataType; } if (foreignMeta.dataLength) { fkDef[this.connector.name].dataLength = foreignMeta.dataLength; } } if (this.connector.defineForeignKey) { var cb = function(err, keyType) { if (err) throw err; fkDef.type = keyType || pkType; // Add the foreign key property to the data source _models this.defineProperty(className, key, fkDef); }.bind(this); switch (this.connector.defineForeignKey.length) { case 4: this.connector.defineForeignKey(className, key, foreignClassName, cb); break; default: case 3: this.connector.defineForeignKey(className, key, cb); break; } } else { // Add the foreign key property to the data source _models this.defineProperty(className, key, fkDef); } }; /** * Close connection to the DataSource. * @param {Function} [cb] The callback function. Optional. */ DataSource.prototype.disconnect = function disconnect(cb) { cb = cb || utils.createPromiseCallback(); var self = this; if (this.connected && (typeof this.connector.disconnect === 'function')) { this.connector.disconnect(function(err, result) { self.connected = false; cb && cb(err, result); }); } else { process.nextTick(function() { self.connected = false; cb && cb(); }); } return cb.promise; }; /** * Copy the model from Master. * @param {Function} Master The model constructor * @returns {Model} A copy of the Model object constructor * * @private */ DataSource.prototype.copyModel = function copyModel(Master) { var dataSource = this; var className = Master.modelName; var md = Master.modelBuilder.getModelDefinition(className); var Slave = function SlaveModel() { Master.apply(this, [].slice.call(arguments)); }; util.inherits(Slave, Master); // Delegating static properties Slave.__proto__ = Master; hiddenProperty(Slave, 'dataSource', dataSource); hiddenProperty(Slave, 'modelName', className); hiddenProperty(Slave, 'relations', Master.relations); if (!(className in dataSource.modelBuilder.models)) { // store class in model pool dataSource.modelBuilder.models[className] = Slave; dataSource.modelBuilder.definitions[className] = new ModelDefinition(dataSource.modelBuilder, md.name, md.properties, md.settings); if ((!dataSource.isTransaction) && dataSource.connector && dataSource.connector.define) { dataSource.connector.define({ model: Slave, properties: md.properties, settings: md.settings, }); } } return Slave; }; /** * Run a transaction against the DataSource. * @returns {EventEmitter} * @private */ DataSource.prototype.transaction = function() { var dataSource = this; var transaction = new EventEmitter(); for (var p in dataSource) { transaction[p] = dataSource[p]; } transaction.isTransaction = true; transaction.origin = dataSource; transaction.name = dataSource.name; transaction.settings = dataSource.settings; transaction.connected = false; transaction.connecting = false; transaction.connector = dataSource.connector.transaction(); // create blank models pool transaction.modelBuilder = new ModelBuilder(); transaction.models = transaction.modelBuilder.models; transaction.definitions = transaction.modelBuilder.definitions; for (var i in dataSource.modelBuilder.models) { dataSource.copyModel.call(transaction, dataSource.modelBuilder.models[i]); } transaction.exec = function(cb) { transaction.connector.exec(cb); }; return transaction; }; /** * Enable remote access to a data source operation. Each [connector](#connector) has its own set of set * remotely enabled and disabled operations. To list the operations, call `dataSource.operations()`. * @param {String} operation The operation name */ DataSource.prototype.enableRemote = function(operation) { var op = this.getOperation(operation); if (op) { op.remoteEnabled = true; } else { throw new Error(g.f('%s is not provided by the attached connector', operation)); } }; /** * Disable remote access to a data source operation. Each [connector](#connector) has its own set of set enabled * and disabled operations. To list the operations, call `dataSource.operations()`. * *```js * var oracle = loopback.createDataSource({ * connector: require('loopback-connector-oracle'), * host: '...', * ... * }); * oracle.disableRemote('destroyAll'); * ``` * **Notes:** * * - Disabled operations will not be added to attached models. * - Disabling the remoting for a method only affects client access (it will still be available from server models). * - Data sources must enable / disable operations before attaching or creating models. * @param {String} operation The operation name */ DataSource.prototype.disableRemote = function(operation) { var op = this.getOperation(operation); if (op) { op.remoteEnabled = false; } else { throw new Error(g.f('%s is not provided by the attached connector', operation)); } }; /** * Get an operation's metadata. * @param {String} operation The operation name */ DataSource.prototype.getOperation = function(operation) { var ops = this.operations(); var opKeys = Object.keys(ops); for (var i = 0; i < opKeys.length; i++) { var op = ops[opKeys[i]]; if (op.name === operation) { return op; } } }; /** * Return JSON object describing all operations. * * Example return value: * ```js * { * find: { * remoteEnabled: true, * accepts: [...], * returns: [...] * enabled: true * }, * save: { * remoteEnabled: true, * prototype: true, * accepts: [...], * returns: [...], * enabled: true * }, * ... * } * ``` */ DataSource.prototype.operations = function() { return this._operations; }; /** * Define an operation to the data source * @param {String} name The operation name * @param {Object} options The options * @param {Function} fn The function */ DataSource.prototype.defineOperation = function(name, options, fn) { options.fn = fn; options.name = name; this._operations[name] = options; }; /** * Check if the backend is a relational DB * @returns {Boolean} */ DataSource.prototype.isRelational = function() { return this.connector && this.connector.relational; }; /** * Check if the data source is ready. * Returns a Boolean value. * @param {Object} obj Deferred method call if the connector is not fully connected yet * @param {Object} args argument passing to method call. */ DataSource.prototype.ready = function(obj, args) { var self = this; if (this.connected) { // Connected return false; } var method = args.callee; // Set up a callback after the connection is established to continue the method call var onConnected = null, onError = null, timeoutHandle = null; onConnected = function() { // Remove the error handler self.removeListener('error', onError); if (timeoutHandle) { clearTimeout(timeoutHandle); } var params = [].slice.call(args); try { method.apply(obj, params); } catch (err) { // Catch the exception and report it via callback var cb = params.pop(); if (typeof cb === 'function') { process.nextTick(function() { cb(err); }); } else { throw err; } } }; onError = function(err) { // Remove the connected listener self.removeListener('connected', onConnected); if (timeoutHandle) { clearTimeout(timeoutHandle); } var params = [].slice.call(args); var cb = params.pop(); if (typeof cb === 'function') { process.nextTick(function() { cb(err); }); } }; this.once('connected', onConnected); this.once('error', onError); // Set up a timeout to cancel the invocation var timeout = this.settings.connectionTimeout || 5000; timeoutHandle = setTimeout(function() { self.removeListener('error', onError); self.removeListener('connected', onConnected); var params = [].slice.call(args); var cb = params.pop(); if (typeof cb === 'function') { cb(new Error(g.f('Timeout in connecting after %s ms', timeout))); } }, timeout); if (!this.connecting) { this.connect(); } return true; }; /** * Ping the underlying connector to test the connections * @param {Function} [cb] Callback function */ DataSource.prototype.ping = function(cb) { var self = this; if (self.connector.ping) { this.connector.ping(cb); } else if (self.connector.discoverModelProperties) { self.discoverModelProperties('dummy', {}, cb); } else { process.nextTick(function() { var err = self.connected ? null : new Error(g.f('Not connected')); cb(err); }); } }; /*! The hidden property call is too expensive so it is not used that much */ /** * Define a hidden property * It is an utility to define a property to the Object with info flags * @param {Object} obj The property owner * @param {String} key The property name * @param {Mixed} value The default value */ function hiddenProperty(obj, key, value) { Object.defineProperty(obj, key, { writable: false, enumerable: false, configurable: false, value: value, }); } /** * Define readonly property on object * * @param {Object} obj The property owner * @param {String} key The property name * @param {Mixed} value The default value */ function defineReadonlyProp(obj, key, value) { Object.defineProperty(obj, key, { writable: false, enumerable: true, configurable: true, value: value, }); } // Carry over a few properties/methods from the ModelBuilder as some tests use them DataSource.Text = ModelBuilder.Text; DataSource.JSON = ModelBuilder.JSON; DataSource.Any = ModelBuilder.Any;