unleash.unleash/lib/featureDb.js

var eventStore      = require('./eventStore');
var eventType       = require('./eventType');
var logger          = require('./logger');
var knex            = require('./dbPool');
var NotFoundError   = require('./error/NotFoundError');
var FEATURE_COLUMNS = ['name', 'description', 'enabled', 'strategy_name', 'parameters'];

eventStore.on(eventType.featureCreated, function (event) {
    return createFeature(event.data);
});

eventStore.on(eventType.featureUpdated, function (event) {
    return updateFeature(event.data);
});

eventStore.on(eventType.featureArchived, function (event) {
    return archiveFeature(event.data);
});

eventStore.on(eventType.featureRevived, function (event) {
    return reviveFeature(event.data);
});

function getFeatures() {
    return knex
        .select(FEATURE_COLUMNS)
        .from('features')
        .where({ archived: 0 })
        .orderBy('name', 'asc')
        .map(rowToFeature);
}

function getFeature(name) {
    return knex
        .first(FEATURE_COLUMNS)
        .from('features')
        .where({ name: name })
        .then(rowToFeature);
}

function getArchivedFeatures() {
    return knex
        .select(FEATURE_COLUMNS)
        .from('features')
        .where({ archived: 1 })
        .orderBy('name', 'asc')
        .map(rowToFeature);
}


function rowToFeature(row) {
    if (!row) {
        throw new NotFoundError('No feature toggle found');
    }

    return {
        name: row.name,
        description: row.description,
        enabled: row.enabled > 0,
        strategy: row.strategy_name, // eslint-disable-line
        parameters: row.parameters
    };
}

function eventDataToRow(data) {
    return {
        name: data.name,
        description: data.description,
        enabled: data.enabled ? 1 : 0,
        archived: data.archived ? 1 :0,
        strategy_name: data.strategy, // eslint-disable-line
        parameters: data.parameters
    };
}

function createFeature(data) {
    return knex('features')
        .insert(eventDataToRow(data))
        .catch(function (err) {
            logger.error('Could not insert feature, error was: ', err);
        });
}

function updateFeature(data) {
    return knex('features')
        .where({ name: data.name })
        .update(eventDataToRow(data))
        .catch(function (err) {
            logger.error('Could not update feature, error was: ', err);
        });
}

function archiveFeature(data) {
    return knex('features')
        .where({ name: data.name })
        .update({ archived: 1 })
        .catch(function (err) {
            logger.error('Could not archive feature, error was: ', err);
        });
}

function reviveFeature(data) {
    return knex('features')
        .where({ name: data.name })
        .update({ archived: 0, enabled: 0 })
        .catch(function (err) {
            logger.error('Could not archive feature, error was: ', err);
        });
}


module.exports = {
    getFeatures: getFeatures,
    getFeature: getFeature,
    getArchivedFeatures: getArchivedFeatures,
    _createFeature: createFeature, // visible for testing
    _updateFeature: updateFeature  // visible for testing
};