unleash.unleash/lib/db/strategy.js

var eventType       = require('../eventType');
var logger          = require('../logger');
var NotFoundError   = require('../error/NotFoundError');
var STRATEGY_COLUMNS = ['name', 'description', 'parameters_template'];

module.exports = function(db, eventStore) {
    eventStore.on(eventType.strategyCreated, function (event) {
        return createStrategy(event.data);
    });

    eventStore.on(eventType.strategyDeleted, function (event) {
        db('strategies')
            .where('name', event.data.name)
            .del()
            .catch(function (err) {
                logger.error('Could not delete strategy, error was: ', err);
            });
    });

    function getStrategies() {
        return db
            .select(STRATEGY_COLUMNS)
            .from('strategies')
            .orderBy('created_at', 'asc')
            .map(rowToStrategy);
    }

    function getStrategy(name) {
        return db
            .first(STRATEGY_COLUMNS)
            .from('strategies')
            .where({ name: name })
            .then(rowToStrategy);
    }

    function rowToStrategy(row) {
        if (!row) {
            throw new NotFoundError('No strategy found');
        }

        return {
            name: row.name,
            description: row.description,
            parametersTemplate: row.parameters_template
        };
    }

    function eventDataToRow(data) {
        return {
            name: data.name,
            description: data.description,
            parameters_template: data.parametersTemplate // eslint-disable-line
        };
    }

    function createStrategy(data) {
        db('strategies')
            .insert(eventDataToRow(data))
            .catch(function (err) {
                logger.error('Could not insert strategy, error was: ', err);
            });
    }

    return {
        getStrategies: getStrategies,
        getStrategy: getStrategy,
        _createStrategy: createStrategy // visible for testing
    };
};