UNPKG

sails-postgresql

Version:

a PostgreSQL adapter for Waterline and Sails.js

210 lines (165 loc) 9.64 kB
// ██████╗ ███████╗███████╗████████╗██████╗ ██████╗ ██╗ ██╗ █████╗ ██████╗████████╗██╗ ██████╗ ███╗ ██╗ // ██╔══██╗██╔════╝██╔════╝╚══██╔══╝██╔══██╗██╔═══██╗╚██╗ ██╔╝ ██╔══██╗██╔════╝╚══██╔══╝██║██╔═══██╗████╗ ██║ // ██║ ██║█████╗ ███████╗ ██║ ██████╔╝██║ ██║ ╚████╔╝ ███████║██║ ██║ ██║██║ ██║██╔██╗ ██║ // ██║ ██║██╔══╝ ╚════██║ ██║ ██╔══██╗██║ ██║ ╚██╔╝ ██╔══██║██║ ██║ ██║██║ ██║██║╚██╗██║ // ██████╔╝███████╗███████║ ██║ ██║ ██║╚██████╔╝ ██║ ██║ ██║╚██████╗ ██║ ██║╚██████╔╝██║ ╚████║ // ╚═════╝ ╚══════╝╚══════╝ ╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═╝ ╚═╝ ╚═════╝ ╚═╝ ╚═══╝ // module.exports = require('machine').build({ friendlyName: 'Destroy', description: 'Destroy record(s) in the database matching a query criteria.', inputs: { datastore: { description: 'The datastore to use for connections.', extendedDescription: 'Datastores represent the config and manager required to obtain an active database connection.', required: true, readOnly: true, type: 'ref' }, models: { description: 'An object containing all of the model definitions that have been registered.', required: true, type: 'ref' }, query: { description: 'A valid stage three Waterline query.', required: true, type: 'ref' } }, exits: { success: { description: 'The results of the destroy query.', outputVariableName: 'records', outputType: 'ref' }, invalidDatastore: { description: 'The datastore used is invalid. It is missing key pieces.' }, badConnection: { friendlyName: 'Bad connection', description: 'A connection either could not be obtained or there was an error using the connection.' } }, fn: function destroy(inputs, exits) { // Dependencies var _ = require('@sailshq/lodash'); var WLUtils = require('waterline-utils'); var Helpers = require('./private'); var Converter = WLUtils.query.converter; // Store the Query input for easier access var query = inputs.query; query.meta = query.meta || {}; // Find the model definition var model = inputs.models[query.using]; if (!model) { return exits.invalidDatastore(); } // Set a flag if a leased connection from outside the adapter was used or not. var leased = _.has(query.meta, 'leasedConnection'); // Set a flag to determine if records are being returned var fetchRecords = false; // ╔═╗╦ ╦╔═╗╔═╗╦╔═ ┌─┐┌─┐┬─┐ ┌─┐ ┌─┐┌─┐ ┌─┐┌─┐┬ ┬┌─┐┌┬┐┌─┐ // ║ ╠═╣║╣ ║ ╠╩╗ ├┤ │ │├┬┘ ├─┤ ├─┘│ ┬ └─┐│ ├─┤├┤ │││├─┤ // ╚═╝╩ ╩╚═╝╚═╝╩ ╩ └ └─┘┴└─ ┴ ┴ ┴ └─┘ └─┘└─┘┴ ┴└─┘┴ ┴┴ ┴ // This is a unique feature of Postgres. It may be passed in on a query // by query basis using the meta input or configured on the datastore. Default // to use the public schema. var schemaName = 'public'; if (_.has(query.meta, 'schemaName')) { schemaName = query.meta.schemaName; } else if (inputs.datastore.config && inputs.datastore.config.schemaName) { schemaName = inputs.datastore.config.schemaName; } // ╔═╗╔═╗╔╗╔╦ ╦╔═╗╦═╗╔╦╗ ┌┬┐┌─┐ ┌─┐┌┬┐┌─┐┌┬┐┌─┐┌┬┐┌─┐┌┐┌┌┬┐ // ║ ║ ║║║║╚╗╔╝║╣ ╠╦╝ ║ │ │ │ └─┐ │ ├─┤ │ ├┤ │││├┤ │││ │ // ╚═╝╚═╝╝╚╝ ╚╝ ╚═╝╩╚═ ╩ ┴ └─┘ └─┘ ┴ ┴ ┴ ┴ └─┘┴ ┴└─┘┘└┘ ┴ // Convert the Waterline criteria into a Waterline Query Statement. This // turns it into something that is declarative and can be easily used to // build a SQL query. // See: https://github.com/treelinehq/waterline-query-docs for more info // on Waterline Query Statements. var statement; try { statement = Converter({ model: query.using, method: 'destroy', criteria: query.criteria, opts: { schema: schemaName } }); } catch (e) { return exits.error(e); } // ╔╦╗╔═╗╔╦╗╔═╗╦═╗╔╦╗╦╔╗╔╔═╗ ┬ ┬┬ ┬┬┌─┐┬ ┬ ┬ ┬┌─┐┬ ┬ ┬┌─┐┌─┐ // ║║║╣ ║ ║╣ ╠╦╝║║║║║║║║╣ │││├─┤││ ├─┤ └┐┌┘├─┤│ │ │├┤ └─┐ // ═╩╝╚═╝ ╩ ╚═╝╩╚═╩ ╩╩╝╚╝╚═╝ └┴┘┴ ┴┴└─┘┴ ┴ └┘ ┴ ┴┴─┘└─┘└─┘└─┘ // ┌┬┐┌─┐ ┬─┐┌─┐┌┬┐┬ ┬┬─┐┌┐┌ // │ │ │ ├┬┘├┤ │ │ │├┬┘│││ // ┴ └─┘ ┴└─└─┘ ┴ └─┘┴└─┘└┘ if (_.has(query.meta, 'fetch') && query.meta.fetch) { fetchRecords = true; // Add the postgres RETURNING * piece to the statement to prevent the // overhead of running two additional queries. statement.returning = '*'; } // Compile the original Waterline Query var compiledQuery; try { compiledQuery = Helpers.query.compileStatement(statement); } catch (e) { return exits.error(e); } // ╔═╗╔═╗╔═╗╦ ╦╔╗╔ ┌─┐┌─┐┌┐┌┌┐┌┌─┐┌─┐┌┬┐┬┌─┐┌┐┌ // ╚═╗╠═╝╠═╣║║║║║║ │ │ │││││││├┤ │ │ ││ ││││ // ╚═╝╩ ╩ ╩╚╩╝╝╚╝ └─┘└─┘┘└┘┘└┘└─┘└─┘ ┴ ┴└─┘┘└┘ // ┌─┐┬─┐ ┬ ┬┌─┐┌─┐ ┬ ┌─┐┌─┐┌─┐┌─┐┌┬┐ ┌─┐┌─┐┌┐┌┌┐┌┌─┐┌─┐┌┬┐┬┌─┐┌┐┌ // │ │├┬┘ │ │└─┐├┤ │ ├┤ ├─┤└─┐├┤ ││ │ │ │││││││├┤ │ │ ││ ││││ // └─┘┴└─ └─┘└─┘└─┘ ┴─┘└─┘┴ ┴└─┘└─┘─┴┘ └─┘└─┘┘└┘┘└┘└─┘└─┘ ┴ ┴└─┘┘└┘ // Spawn a new connection for running queries on. Helpers.connection.spawnOrLeaseConnection(inputs.datastore, query.meta, function spawnConnectionCb(err, connection) { if (err) { return exits.badConnection(err); } // ╦═╗╦ ╦╔╗╔ ┌┬┐┌─┐┌─┐┌┬┐┬─┐┌─┐┬ ┬ ┌─┐ ┬ ┬┌─┐┬─┐┬ ┬ // ╠╦╝║ ║║║║ ││├┤ └─┐ │ ├┬┘│ │└┬┘ │─┼┐│ │├┤ ├┬┘└┬┘ // ╩╚═╚═╝╝╚╝ ─┴┘└─┘└─┘ ┴ ┴└─└─┘ ┴ └─┘└└─┘└─┘┴└─ ┴ Helpers.query.runQuery({ queryType: 'select', connection: connection, nativeQuery: compiledQuery.nativeQuery, valuesToEscape: compiledQuery.valuesToEscape, disconnectOnError: leased ? false : true }, function runQueryCb(err, report) { // The connection will have been disconnected on error already if needed. if (err) { return exits.error(err); } // Always release the connection unless a leased connection from outside // the adapter was used. Helpers.connection.releaseConnection(connection, leased, function cb() { if (fetchRecords) { var selectRecords = report.result; var orm = { collections: inputs.models }; // Process each record to normalize output try { Helpers.query.processEachRecord({ records: selectRecords, identity: model.identity, orm: orm }); } catch (e) { return exits.error(e); } return exits.success({ records: selectRecords }); } return exits.success(); }); // </ releaseConnection > }); // </ runQuery > }); // </ spawnConnection > } });